Refactor jbang-example.java: streamline session creation and message sending

Author: brunoborges

jbang-example.java

@@ -12,27 +12,27 @@ public static void main(String[] args) throws Exception {
             client.start().get();
 
             // Create a session
-            var sessionConfig = new SessionConfig().setModel("claude-sonnet-4.5");
-            var session = client.createSession(sessionConfig).get();
-
-            // Wait for response using session.idle event
-            var done = new CompletableFuture<Void>();
+            var session = client.createSession(
+                new SessionConfig().setModel("claude-sonnet-4.5")).get();
 
+            // Handle assistant message events
             session.on(AssistantMessageEvent.class, msg -> {
                 System.out.println(msg.getData().getContent());
             });
+
+            // Handle session usage info events
             session.on(SessionUsageInfoEvent.class, usage -> {
                 var data = usage.getData();
                 System.out.println("\n--- Usage Metrics ---");
                 System.out.println("Current tokens: " + (int) data.getCurrentTokens());
                 System.out.println("Token limit: " + (int) data.getTokenLimit());
                 System.out.println("Messages count: " + (int) data.getMessagesLength());
             });
-            session.on(SessionIdleEvent.class, idle -> done.complete(null));
 
-            // Send a message and wait for completion
-            session.send(new MessageOptions().setPrompt("What is 2+2?")).get();
-            done.get();
+            // Send a message
+            var completable = session.sendAndWait(new MessageOptions().setPrompt("What is 2+2?"));
+            // and wait for completion
+            completable.get();
         }
     }
 }