dvc890 commited on
Commit
e3dd2a9
·
1 Parent(s): d502ffa

Update api/common.go

Browse files
Files changed (1) hide show
  1. api/common.go +4 -3
api/common.go CHANGED
@@ -97,12 +97,13 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) {
97
  c.Writer.Write([]byte(line + "\n\n"))
98
  c.Writer.Flush()
99
 
 
 
 
100
  data := line[6:]
101
  var result map[string]interface{}
102
  err := json.Unmarshal([]byte(data), &result)
103
- if err != nil {
104
- continue
105
- }
106
  message := result["message"].(map[string]interface{})
107
  status := message["status"].(string)
108
 
 
97
  c.Writer.Write([]byte(line + "\n\n"))
98
  c.Writer.Flush()
99
 
100
+ if strings.HasPrefix(line, "[DONE]") {
101
+ continue
102
+ }
103
  data := line[6:]
104
  var result map[string]interface{}
105
  err := json.Unmarshal([]byte(data), &result)
106
+ if err != nil
 
 
107
  message := result["message"].(map[string]interface{})
108
  status := message["status"].(string)
109