kernl 0.11.2 → 0.12.0

package/src/tool/__tests__/tool.test.ts

@@ -186,7 +186,7 @@ describe("FunctionTool", () => {
   describe("isEnabled", () => {
     it("should default to enabled", async () => {
       const ctx = mockContext();
-      const enabled = await simpleStringTool.isEnabled(ctx, null as any);
+      const enabled = await simpleStringTool.isEnabled(ctx);
 
       expect(enabled).toBe(true);
     });
@@ -201,7 +201,7 @@ describe("FunctionTool", () => {
       });
 
       const ctx = mockContext();
-      const enabled = await disabledTool.isEnabled(ctx, null as any);
+      const enabled = await disabledTool.isEnabled(ctx);
 
       expect(enabled).toBe(false);
     });
@@ -224,12 +224,8 @@ describe("FunctionTool", () => {
       const enabledCtx = mockContext<MyContext>({ enabled: true });
       const disabledCtx = mockContext<MyContext>({ enabled: false });
 
-      expect(await conditionalTool.isEnabled(enabledCtx, null as any)).toBe(
-        true,
-      );
-      expect(await conditionalTool.isEnabled(disabledCtx, null as any)).toBe(
-        false,
-      );
+      expect(await conditionalTool.isEnabled(enabledCtx)).toBe(true);
+      expect(await conditionalTool.isEnabled(disabledCtx)).toBe(false);
     });
   });
 });

package/src/tool/tool.ts

@@ -62,7 +62,7 @@ export abstract class BaseTool<TContext = UnknownContext> {
   /**
    * Determines whether the tool should be exposed to the model for the current run.
    */
-  abstract isEnabled(context: Context<TContext>, agent: BaseAgent<TContext>): Promise<boolean>;
+  abstract isEnabled(context: Context<TContext>): Promise<boolean>;
 
   /**
    * Serialize this tool for sending to the model
@@ -117,10 +117,10 @@ export class FunctionTool<
     // setup enabled function
     this.isEnabled =
       typeof config.isEnabled === "function"
-        ? async (context, agent) => {
+        ? async (context) => {
             const predicate =
               config.isEnabled as ToolEnabledPredicate<TContext>;
-            const result = await predicate({ context, agent });
+            const result = await predicate({ context });
             return Boolean(result);
           }
         : async () =>

package/src/tool/toolkit.ts

@@ -30,15 +30,17 @@ export abstract class BaseToolkit<TContext = UnknownContext> {
   abstract readonly description: string;
 
   /**
-   * The agent this toolkit is bound to (if any)
+   * The agent this toolkit is bound to (if any).
+   * Uses `any` to allow toolkits with different context types
+   * to be composed in the same agent.
    */
-  protected agent?: BaseAgent<TContext>;
+  protected agent?: BaseAgent<any>;
 
   /**
    * Bind this toolkit to an agent.
    * Called by agent constructor.
    */
-  bind(agent: BaseAgent<TContext>): void {
+  bind(agent: BaseAgent<any>): void {
     this.agent = agent;
   }
 

package/src/tool/types.ts

@@ -227,12 +227,10 @@ export type ToolApprovalFunction<TParameters extends ToolInputParameters> = (
 
 export type ToolEnabledFunction<TContext = UnknownContext> = (
   context: Context<TContext>,
-  agent: BaseAgent<TContext>,
 ) => Promise<boolean>;
 
 export type ToolEnabledPredicate<TContext = UnknownContext> = (args: {
   context: Context<TContext>;
-  agent: BaseAgent<TContext>;
 }) => boolean | Promise<boolean>;
 
 type ToolEnabledOption<Context = UnknownContext> =