Refactor chatStreamClaude3 to use firstMessage variable for clari…

…ty. (#543)
swuecho · Sep 14, 2024 · fc18baf · fc18baf
1 parent 2ce0f3c
commit fc18baf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go
@@ -957,10 +957,11 @@ func (h *ChatHandler) chatStreamClaude3(w http.ResponseWriter, chatSession sqlc_
 		}
 		defer resp.Body.Close()
 		uuid := message.ID
-		answer := constructChatCompletionStreamReponse(uuid, message.Content[0].Text)
+		firstMessage := message.Content[0].Text
+		answer := constructChatCompletionStreamReponse(uuid, firstMessage)
 		data, _ := json.Marshal(answer)
 		fmt.Fprint(w, string(data))
-		return string(data), uuid, false
+		return firstMessage, uuid, false
 	}
 
 	ioreader := bufio.NewReader(resp.Body)