test(image-to-diagram): finalize subgraph support

work on #39
bsorrentino · Nov 26, 2024 · 59f2a05 · 59f2a05
1 parent 3898878
commit 59f2a05
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...mage-to-diagram/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java b/...mage-to-diagram/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java
@@ -11,8 +11,6 @@
 import org.bsc.async.AsyncGenerator;
 import org.bsc.langgraph4j.StateGraph;
 import org.bsc.langgraph4j.NodeOutput;
-import org.bsc.langgraph4j.subgraph.*;
-
 import java.net.URI;
 import java.util.*;
 
@@ -90,7 +88,7 @@ public AsyncGenerator<NodeOutput<State>> executeWithCorrection(  Map<String, Obj
                 .addNode("agent_describer", DescribeDiagramImage.of(imageUrlOrData, getVisionModel() )  )
                 .addNode("agent_sequence_plantuml", TranslateSequenceDiagramToPlantUML.of(getModel()) )
                 .addNode("agent_generic_plantuml", TranslateGenericDiagramToPlantUML.of(getModel()) )
-                .addNode( "agent_diagram_correction", SubgraphNodeAction.of(diagramCorrectionProcess) )
+                .addSubgraph( "agent_diagram_correction", diagramCorrectionProcess )
                 .addNode( "evaluate_result", EvaluateResult.of(getModel()))
                 .addConditionalEdges(
                         "agent_describer",