feat: provide messages to tools (#26)

callstackincubator · Dec 7, 2024 · 4378a18 · 4378a18
1 parent 9217c83
commit 4378a18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/framework/src/executor.ts b/packages/framework/src/executor.ts
@@ -75,6 +75,7 @@ export async function executeTaskWithAgent(
 
           const content = await tool.execute(toolCall.function.parsed_arguments, {
             provider: agent.provider,
+            messages,
           })
           return {
             role: 'tool' as const,

diff --git a/packages/framework/src/tool.ts b/packages/framework/src/tool.ts
@@ -1,11 +1,15 @@
 import z, { ZodTypeAny } from 'zod'
 
 import { Provider } from './models/openai.js'
+import { Message } from './types.js'
 
 export type Tool<P extends ZodTypeAny = any> = {
   description: string
   parameters: P
-  execute: (parameters: z.infer<P>, context: { provider: Provider }) => Promise<string>
+  execute: (
+    parameters: z.infer<P>,
+    context: { provider: Provider; messages: Message[] }
+  ) => Promise<string>
 }
 
 /**