diff --git a/adaptive-rag/src/main/java/dev/langchain4j/adaptiverag/AdaptiveRag.java b/adaptive-rag/src/main/java/dev/langchain4j/adaptiverag/AdaptiveRag.java index 08e4677..19f8872 100644 --- 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 ); + } }