diff options
Diffstat (limited to 'internal/llm/provider')
| -rw-r--r-- | internal/llm/provider/provider.go | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/internal/llm/provider/provider.go b/internal/llm/provider/provider.go index 938a8c0ad..34d91f2b7 100644 --- a/internal/llm/provider/provider.go +++ b/internal/llm/provider/provider.go @@ -57,7 +57,9 @@ func cleanupMessages(messages []message.Message) []message.Message { // First pass: filter out canceled messages var cleanedMessages []message.Message for _, msg := range messages { - if msg.FinishReason() != "canceled" { + if msg.FinishReason() != "canceled" || len(msg.ToolCalls()) > 0 { + // if there are toolCalls this means we want to return it to the LLM telling it that those tools have been + // cancelled cleanedMessages = append(cleanedMessages, msg) } } |