feat: Make websockets an optional dependency for live module

google/genai/live.py

@@ -25,7 +25,24 @@
 
 import google.auth
 import pydantic
-from websockets import ConnectionClosed
+
+# MODIFIED: Wrap all websockets imports in a single try-except ImportError block
+try:
+    from websockets import ConnectionClosed
+
+    # Original try-except block for different websockets APIs/versions is preserved inside
+    try:
+      from websockets.asyncio.client import ClientConnection
+      from websockets.asyncio.client import connect
+    except ModuleNotFoundError:
+      # This try/except is for TAP, mypy complains about it which is why we have the type: ignore
+      from websockets.client import ClientConnection  # type: ignore
+      from websockets.client import connect  # type: ignore
+except ImportError:
+    raise ImportError(
+        "The `websockets` library is required for live, interactive sessions. "
+        "Please install it using `pip install google-genai[live]`."
+    )
 
 from . import _api_module
 from . import _common
@@ -40,14 +57,6 @@
 from .models import _Content_to_vertex
 
 
-try:
-  from websockets.asyncio.client import ClientConnection
-  from websockets.asyncio.client import connect
-except ModuleNotFoundError:
-  # This try/except is for TAP, mypy complains about it which is why we have the type: ignore
-  from websockets.client import ClientConnection  # type: ignore
-  from websockets.client import connect  # type: ignore
-
 logger = logging.getLogger('google_genai.live')
 
 _FUNCTION_RESPONSE_REQUIRES_ID = (

pyproject.toml

@@ -31,10 +31,12 @@ dependencies = [
     "httpx>=0.28.1, <1.0.0",
     "pydantic>=2.0.0, <3.0.0",
     "requests>=2.28.1, <3.0.0",
-    "websockets>=13.0.0, <15.1.0",
     "typing-extensions>=4.11.0, <5.0.0",
 ]
 
+[project.optional-dependencies]
+live = ["websockets>=13.0.0, <15.1.0"]
+
 [project.urls]
 Homepage = "https://github.com/googleapis/python-genai"