Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-26 03:31:24 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Update transform.ts

Browse Source
This commit is contained in:
suyao 2025-09-22 16:11:19 +08:00
parent 335bf47dbd
commit 634c478e18
No known key found for this signature in database
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/services/agents/services/claudecode/transform.ts
View File

@ -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