google-genai 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

google/genai/chats.py

@@ -109,7 +109,7 @@ class Chat(_BaseChat):
     .. code-block:: python
 
       chat = client.chats.create(model='gemini-1.5-flash')
-      for chunk in chat.send_message_stream('tell me a story')
+      for chunk in chat.send_message_stream('tell me a story'):
         print(chunk.text)
     """
 
@@ -215,7 +215,7 @@ class AsyncChat(_BaseChat):
 
     .. code-block:: python
       chat = client.aio.chats.create(model='gemini-1.5-flash')
-      async for chunk in chat.send_message_stream('tell me a story')
+      async for chunk in chat.send_message_stream('tell me a story'):
         print(chunk.text)
     """
 

google/genai/client.py

@@ -14,12 +14,12 @@
 #
 
 import os
-from typing import Optional
+from typing import Optional, Union
 
 import google.auth
 import pydantic
 
-from ._api_client import ApiClient, HttpOptions
+from ._api_client import ApiClient, HttpOptions, HttpOptionsDict
 from ._replay_api_client import ReplayApiClient
 from .batches import AsyncBatches, Batches
 from .caches import AsyncCaches, Caches
@@ -145,7 +145,7 @@ class Client:
       project: Optional[str] = None,
       location: Optional[str] = None,
       debug_config: Optional[DebugConfig] = None,
-      http_options: Optional[HttpOptions] = None,
+      http_options: Optional[Union[HttpOptions, HttpOptionsDict]] = None,
   ):
     """Initializes the client.
 
@@ -179,6 +179,9 @@ class Client:
           debug_config (DebugConfig):
               Config settings that control network
               behavior of the client. This is typically used when running test code.
+          http_options (Union[HttpOptions, HttpOptionsDict]):
+              Http options to use for the client. Response_payload can't be
+              set when passing to the client constructor.
     """
 
     self._debug_config = debug_config or DebugConfig()

google/genai/files.py

@@ -98,7 +98,7 @@ def _ListFilesParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -128,13 +128,13 @@ def _FileStatus_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['details']):
+  if getv(from_object, ['details']) is not None:
     raise ValueError('details parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['message']):
+  if getv(from_object, ['message']) is not None:
     raise ValueError('message parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['code']):
+  if getv(from_object, ['code']) is not None:
     raise ValueError('code parameter is not supported in Vertex AI.')
 
   return to_object
@@ -197,40 +197,40 @@ def _File_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['display_name']):
+  if getv(from_object, ['display_name']) is not None:
     raise ValueError('display_name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['mime_type']):
+  if getv(from_object, ['mime_type']) is not None:
     raise ValueError('mime_type parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['size_bytes']):
+  if getv(from_object, ['size_bytes']) is not None:
     raise ValueError('size_bytes parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['create_time']):
+  if getv(from_object, ['create_time']) is not None:
     raise ValueError('create_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['expiration_time']):
+  if getv(from_object, ['expiration_time']) is not None:
     raise ValueError('expiration_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['update_time']):
+  if getv(from_object, ['update_time']) is not None:
     raise ValueError('update_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['sha256_hash']):
+  if getv(from_object, ['sha256_hash']) is not None:
     raise ValueError('sha256_hash parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['uri']):
+  if getv(from_object, ['uri']) is not None:
     raise ValueError('uri parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['state']):
+  if getv(from_object, ['state']) is not None:
     raise ValueError('state parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['video_metadata']):
+  if getv(from_object, ['video_metadata']) is not None:
     raise ValueError('video_metadata parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['error']):
+  if getv(from_object, ['error']) is not None:
     raise ValueError('error parameter is not supported in Vertex AI.')
 
   return to_object
@@ -291,10 +291,10 @@ def _CreateFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['file']):
+  if getv(from_object, ['file']) is not None:
     raise ValueError('file parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -355,10 +355,10 @@ def _GetFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -419,10 +419,10 @@ def _DeleteFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object

google/genai/live.py

@@ -68,6 +68,7 @@ class AsyncSession:
 
   async def send(
       self,
+      *,
       input: Union[
           types.ContentListUnion,
           types.ContentListUnionDict,
@@ -80,6 +81,25 @@ class AsyncSession:
       ],
       end_of_turn: Optional[bool] = False,
   ):
+    """Send input to the model.
+
+    The method will send the input request to the server.
+
+    Args:
+      input: The input request to the model.
+      end_of_turn: Whether the input is the last message in a turn.
+
+    Example usage:
+
+    .. code-block:: python
+
+      client = genai.Client(api_key=API_KEY)
+
+      async with client.aio.live.connect(model='...') as session:
+        await session.send(input='Hello world!', end_of_turn=True)
+        async for message in session.receive():
+          print(message)
+    """
     client_message = self._parse_client_message(input, end_of_turn)
     await self._ws.send(json.dumps(client_message))
 
@@ -113,7 +133,7 @@ class AsyncSession:
       yield result
 
   async def start_stream(
-      self, stream: AsyncIterator[bytes], mime_type: str
+      self, *, stream: AsyncIterator[bytes], mime_type: str
   ) -> AsyncIterator[types.LiveServerMessage]:
     """start a live session from a data stream.
 
@@ -199,7 +219,7 @@ class AsyncSession:
   ):
     async for data in data_stream:
       input = {'data': data, 'mimeType': mime_type}
-      await self.send(input)
+      await self.send(input=input)
       # Give a chance for the receive loop to process responses.
       await asyncio.sleep(10**-12)
     # Give a chance for the receiver to process the last response.
@@ -599,7 +619,10 @@ class AsyncLive(_common.BaseModule):
 
   @contextlib.asynccontextmanager
   async def connect(
-      self, model: str, config: Optional[types.LiveConnectConfigOrDict] = None
+      self,
+      *,
+      model: str,
+      config: Optional[types.LiveConnectConfigOrDict] = None,
   ) -> AsyncSession:
     """Connect to the live server.
 
@@ -609,9 +632,9 @@ class AsyncLive(_common.BaseModule):
 
       client = genai.Client(api_key=API_KEY)
       config = {}
-      async with client.aio.live.connect(model='gemini-1.0-pro-002', config=config) as session:
+      async with client.aio.live.connect(model='...', config=config) as session:
         await session.send(input='Hello world!', end_of_turn=True)
-        async for message in session:
+        async for message in session.receive():
           print(message)
     """
     base_url = self.api_client._websocket_base_url()