@agentforge/patterns 0.4.1 → 0.5.1

package/dist/index.cjs

@@ -2046,6 +2046,90 @@ function getRoutingStrategy(name) {
   }
 }
 
+// src/multi-agent/utils.ts
+function isReActAgent(obj) {
+  return obj && typeof obj === "object" && typeof obj.invoke === "function" && typeof obj.stream === "function" && // Additional check to ensure it's not just any object with invoke/stream
+  (obj.constructor?.name === "CompiledGraph" || obj.constructor?.name === "CompiledStateGraph");
+}
+function wrapReActAgent(workerId, agent, verbose = false) {
+  return async (state) => {
+    try {
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Wrapping ReAct agent execution`);
+      }
+      const task = state.messages[state.messages.length - 1]?.content || state.input;
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Task:`, task.substring(0, 100) + "...");
+      }
+      const currentAssignment = state.activeAssignments.find(
+        (assignment) => assignment.workerId === workerId && !state.completedTasks.some((task2) => task2.assignmentId === assignment.id)
+      );
+      if (!currentAssignment) {
+        if (verbose) {
+          console.log(`[ReActWrapper:${workerId}] No active assignment found`);
+        }
+        return {
+          currentAgent: "supervisor",
+          status: "routing"
+        };
+      }
+      const result = await agent.invoke({
+        messages: [{ role: "user", content: task }]
+      });
+      const response = result.messages?.[result.messages.length - 1]?.content || "No response";
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Response:`, response.substring(0, 100) + "...");
+      }
+      const toolsUsed = result.actions?.map((action) => action.name).filter(Boolean) || [];
+      const uniqueTools = [...new Set(toolsUsed)];
+      if (verbose && uniqueTools.length > 0) {
+        console.log(`[ReActWrapper:${workerId}] Tools used:`, uniqueTools.join(", "));
+      }
+      const taskResult = {
+        assignmentId: currentAssignment.id,
+        workerId,
+        result: response,
+        completedAt: Date.now(),
+        success: true,
+        metadata: {
+          agent_type: "react",
+          iterations: result.iteration || 0,
+          tools_used: uniqueTools
+        }
+      };
+      return {
+        completedTasks: [taskResult],
+        currentAgent: "supervisor",
+        status: "routing"
+      };
+    } catch (error) {
+      console.error(`[ReActWrapper:${workerId}] Error:`, error);
+      const currentAssignment = state.activeAssignments.find(
+        (assignment) => assignment.workerId === workerId
+      );
+      if (currentAssignment) {
+        const errorResult = {
+          assignmentId: currentAssignment.id,
+          workerId,
+          success: false,
+          result: "",
+          error: error instanceof Error ? error.message : "Unknown error in ReAct agent",
+          completedAt: Date.now()
+        };
+        return {
+          completedTasks: [errorResult],
+          currentAgent: "supervisor",
+          status: "routing"
+        };
+      }
+      return {
+        status: "failed",
+        error: error instanceof Error ? error.message : `Unknown error in ReAct wrapper for ${workerId}`
+      };
+    }
+  };
+}
+
 // src/multi-agent/nodes.ts
 var import_messages5 = require("@langchain/core/messages");
 var import_core7 = require("@agentforge/core");
@@ -2139,7 +2223,8 @@ function createWorkerNode(config) {
     tools = [],
     systemPrompt,
     verbose = false,
-    executeFn
+    executeFn,
+    agent
   } = config;
   return async (state) => {
     try {
@@ -2159,10 +2244,26 @@ function createWorkerNode(config) {
         };
       }
       if (executeFn) {
+        if (verbose) {
+          console.log(`[Worker:${id}] Using custom executeFn`);
+        }
         return await executeFn(state);
       }
+      if (agent) {
+        if (isReActAgent(agent)) {
+          if (verbose) {
+            console.log(`[Worker:${id}] Using ReAct agent (auto-wrapped)`);
+          }
+          const wrappedFn = wrapReActAgent(id, agent, verbose);
+          return await wrappedFn(state);
+        } else {
+          console.warn(`[Worker:${id}] Agent provided but does not appear to be a ReAct agent. Falling back to default execution.`);
+        }
+      }
       if (!model) {
-        throw new Error(`Worker ${id} requires either a model or custom execution function`);
+        throw new Error(
+          `Worker ${id} requires either a model, an agent, or a custom execution function. Provide one of: config.model, config.agent, or config.executeFn`
+        );
       }
       const defaultSystemPrompt = `You are a specialized worker agent with the following capabilities:
 Skills: ${capabilities.skills.join(", ")}
@@ -2386,6 +2487,17 @@ function createMultiAgentSystem(config) {
     };
     return originalInvoke(mergedInput, config2);
   };
+  const originalStream = compiled.stream.bind(compiled);
+  compiled.stream = async function(input, config2) {
+    const mergedInput = {
+      ...input,
+      workers: {
+        ...workerCapabilities,
+        ...input.workers || {}
+      }
+    };
+    return originalStream(mergedInput, config2);
+  };
   return compiled;
 }
 var MultiAgentSystemBuilder = class {

package/dist/index.d.cts

@@ -2313,8 +2313,34 @@ interface WorkerConfig {
     verbose?: boolean;
     /**
      * Custom execution function
+     *
+     * If provided, this function will be used to execute tasks for this worker.
+     * Takes precedence over the `agent` property.
      */
     executeFn?: (state: MultiAgentStateType) => Promise<Partial<MultiAgentStateType>>;
+    /**
+     * ReAct agent instance
+     *
+     * If provided, the Multi-Agent pattern will automatically wrap this ReAct agent
+     * to work as a worker. The agent should be a compiled LangGraph StateGraph
+     * (e.g., created with `createReActAgent()`).
+     *
+     * Note: `executeFn` takes precedence over `agent` if both are provided.
+     *
+     * @example
+     * ```typescript
+     * const hrAgent = createReActAgent({ model, tools, systemPrompt });
+     *
+     * const system = createMultiAgentSystem({
+     *   workers: [{
+     *     id: 'hr',
+     *     capabilities: { skills: ['hr'], ... },
+     *     agent: hrAgent,  // Automatically wrapped!
+     *   }]
+     * });
+     * ```
+     */
+    agent?: CompiledStateGraph<any, any>;
 }
 /**
  * Configuration for the aggregator node

package/dist/index.d.ts

@@ -2313,8 +2313,34 @@ interface WorkerConfig {
     verbose?: boolean;
     /**
      * Custom execution function
+     *
+     * If provided, this function will be used to execute tasks for this worker.
+     * Takes precedence over the `agent` property.
      */
     executeFn?: (state: MultiAgentStateType) => Promise<Partial<MultiAgentStateType>>;
+    /**
+     * ReAct agent instance
+     *
+     * If provided, the Multi-Agent pattern will automatically wrap this ReAct agent
+     * to work as a worker. The agent should be a compiled LangGraph StateGraph
+     * (e.g., created with `createReActAgent()`).
+     *
+     * Note: `executeFn` takes precedence over `agent` if both are provided.
+     *
+     * @example
+     * ```typescript
+     * const hrAgent = createReActAgent({ model, tools, systemPrompt });
+     *
+     * const system = createMultiAgentSystem({
+     *   workers: [{
+     *     id: 'hr',
+     *     capabilities: { skills: ['hr'], ... },
+     *     agent: hrAgent,  // Automatically wrapped!
+     *   }]
+     * });
+     * ```
+     */
+    agent?: CompiledStateGraph<any, any>;
 }
 /**
  * Configuration for the aggregator node

package/dist/index.js

@@ -1947,6 +1947,90 @@ function getRoutingStrategy(name) {
   }
 }
 
+// src/multi-agent/utils.ts
+function isReActAgent(obj) {
+  return obj && typeof obj === "object" && typeof obj.invoke === "function" && typeof obj.stream === "function" && // Additional check to ensure it's not just any object with invoke/stream
+  (obj.constructor?.name === "CompiledGraph" || obj.constructor?.name === "CompiledStateGraph");
+}
+function wrapReActAgent(workerId, agent, verbose = false) {
+  return async (state) => {
+    try {
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Wrapping ReAct agent execution`);
+      }
+      const task = state.messages[state.messages.length - 1]?.content || state.input;
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Task:`, task.substring(0, 100) + "...");
+      }
+      const currentAssignment = state.activeAssignments.find(
+        (assignment) => assignment.workerId === workerId && !state.completedTasks.some((task2) => task2.assignmentId === assignment.id)
+      );
+      if (!currentAssignment) {
+        if (verbose) {
+          console.log(`[ReActWrapper:${workerId}] No active assignment found`);
+        }
+        return {
+          currentAgent: "supervisor",
+          status: "routing"
+        };
+      }
+      const result = await agent.invoke({
+        messages: [{ role: "user", content: task }]
+      });
+      const response = result.messages?.[result.messages.length - 1]?.content || "No response";
+      if (verbose) {
+        console.log(`[ReActWrapper:${workerId}] Response:`, response.substring(0, 100) + "...");
+      }
+      const toolsUsed = result.actions?.map((action) => action.name).filter(Boolean) || [];
+      const uniqueTools = [...new Set(toolsUsed)];
+      if (verbose && uniqueTools.length > 0) {
+        console.log(`[ReActWrapper:${workerId}] Tools used:`, uniqueTools.join(", "));
+      }
+      const taskResult = {
+        assignmentId: currentAssignment.id,
+        workerId,
+        result: response,
+        completedAt: Date.now(),
+        success: true,
+        metadata: {
+          agent_type: "react",
+          iterations: result.iteration || 0,
+          tools_used: uniqueTools
+        }
+      };
+      return {
+        completedTasks: [taskResult],
+        currentAgent: "supervisor",
+        status: "routing"
+      };
+    } catch (error) {
+      console.error(`[ReActWrapper:${workerId}] Error:`, error);
+      const currentAssignment = state.activeAssignments.find(
+        (assignment) => assignment.workerId === workerId
+      );
+      if (currentAssignment) {
+        const errorResult = {
+          assignmentId: currentAssignment.id,
+          workerId,
+          success: false,
+          result: "",
+          error: error instanceof Error ? error.message : "Unknown error in ReAct agent",
+          completedAt: Date.now()
+        };
+        return {
+          completedTasks: [errorResult],
+          currentAgent: "supervisor",
+          status: "routing"
+        };
+      }
+      return {
+        status: "failed",
+        error: error instanceof Error ? error.message : `Unknown error in ReAct wrapper for ${workerId}`
+      };
+    }
+  };
+}
+
 // src/multi-agent/nodes.ts
 import { HumanMessage as HumanMessage5, SystemMessage as SystemMessage5 } from "@langchain/core/messages";
 import { toLangChainTools as toLangChainTools2 } from "@agentforge/core";
@@ -2040,7 +2124,8 @@ function createWorkerNode(config) {
     tools = [],
     systemPrompt,
     verbose = false,
-    executeFn
+    executeFn,
+    agent
   } = config;
   return async (state) => {
     try {
@@ -2060,10 +2145,26 @@ function createWorkerNode(config) {
         };
       }
       if (executeFn) {
+        if (verbose) {
+          console.log(`[Worker:${id}] Using custom executeFn`);
+        }
         return await executeFn(state);
       }
+      if (agent) {
+        if (isReActAgent(agent)) {
+          if (verbose) {
+            console.log(`[Worker:${id}] Using ReAct agent (auto-wrapped)`);
+          }
+          const wrappedFn = wrapReActAgent(id, agent, verbose);
+          return await wrappedFn(state);
+        } else {
+          console.warn(`[Worker:${id}] Agent provided but does not appear to be a ReAct agent. Falling back to default execution.`);
+        }
+      }
       if (!model) {
-        throw new Error(`Worker ${id} requires either a model or custom execution function`);
+        throw new Error(
+          `Worker ${id} requires either a model, an agent, or a custom execution function. Provide one of: config.model, config.agent, or config.executeFn`
+        );
       }
       const defaultSystemPrompt = `You are a specialized worker agent with the following capabilities:
 Skills: ${capabilities.skills.join(", ")}
@@ -2287,6 +2388,17 @@ function createMultiAgentSystem(config) {
     };
     return originalInvoke(mergedInput, config2);
   };
+  const originalStream = compiled.stream.bind(compiled);
+  compiled.stream = async function(input, config2) {
+    const mergedInput = {
+      ...input,
+      workers: {
+        ...workerCapabilities,
+        ...input.workers || {}
+      }
+    };
+    return originalStream(mergedInput, config2);
+  };
   return compiled;
 }
 var MultiAgentSystemBuilder = class {

package/package.json

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