dvc890 commited on
Commit
598c85b
·
1 Parent(s): 7fd9203

Upload 2 files

Browse files
Files changed (2) hide show
  1. api/chatgpt/api.go +0 -1
  2. api/common.go +3 -3
api/chatgpt/api.go CHANGED
@@ -83,7 +83,6 @@ func CreateConversation(c *gin.Context) {
83
  c.Set("oldpart", "")
84
  Status, ConversationID, part := api.HandleConversationResponse(c, resp)
85
  if Status {
86
- logger.Info(fmt.Sprintf("part的值: %s", part))
87
  ContinueConversation(c, ConversationID, request.ParentMessageID, request.Model, part)
88
  }
89
  }
 
83
  c.Set("oldpart", "")
84
  Status, ConversationID, part := api.HandleConversationResponse(c, resp)
85
  if Status {
 
86
  ContinueConversation(c, ConversationID, request.ParentMessageID, request.Model, part)
87
  }
88
  }
api/common.go CHANGED
@@ -104,11 +104,12 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
104
  continue
105
  }
106
 
 
107
  if strings.HasPrefix(line, "[DONE]") {
108
  if Status {
109
  break
110
  }
111
- } else if oldpart != "" {
112
  var result map[string]interface{}
113
  err := json.Unmarshal([]byte(line[6:]), &result)
114
  if err == nil {
@@ -127,6 +128,7 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
127
 
128
  c.Writer.Write([]byte(line + "\n\n"))
129
  c.Writer.Flush()
 
130
  if strings.HasPrefix(line, "[DONE]") {
131
  break
132
  }
@@ -141,8 +143,6 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
141
  status := message["status"].(string)
142
 
143
  if status == "finished_successfully" {
144
- logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
145
- logger.Info(fmt.Sprintf("status的值: %s", status))
146
  if message["metadata"] != nil {
147
  metadata := message["metadata"].(map[string]interface{})
148
  if metadata["finish_details"] != nil {
 
104
  continue
105
  }
106
 
107
+ logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
108
  if strings.HasPrefix(line, "[DONE]") {
109
  if Status {
110
  break
111
  }
112
+ } else if len(oldpart) > 0 {
113
  var result map[string]interface{}
114
  err := json.Unmarshal([]byte(line[6:]), &result)
115
  if err == nil {
 
128
 
129
  c.Writer.Write([]byte(line + "\n\n"))
130
  c.Writer.Flush()
131
+
132
  if strings.HasPrefix(line, "[DONE]") {
133
  break
134
  }
 
143
  status := message["status"].(string)
144
 
145
  if status == "finished_successfully" {
 
 
146
  if message["metadata"] != nil {
147
  metadata := message["metadata"].(map[string]interface{})
148
  if metadata["finish_details"] != nil {