Merge pull request #137 from ai-action/fix/tools

Author: remarkablemark

src/cli.test.ts

@@ -1,3 +1,5 @@
+import type { ToolResult } from './types';
+
 type RunAction = (model: string, prompt: string) => Promise<void>;
 type ResumeAction = (sessionId: string) => Promise<void>;
 
@@ -59,14 +61,11 @@ vi.mock('./utils', () => ({
       const result = (await executeTool(
         toolCall.function.name,
         toolCall.function.arguments,
-      )) as { content: string; error?: string };
+      )) as ToolResult;
       return result;
     },
-    formatToolResultContent: (
-      toolName: string,
-      result: { content: string; error?: string },
-    ) =>
-      `Tool ${toolName} result:\n${result.content}${result.error ? `\nError: ${result.error}` : ''}`,
+    formatToolResultContent: (toolName: string, result: ToolResult) =>
+      `Tool ${toolName} result:\n${result.content}${result.error ? `\nError: ${result.error}` : ''}${result.stack ? `\nStack trace:\n${result.stack}` : ''}`,
   },
 }));
 vi.mock('./tui', () => ({ renderApp }));
@@ -239,6 +238,7 @@ describe('cli', () => {
     executeTool.mockResolvedValueOnce({
       content: 'partial output',
       error: 'shell failed',
+      stack: 'Error: shell failed\n    at runShell',
     });
 
     await commandState.runAction?.('gemma4', 'review diff');
@@ -252,7 +252,7 @@ describe('cli', () => {
         {
           role: 'system',
           content:
-            'Tool run_shell result:\npartial output\nError: shell failed',
+            'Tool run_shell result:\npartial output\nError: shell failed\nStack trace:\nError: shell failed\n    at runShell',
         },
       ],
       'gemma4',