@agentforge/patterns 0.4.0 → 0.5.0

package/dist/index.cjs

@@ -2046,6 +2046,84 @@ 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 taskResult = {
+        assignmentId: currentAssignment.id,
+        workerId,
+        result: response,
+        completedAt: Date.now(),
+        success: true,
+        metadata: {
+          agent_type: "react",
+          iterations: result.iteration || 0
+        }
+      };
+      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 +2217,8 @@ function createWorkerNode(config) {
     tools = [],
     systemPrompt,
     verbose = false,
-    executeFn
+    executeFn,
+    agent
   } = config;
   return async (state) => {
     try {
@@ -2159,10 +2238,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(", ")}

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,84 @@ 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 taskResult = {
+        assignmentId: currentAssignment.id,
+        workerId,
+        result: response,
+        completedAt: Date.now(),
+        success: true,
+        metadata: {
+          agent_type: "react",
+          iterations: result.iteration || 0
+        }
+      };
+      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 +2118,8 @@ function createWorkerNode(config) {
     tools = [],
     systemPrompt,
     verbose = false,
-    executeFn
+    executeFn,
+    agent
   } = config;
   return async (state) => {
     try {
@@ -2060,10 +2139,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(", ")}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge/patterns",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "Agent patterns (ReAct, Planner-Executor) for AgentForge framework",
   "type": "module",
   "main": "./dist/index.cjs",
@@ -16,6 +16,17 @@
   "files": [
     "dist"
   ],
+  "scripts": {
+    "build": "tsup src/index.ts --format esm,cjs --dts --clean",
+    "dev": "tsup src/index.ts --format esm,cjs --dts --watch",
+    "test": "vitest",
+    "test:coverage": "vitest --coverage",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint src",
+    "lint:fix": "eslint src --fix",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "clean": "rm -rf dist *.tsbuildinfo"
+  },
   "keywords": [
     "agents",
     "ai",
@@ -36,10 +47,10 @@
     "url": "https://github.com/TVScoundrel/agentforge/issues"
   },
   "dependencies": {
+    "@agentforge/core": "workspace:*",
     "@langchain/core": "^1.1.0",
     "@langchain/langgraph": "^1.0.0",
-    "zod": "^3.23.8",
-    "@agentforge/core": "0.4.0"
+    "zod": "^3.23.8"
   },
   "devDependencies": {
     "@eslint/js": "^9.17.0",
@@ -54,16 +65,5 @@
   "peerDependencies": {
     "@langchain/core": "^1.1.0",
     "@langchain/langgraph": "^1.0.0"
-  },
-  "scripts": {
-    "build": "tsup src/index.ts --format esm,cjs --dts --clean",
-    "dev": "tsup src/index.ts --format esm,cjs --dts --watch",
-    "test": "vitest",
-    "test:coverage": "vitest --coverage",
-    "typecheck": "tsc --noEmit",
-    "lint": "eslint src",
-    "lint:fix": "eslint src --fix",
-    "format": "prettier --write \"src/**/*.ts\"",
-    "clean": "rm -rf dist *.tsbuildinfo"
   }
-}
+}

package/LICENSE

@@ -1,22 +0,0 @@
-MIT License
-
-Copyright (c) 2024 Tom Van Schoor
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
-