feat(vscode-ide-companion): add specialized tool call components

- Added ExecuteNodeToolCall for specialized node/npm command execution
- Added UpdatedPlanToolCall for enhanced plan visualization with checkboxes
- Created ExecuteNode CSS styling
- Refactored ReadToolCall to new directory structure
- Updated tool call router to handle new component types
- Enhanced LayoutComponents with className prop support
- Added specialized handling for todo_write and updated_plan tool kinds

These additions improve the visualization and handling of various tool call types in the UI.
This commit is contained in:
yiliang114
2025-12-04 08:29:07 +08:00
parent 3053e6c41f
commit 5dec3e653c
7 changed files with 376 additions and 9 deletions

View File

@@ -10,11 +10,13 @@ import type React from 'react';
import type { BaseToolCallProps } from './shared/types.js';
import { shouldShowToolCall } from './shared/utils.js';
import { GenericToolCall } from './GenericToolCall.js';
import { ReadToolCall } from './ReadToolCall.js';
import { ReadToolCall } from './Read/ReadToolCall.js';
import { WriteToolCall } from './WriteToolCall.js';
import { EditToolCall } from './Edit/EditToolCall.js';
import { ExecuteToolCall as BashExecuteToolCall } from './Bash/Bash.js';
import { ExecuteToolCall } from './Execute/Execute.js';
import { UpdatedPlanToolCall } from './UpdatedPlan/UpdatedPlanToolCall.js';
import { ExecuteNodeToolCall } from './ExecuteNode/ExecuteNodeToolCall.js';
import { SearchToolCall } from './SearchToolCall.js';
import { ThinkToolCall } from './ThinkToolCall.js';
import { TodoWriteToolCall } from './TodoWriteToolCall.js';
@@ -24,6 +26,7 @@ import { TodoWriteToolCall } from './TodoWriteToolCall.js';
*/
export const getToolCallComponent = (
kind: string,
toolCall?: import('./shared/types.js').ToolCallData,
): React.FC<BaseToolCallProps> => {
const normalizedKind = kind.toLowerCase();
@@ -39,12 +42,35 @@ export const getToolCallComponent = (
return EditToolCall;
case 'execute':
// Check if this is a node/npm version check command
if (toolCall) {
const commandText =
typeof toolCall.rawInput === 'string'
? toolCall.rawInput
: typeof toolCall.rawInput === 'object' &&
toolCall.rawInput !== null
? (toolCall.rawInput as { command?: string }).command || ''
: '';
// TODO:
if (
commandText.includes('node --version') ||
commandText.includes('npm --version')
) {
return ExecuteNodeToolCall;
}
}
return ExecuteToolCall;
case 'bash':
case 'command':
return BashExecuteToolCall;
case 'updated_plan':
case 'updatedplan':
case 'todo_write':
return UpdatedPlanToolCall;
case 'search':
case 'grep':
case 'glob':
@@ -56,7 +82,8 @@ export const getToolCallComponent = (
return ThinkToolCall;
case 'todowrite':
case 'todo_write':
return TodoWriteToolCall;
// case 'todo_write':
case 'update_todos':
return TodoWriteToolCall;
@@ -82,7 +109,7 @@ export const ToolCallRouter: React.FC<BaseToolCallProps> = ({ toolCall }) => {
}
// Get the appropriate component for this kind
const Component = getToolCallComponent(toolCall.kind);
const Component = getToolCallComponent(toolCall.kind, toolCall);
// Render the specialized component
return <Component toolCall={toolCall} />;