diff options
Diffstat (limited to 'packages/plugin/src')
| -rw-r--r-- | packages/plugin/src/tool.ts | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/packages/plugin/src/tool.ts b/packages/plugin/src/tool.ts index b568d0371..3105bf534 100644 --- a/packages/plugin/src/tool.ts +++ b/packages/plugin/src/tool.ts @@ -27,10 +27,12 @@ type AskInput = { metadata: { [key: string]: any } } +export type ToolResult = string | { output: string; metadata?: { [key: string]: any } } + export function tool<Args extends z.ZodRawShape>(input: { description: string args: Args - execute(args: z.infer<z.ZodObject<Args>>, context: ToolContext): Promise<string> + execute(args: z.infer<z.ZodObject<Args>>, context: ToolContext): Promise<ToolResult> }) { return input } |
