From 09d4acce4c27bd301e7f7a3b283441c1a48dc535 Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Mon, 29 Dec 2025 11:02:02 +0700
Subject: [PATCH] Clean up Claude agent SDK message conversion
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
---
 .../server/agent/providers/claude-agent.ts    | 40 +++++++++----------
 1 file changed, 19 insertions(+), 21 deletions(-)

diff --git a/packages/server/src/server/agent/providers/claude-agent.ts b/packages/server/src/server/agent/providers/claude-agent.ts
index 101094cf5..a71ec82b8 100644
--- a/packages/server/src/server/agent/providers/claude-agent.ts
+++ b/packages/server/src/server/agent/providers/claude-agent.ts
@@ -29,7 +29,6 @@ import type {
   AgentPermissionResponse,
   AgentPermissionUpdate,
   AgentPersistenceHandle,
-  AgentPromptContentBlock,
   AgentPromptInput,
   AgentRunOptions,
   AgentRunResult,
@@ -788,26 +787,25 @@ class ClaudeAgentSession implements AgentSession {
   }
 
   private toSdkUserMessage(prompt: AgentPromptInput): SDKUserMessage {
-    const content = Array.isArray(prompt)
-      ? prompt.flatMap((chunk: AgentPromptContentBlock) => {
-          if (chunk.type === "text") {
-            return [{ type: "text", text: chunk.text }];
-          }
-          if (chunk.type === "image") {
-            return [
-              {
-                type: "image",
-                source: {
-                  type: "base64",
-                  media_type: chunk.mimeType,
-                  data: chunk.data,
-                },
-              },
-            ];
-          }
-          return [];
-        })
-      : [{ type: "text", text: prompt }];
+    const content: Array<{ type: "text"; text: string } | { type: "image"; source: { type: "base64"; media_type: string; data: string } }> = [];
+    if (Array.isArray(prompt)) {
+      for (const chunk of prompt) {
+        if (chunk.type === "text") {
+          content.push({ type: "text", text: chunk.text });
+        } else if (chunk.type === "image") {
+          content.push({
+            type: "image",
+            source: {
+              type: "base64",
+              media_type: chunk.mimeType,
+              data: chunk.data,
+            },
+          });
+        }
+      }
+    } else {
+      content.push({ type: "text", text: prompt });
+    }
 
     return {
       type: "user",