fix: stream have not return usage.
This commit is contained in:
@@ -452,6 +452,9 @@ func (tc *TrainingController) streamAIAnalysis(c *gin.Context, prompt string,
|
|||||||
TopP: 0.6,
|
TopP: 0.6,
|
||||||
MaxTokens: 4000,
|
MaxTokens: 4000,
|
||||||
Stream: true,
|
Stream: true,
|
||||||
|
StreamOptions: &openai.StreamOptions{
|
||||||
|
IncludeUsage: true,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
Reference in New Issue
Block a user