fix: Set event loop before asyncio.Lock() to ensure threading safety

yinghsienwu · copybara-github · commit be2d9c61176f · 2025-03-24T20:22:20.000-07:00
PiperOrigin-RevId: 740168890
diff --git a/google/genai/_api_client.py b/google/genai/_api_client.py
@@ -332,6 +332,10 @@ def __init__(
     # Initialize the lock. This lock will be used to protect access to the
     # credentials. This is crucial for thread safety when multiple coroutines
     # might be accessing the credentials at the same time.
+    try:
+      asyncio.get_event_loop()
+    except RuntimeError:
+      asyncio.set_event_loop(asyncio.new_event_loop())
     self._auth_lock = asyncio.Lock()
 
     # Handle when to use Vertex AI in express mode (api key).
