@@ -379,6 +379,9 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
379379 }
380380 a.updateSessionUsage(largeModel, &updatedSession, stepResult.Usage, a.openrouterCost(stepResult.ProviderMetadata))
381381 _, sessionErr := a.sessions.Save(genCtx, updatedSession)
382+ if sessionErr == nil {
383+ currentSession = updatedSession
384+ }
382385 sessionLock.Unlock()
383386 if sessionErr != nil {
384387 return sessionErr