From 3983c668d1b30c51510ff07fcec72870b6c7100e Mon Sep 17 00:00:00 2001
From: Sandeep Belgavi <belgavi.sandeep@gmail.com>
Date: Fri, 19 Jun 2026 19:37:48 +0530
Subject: [PATCH] test: update LiveAudioTokensTest to use BIDI connection

Author: Sandeep Belgavi
Date: June 19, 2026

- Reconfigured LiveAudioTokensTest to use streamingMode(RunConfig.StreamingMode.BIDI).
- Utilized Runner.runLive with a LiveRequestQueue to communicate over WebSockets.
- Updated model to gemini-3.1-flash-live-preview for correct live audio API testing.
- Ensured Completion Modality metrics output properly in console.

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 .../adk/tutorials/LiveAudioTokensTest.java    | 23 +++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/tutorials/live-audio-single-agent/src/main/java/com/google/adk/tutorials/LiveAudioTokensTest.java b/tutorials/live-audio-single-agent/src/main/java/com/google/adk/tutorials/LiveAudioTokensTest.java
index 4bf39c0de..c4e798bb9 100644
--- a/tutorials/live-audio-single-agent/src/main/java/com/google/adk/tutorials/LiveAudioTokensTest.java
+++ b/tutorials/live-audio-single-agent/src/main/java/com/google/adk/tutorials/LiveAudioTokensTest.java
@@ -1,9 +1,10 @@
 /*
  * Author: Sandeep Belgavi
- * Date: June 18, 2026
+ * Date: June 19, 2026
  */
 package com.google.adk.tutorials;
 
+import com.google.adk.agents.LiveRequestQueue;
 import com.google.adk.agents.LlmAgent;
 import com.google.adk.agents.RunConfig;
 import com.google.adk.runner.Runner;
@@ -19,7 +20,7 @@ public static void main(String[] args) {
     LlmAgent agent =
         LlmAgent.builder()
             .name("audio_agent")
-            .model("gemini-2.5-pro") // using the requested model
+            .model("gemini-3.1-flash-live-preview")
             .instruction(
                 "You are a helpful assistant. Please say 'Hello, how can I help you today?'")
             .build();
@@ -29,6 +30,7 @@ public static void main(String[] args) {
     RunConfig runConfig =
         RunConfig.builder()
             .autoCreateSession(true)
+            .streamingMode(RunConfig.StreamingMode.BIDI)
             .responseModalities(ImmutableList.of(new Modality(Modality.Known.AUDIO)))
             .build();
 
@@ -38,13 +40,25 @@ public static void main(String[] args) {
             .parts(ImmutableList.of(Part.fromText("Please introduce yourself.")))
             .build();
 
+    LiveRequestQueue liveRequestQueue = new LiveRequestQueue();
+    liveRequestQueue.content(userMessage);
+
     System.out.println("Sending request to model...");
 
     runner
-        .runAsync("user1", "session1", userMessage, runConfig)
+        .runLive("user1", "session1", liveRequestQueue, runConfig)
         .doOnNext(
             event -> {
-              if (event.author() != null && event.author().equals("model")) {
+              System.out.println(
+                  "Got event from author: "
+                      + event.author()
+                      + " content: "
+                      + event.content().map(c -> c.parts().map(ps -> ps.size()).orElse(0)).orElse(0)
+                      + " parts");
+              if (event.author() != null && event.author().equals("audio_agent")) {
+                if (event.turnComplete().orElse(false)) {
+                  liveRequestQueue.close();
+                }
                 if (event.content().isPresent()) {
                   Content c = event.content().get();
                   for (Part p : c.parts().get()) {
@@ -56,6 +70,7 @@ public static void main(String[] args) {
                 if (event.usageMetadata().isPresent()) {
                   GenerateContentResponseUsageMetadata usage = event.usageMetadata().get();
                   System.out.println("Total Tokens: " + usage.totalTokenCount().orElse(0));
+                  System.out.println("Usage details: " + usage);
 
                   if (usage.promptTokensDetails().isPresent()) {
                     for (ModalityTokenCount mtc : usage.promptTokensDetails().get()) {