From 3f6b50f8b9dc987c2b7a5bafd1a4862ba73daf2d Mon Sep 17 00:00:00 2001 From: Carlos Alexandro Becker Date: Tue, 27 Jan 2026 16:38:29 -0300 Subject: [PATCH] fix: simplify Signed-off-by: Carlos Alexandro Becker --- internal/agent/coordinator.go | 4 ++-- internal/agent/coordinator_test.go | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/internal/agent/coordinator.go b/internal/agent/coordinator.go index d51d90a0349ce58d6e397c93af354e241a9bb71f..71fd3b3375898aa8820bb9ebe70ae27eae494baf 100644 --- a/internal/agent/coordinator.go +++ b/internal/agent/coordinator.go @@ -851,17 +851,17 @@ func (c *coordinator) RecoverSession(ctx context.Context, sessionID string) erro } for _, msg := range msgs { - if msg.IsFinished() || msg.Role == message.Assistant { + if msg.IsFinished() { continue } + msg.FinishThinking() for _, tc := range msg.ToolCalls() { if !tc.Finished { msg.FinishToolCall(tc.ID) } } - msg.FinishThinking() msg.AddFinish(message.FinishReasonError, "Session interrupted", "The session was previously interrupted") if updateErr := c.messages.Update(ctx, msg); updateErr != nil { slog.Error("Failed to recover message", "message_id", msg.ID, "error", updateErr) diff --git a/internal/agent/coordinator_test.go b/internal/agent/coordinator_test.go index 564a3ac8e908294c7e2d061a5972a5d896e2e08b..5464a420fdd78e36f119cf5ccafd925e58ae0211 100644 --- a/internal/agent/coordinator_test.go +++ b/internal/agent/coordinator_test.go @@ -93,7 +93,7 @@ func TestRecoverSession(t *testing.T) { require.NoError(t, err) require.True(t, recoveredMsg.IsFinished()) require.Equal(t, message.FinishReasonError, recoveredMsg.FinishReason()) - require.Contains(t, recoveredMsg.FinishPart().Message, "Summarization interrupted") + require.Contains(t, recoveredMsg.FinishPart().Message, "Session interrupted") }) t.Run("incomplete assistant message with tool calls", func(t *testing.T) { @@ -136,7 +136,7 @@ func TestRecoverSession(t *testing.T) { require.NoError(t, err) require.True(t, recoveredMsg.IsFinished()) require.Equal(t, message.FinishReasonError, recoveredMsg.FinishReason()) - require.Contains(t, recoveredMsg.FinishPart().Message, "Response interrupted") + require.Contains(t, recoveredMsg.FinishPart().Message, "Session interrupted") // Verify the tool call was marked as finished toolCalls := recoveredMsg.ToolCalls() @@ -175,7 +175,7 @@ func TestRecoverSession(t *testing.T) { require.NoError(t, err) require.True(t, recoveredMsg.IsFinished()) require.Equal(t, message.FinishReasonError, recoveredMsg.FinishReason()) - require.Contains(t, recoveredMsg.FinishPart().Message, "Response interrupted") + require.Contains(t, recoveredMsg.FinishPart().Message, "Session interrupted") require.Equal(t, "This is a partial response...", recoveredMsg.Content().Text) })