diff options
Diffstat (limited to 'packages/cli/src')
| -rw-r--r-- | packages/cli/src/ui/hooks/useReactToolScheduler.ts | 6 | ||||
| -rw-r--r-- | packages/cli/src/ui/hooks/useToolScheduler.test.ts | 13 |
2 files changed, 5 insertions, 14 deletions
diff --git a/packages/cli/src/ui/hooks/useReactToolScheduler.ts b/packages/cli/src/ui/hooks/useReactToolScheduler.ts index ae58ed38..8dbcfb87 100644 --- a/packages/cli/src/ui/hooks/useReactToolScheduler.ts +++ b/packages/cli/src/ui/hooks/useReactToolScheduler.ts @@ -249,10 +249,8 @@ export function mapToDisplay( trackedCall.request.args, ); renderOutputAsMarkdown = currentToolInstance.isOutputMarkdown; - } - - if (trackedCall.status === 'error') { - description = ''; + } else if ('request' in trackedCall && 'args' in trackedCall.request) { + description = JSON.stringify(trackedCall.request.args); } const baseDisplayProperties: Omit< diff --git a/packages/cli/src/ui/hooks/useToolScheduler.test.ts b/packages/cli/src/ui/hooks/useToolScheduler.test.ts index f5a3529c..8e3f139b 100644 --- a/packages/cli/src/ui/hooks/useToolScheduler.test.ts +++ b/packages/cli/src/ui/hooks/useToolScheduler.test.ts @@ -924,7 +924,7 @@ describe('mapToDisplay', () => { expectedStatus: ToolCallStatus.Error, expectedResultDisplay: 'Error display tool not found', expectedName: baseRequest.name, - expectedDescription: '', + expectedDescription: JSON.stringify(baseRequest.args), }, { name: 'error tool execution failed', @@ -940,7 +940,7 @@ describe('mapToDisplay', () => { expectedStatus: ToolCallStatus.Error, expectedResultDisplay: 'Execution failed display', expectedName: baseTool.displayName, // Changed from baseTool.name - expectedDescription: '', + expectedDescription: baseTool.getDescription(baseRequest.args), }, { name: 'cancelled', @@ -986,14 +986,7 @@ describe('mapToDisplay', () => { expect(toolDisplay.resultDisplay).toBe(expectedResultDisplay); expect(toolDisplay.name).toBe(expectedName); - - if (status === 'error' && !extraProps?.tool) { - expect(toolDisplay.description).toBe(''); - } else { - expect(toolDisplay.description).toBe( - expectedDescription ?? baseTool.getDescription(baseRequest.args), - ); - } + expect(toolDisplay.description).toBe(expectedDescription); expect(toolDisplay.renderOutputAsMarkdown).toBe( extraProps?.tool?.isOutputMarkdown ?? false, |
