feat: move main implementation of getGraph() on StateGraph

work on #24
bsorrentino · Sep 11, 2024 · 9c1b39b · 9c1b39b
1 parent 52bfbec
commit 9c1b39b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/CompiledGraph.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/CompiledGraph.java
@@ -396,7 +396,7 @@ public Optional<State> invoke(Map<String,Object> inputs ) throws Exception {
      */
     public GraphRepresentation getGraph( GraphRepresentation.Type type, String title, boolean printConditionalEdges ) {
 
-        String content = type.generator.generate( this, title, printConditionalEdges);
+        String content = type.generator.generate( this.stateGraph, title, printConditionalEdges);
 
         return new GraphRepresentation( type, content );
     }
@@ -410,7 +410,7 @@ public GraphRepresentation getGraph( GraphRepresentation.Type type, String title
      */
     public GraphRepresentation getGraph( GraphRepresentation.Type type, String title ) {
 
-        String content = type.generator.generate( this, title, true);
+        String content = type.generator.generate( this.stateGraph, title, true);
 
         return new GraphRepresentation( type, content );
     }