@@ -379,6 +379,9 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
380 a.updateSessionUsage(largeModel, &updatedSession, stepResult.Usage, a.openrouterCost(stepResult.ProviderMetadata))
380 a.updateSessionUsage(largeModel, &updatedSession, stepResult.Usage, a.openrouterCost(stepResult.ProviderMetadata))
381 _, sessionErr := a.sessions.Save(genCtx, updatedSession)
381 _, sessionErr := a.sessions.Save(genCtx, updatedSession)
382+ if sessionErr == nil {
383+ currentSession = updatedSession
382 sessionLock.Unlock()
385 sessionLock.Unlock()
383 if sessionErr != nil {
386 if sessionErr != nil {
384 return sessionErr
387 return sessionErr