refactor(DiagramGenerator.java): access node elements directly

work on #73
bsorrentino · Feb 11, 2025 · 770c003 · 770c003
1 parent 2265238
commit 770c003
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/java/org/bsc/langgraph4j/DiagramGenerator.java b/core/src/main/java/org/bsc/langgraph4j/DiagramGenerator.java
@@ -146,7 +146,7 @@ protected final <State extends AgentState> Context generate( StateGraph<State> s
 
         appendHeader( ctx );
 
-        stateGraph.nodes
+        stateGraph.nodes.elements
                 .forEach( n -> {
 
                     try {