Compare commits

..
1 Commits
+4
View File
@@ -168,6 +168,7 @@ func (e *AnyLLMEngine) Stream(
Extra: extra,
})
if params.OnChunk != nil {
params.OnChunk(&domain.Chunk{
Done: false,
ToolName: tc.Function.Name,
@@ -175,6 +176,7 @@ func (e *AnyLLMEngine) Stream(
Content: tc.Function.Arguments,
Reasoning: false,
})
}
}
@@ -182,12 +184,14 @@ func (e *AnyLLMEngine) Stream(
}
if !toolCalling {
if params.OnChunk != nil {
params.OnChunk(&domain.Chunk{
Done: true,
Role: anyllm.RoleAssistant,
Content: "",
})
}
}
if err := <-errChan; err != nil {
return &domain.Message{