refactor: adds Node constructors

work on #39
bsorrentino · Nov 25, 2024 · 63dd250 · 63dd250
1 parent 14fc442
commit 63dd250
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/Node.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/Node.java
@@ -30,6 +30,23 @@ class Node<State extends AgentState> {
 
     AsyncNodeActionWithConfig<State> actionWithConfig;
 
+    public Node( String id ) {
+        this.id = id;
+        this.action = null;
+        this.actionWithConfig = null;
+
+    }
+    public Node( String id, AsyncNodeAction<State> action ) {
+        this.id = id;
+        this.action = action;
+        this.actionWithConfig = null;
+    }
+    public Node( String id, AsyncNodeActionWithConfig<State> actionWithConfig ) {
+        this.id = id;
+        this.action = null;
+        this.actionWithConfig = actionWithConfig;
+
+    }
     /**
      * Checks if this node is equal to another object.
      *

diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/StateGraph.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/StateGraph.java
@@ -203,7 +203,7 @@ public StateGraph<State> addNode(String id, AsyncNodeAction<State> action) throw
         if (Objects.equals(id, END)) {
             throw Errors.invalidNodeIdentifier.exception(END);
         }
-        Node<State> node = new Node<State>(id, action, null);
+        Node<State> node = new Node<State>(id, action);
 
         if (nodes.contains(node)) {
             throw Errors.duplicateNodeError.exception(id);
@@ -217,7 +217,7 @@ public StateGraph<State> addNode(String id, AsyncNodeActionWithConfig<State> act
         if (Objects.equals(id, END)) {
             throw Errors.invalidNodeIdentifier.exception(END);
         }
-        Node<State> node = new Node<State>(id, null, actionWithConfig);
+        Node<State> node = new Node<State>(id, actionWithConfig);
 
         if (nodes.contains(node)) {
             throw Errors.duplicateNodeError.exception(id);
@@ -292,7 +292,7 @@ public StateGraph<State> addConditionalEdges(String sourceId, AsyncEdgeAction<St
      * @return a new fake node
      */
     private Node<State> nodeById(String id) {
-        return new Node<>(id, null, null);
+        return new Node<>(id);
     }
 
     /**