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