Add context to tool enabled definitions

Author: coachjackferg

src/agent.ts

@@ -316,7 +316,7 @@ export class MagmaAgent {
                             }),
                             tools: Object.fromEntries(
                                 this.tools
-                                    .filter((t) => t.enabled(this))
+                                    .filter((t) => t.enabled(this, ctx))
                                     .map((t) => [t.name, convertMagmaToolToAISDKTool(t)])
                             ),
                             messages: [...this.getSystemPrompts(ctx), ...completionMessages],
@@ -623,14 +623,12 @@ export class MagmaAgent {
      */
     private async executeTools({
         message,
-        allowList = [],
         trace,
         requestId,
         send,
         ctx,
     }: {
         message: AssistantModelMessage;
-        allowList?: string[];
         trace: TraceEvent[];
         requestId: string;
         send: MagmaSendFunction;
@@ -653,7 +651,7 @@ export class MagmaAgent {
 
                 try {
                     const tool = this.tools
-                        .filter((t) => t.enabled(this) || allowList.includes(t.name))
+                        .filter((t) => t.enabled(this, ctx))
                         .find((t) => t.name === toolCall.toolName);
                     if (!tool)
                         throw new Error(`No tool found to handle call for ${toolCall.toolName}()`);

src/decorators.ts

@@ -21,7 +21,7 @@ export function tool(args?: {
     name?: string;
     description?: string;
     cache?: boolean;
-    enabled?: (agent: MagmaAgent) => boolean;
+    enabled?: (agent: MagmaAgent, ctx: Record<string, any>) => boolean;
 }) {
     return function <R extends MagmaToolReturnType | Promise<MagmaToolReturnType>>(
         target: object,
@@ -32,7 +32,7 @@ export function tool(args?: {
                     name?: string;
                     description?: string;
                     cache?: boolean;
-                    enabled?: (agent: MagmaAgent) => boolean;
+                    enabled?: (agent: MagmaAgent, ctx: Record<string, any>) => boolean;
                 };
             }
         >

src/types/utilities/tools.ts

@@ -54,6 +54,6 @@ export type MagmaTool = {
     description: string;
     params: MagmaToolObjectParam['properties'];
     target: MagmaToolTarget;
-    enabled: (agent: MagmaAgent) => boolean;
+    enabled: (agent: MagmaAgent, ctx: Record<string, any>) => boolean;
     cache?: boolean;
 };