Compare commits

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