Merge pull request #878 from isamu/map_type_inference

type inference
receptron · Jan 8, 2025 · 990e87a · 990e87a
2 parents 945c0f4 + aa8dda0
commit 990e87a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/agents/vanilla_agents/src/graph_agents/map_agent.ts b/agents/vanilla_agents/src/graph_agents/map_agent.ts
@@ -1,4 +1,4 @@
-import { GraphAI, AgentFunction, AgentFunctionInfo, StaticNodeData, assert, graphDataLatestVersion } from "graphai";
+import { GraphAI, AgentFunction, AgentFunctionInfo, assert, graphDataLatestVersion } from "graphai";
 
 export const mapAgent: AgentFunction<
   {
@@ -39,9 +39,9 @@ export const mapAgent: AgentFunction<
     if (nestedGraphData.nodes[mappedNodeId] === undefined) {
       // If the input node does not exist, automatically create a static node
       nestedGraphData.nodes[mappedNodeId] = { value: namedInputs[nodeId] };
-    } else {
+    } else if (!("agent" in nestedGraphData.nodes[mappedNodeId])){
       // Otherwise, inject the proper data here (instead of calling injectTo method later)
-      (nestedGraphData.nodes[mappedNodeId] as StaticNodeData)["value"] = namedInputs[nodeId];
+      nestedGraphData.nodes[mappedNodeId]["value"] = namedInputs[nodeId];
     }
   });