update conversation

api/chat.go

@@ -73,12 +73,23 @@ func ChatAPI(c *gin.Context) {
 			keyword, segment := ChatWithWeb(instance.GetMessageSegment(12), true)
 			SendSegmentMessage(conn, types.ChatGPTSegmentResponse{Keyword: keyword, End: false})
 
+			msg := ""
 			StreamRequest("gpt-3.5-turbo-16k-0613", segment, 2000, func(resp string) {
+				msg += resp
 				SendSegmentMessage(conn, types.ChatGPTSegmentResponse{
 					Message: resp,
 					End:     false,
 				})
 			})
+			if msg == "" {
+				msg = "There was something wrong... Please try again later."
+				SendSegmentMessage(conn, types.ChatGPTSegmentResponse{
+					Message: msg,
+					End:     false,
+				})
+			}
+
+			instance.SaveResponse(db, msg)
 			SendSegmentMessage(conn, types.ChatGPTSegmentResponse{End: true})
 		}
 	}

api/utils.go

@@ -70,6 +70,12 @@ func SearchWeb(message []types.ChatGPTMessage) string {
 	}, {
 		Role:    "assistant",
 		Content: "{\"keyword\":\"\"}",
+	}, {
+		Role:    "user",
+		Content: "?",
+	}, {
+		Role:    "assistant",
+		Content: "{\"keyword\":\"\"}",
 	}, {
 		Role:    "user",
 		Content: message[len(message)-1].Content,

conversation/conversation.go

@@ -124,3 +124,8 @@ func (c *Conversation) HandleMessage(db *sql.DB, data []byte) bool {
 	c.SaveConversation(db)
 	return true
 }
+
+func (c *Conversation) SaveResponse(db *sql.DB, message string) {
+	c.AddMessageFromAssistant(message)
+	c.SaveConversation(db)
+}