@agentforge/core 0.16.25 → 0.16.27

package/dist/index.cjs

@@ -2244,6 +2244,12 @@ function cache(tool, ttl) {
 }
 
 // src/tools/testing.ts
+function matchesMockInput(matcher, input) {
+  if (typeof matcher === "function") {
+    return matcher(input);
+  }
+  return JSON.stringify(matcher) === JSON.stringify(input);
+}
 function createMockTool(config) {
   const {
     name,
@@ -2273,12 +2279,7 @@ function createMockTool(config) {
         });
         throw error2;
       }
-      const matchingResponse = responses.find((r) => {
-        if (typeof r.input === "function") {
-          return r.input(input);
-        }
-        return JSON.stringify(r.input) === JSON.stringify(input);
-      });
+      const matchingResponse = responses.find((response) => matchesMockInput(response.input, input));
       if (matchingResponse) {
         if (matchingResponse.error) {
           invocations.push({
@@ -2382,7 +2383,8 @@ function createToolSimulator(config) {
       }
     },
     getInvocations: (toolName) => {
-      return invocations.get(toolName) ? [...invocations.get(toolName)] : [];
+      const toolInvocations = invocations.get(toolName) ?? [];
+      return [...toolInvocations];
     },
     getAllInvocations: () => {
       const all = {};

package/dist/index.d.cts

@@ -1748,34 +1748,48 @@ declare function cache<TInput = unknown, TOutput = unknown>(tool: ComposedTool<T
  * Tool Mocking & Testing - Mock tools for testing
  * @module tools/testing
  */
-interface MockToolResponse {
-    input: any;
-    output?: any;
-    error?: Error;
-}
-interface MockToolConfig {
-    name: string;
+type MockToolMatcher<TInput> = TInput | ((input: TInput) => boolean);
+interface MockToolSuccessResponse<TInput, TOutput> {
+    input: MockToolMatcher<TInput>;
+    output: TOutput;
+    error?: never;
+}
+interface MockToolErrorResponse<TInput> {
+    input: MockToolMatcher<TInput>;
+    output?: never;
+    error: Error;
+}
+type MockToolResponse<TInput = unknown, TOutput = unknown> = MockToolSuccessResponse<TInput, TOutput> | MockToolErrorResponse<TInput>;
+interface MockToolConfig<TName extends string = string, TInput = unknown, TOutput = unknown> {
+    name: TName;
     description?: string;
-    responses?: MockToolResponse[];
-    defaultResponse?: any;
+    responses?: MockToolResponse<TInput, TOutput>[];
+    defaultResponse?: TOutput;
     latency?: {
         min: number;
         max: number;
     } | number;
     errorRate?: number;
 }
-interface ToolInvocation {
-    input: any;
-    output?: any;
+interface ToolInvocation<TInput = unknown, TOutput = unknown> {
+    input: TInput;
+    output?: TOutput;
     error?: Error;
     timestamp: number;
     duration: number;
 }
-interface ToolSimulatorConfig {
-    tools: Array<{
-        name: string;
-        invoke: (input: any) => Promise<any>;
-    }>;
+interface SimulatedTool<TName extends string = string, TInput = unknown, TOutput = unknown> {
+    name: TName;
+    invoke(input: TInput): Promise<TOutput>;
+}
+type ToolName<TTools extends readonly SimulatedTool[]> = TTools[number]['name'] & string;
+type ToolByName<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = Extract<TTools[number], {
+    name: TName;
+}>;
+type ToolInputFor<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = ToolByName<TTools, TName> extends SimulatedTool<string, infer TInput, unknown> ? TInput : never;
+type ToolOutputFor<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = ToolByName<TTools, TName> extends SimulatedTool<string, unknown, infer TOutput> ? TOutput : never;
+interface ToolSimulatorConfig<TTools extends readonly SimulatedTool[] = readonly SimulatedTool[]> {
+    tools: TTools;
     errorRate?: number;
     latency?: {
         mean: number;
@@ -1783,24 +1797,23 @@ interface ToolSimulatorConfig {
     };
     recordInvocations?: boolean;
 }
+interface MockTool<TName extends string = string, TInput = unknown, TOutput = unknown> extends SimulatedTool<TName, TInput, TOutput> {
+    description: string;
+    getInvocations: () => ToolInvocation<TInput, TOutput>[];
+    clearInvocations: () => void;
+}
 /**
  * Create a mock tool for testing
  */
-declare function createMockTool(config: MockToolConfig): {
-    name: string;
-    description: string;
-    invoke: (input: any) => Promise<any>;
-    getInvocations: () => ToolInvocation[];
-    clearInvocations: () => void;
-};
+declare function createMockTool<TName extends string, TInput = unknown, TOutput = unknown>(config: MockToolConfig<TName, TInput, TOutput>): MockTool<TName, TInput, TOutput>;
 /**
  * Create a tool simulator for testing
  */
-declare function createToolSimulator(config: ToolSimulatorConfig): {
-    execute: (toolName: string, input: any) => Promise<any>;
-    getInvocations: (toolName: string) => ToolInvocation[];
-    getAllInvocations: () => Record<string, ToolInvocation[]>;
-    clearInvocations: (toolName?: string) => void;
+declare function createToolSimulator<const TTools extends readonly SimulatedTool[]>(config: ToolSimulatorConfig<TTools>): {
+    execute: <TName extends ToolName<TTools>>(toolName: TName, input: ToolInputFor<TTools, TName>) => Promise<ToolOutputFor<TTools, TName>>;
+    getInvocations: <TName extends ToolName<TTools>>(toolName: TName) => ToolInvocation<ToolInputFor<TTools, TName>, ToolOutputFor<TTools, TName>>[];
+    getAllInvocations: () => Partial<Record<TTools[number]["name"], ToolInvocation<unknown, unknown>[]>>;
+    clearInvocations: (toolName?: ToolName<TTools>) => void;
 };
 
 /**
@@ -4901,7 +4914,7 @@ interface InterruptEventData {
 interface ResumeEventData {
     type: 'resume';
     interruptId: string;
-    value: any;
+    value: InterruptPayload;
     threadId: string;
 }
 /**
@@ -4963,7 +4976,7 @@ declare function formatInterruptEvent(interrupt: AnyInterrupt, threadId: string)
  * @param threadId - The thread ID
  * @returns An SSE event
  */
-declare function formatResumeEvent(interruptId: string, value: any, threadId: string): SSEEvent;
+declare function formatResumeEvent(interruptId: string, value: InterruptPayload, threadId: string): SSEEvent;
 /**
  * Format an agent waiting event as an SSE event
  *

package/dist/index.d.ts

@@ -1748,34 +1748,48 @@ declare function cache<TInput = unknown, TOutput = unknown>(tool: ComposedTool<T
  * Tool Mocking & Testing - Mock tools for testing
  * @module tools/testing
  */
-interface MockToolResponse {
-    input: any;
-    output?: any;
-    error?: Error;
-}
-interface MockToolConfig {
-    name: string;
+type MockToolMatcher<TInput> = TInput | ((input: TInput) => boolean);
+interface MockToolSuccessResponse<TInput, TOutput> {
+    input: MockToolMatcher<TInput>;
+    output: TOutput;
+    error?: never;
+}
+interface MockToolErrorResponse<TInput> {
+    input: MockToolMatcher<TInput>;
+    output?: never;
+    error: Error;
+}
+type MockToolResponse<TInput = unknown, TOutput = unknown> = MockToolSuccessResponse<TInput, TOutput> | MockToolErrorResponse<TInput>;
+interface MockToolConfig<TName extends string = string, TInput = unknown, TOutput = unknown> {
+    name: TName;
     description?: string;
-    responses?: MockToolResponse[];
-    defaultResponse?: any;
+    responses?: MockToolResponse<TInput, TOutput>[];
+    defaultResponse?: TOutput;
     latency?: {
         min: number;
         max: number;
     } | number;
     errorRate?: number;
 }
-interface ToolInvocation {
-    input: any;
-    output?: any;
+interface ToolInvocation<TInput = unknown, TOutput = unknown> {
+    input: TInput;
+    output?: TOutput;
     error?: Error;
     timestamp: number;
     duration: number;
 }
-interface ToolSimulatorConfig {
-    tools: Array<{
-        name: string;
-        invoke: (input: any) => Promise<any>;
-    }>;
+interface SimulatedTool<TName extends string = string, TInput = unknown, TOutput = unknown> {
+    name: TName;
+    invoke(input: TInput): Promise<TOutput>;
+}
+type ToolName<TTools extends readonly SimulatedTool[]> = TTools[number]['name'] & string;
+type ToolByName<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = Extract<TTools[number], {
+    name: TName;
+}>;
+type ToolInputFor<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = ToolByName<TTools, TName> extends SimulatedTool<string, infer TInput, unknown> ? TInput : never;
+type ToolOutputFor<TTools extends readonly SimulatedTool[], TName extends ToolName<TTools>> = ToolByName<TTools, TName> extends SimulatedTool<string, unknown, infer TOutput> ? TOutput : never;
+interface ToolSimulatorConfig<TTools extends readonly SimulatedTool[] = readonly SimulatedTool[]> {
+    tools: TTools;
     errorRate?: number;
     latency?: {
         mean: number;
@@ -1783,24 +1797,23 @@ interface ToolSimulatorConfig {
     };
     recordInvocations?: boolean;
 }
+interface MockTool<TName extends string = string, TInput = unknown, TOutput = unknown> extends SimulatedTool<TName, TInput, TOutput> {
+    description: string;
+    getInvocations: () => ToolInvocation<TInput, TOutput>[];
+    clearInvocations: () => void;
+}
 /**
  * Create a mock tool for testing
  */
-declare function createMockTool(config: MockToolConfig): {
-    name: string;
-    description: string;
-    invoke: (input: any) => Promise<any>;
-    getInvocations: () => ToolInvocation[];
-    clearInvocations: () => void;
-};
+declare function createMockTool<TName extends string, TInput = unknown, TOutput = unknown>(config: MockToolConfig<TName, TInput, TOutput>): MockTool<TName, TInput, TOutput>;
 /**
  * Create a tool simulator for testing
  */
-declare function createToolSimulator(config: ToolSimulatorConfig): {
-    execute: (toolName: string, input: any) => Promise<any>;
-    getInvocations: (toolName: string) => ToolInvocation[];
-    getAllInvocations: () => Record<string, ToolInvocation[]>;
-    clearInvocations: (toolName?: string) => void;
+declare function createToolSimulator<const TTools extends readonly SimulatedTool[]>(config: ToolSimulatorConfig<TTools>): {
+    execute: <TName extends ToolName<TTools>>(toolName: TName, input: ToolInputFor<TTools, TName>) => Promise<ToolOutputFor<TTools, TName>>;
+    getInvocations: <TName extends ToolName<TTools>>(toolName: TName) => ToolInvocation<ToolInputFor<TTools, TName>, ToolOutputFor<TTools, TName>>[];
+    getAllInvocations: () => Partial<Record<TTools[number]["name"], ToolInvocation<unknown, unknown>[]>>;
+    clearInvocations: (toolName?: ToolName<TTools>) => void;
 };
 
 /**
@@ -4901,7 +4914,7 @@ interface InterruptEventData {
 interface ResumeEventData {
     type: 'resume';
     interruptId: string;
-    value: any;
+    value: InterruptPayload;
     threadId: string;
 }
 /**
@@ -4963,7 +4976,7 @@ declare function formatInterruptEvent(interrupt: AnyInterrupt, threadId: string)
  * @param threadId - The thread ID
  * @returns An SSE event
  */
-declare function formatResumeEvent(interruptId: string, value: any, threadId: string): SSEEvent;
+declare function formatResumeEvent(interruptId: string, value: InterruptPayload, threadId: string): SSEEvent;
 /**
  * Format an agent waiting event as an SSE event
  *

package/dist/index.js

@@ -2069,6 +2069,12 @@ function cache(tool, ttl) {
 }
 
 // src/tools/testing.ts
+function matchesMockInput(matcher, input) {
+  if (typeof matcher === "function") {
+    return matcher(input);
+  }
+  return JSON.stringify(matcher) === JSON.stringify(input);
+}
 function createMockTool(config) {
   const {
     name,
@@ -2098,12 +2104,7 @@ function createMockTool(config) {
         });
         throw error2;
       }
-      const matchingResponse = responses.find((r) => {
-        if (typeof r.input === "function") {
-          return r.input(input);
-        }
-        return JSON.stringify(r.input) === JSON.stringify(input);
-      });
+      const matchingResponse = responses.find((response) => matchesMockInput(response.input, input));
       if (matchingResponse) {
         if (matchingResponse.error) {
           invocations.push({
@@ -2207,7 +2208,8 @@ function createToolSimulator(config) {
       }
     },
     getInvocations: (toolName) => {
-      return invocations.get(toolName) ? [...invocations.get(toolName)] : [];
+      const toolInvocations = invocations.get(toolName) ?? [];
+      return [...toolInvocations];
     },
     getAllInvocations: () => {
       const all = {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge/core",
-  "version": "0.16.25",
+  "version": "0.16.27",
   "description": "Production-ready TypeScript agent framework built on LangGraph with orchestration, middleware, and typed abstractions.",
   "type": "module",
   "main": "./dist/index.cjs",