google-genai 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

google/genai/live.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 
-"""Live client."""
+"""Live client. The live module is experimental."""
 
 import asyncio
 import base64
@@ -25,6 +25,7 @@ from typing import AsyncIterator, Optional, Sequence, Union
 import google.auth
 from websockets import ConnectionClosed
 
+from . import _api_module
 from . import _common
 from . import _transformers as t
 from . import client
@@ -60,7 +61,7 @@ _FUNCTION_RESPONSE_REQUIRES_ID = (
 
 
 class AsyncSession:
-  """AsyncSession."""
+  """AsyncSession. The live module is experimental."""
 
   def __init__(self, api_client: client.ApiClient, websocket: ClientConnection):
     self._api_client = api_client
@@ -74,7 +75,6 @@ class AsyncSession:
           types.ContentListUnionDict,
           types.LiveClientContentOrDict,
           types.LiveClientRealtimeInputOrDict,
-          types.LiveClientRealtimeInputOrDict,
           types.LiveClientToolResponseOrDict,
           types.FunctionResponseOrDict,
           Sequence[types.FunctionResponseOrDict],
@@ -108,9 +108,11 @@ class AsyncSession:
 
     The method will yield the model responses from the server. The returned
     responses will represent a complete model turn. When the returned message
-    is fuction call, user must call `send` with the function response to
+    is function call, user must call `send` with the function response to
     continue the turn.
 
+    The live module is experimental.
+
     Yields:
       The model responses from the server.
 
@@ -142,6 +144,8 @@ class AsyncSession:
     input stream to the model and the other task will be used to receive the
     responses from the model.
 
+    The live module is experimental.
+
     Args:
       stream: An iterator that yields the model response.
       mime_type: The MIME type of the data in the stream.
@@ -380,6 +384,10 @@ class AsyncSession:
         isinstance(c, dict) and 'name' in c and 'response' in c for c in input
     ):
       # ToolResponse.FunctionResponse
+      if not (self._api_client.vertexai):
+        for item in input:
+          if 'id' not in item:
+            raise ValueError(_FUNCTION_RESPONSE_REQUIRES_ID)
       client_message = {'tool_response': {'function_responses': input}}
     elif isinstance(input, Sequence) and any(isinstance(c, str) for c in input):
       to_object = {}
@@ -466,8 +474,8 @@ class AsyncSession:
     await self._ws.close()
 
 
-class AsyncLive(_common.BaseModule):
-  """AsyncLive."""
+class AsyncLive(_api_module.BaseModule):
+  """AsyncLive. The live module is experimental."""
 
   def _LiveSetup_to_mldev(
       self, model: str, config: Optional[types.LiveConnectConfigOrDict] = None
@@ -634,6 +642,8 @@ class AsyncLive(_common.BaseModule):
   ) -> AsyncSession:
     """Connect to the live server.
 
+    The live module is experimental.
+
     Usage:
 
     .. code-block:: python