langgraph[patch]: Fix typing for conditional edge array route maps (#408

)
langchain-ai · Aug 28, 2024 · eae2321 · eae2321
1 parent 4298a20
commit eae2321
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/libs/langgraph/src/graph/graph.ts b/libs/langgraph/src/graph/graph.ts
@@ -35,7 +35,7 @@ export const END = "__end__";
 export interface BranchOptions<IO, N extends string> {
   source: N;
   path: Branch<IO, N>["condition"];
-  pathMap?: Record<string, N | typeof END> | N[];
+  pathMap?: Record<string, N | typeof END> | (N | typeof END)[];
 }
 
 export class Branch<IO, N extends string> {

diff --git a/libs/langgraph/src/tests/pregel.test.ts b/libs/langgraph/src/tests/pregel.test.ts
@@ -2036,9 +2036,9 @@ describe("StateGraph", () => {
     data: typeof AgentAnnotation.State
   ): Promise<string> => {
     if (data.agentOutcome && "returnValues" in data.agentOutcome) {
-      return "exit";
+      return "__end__";
     }
-    return "continue";
+    return "tools";
   };
 
   it("can invoke", async () => {
@@ -2088,10 +2088,7 @@ describe("StateGraph", () => {
       .addNode("tools", executeTools)
       .addEdge(START, "agent")
       .addEdge("agent", "passthrough")
-      .addConditionalEdges("passthrough", shouldContinue, {
-        continue: "tools",
-        exit: END,
-      })
+      .addConditionalEdges("passthrough", shouldContinue, ["tools", "__end__"])
       .addEdge("tools", "agent")
       .compile();
 
@@ -2182,10 +2179,7 @@ describe("StateGraph", () => {
       .addNode("agent", agent)
       .addNode("tools", executeTools)
       .addEdge(START, "agent")
-      .addConditionalEdges("agent", shouldContinue, {
-        continue: "tools",
-        exit: END,
-      })
+      .addConditionalEdges("agent", shouldContinue, ["tools", "__end__"])
       .addEdge("tools", "agent")
       .compile();