refactor: clean code

bsorrentino · May 19, 2024 · daac0e8 · daac0e8
1 parent 7e8a739
commit daac0e8
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/agents-jdk8/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java b/agents-jdk8/src/main/java/dev/langchain4j/image_to_diagram/ImageToDiagramProcess.java
@@ -184,21 +184,12 @@ public AsyncGenerator<NodeOutput<State>> execute(  Map<String, Object> inputs )
                     "generic", "agent_generic_plantuml" )
         );
 
-        //workflow.addNode( "agent_review", this::reviewResult );
         workflow.addNode( "evaluate_result", this::evaluateResult);
         workflow.addEdge("agent_sequence_plantuml", "evaluate_result");
         workflow.addEdge("agent_generic_plantuml", "evaluate_result");
-        workflow.addEdge( "agent_review", "evaluate_result" );
-        /*
-        workflow.addConditionalEdges(
-                "evaluate_result",
-                edge_async(this::routeEvaluationResult),
-                mapOf(  "OK", END,
-                        "ERROR", "agent_review",
-                        "UNKNOWN", END )
-        );
-        */
         workflow.setEntryPoint("agent_describer");
+        workflow.setFinishPoint("evaluate_result");
+
 
         var app = workflow.compile();