feat(tool): add execute options

packages/generate-text/src/index.ts

@@ -43,7 +43,7 @@
 
 export interface StepResult {
   text?: string
   // TODO: step type
   // type: 'continue' | 'initial' | 'tool-result'
   toolCalls: ToolCall[]
   toolResults: ToolResult[]
@@ -117,7 +117,7 @@
       } of message.tool_calls) {
         const tool = (options.tools as Tool[]).find(tool => tool.function.name === toolName)!
         const parsedArgs = JSON.parse(toolArgs) as Record<string, unknown>
-        const result = await tool.execute(parsedArgs)
+        const result = await tool.execute(parsedArgs, { abortSignal: options.abortSignal, messages, toolCallId })
 
         toolCalls.push({
           args: toolArgs,

packages/shared-chat/src/types/tool.ts

@@ -1,5 +1,7 @@
+import type { Message } from './message'
+
 export interface Tool {
-  execute: (input: unknown) => Promise<string> | string
+  execute: (input: unknown, options: ToolExecuteOptions) => Promise<string> | string
   function: {
     description?: string
     name: string
@@ -8,3 +10,9 @@ export interface Tool {
   }
   type: 'function'
 }
+
+interface ToolExecuteOptions {
+  abortSignal?: AbortSignal
+  messages: Message[]
+  toolCallId: string
+}

packages/tool/src/generate-text.ts

@@ -1,3 +1,5 @@
+/// <reference types="@xsai/generate-text" />
+
 import type { Schema } from '@typeschema/main'
 
 import type { ToolResult } from '.'