refactor(adaptive-rag): clean code

bsorrentino · Jun 20, 2024 · 5391138 · 5391138
1 parent 985275c
commit 5391138
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/adaptive-rag/src/main/java/dev/langchain4j/adaptiverag/AdaptiveRag.java b/adaptive-rag/src/main/java/dev/langchain4j/adaptiverag/AdaptiveRag.java
@@ -5,6 +5,7 @@
 import lombok.extern.slf4j.Slf4j;
 import lombok.var;
 import org.bsc.langgraph4j.CompiledGraph;
+import org.bsc.langgraph4j.GraphRepresentation;
 import org.bsc.langgraph4j.StateGraph;
 import org.bsc.langgraph4j.state.AgentState;
 
@@ -290,6 +291,7 @@ public static void main( String[] args ) throws Exception {
         var graph = adaptiveRagTest.buildGraph();
 
         var result = graph.stream( mapOf( "question", "What player at the Bears expected to draft first in the 2024 NFL draft?" ) );
+        // var result = graph.stream( mapOf( "question", "What kind the agent memory do iu know?" ) );
 
         String generation = "";
         for( var r : result ) {
@@ -301,6 +303,10 @@ public static void main( String[] args ) throws Exception {
 
         System.out.println( generation );
 
+        // generate plantuml script
+        // var plantUml = graph.getGraph( GraphRepresentation.Type.PLANTUML );
+        // System.out.println( plantUml );
+
     }
 
 }