@@ -359,6 +359,10 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*ai.Agen
359 tc.Finished = true
359 tc.Finished = true
360 tc.Input = "{}"
360 tc.Input = "{}"
361 currentAssistant.AddToolCall(tc)
361 currentAssistant.AddToolCall(tc)
362+ updateErr := a.messages.Update(ctx, *currentAssistant)
363+ if updateErr != nil {
364+ return nil, updateErr
364 found := false
368 found := false