From cdc5f209dccdc980714f2ca1aeb52133d6e93cce Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Sun, 13 Apr 2025 14:37:05 +0200 Subject: cleanup diff, cleanup agent --- internal/llm/provider/provider.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'internal/llm/provider') 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) } } -- cgit v1.2.3