From 7cf0bfb1bd91312bcf0b3f383cf949123dbad72f Mon Sep 17 00:00:00 2001
From: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>
Date: Mon, 19 Jan 2026 15:48:15 -0300
Subject: [PATCH] updates

Signed-off-by: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>
---
 internal/agent/coordinator.go      | 12 ++++++++++
 internal/agent/coordinator_test.go | 35 ++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/internal/agent/coordinator.go b/internal/agent/coordinator.go
index ba88a961bd286ede2fa308ed2d52d13b86359739..79fce220d0e98bb1ad031349cf97bebd5e86a6be 100644
--- a/internal/agent/coordinator.go
+++ b/internal/agent/coordinator.go
@@ -881,6 +881,18 @@ func (c *coordinator) RecoverSession(ctx context.Context, sessionID string) erro
 			if updateErr := c.messages.Update(ctx, msg); updateErr != nil {
 				slog.Error("Failed to recover assistant message", "message_id", msg.ID, "error", updateErr)
 			}
+			continue
+		}
+
+		// Handle incomplete assistant messages without tool calls
+		// (e.g., message content was partially generated but interrupted)
+		if msg.Role == message.Assistant && msg.Content().Text != "" {
+			msg.FinishThinking()
+			msg.AddFinish(message.FinishReasonError, "Response interrupted", "Session was interrupted during response generation")
+			if updateErr := c.messages.Update(ctx, msg); updateErr != nil {
+				slog.Error("Failed to recover assistant message", "message_id", msg.ID, "error", updateErr)
+			}
+			continue
 		}
 	}
 
diff --git a/internal/agent/coordinator_test.go b/internal/agent/coordinator_test.go
index 5f1d97995ff78457365283e77572720aab177e0e..6bad20f4cacf1c8b96666079dbd20a515cd23fe4 100644
--- a/internal/agent/coordinator_test.go
+++ b/internal/agent/coordinator_test.go
@@ -144,6 +144,41 @@ func TestRecoverSession(t *testing.T) {
 		require.True(t, toolCalls[0].Finished)
 	})
 
+	t.Run("incomplete assistant message without tool calls", func(t *testing.T) {
+		env := testEnv(t)
+
+		sess, err := env.sessions.Create(t.Context(), "Test Session")
+		require.NoError(t, err)
+
+		// Create an incomplete assistant message with partial content but no tool calls
+		assistantMsg, err := env.messages.Create(t.Context(), sess.ID, message.CreateMessageParams{
+			Role:  message.Assistant,
+			Parts: []message.ContentPart{message.TextContent{Text: "This is a partial response..."}},
+			Model: "test-model",
+		})
+		require.NoError(t, err)
+
+		// Verify the message is not finished
+		require.False(t, assistantMsg.IsFinished())
+
+		// Create coordinator with mock services
+		coordinator := &coordinator{
+			sessions: env.sessions,
+			messages: env.messages,
+		}
+
+		err = coordinator.RecoverSession(t.Context(), sess.ID)
+		require.NoError(t, err)
+
+		// Verify the assistant message was recovered
+		recoveredMsg, err := env.messages.Get(t.Context(), assistantMsg.ID)
+		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.Equal(t, "This is a partial response...", recoveredMsg.Content().Text)
+	})
+
 	t.Run("session is busy - skips recovery", func(t *testing.T) {
 		env := testEnv(t)