fix: preserve plugin tool metadata in execute result (#22827)

Co-authored-by: jquense <[email protected]>
Co-authored-by: Aiden Cline <[email protected]>

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
 }