fix stop reason missing problem

linexjlin · Mar 26, 2024 · b90aca6 · b90aca6
1 parent 8da53a6
commit b90aca6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/userCore.go b/userCore.go
@@ -161,14 +161,13 @@ func (u *UserCore) QueryGPT(ctx context.Context, txtChan chan string, messages [
 				txtChan <- response.Choices[0].Delta.Content
 			}
 			if response.Choices[0].FinishReason == "stop" {
-				close(txtChan)
+				fmt.Println("stop")
 			}
 		}
 	})
 	if err != nil {
 		fmt.Printf("ChatCompletionStream error: %v\n", err)
 		txtChan <- fmt.Sprintf("ChatCompletionStream error: %v\n", err)
-		close(txtChan)
-		return
 	}
+	close(txtChan)
 }