From 634c478e18e8e3202d3bca62e7d1dd86b9d3e81b Mon Sep 17 00:00:00 2001
From: suyao <sy20010504@gmail.com>
Date: Mon, 22 Sep 2025 16:11:19 +0800
Subject: [PATCH] Update transform.ts

---
 .../services/agents/services/claudecode/transform.ts   | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/main/services/agents/services/claudecode/transform.ts b/src/main/services/agents/services/claudecode/transform.ts
index af393762d3..8c0d429a14 100644
--- a/src/main/services/agents/services/claudecode/transform.ts
+++ b/src/main/services/agents/services/claudecode/transform.ts
@@ -21,7 +21,7 @@ type contentBlock =
       type: 'tool-call'
       toolCallId: string
       toolName: string
-      input?: string
+      input: unknown
     }
 
 const contentBlockState = new Map<string, contentBlock>()
@@ -122,18 +122,20 @@ function handleUserOrAssistantMessage(message: Extract<SDKMessage, { type: 'assi
           contentBlockState.set(block.id, {
             type: 'tool-call',
             toolCallId: block.id,
-            toolName: block.name
+            toolName: block.name,
+            input: block.input
           })
           break
         case 'tool_result': {
           logger.debug('Handling tool result:', { block })
           logger.debug('contentblockState', { content: contentBlockState })
+          const hasToolCall = contentBlockState.has(block.tool_use_id)
           const toolCall = contentBlockState.get(block.tool_use_id) as toolCallBlock
           chunks.push({
             type: 'tool-result',
             toolCallId: block.tool_use_id,
-            toolName: contentBlockState.has(block.tool_use_id) ? toolCall.toolName : 'Unknown',
-            input: '',
+            toolName: hasToolCall ? toolCall.toolName : 'Unknown',
+            input: hasToolCall ? toolCall.input : '',
             output: block.content
           })
           break