fix(agent): read step data for summarization check (#1787)

Amolith created 2 weeks ago

Change summary

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

Detailed changes

internal/agent/agent.go

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