Update api/common.go

api/common.go

@@ -77,21 +77,20 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
 	ConversationID := ""
 	oldpart := c.GetString("oldpart")
 	part := ""
-	logger.Info(fmt.Sprintf("oldpart的值: %s", oldpart))
 
 	c.Writer.Header().Set("Content-Type", "text/event-stream; charset=utf-8")
 
 	reader := bufio.NewReader(resp.Body)
 	for {
 		if c.Request.Context().Err() != nil {
-			c.Writer.Write([]byte("data: " + "\n\n"))
+			c.Writer.Write([]byte("data: [DONE]" + "\n\n"))
 			c.Writer.Flush()
 			break
 		}
 
 		line, err := reader.ReadString('\n')
 		if err != nil {
-			c.Writer.Write([]byte("data: " + "\n\n"))
+			c.Writer.Write([]byte("data: [DONE]" + "\n\n"))
 			c.Writer.Flush()
 			break
 		}
@@ -104,7 +103,7 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
 			continue
 		}
 
-		logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
+		// logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
 		if strings.HasPrefix(line, "[DONE]") {
 			if Status {
 				break
@@ -117,7 +116,12 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
 				content := message["content"].(map[string]interface{})
 				parts := content["parts"].([]interface{})
 				part = parts[0].(string)
+
 				parts[0] = oldpart + part
+				message["id"] = c.GetString("msg_id")
+				metadata := message["metadata"].(map[string]interface{})
+				metadata["message_type"] = "next"
+
 				resultJSON, err2 := json.Marshal(result)
 				if err2 == nil {
 					line = "data: " + string(resultJSON)
@@ -155,6 +159,11 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
 						logger.Info(fmt.Sprintf("finish_details中type的值: %s", finishType))
 						Status = true
 						ConversationID = result["conversation_id"].(string)
+
+						if len(oldpart) == 0 {
+							msg_id := message["id"].(string)
+							c.Set("msg_id", msg_id)
+						}
 					}
 				}
 			}