refactor: rename method

addConditionalEdge to addConditionalEdges
bsorrentino · Apr 26, 2024 · 4c196bf · 4c196bf
1 parent d80fdc6
commit 4c196bf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/GraphState.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/GraphState.java
@@ -1,10 +1,10 @@
 package org.bsc.langgraph4j;
 
 import lombok.var;
+import org.bsc.async.AsyncGenerator;
+import org.bsc.async.AsyncGeneratorQueue;
 import org.bsc.langgraph4j.action.AsyncEdgeAction;
 import org.bsc.langgraph4j.action.AsyncNodeAction;
-import org.bsc.langgraph4j.async.AsyncGenerator;
-import org.bsc.langgraph4j.async.AsyncGeneratorQueue;
 import org.bsc.langgraph4j.state.AgentState;
 import org.bsc.langgraph4j.state.AgentStateFactory;
 import org.bsc.langgraph4j.state.AppendableValue;
@@ -227,7 +227,7 @@ public void addEdge(String sourceId, String targetId) throws GraphStateException
         edges.add( edge );
     }
 
-    public void addConditionalEdge(String sourceId, AsyncEdgeAction<State> condition, Map<String,String> mappings ) throws GraphStateException {
+    public void addConditionalEdges(String sourceId, AsyncEdgeAction<State> condition, Map<String,String> mappings ) throws GraphStateException {
         if( Objects.equals( sourceId, END)) {
             throw Errors.invalidEdgeIdentifier.exception(END);
         }

diff --git a/core-jdk8/src/test/java/org/bsc/langgraph4j/AsyncTest.java b/core-jdk8/src/test/java/org/bsc/langgraph4j/AsyncTest.java
@@ -1,8 +1,8 @@
 package org.bsc.langgraph4j;
 
 import lombok.var;
-import org.bsc.langgraph4j.async.AsyncGenerator;
-import org.bsc.langgraph4j.async.AsyncGeneratorQueue;
+import org.bsc.async.AsyncGenerator;
+import org.bsc.async.AsyncGeneratorQueue;
 import org.junit.jupiter.api.Test;
 
 import java.util.ArrayList;

diff --git a/core-jdk8/src/test/java/org/bsc/langgraph4j/LangGraphTest.java b/core-jdk8/src/test/java/org/bsc/langgraph4j/LangGraphTest.java
@@ -71,7 +71,7 @@ void testValidation() throws Exception {
         System.out.println(exception.getMessage());
 
         exception = assertThrows(GraphStateException.class, () ->
-            workflow.addConditionalEdge("agent_1", edge_async( state  -> "agent_3" ), mapOf() )
+            workflow.addConditionalEdges("agent_1", edge_async(state  -> "agent_3" ), mapOf() )
         );
         System.out.println(exception.getMessage());