From 5e68e50aa1e359623adfcb8d6ef3c302b72b8d9f Mon Sep 17 00:00:00 2001
From: laoboli <1293528695@qq.com>
Date: Sat, 2 May 2026 21:39:44 +0800
Subject: [PATCH] fix: stream have not return usage.

---
 controllers/ai.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controllers/ai.go b/controllers/ai.go
index 6dfb890..e08ed5c 100644
--- a/controllers/ai.go
+++ b/controllers/ai.go
@@ -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 {