From e2e39202185f674473fda114132baae9761c55dc Mon Sep 17 00:00:00 2001
From: kujtimiihoxha <kujtimii.h@gmail.com>
Date: Fri, 10 Oct 2025 06:58:35 +0200
Subject: [PATCH] fix(agent): save assistant message

---
 internal/agent/agent.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internal/agent/agent.go b/internal/agent/agent.go
index 8ab2b8010bca4fef73c02f904bbbc80fdf05b339..3b2f39fc11ad17a2af2e1e51f3275073734ad589 100644
--- a/internal/agent/agent.go
+++ b/internal/agent/agent.go
@@ -359,6 +359,10 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*ai.Agen
 				tc.Finished = true
 				tc.Input = "{}"
 				currentAssistant.AddToolCall(tc)
+				updateErr := a.messages.Update(ctx, *currentAssistant)
+				if updateErr != nil {
+					return nil, updateErr
+				}
 			}
 
 			found := false