live output from shell tool (#573)

1 files changed, 10 insertions, 2 deletions

diff --git a/packages/server/src/tools/tools.ts b/packages/server/src/tools/tools.ts
index 2f5a4095..c57bbd39 100644
--- a/packages/server/src/tools/tools.ts
+++ b/packages/server/src/tools/tools.ts
@@ -64,7 +64,11 @@ export interface Tool<
    * @param params Parameters for the tool execution
    * @returns Result of the tool execution
    */
-  execute(params: TParams, signal: AbortSignal): Promise<TResult>;
+  execute(
+    params: TParams,
+    signal: AbortSignal,
+    onOutputChunk?: (chunk: string) => void,
+  ): Promise<TResult>;
 }
 
 /**
@@ -144,7 +148,11 @@ export abstract class BaseTool<
    * @param signal AbortSignal for tool cancellation
    * @returns Result of the tool execution
    */
-  abstract execute(params: TParams, signal: AbortSignal): Promise<TResult>;
+  abstract execute(
+    params: TParams,
+    signal: AbortSignal,
+    onOutputChunk?: (chunk: string) => void,
+  ): Promise<TResult>;
 }
 
 export interface ToolResult {