From 59f2a056b61c5b07828912ea076abe882099d29c Mon Sep 17 00:00:00 2001
From: bsorrentino <bartolomeo.sorrentino@gmail.com>
Date: Tue, 26 Nov 2024 10:41:28 +0100
Subject: [PATCH] test(image-to-diagram): finalize subgraph support

work on #39
---
 .../langchain4j/image_to_diagram/ImageToDiagramProcess.java   | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/samples/image-to-diagram/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java b/samples/image-to-diagram/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java
index f556407b..83f589fe 100644
--- a/samples/image-to-diagram/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java
+++ b/samples/image-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",