diff options
Diffstat (limited to 'packages/cli/src')
| -rw-r--r-- | packages/cli/src/tools/read-file.tool.ts | 5 | ||||
| -rw-r--r-- | packages/cli/src/ui/hooks/useGeminiStream.ts | 13 |
2 files changed, 13 insertions, 5 deletions
diff --git a/packages/cli/src/tools/read-file.tool.ts b/packages/cli/src/tools/read-file.tool.ts index bbf9eafb..206267be 100644 --- a/packages/cli/src/tools/read-file.tool.ts +++ b/packages/cli/src/tools/read-file.tool.ts @@ -38,8 +38,7 @@ export class ReadFileTool extends BaseTool<ReadFileToolParams, ToolResult> { * Delegates validation to the core logic */ validateToolParams(_params: ReadFileToolParams): string | null { - // Currently allowing any path. Add validation if needed. - return null; + return this.coreLogic.validateToolParams(_params); } /** @@ -62,7 +61,7 @@ export class ReadFileTool extends BaseTool<ReadFileToolParams, ToolResult> { /** * Delegates execution to the core logic */ - async execute(params: ReadFileToolParams): Promise<ToolResult> { + execute(params: ReadFileToolParams): Promise<ToolResult> { return this.coreLogic.execute(params); } } diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts index 6e3e8e84..21a9f508 100644 --- a/packages/cli/src/ui/hooks/useGeminiStream.ts +++ b/packages/cli/src/ui/hooks/useGeminiStream.ts @@ -330,7 +330,13 @@ export const useGeminiStream = ( ...item, tools: item.tools.map((tool) => tool.callId === callId - ? { ...tool, status: ToolCallStatus.Invoked } + ? { + ...tool, + status: + tool.status === ToolCallStatus.Error + ? ToolCallStatus.Error + : ToolCallStatus.Invoked, + } : tool, ), }; @@ -362,7 +368,10 @@ export const useGeminiStream = ( tool.callId === callId ? { ...tool, - status: ToolCallStatus.Success, + status: + tool.status === ToolCallStatus.Error + ? ToolCallStatus.Error + : ToolCallStatus.Success, resultDisplay: result.returnDisplay, } : tool, |