@agentforge/patterns 0.5.4 → 0.6.0

package/dist/index.cjs

@@ -1891,6 +1891,33 @@ var MultiAgentState = (0, import_core6.createStateAnnotation)(MultiAgentStateCon
 
 // src/multi-agent/routing.ts
 var import_messages4 = require("@langchain/core/messages");
+async function executeTools(toolCalls, tools) {
+  const results = [];
+  for (const toolCall of toolCalls) {
+    const tool = tools.find((t) => t.metadata.name === toolCall.name);
+    if (!tool) {
+      results.push(new import_messages4.ToolMessage({
+        content: `Error: Tool '${toolCall.name}' not found`,
+        tool_call_id: toolCall.id
+      }));
+      continue;
+    }
+    try {
+      const result = await tool.execute(toolCall.args);
+      const content = typeof result === "string" ? result : JSON.stringify(result);
+      results.push(new import_messages4.ToolMessage({
+        content,
+        tool_call_id: toolCall.id
+      }));
+    } catch (error) {
+      results.push(new import_messages4.ToolMessage({
+        content: `Error executing tool: ${error.message}`,
+        tool_call_id: toolCall.id
+      }));
+    }
+  }
+  return results;
+}
 var DEFAULT_SUPERVISOR_SYSTEM_PROMPT = `You are a supervisor agent responsible for routing tasks to specialized worker agents.
 
 Your job is to:
@@ -1913,11 +1940,13 @@ var llmBasedRouting = {
       throw new Error("LLM-based routing requires a model to be configured");
     }
     const systemPrompt = config.systemPrompt || DEFAULT_SUPERVISOR_SYSTEM_PROMPT;
+    const maxRetries = config.maxToolRetries || 3;
+    const tools = config.tools || [];
     const workerInfo = Object.entries(state.workers).map(([id, caps]) => {
       const skills = caps.skills.join(", ");
-      const tools = caps.tools.join(", ");
+      const tools2 = caps.tools.join(", ");
       const available = caps.available ? "available" : "busy";
-      return `- ${id}: Skills: [${skills}], Tools: [${tools}], Status: ${available}, Workload: ${caps.currentWorkload}`;
+      return `- ${id}: Skills: [${skills}], Tools: [${tools2}], Status: ${available}, Workload: ${caps.currentWorkload}`;
     }).join("\n");
     const lastMessage = state.messages[state.messages.length - 1];
     const taskContext = lastMessage?.content || state.input;
@@ -1927,24 +1956,42 @@ Available workers:
 ${workerInfo}
 
 Select the best worker for this task and explain your reasoning.`;
-    const messages = [
-      new import_messages4.SystemMessage(systemPrompt),
-      new import_messages4.HumanMessage(userPrompt)
-    ];
-    const response = await config.model.invoke(messages);
-    const content = typeof response.content === "string" ? response.content : JSON.stringify(response.content);
-    try {
-      const decision = JSON.parse(content);
-      return {
-        targetAgent: decision.targetAgent,
-        reasoning: decision.reasoning,
-        confidence: decision.confidence,
-        strategy: "llm-based",
-        timestamp: Date.now()
-      };
-    } catch (error) {
-      throw new Error(`Failed to parse routing decision from LLM: ${error}`);
+    const conversationHistory = [];
+    let attempt = 0;
+    while (attempt < maxRetries) {
+      const messages = [
+        new import_messages4.SystemMessage(systemPrompt),
+        new import_messages4.HumanMessage(userPrompt),
+        ...conversationHistory
+      ];
+      const response = await config.model.invoke(messages);
+      if (response.tool_calls && response.tool_calls.length > 0) {
+        if (tools.length === 0) {
+          throw new Error("LLM requested tool calls but no tools are configured");
+        }
+        const toolResults = await executeTools(response.tool_calls, tools);
+        conversationHistory.push(
+          new import_messages4.AIMessage({ content: response.content || "", tool_calls: response.tool_calls }),
+          ...toolResults
+        );
+        attempt++;
+        continue;
+      }
+      const content = typeof response.content === "string" ? response.content : JSON.stringify(response.content);
+      try {
+        const decision = JSON.parse(content);
+        return {
+          targetAgent: decision.targetAgent,
+          reasoning: decision.reasoning,
+          confidence: decision.confidence,
+          strategy: "llm-based",
+          timestamp: Date.now()
+        };
+      } catch (error) {
+        throw new Error(`Failed to parse routing decision from LLM: ${error}`);
+      }
     }
+    throw new Error(`Max tool retries (${maxRetries}) exceeded without routing decision`);
   }
 };
 var roundRobinRouting = {
@@ -2419,6 +2466,7 @@ Please synthesize these results into a comprehensive response that addresses the
 
 // src/multi-agent/agent.ts
 var import_langgraph4 = require("@langchain/langgraph");
+var import_core8 = require("@agentforge/core");
 function createMultiAgentSystem(config) {
   const {
     supervisor,
@@ -2429,11 +2477,13 @@ function createMultiAgentSystem(config) {
     checkpointer
   } = config;
   const workflow = new import_langgraph4.StateGraph(MultiAgentState);
-  const supervisorNode = createSupervisorNode({
-    ...supervisor,
-    maxIterations,
-    verbose
-  });
+  let supervisorConfig = { ...supervisor, maxIterations, verbose };
+  if (supervisor.model && supervisor.tools && supervisor.tools.length > 0) {
+    const langchainTools = (0, import_core8.toLangChainTools)(supervisor.tools);
+    const modelWithTools = supervisor.model.bindTools(langchainTools);
+    supervisorConfig.model = modelWithTools;
+  }
+  const supervisorNode = createSupervisorNode(supervisorConfig);
   workflow.addNode("supervisor", supervisorNode);
   const workerIds = [];
   const workerCapabilities = {};

package/dist/index.d.cts

@@ -2411,6 +2411,37 @@ interface SupervisorConfig {
      * Maximum number of routing iterations
      */
     maxIterations?: number;
+    /**
+     * Optional tools the supervisor can use during routing
+     *
+     * Enables the supervisor to gather additional information before making routing decisions.
+     * Common use case: askHuman tool for clarifying ambiguous queries.
+     *
+     * Note: Only works with LLM-based routing strategy.
+     *
+     * @example
+     * ```typescript
+     * import { createAskHumanTool } from '@agentforge/tools';
+     *
+     * const system = createMultiAgentSystem({
+     *   supervisor: {
+     *     strategy: 'llm-based',
+     *     model: chatModel,
+     *     tools: [createAskHumanTool()],
+     *   },
+     *   // ...
+     * });
+     * ```
+     */
+    tools?: Tool<any, any>[];
+    /**
+     * Maximum number of tool call retries before requiring routing decision
+     *
+     * Prevents infinite loops where the supervisor keeps calling tools without making a routing decision.
+     *
+     * @default 3
+     */
+    maxToolRetries?: number;
 }
 /**
  * Configuration for a worker agent node
@@ -2576,6 +2607,8 @@ declare const DEFAULT_SUPERVISOR_SYSTEM_PROMPT = "You are a supervisor agent res
 /**
  * LLM-based routing strategy
  * Uses an LLM to intelligently route tasks based on worker capabilities
+ *
+ * Supports tool calls (e.g., askHuman) for gathering additional information before routing.
  */
 declare const llmBasedRouting: RoutingStrategyImpl;
 /**

package/dist/index.d.ts

@@ -2411,6 +2411,37 @@ interface SupervisorConfig {
      * Maximum number of routing iterations
      */
     maxIterations?: number;
+    /**
+     * Optional tools the supervisor can use during routing
+     *
+     * Enables the supervisor to gather additional information before making routing decisions.
+     * Common use case: askHuman tool for clarifying ambiguous queries.
+     *
+     * Note: Only works with LLM-based routing strategy.
+     *
+     * @example
+     * ```typescript
+     * import { createAskHumanTool } from '@agentforge/tools';
+     *
+     * const system = createMultiAgentSystem({
+     *   supervisor: {
+     *     strategy: 'llm-based',
+     *     model: chatModel,
+     *     tools: [createAskHumanTool()],
+     *   },
+     *   // ...
+     * });
+     * ```
+     */
+    tools?: Tool<any, any>[];
+    /**
+     * Maximum number of tool call retries before requiring routing decision
+     *
+     * Prevents infinite loops where the supervisor keeps calling tools without making a routing decision.
+     *
+     * @default 3
+     */
+    maxToolRetries?: number;
 }
 /**
  * Configuration for a worker agent node
@@ -2576,6 +2607,8 @@ declare const DEFAULT_SUPERVISOR_SYSTEM_PROMPT = "You are a supervisor agent res
 /**
  * LLM-based routing strategy
  * Uses an LLM to intelligently route tasks based on worker capabilities
+ *
+ * Supports tool calls (e.g., askHuman) for gathering additional information before routing.
  */
 declare const llmBasedRouting: RoutingStrategyImpl;
 /**

package/dist/index.js

@@ -1791,7 +1791,34 @@ var MultiAgentStateConfig = {
 var MultiAgentState = createStateAnnotation4(MultiAgentStateConfig);
 
 // src/multi-agent/routing.ts
-import { HumanMessage as HumanMessage4, SystemMessage as SystemMessage4 } from "@langchain/core/messages";
+import { HumanMessage as HumanMessage4, SystemMessage as SystemMessage4, AIMessage as AIMessage2, ToolMessage as ToolMessage2 } from "@langchain/core/messages";
+async function executeTools(toolCalls, tools) {
+  const results = [];
+  for (const toolCall of toolCalls) {
+    const tool = tools.find((t) => t.metadata.name === toolCall.name);
+    if (!tool) {
+      results.push(new ToolMessage2({
+        content: `Error: Tool '${toolCall.name}' not found`,
+        tool_call_id: toolCall.id
+      }));
+      continue;
+    }
+    try {
+      const result = await tool.execute(toolCall.args);
+      const content = typeof result === "string" ? result : JSON.stringify(result);
+      results.push(new ToolMessage2({
+        content,
+        tool_call_id: toolCall.id
+      }));
+    } catch (error) {
+      results.push(new ToolMessage2({
+        content: `Error executing tool: ${error.message}`,
+        tool_call_id: toolCall.id
+      }));
+    }
+  }
+  return results;
+}
 var DEFAULT_SUPERVISOR_SYSTEM_PROMPT = `You are a supervisor agent responsible for routing tasks to specialized worker agents.
 
 Your job is to:
@@ -1814,11 +1841,13 @@ var llmBasedRouting = {
       throw new Error("LLM-based routing requires a model to be configured");
     }
     const systemPrompt = config.systemPrompt || DEFAULT_SUPERVISOR_SYSTEM_PROMPT;
+    const maxRetries = config.maxToolRetries || 3;
+    const tools = config.tools || [];
     const workerInfo = Object.entries(state.workers).map(([id, caps]) => {
       const skills = caps.skills.join(", ");
-      const tools = caps.tools.join(", ");
+      const tools2 = caps.tools.join(", ");
       const available = caps.available ? "available" : "busy";
-      return `- ${id}: Skills: [${skills}], Tools: [${tools}], Status: ${available}, Workload: ${caps.currentWorkload}`;
+      return `- ${id}: Skills: [${skills}], Tools: [${tools2}], Status: ${available}, Workload: ${caps.currentWorkload}`;
     }).join("\n");
     const lastMessage = state.messages[state.messages.length - 1];
     const taskContext = lastMessage?.content || state.input;
@@ -1828,24 +1857,42 @@ Available workers:
 ${workerInfo}
 
 Select the best worker for this task and explain your reasoning.`;
-    const messages = [
-      new SystemMessage4(systemPrompt),
-      new HumanMessage4(userPrompt)
-    ];
-    const response = await config.model.invoke(messages);
-    const content = typeof response.content === "string" ? response.content : JSON.stringify(response.content);
-    try {
-      const decision = JSON.parse(content);
-      return {
-        targetAgent: decision.targetAgent,
-        reasoning: decision.reasoning,
-        confidence: decision.confidence,
-        strategy: "llm-based",
-        timestamp: Date.now()
-      };
-    } catch (error) {
-      throw new Error(`Failed to parse routing decision from LLM: ${error}`);
+    const conversationHistory = [];
+    let attempt = 0;
+    while (attempt < maxRetries) {
+      const messages = [
+        new SystemMessage4(systemPrompt),
+        new HumanMessage4(userPrompt),
+        ...conversationHistory
+      ];
+      const response = await config.model.invoke(messages);
+      if (response.tool_calls && response.tool_calls.length > 0) {
+        if (tools.length === 0) {
+          throw new Error("LLM requested tool calls but no tools are configured");
+        }
+        const toolResults = await executeTools(response.tool_calls, tools);
+        conversationHistory.push(
+          new AIMessage2({ content: response.content || "", tool_calls: response.tool_calls }),
+          ...toolResults
+        );
+        attempt++;
+        continue;
+      }
+      const content = typeof response.content === "string" ? response.content : JSON.stringify(response.content);
+      try {
+        const decision = JSON.parse(content);
+        return {
+          targetAgent: decision.targetAgent,
+          reasoning: decision.reasoning,
+          confidence: decision.confidence,
+          strategy: "llm-based",
+          timestamp: Date.now()
+        };
+      } catch (error) {
+        throw new Error(`Failed to parse routing decision from LLM: ${error}`);
+      }
     }
+    throw new Error(`Max tool retries (${maxRetries}) exceeded without routing decision`);
   }
 };
 var roundRobinRouting = {
@@ -2320,6 +2367,7 @@ Please synthesize these results into a comprehensive response that addresses the
 
 // src/multi-agent/agent.ts
 import { StateGraph as StateGraph4, END as END4 } from "@langchain/langgraph";
+import { toLangChainTools as toLangChainTools3 } from "@agentforge/core";
 function createMultiAgentSystem(config) {
   const {
     supervisor,
@@ -2330,11 +2378,13 @@ function createMultiAgentSystem(config) {
     checkpointer
   } = config;
   const workflow = new StateGraph4(MultiAgentState);
-  const supervisorNode = createSupervisorNode({
-    ...supervisor,
-    maxIterations,
-    verbose
-  });
+  let supervisorConfig = { ...supervisor, maxIterations, verbose };
+  if (supervisor.model && supervisor.tools && supervisor.tools.length > 0) {
+    const langchainTools = toLangChainTools3(supervisor.tools);
+    const modelWithTools = supervisor.model.bindTools(langchainTools);
+    supervisorConfig.model = modelWithTools;
+  }
+  const supervisorNode = createSupervisorNode(supervisorConfig);
   workflow.addNode("supervisor", supervisorNode);
   const workerIds = [];
   const workerCapabilities = {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge/patterns",
-  "version": "0.5.4",
+  "version": "0.6.0",
   "description": "Agent patterns (ReAct, Planner-Executor) for AgentForge framework",
   "type": "module",
   "main": "./dist/index.cjs",