fix: stream have not return usage.

2026-05-02 21:39:44 +08:00
parent ae5d70b951
commit 5e68e50aa1
1 changed files with 3 additions and 0 deletions
@@ -452,6 +452,9 @@ func (tc *TrainingController) streamAIAnalysis(c *gin.Context, prompt string,
 			TopP:        0.6,
 			MaxTokens:   4000,
 			Stream:      true,
+			StreamOptions: &openai.StreamOptions{
+				IncludeUsage: true,
+			},
 		},
 	)
 	if err != nil {