test: reset steps counter after ending graph

work on #9
bsorrentino · Jul 17, 2024 · acf32a7 · acf32a7
1 parent 17e5750
commit acf32a7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/jetty/src/test/java/org/bsc/langgraph4j/LangGraphStreamingServerTest.java b/jetty/src/test/java/org/bsc/langgraph4j/LangGraphStreamingServerTest.java
@@ -41,6 +41,7 @@ public static void main(String[] args) throws Exception {
             @Override
             public String apply(AgentState state) {
                 if( ++steps == 2 ) {
+                    steps = 0;
                     return "end";
                 }
                 return "a2";
@@ -54,6 +55,7 @@ public String apply(AgentState state) {
 
         LangGraphStreamingServer server = LangGraphStreamingServer.builder()
                                                                     .port(8080)
+                                                                    .title("LANGGRAPH4j - TEST")
                                                                     .addInputStringArg("input")
                                                                     .build(app);