From 23c4117d6f7def21bbd07dcbd2f5bcee88e435a0 Mon Sep 17 00:00:00 2001
From: suyao <sy20010504@gmail.com>
Date: Mon, 22 Sep 2025 16:01:58 +0800
Subject: [PATCH] get toolName

---
 .../agents/services/claudecode/transform.ts   | 36 +++++++++++++------
 1 file changed, 25 insertions(+), 11 deletions(-)

diff --git a/src/main/services/agents/services/claudecode/transform.ts b/src/main/services/agents/services/claudecode/transform.ts
index 61ab891727..af393762d3 100644
--- a/src/main/services/agents/services/claudecode/transform.ts
+++ b/src/main/services/agents/services/claudecode/transform.ts
@@ -13,15 +13,20 @@ const logger = loggerService.withContext('ClaudeCodeTransform')
 
 type AgentStreamPart = TextStreamPart<Record<string, any>>
 
-const contentBlockState = new Map<
-  string,
-  {
-    type: 'text' | 'tool-call'
-    toolCallId?: string
-    toolName?: string
-    input?: string
-  }
->()
+type contentBlock =
+  | {
+      type: 'text'
+    }
+  | {
+      type: 'tool-call'
+      toolCallId: string
+      toolName: string
+      input?: string
+    }
+
+const contentBlockState = new Map<string, contentBlock>()
+
+type toolCallBlock = Extract<contentBlock, { type: 'tool-call' }>
 
 // Helper function to generate unique IDs for text blocks
 const generateMessageId = (): string => `msg_${uuidv4().replace(/-/g, '')}`
@@ -114,16 +119,25 @@ function handleUserOrAssistantMessage(message: Extract<SDKMessage, { type: 'assi
             providerExecuted: true,
             providerMetadata: sdkMessageToProviderMetadata(message)
           })
+          contentBlockState.set(block.id, {
+            type: 'tool-call',
+            toolCallId: block.id,
+            toolName: block.name
+          })
           break
-        case 'tool_result':
+        case 'tool_result': {
+          logger.debug('Handling tool result:', { block })
+          logger.debug('contentblockState', { content: contentBlockState })
+          const toolCall = contentBlockState.get(block.tool_use_id) as toolCallBlock
           chunks.push({
             type: 'tool-result',
             toolCallId: block.tool_use_id,
-            toolName: contentBlockState[block.tool_use_id].toolName,
+            toolName: contentBlockState.has(block.tool_use_id) ? toolCall.toolName : 'Unknown',
             input: '',
             output: block.content
           })
           break
+        }
         default:
           logger.warn('Unknown content block type in user/assistant message:', {
             type: block.type