Merge pull request #907 from isamu/updateToolsResult

Update tools result
receptron · Jan 28, 2025 · 6b808fb · 6b808fb
2 parents eeaceea + e20b91e
commit 6b808fb
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 6 deletions.
diff --git a/llm_agents/tools_agent/lib/tools_agent.js b/llm_agents/tools_agent/lib/tools_agent.js
@@ -115,13 +115,17 @@ const toolWorkFlowStep = {
             inputs: { array: [":textMessage.messages", ":tool_call_response.mergeToolsResponse.array.$0"] },
         },
         reducer: {
-            isResult: true,
             agent: "pushAgent",
             inputs: { array: ":messages", items: ":buffer.array.$0" },
         },
+        result: {
+            agent: "copyAgent",
+            isResult: true,
+            inputs: { messages: ":reducer.array" }
+        },
     },
 };
-const toolsAgent = (0, generator_1.nestedAgentGenerator)(toolWorkFlowStep);
+const toolsAgent = (0, generator_1.nestedAgentGenerator)(toolWorkFlowStep, { resultNodeId: "result" });
 const toolsAgentInfo = {
     name: "toolsAgent",
     agent: toolsAgent,

diff --git a/llm_agents/tools_agent/package.json b/llm_agents/tools_agent/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@graphai/tools_agent",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "General tools agents",
   "main": "lib/index.js",
   "files": [

diff --git a/llm_agents/tools_agent/src/tools_agent.ts b/llm_agents/tools_agent/src/tools_agent.ts
@@ -114,14 +114,18 @@ const toolWorkFlowStep = {
       inputs: { array: [":textMessage.messages", ":tool_call_response.mergeToolsResponse.array.$0"] },
     },
     reducer: {
-      isResult: true,
       agent: "pushAgent",
       inputs: { array: ":messages", items: ":buffer.array.$0" },
     },
+    result: {
+      agent: "copyAgent",
+      isResult: true,
+      inputs: { messages: ":reducer.array"}
+    },
   },
 };
 
-const toolsAgent = nestedAgentGenerator(toolWorkFlowStep);
+const toolsAgent = nestedAgentGenerator(toolWorkFlowStep, {resultNodeId: "result"});
 
 const toolsAgentInfo = {
   name: "toolsAgent",

diff --git a/llm_agents/tools_agent/tests/run_tools.ts b/llm_agents/tools_agent/tests/run_tools.ts
@@ -24,7 +24,7 @@ const main = async (agentInfo: AgentFunctionInfo) => {
   const { agent, samples, inputs: inputSchema } = agentInfo;
   for await (const sampleKey of samples.keys()) {
     const { params, inputs } = samples[sampleKey];
-    await agent({
+    const result = await agent({
       ...defaultTestContext,
       params,
       inputSchema,
@@ -34,6 +34,7 @@ const main = async (agentInfo: AgentFunctionInfo) => {
         graphOptions: {},
       },
     });
+    console.log(result);
   }
 };