fix: Use none check to avoid 30s delay in agent run.

vertex-sdk-bot · copybara-github · commit 84895b6c6cd8 · 2025-06-11T12:59:10.000-07:00
PiperOrigin-RevId: 770280829
diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py
@@ -627,15 +627,17 @@ def _asyncio_thread_main():
                 asyncio.run(_invoke_agent_async())
             except RuntimeError as e:
                 event_queue.put(e)
+            finally:
+                # Use None as a sentinel to stop the main thread.
+                event_queue.put(None)
 
         thread = threading.Thread(target=_asyncio_thread_main)
         thread.start()
 
         try:
             while True:
-                try:
-                    event = event_queue.get(timeout=30)
-                except queue.Empty:
+                event = event_queue.get()
+                if event is None:
                     break
                 if isinstance(event, RuntimeError):
                     raise event
