google-genai 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

google/genai/__init__.py

@@ -17,6 +17,6 @@
 
 from .client import Client
 
-__version__ = '0.1.0'
+__version__ = '0.2.1'
 
 __all__ = ['Client']

google/genai/_api_client.py

@@ -24,7 +24,7 @@ import json
 import os
 import sys
 from typing import Any, Optional, TypedDict, Union
-import urllib
+from urllib.parse import urlparse, urlunparse
 
 import google.auth
 import google.auth.credentials
@@ -51,7 +51,7 @@ class HttpOptions(TypedDict):
 def _append_library_version_headers(headers: dict[str, str]) -> None:
   """Appends the telemetry header to the headers dict."""
   # TODO: Automate revisions to the SDK library version.
-  library_label = f'google-genai-sdk/0.1.0'
+  library_label = f'google-genai-sdk/0.2.1'
   language_label = 'gl-python/' + sys.version.split()[0]
   version_header_value = f'{library_label} {language_label}'
   if (
@@ -89,6 +89,13 @@ def _patch_http_options(
   return copy_option
 
 
+def _join_url_path(base_url: str, path: str) -> str:
+  parsed_base = urlparse(base_url)
+  base_path = parsed_base.path[:-1] if parsed_base.path.endswith('/') else parsed_base.path
+  path = path[1:] if path.startswith('/') else path
+  return urlunparse(parsed_base._replace(path=base_path + '/' + path))
+
+
 @dataclass
 class HttpRequest:
   headers: dict[str, str]
@@ -193,7 +200,7 @@ class ApiClient:
       _append_library_version_headers(self._http_options['headers'])
 
   def _websocket_base_url(self):
-    url_parts = urllib.parse.urlparse(self._http_options['base_url'])
+    url_parts = urlparse(self._http_options['base_url'])
     return url_parts._replace(scheme='wss').geturl()
 
   def _build_request(
@@ -216,9 +223,10 @@ class ApiClient:
       patched_http_options = self._http_options
     if self.vertexai and not path.startswith('projects/'):
       path = f'projects/{self.project}/locations/{self.location}/' + path
-    url = urllib.parse.urljoin(
+    url = _join_url_path(
         patched_http_options['base_url'],
-        patched_http_options['api_version'] + '/' + path)
+        patched_http_options['api_version'] + '/' + path,
+    )
     return HttpRequest(
         method=http_method,
         url=url,
@@ -265,13 +273,13 @@ class ApiClient:
         data = http_request.data
 
     http_session = requests.Session()
-    async_request = requests.Request(
+    request = requests.Request(
         method=http_request.method,
         url=http_request.url,
         headers=http_request.headers,
         data=data,
     ).prepare()
-    response = http_session.send(async_request, stream=stream)
+    response = http_session.send(request, stream=stream)
     errors.APIError.raise_for_response(response)
     return HttpResponse(
         response.headers, response if stream else [response.text]

google/genai/batches.py

@@ -213,6 +213,30 @@ def _CreateBatchJobParameters_to_vertex(
   return to_object
 
 
+def _GetBatchJobConfig_to_mldev(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
+  return to_object
+
+
+def _GetBatchJobConfig_to_vertex(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
+  return to_object
+
+
 def _GetBatchJobParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -222,6 +246,15 @@ def _GetBatchJobParameters_to_mldev(
   if getv(from_object, ['name']):
     raise ValueError('name parameter is not supported in Google AI.')
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _GetBatchJobConfig_to_mldev(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
   return to_object
 
 
@@ -238,6 +271,39 @@ def _GetBatchJobParameters_to_vertex(
         t.t_batch_job_name(api_client, getv(from_object, ['name'])),
     )
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _GetBatchJobConfig_to_vertex(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
+  return to_object
+
+
+def _CancelBatchJobConfig_to_mldev(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
+  return to_object
+
+
+def _CancelBatchJobConfig_to_vertex(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
   return to_object
 
 
@@ -250,6 +316,15 @@ def _CancelBatchJobParameters_to_mldev(
   if getv(from_object, ['name']):
     raise ValueError('name parameter is not supported in Google AI.')
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _CancelBatchJobConfig_to_mldev(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
   return to_object
 
 
@@ -266,6 +341,15 @@ def _CancelBatchJobParameters_to_vertex(
         t.t_batch_job_name(api_client, getv(from_object, ['name'])),
     )
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _CancelBatchJobConfig_to_vertex(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
   return to_object
 
 
@@ -660,7 +744,9 @@ class Batches(_common.BaseModule):
     self.api_client._verify_response(return_value)
     return return_value
 
-  def get(self, *, name: str) -> types.BatchJob:
+  def get(
+      self, *, name: str, config: Optional[types.GetBatchJobConfigOrDict] = None
+  ) -> types.BatchJob:
     """Gets a batch job.
 
     Args:
@@ -681,6 +767,7 @@ class Batches(_common.BaseModule):
 
     parameter_model = types._GetBatchJobParameters(
         name=name,
+        config=config,
     )
 
     if not self.api_client.vertexai:
@@ -713,9 +800,15 @@ class Batches(_common.BaseModule):
     self.api_client._verify_response(return_value)
     return return_value
 
-  def cancel(self, *, name: str) -> None:
+  def cancel(
+      self,
+      *,
+      name: str,
+      config: Optional[types.CancelBatchJobConfigOrDict] = None,
+  ) -> None:
     parameter_model = types._CancelBatchJobParameters(
         name=name,
+        config=config,
     )
 
     if not self.api_client.vertexai:
@@ -947,7 +1040,9 @@ class AsyncBatches(_common.BaseModule):
     self.api_client._verify_response(return_value)
     return return_value
 
-  async def get(self, *, name: str) -> types.BatchJob:
+  async def get(
+      self, *, name: str, config: Optional[types.GetBatchJobConfigOrDict] = None
+  ) -> types.BatchJob:
     """Gets a batch job.
 
     Args:
@@ -968,6 +1063,7 @@ class AsyncBatches(_common.BaseModule):
 
     parameter_model = types._GetBatchJobParameters(
         name=name,
+        config=config,
     )
 
     if not self.api_client.vertexai:
@@ -1000,9 +1096,15 @@ class AsyncBatches(_common.BaseModule):
     self.api_client._verify_response(return_value)
     return return_value
 
-  async def cancel(self, *, name: str) -> None:
+  async def cancel(
+      self,
+      *,
+      name: str,
+      config: Optional[types.CancelBatchJobConfigOrDict] = None,
+  ) -> None:
     parameter_model = types._CancelBatchJobParameters(
         name=name,
+        config=config,
     )
 
     if not self.api_client.vertexai:

google/genai/chats.py

@@ -54,6 +54,7 @@ class Chat(_BaseChat):
     Usage:
 
     .. code-block:: python
+
       chat = client.chats.create(model='gemini-1.5-flash')
       response = chat.send_message('tell me a story')
     """
@@ -132,6 +133,7 @@ class AsyncChat(_BaseChat):
     Usage:
 
     .. code-block:: python
+
       chat = client.chats.create(model='gemini-1.5-flash')
       response = chat.send_message('tell me a story')
     """

google/genai/files.py

@@ -605,6 +605,7 @@ class Files(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       pager = client.files.list(config={'page_size': 10})
       for file in pager.page:
         print(file.name)
@@ -712,6 +713,7 @@ class Files(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       file = client.files.get(name='files/...')
       print(file.uri)
     """
@@ -766,6 +768,7 @@ class Files(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       client.files.delete(name='files/...')
     """
 
@@ -916,6 +919,7 @@ class AsyncFiles(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       pager = client.files.list(config={'page_size': 10})
       for file in pager.page:
         print(file.name)
@@ -1023,6 +1027,7 @@ class AsyncFiles(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       file = client.files.get(name='files/...')
       print(file.uri)
     """
@@ -1077,6 +1082,7 @@ class AsyncFiles(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       client.files.delete(name='files/...')
     """
 

google/genai/models.py

@@ -880,6 +880,11 @@ def _GenerateContentConfig_to_mldev(
         getv(from_object, ['response_modalities']),
     )
 
+  if getv(from_object, ['media_resolution']):
+    raise ValueError(
+        'media_resolution parameter is not supported in Google AI.'
+    )
+
   if getv(from_object, ['speech_config']) is not None:
     setv(
         to_object,
@@ -1022,6 +1027,11 @@ def _GenerateContentConfig_to_vertex(
         getv(from_object, ['response_modalities']),
     )
 
+  if getv(from_object, ['media_resolution']) is not None:
+    setv(
+        to_object, ['mediaResolution'], getv(from_object, ['media_resolution'])
+    )
+
   if getv(from_object, ['speech_config']) is not None:
     setv(
         to_object,
@@ -3733,6 +3743,7 @@ class Models(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       embeddings = client.models.embed_content(
           model= 'text-embedding-004',
           contents=[
@@ -4630,6 +4641,7 @@ class AsyncModels(_common.BaseModule):
     Usage:
 
     .. code-block:: python
+
       embeddings = client.models.embed_content(
           model= 'text-embedding-004',
           contents=[

google/genai/tunings.py

@@ -24,6 +24,30 @@ from ._common import set_value_by_path as setv
 from .pagers import AsyncPager, Pager
 
 
+def _GetTuningJobConfig_to_mldev(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
+  return to_object
+
+
+def _GetTuningJobConfig_to_vertex(
+    api_client: ApiClient,
+    from_object: Union[dict, object],
+    parent_object: dict = None,
+) -> dict:
+  to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
+  return to_object
+
+
 def _GetTuningJobParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -33,6 +57,15 @@ def _GetTuningJobParameters_to_mldev(
   if getv(from_object, ['name']) is not None:
     setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _GetTuningJobConfig_to_mldev(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
   return to_object
 
 
@@ -45,6 +78,15 @@ def _GetTuningJobParameters_to_vertex(
   if getv(from_object, ['name']) is not None:
     setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
+  if getv(from_object, ['config']) is not None:
+    setv(
+        to_object,
+        ['config'],
+        _GetTuningJobConfig_to_vertex(
+            api_client, getv(from_object, ['config']), to_object
+        ),
+    )
+
   return to_object
 
 
@@ -233,6 +275,9 @@ def _CreateTuningJobConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
   if getv(from_object, ['validation_dataset']):
     raise ValueError(
         'validation_dataset parameter is not supported in Google AI.'
@@ -288,6 +333,9 @@ def _CreateTuningJobConfig_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
   if getv(from_object, ['validation_dataset']) is not None:
     setv(
         parent_object,
@@ -455,6 +503,9 @@ def _CreateDistillationJobConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
   if getv(from_object, ['validation_dataset']):
     raise ValueError(
         'validation_dataset parameter is not supported in Google AI.'
@@ -498,6 +549,9 @@ def _CreateDistillationJobConfig_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+  if getv(from_object, ['http_options']) is not None:
+    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
+
   if getv(from_object, ['validation_dataset']) is not None:
     setv(
         parent_object,
@@ -896,7 +950,12 @@ def _TuningJobOrOperation_from_vertex(
 
 class Tunings(_common.BaseModule):
 
-  def _get(self, *, name: str) -> types.TuningJob:
+  def _get(
+      self,
+      *,
+      name: str,
+      config: Optional[types.GetTuningJobConfigOrDict] = None,
+  ) -> types.TuningJob:
     """Gets a TuningJob.
 
     Args:
@@ -908,6 +967,7 @@ class Tunings(_common.BaseModule):
 
     parameter_model = types._GetTuningJobParameters(
         name=name,
+        config=config,
     )
 
     if self.api_client.vertexai:
@@ -1129,8 +1189,13 @@ class Tunings(_common.BaseModule):
         config,
     )
 
-  def get(self, *, name: str) -> types.TuningJob:
-    job = self._get(name=name)
+  def get(
+      self,
+      *,
+      name: str,
+      config: Optional[types.GetTuningJobConfigOrDict] = None,
+  ) -> types.TuningJob:
+    job = self._get(name=name, config=config)
     if job.experiment and self.api_client.vertexai:
       _IpythonUtils.display_experiment_button(
           experiment=job.experiment,
@@ -1157,7 +1222,12 @@ class Tunings(_common.BaseModule):
 
 class AsyncTunings(_common.BaseModule):
 
-  async def _get(self, *, name: str) -> types.TuningJob:
+  async def _get(
+      self,
+      *,
+      name: str,
+      config: Optional[types.GetTuningJobConfigOrDict] = None,
+  ) -> types.TuningJob:
     """Gets a TuningJob.
 
     Args:
@@ -1169,6 +1239,7 @@ class AsyncTunings(_common.BaseModule):
 
     parameter_model = types._GetTuningJobParameters(
         name=name,
+        config=config,
     )
 
     if self.api_client.vertexai:
@@ -1390,8 +1461,13 @@ class AsyncTunings(_common.BaseModule):
         config,
     )
 
-  async def get(self, *, name: str) -> types.TuningJob:
-    job = await self._get(name=name)
+  async def get(
+      self,
+      *,
+      name: str,
+      config: Optional[types.GetTuningJobConfigOrDict] = None,
+  ) -> types.TuningJob:
+    job = await self._get(name=name, config=config)
     if job.experiment and self.api_client.vertexai:
       _IpythonUtils.display_experiment_button(
           experiment=job.experiment,

google/genai/types.py

@@ -154,6 +154,14 @@ DynamicRetrievalConfigMode = Literal["MODE_UNSPECIFIED", "MODE_DYNAMIC"]
 FunctionCallingConfigMode = Literal["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]
 
 
+MediaResolution = Literal[
+    "MEDIA_RESOLUTION_UNSPECIFIED",
+    "MEDIA_RESOLUTION_LOW",
+    "MEDIA_RESOLUTION_MEDIUM",
+    "MEDIA_RESOLUTION_HIGH",
+]
+
+
 SafetyFilterLevel = Literal[
     "BLOCK_LOW_AND_ABOVE",
     "BLOCK_MEDIUM_AND_ABOVE",
@@ -1524,6 +1532,11 @@ class GenerateContentConfig(_common.BaseModel):
       modalities that the model can return.
       """,
   )
+  media_resolution: Optional[MediaResolution] = Field(
+      default=None,
+      description="""If specified, the media resolution specified will be used.
+    """,
+  )
   speech_config: Optional[SpeechConfigUnion] = Field(
       default=None,
       description="""The speech generation configuration.
@@ -1647,6 +1660,10 @@ class GenerateContentConfigDict(TypedDict, total=False):
       modalities that the model can return.
       """
 
+  media_resolution: Optional[MediaResolution]
+  """If specified, the media resolution specified will be used.
+    """
+
   speech_config: Optional[SpeechConfigUnionDict]
   """The speech generation configuration.
       """
@@ -4210,10 +4227,31 @@ ComputeTokensResponseOrDict = Union[
 ]
 
 
+class GetTuningJobConfig(_common.BaseModel):
+  """Optional parameters for tunings.get method."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetTuningJobConfigDict(TypedDict, total=False):
+  """Optional parameters for tunings.get method."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+GetTuningJobConfigOrDict = Union[GetTuningJobConfig, GetTuningJobConfigDict]
+
+
 class _GetTuningJobParameters(_common.BaseModel):
   """Parameters for the get method."""
 
   name: Optional[str] = Field(default=None, description="""""")
+  config: Optional[GetTuningJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _GetTuningJobParametersDict(TypedDict, total=False):
@@ -4222,6 +4260,9 @@ class _GetTuningJobParametersDict(TypedDict, total=False):
   name: Optional[str]
   """"""
 
+  config: Optional[GetTuningJobConfigDict]
+  """Optional parameters for the request."""
+
 
 _GetTuningJobParametersOrDict = Union[
     _GetTuningJobParameters, _GetTuningJobParametersDict
@@ -5242,6 +5283,9 @@ TuningValidationDatasetOrDict = Union[
 class CreateTuningJobConfig(_common.BaseModel):
   """Supervised fine-tuning job creation request - optional fields."""
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   validation_dataset: Optional[TuningValidationDataset] = Field(
       default=None,
       description="""Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
@@ -5277,6 +5321,9 @@ class CreateTuningJobConfig(_common.BaseModel):
 class CreateTuningJobConfigDict(TypedDict, total=False):
   """Supervised fine-tuning job creation request - optional fields."""
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   validation_dataset: Optional[TuningValidationDatasetDict]
   """Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
 
@@ -5402,6 +5449,9 @@ DistillationValidationDatasetOrDict = Union[
 class CreateDistillationJobConfig(_common.BaseModel):
   """Distillation job creation request - optional fields."""
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   validation_dataset: Optional[DistillationValidationDataset] = Field(
       default=None,
       description="""Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.""",
@@ -5430,6 +5480,9 @@ class CreateDistillationJobConfig(_common.BaseModel):
 class CreateDistillationJobConfigDict(TypedDict, total=False):
   """Distillation job creation request - optional fields."""
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   validation_dataset: Optional[DistillationValidationDatasetDict]
   """Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file."""
 
@@ -6621,6 +6674,24 @@ class BatchJobDict(TypedDict, total=False):
 BatchJobOrDict = Union[BatchJob, BatchJobDict]
 
 
+class GetBatchJobConfig(_common.BaseModel):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetBatchJobConfigDict(TypedDict, total=False):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+GetBatchJobConfigOrDict = Union[GetBatchJobConfig, GetBatchJobConfigDict]
+
+
 class _GetBatchJobParameters(_common.BaseModel):
   """Config class for batches.get parameters."""
 
@@ -6631,6 +6702,9 @@ class _GetBatchJobParameters(_common.BaseModel):
     or "456" when project and location are initialized in the client.
     """,
   )
+  config: Optional[GetBatchJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _GetBatchJobParametersDict(TypedDict, total=False):
@@ -6642,12 +6716,35 @@ class _GetBatchJobParametersDict(TypedDict, total=False):
     or "456" when project and location are initialized in the client.
     """
 
+  config: Optional[GetBatchJobConfigDict]
+  """Optional parameters for the request."""
+
 
 _GetBatchJobParametersOrDict = Union[
     _GetBatchJobParameters, _GetBatchJobParametersDict
 ]
 
 
+class CancelBatchJobConfig(_common.BaseModel):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class CancelBatchJobConfigDict(TypedDict, total=False):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+CancelBatchJobConfigOrDict = Union[
+    CancelBatchJobConfig, CancelBatchJobConfigDict
+]
+
+
 class _CancelBatchJobParameters(_common.BaseModel):
   """Config class for batches.cancel parameters."""
 
@@ -6658,6 +6755,9 @@ class _CancelBatchJobParameters(_common.BaseModel):
     or "456" when project and location are initialized in the client.
     """,
   )
+  config: Optional[CancelBatchJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _CancelBatchJobParametersDict(TypedDict, total=False):
@@ -6669,6 +6769,9 @@ class _CancelBatchJobParametersDict(TypedDict, total=False):
     or "456" when project and location are initialized in the client.
     """
 
+  config: Optional[CancelBatchJobConfigDict]
+  """Optional parameters for the request."""
+
 
 _CancelBatchJobParametersOrDict = Union[
     _CancelBatchJobParameters, _CancelBatchJobParametersDict

{google_genai-0.1.0.dist-info → google_genai-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-genai
-Version: 0.1.0
+Version: 0.2.1
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -28,6 +28,13 @@ Requires-Dist: websockets<15.0dev,>=13.0
 
 # Google Gen AI SDK
 
+[![PyPI version](https://img.shields.io/pypi/v/google-genai.svg)](https://pypi.org/project/google-genai/)
+
+--------
+**Documentation:** https://googleapis.github.io/python-genai/
+
+-----
+
 ## Imports
 
 ``` python
@@ -504,7 +511,7 @@ client.models.generate_content(
 ## Tunings
 
 `client.tunings` contains tuning job APIs and supports supervised fine
-tuning through `tune` and distiallation through `distill`
+tuning through `tune` and distillation through `distill`
 
 ### Tune
 
@@ -512,7 +519,7 @@ tuning through `tune` and distiallation through `distill`
 -   Google AI supports tuning from inline examples
 
 ``` python
-if client._api_client.vertexai:
+if client.vertexai:
   model = 'gemini-1.5-pro-002'
   training_dataset=types.TuningDataset(
         gcs_uri='gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl',
@@ -591,9 +598,9 @@ for model in client.models.list(config={'page_size': 10}):
 ``` python
 pager = client.models.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -606,15 +613,15 @@ async for job in await client.aio.models.list(config={'page_size': 10}):
 ``` python
 async_pager = await client.aio.models.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ### Update Tuned Model
 
 ``` python
-model = pager.page[0]
+model = pager[0]
 
 model = client.models.update(
     model=model.name,
@@ -674,9 +681,9 @@ for job in client.tunings.list(config={'page_size': 10}):
 ``` python
 pager = client.tunings.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -689,9 +696,9 @@ async for job in await client.aio.tunings.list(config={'page_size': 10}):
 ``` python
 async_pager = await client.aio.tunings.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ## Batch Prediction
@@ -705,6 +712,8 @@ Only supported in Vertex AI.
 job = client.batches.create(
     model='gemini-1.5-flash-002',
     src='bq://my-project.my-dataset.my-table',
+)
+
 job
 ```
 
@@ -741,9 +750,9 @@ for job in client.batches.list(config={'page_size': 10}):
 ``` python
 pager = client.batches.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -754,11 +763,11 @@ async for job in await client.aio.batches.list(config={'page_size': 10}):
 ```
 
 ``` python
-async_pager = await client.aio.tunings.list(config={'page_size': 10})
+async_pager = await client.aio.batches.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ### Delete

{google_genai-0.1.0.dist-info → google_genai-0.2.1.dist-info}/RECORD

@@ -1,24 +1,24 @@
-google/genai/__init__.py,sha256=BkkJfmiA_SwVZvz_tSjJaEYEVh75HXC9UcH91FRSjUU,674
-google/genai/_api_client.py,sha256=9fSVC8elIVV-EwU1-GmC95OxHptdWrIrLRNs6Dvcs8Y,15638
+google/genai/__init__.py,sha256=AfCfM5DuvXAhgpzeb3XgTna3xu11Kpom6mSzjUeo91M,674
+google/genai/_api_client.py,sha256=jtiWyDID-keembuyvSbFeQaxlQXKef75MK-ofhHMny8,15950
 google/genai/_automatic_function_calling_util.py,sha256=E25_66RH3DbDIucq7x-93XWPPBwB9FnzwD1NCGyPrjM,10242
 google/genai/_common.py,sha256=Yj5cBkq5QRNFSBqvpB949Rjo7cbIhdtKp5dJxMW_I6I,7971
 google/genai/_extra_utils.py,sha256=GQZnraFCrMffqrBEpurdcBmgrltRsnYgMizt-Ok6xX8,11098
 google/genai/_replay_api_client.py,sha256=QPNg4SBpOLS58bx-kuJQngxy1tbjMpCpJzmImCwYePA,16226
 google/genai/_test_api_client.py,sha256=p771T27icmzENxKtyNDwPG1sTI7jaoJNFPwlwq9GK6o,4759
 google/genai/_transformers.py,sha256=_zWNr7zFTrUFniECYaZUn0n4TdioLpj783l3-z1XvIE,13443
-google/genai/batches.py,sha256=gAuFZOKosJj-GYHvftFdbBuw6Y9VkFC9scOjeUssAC0,34632
+google/genai/batches.py,sha256=Wi4Kptampp2WepAqv_AawwNCR6MKVhLKmzJdYXDQ_aE,37148
 google/genai/caches.py,sha256=LJm2raykec7_iCHsVbEtX4v942mR-OSQvxTVKcBN2RA,53434
-google/genai/chats.py,sha256=QcOqW87D5bpGQ_78xTu9y1G28dQx9iUEyqNFwPsAFdU,5686
+google/genai/chats.py,sha256=x-vCXrsxZ8kdEZ_0ZDfrBQnQ9urCr42x3urP0OXHyTo,5688
 google/genai/client.py,sha256=HH_lYnjPOwW-4Vgynyw4K8cwurT2g578Dc51H_uk7GY,9244
 google/genai/errors.py,sha256=TrlUk1jz7r1aN1lrL3FZZ30LU4iMfSonm1ZwEAk07k4,3048
-google/genai/files.py,sha256=9B1Xsb2R7jBAU3Ot-1Z4YJjC9gQEywzAfLT24yHN4EI,35567
+google/genai/files.py,sha256=dn3q8P9aTN9OG3PtA4AYDs9hF6Uk-jkMjgAW7dSlt_4,35573
 google/genai/live.py,sha256=T-pOtq7k43wE2VjQzqLrx-kqhotS66I2PY_NHBdv9G8,22056
-google/genai/models.py,sha256=1slDoZuGq91B7erpxeBaFGfv1PAcIPV4j1Gqg-aHhEs,154534
+google/genai/models.py,sha256=t5XgwlgkNrQKb6eww0oBGzjMiMQaj-BQedc8lVdJHz4,154834
 google/genai/pagers.py,sha256=hSHd-gLvEzYWwK85i8EcFNWUMKtszUs7Nw2r3L7d6_U,6686
-google/genai/tunings.py,sha256=EYA9FykWZSkhcHkHcCO7VvYkFBVzQCWYX2umtIhlhQQ,47018
-google/genai/types.py,sha256=73tKrlpCmV4PRb2HMYHVZXmiHTS2iK-CL89uJvNE54Q,260869
-google_genai-0.1.0.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-0.1.0.dist-info/METADATA,sha256=1AAUKL2kZ6sGbkfQdUr-rSWttYg1qFAVXe9mneSikSc,17260
-google_genai-0.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-google_genai-0.1.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-0.1.0.dist-info/RECORD,,
+google/genai/tunings.py,sha256=tFTSEaECKZ6xeYcxUTIKUmXqPoDymYP3eyTcEKjnPa4,49010
+google/genai/types.py,sha256=mIjtCSXbp6CRL5iEhtdxczoMTtyQ1EKYpBlzLvGIedY,263841
+google_genai-0.2.1.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-0.2.1.dist-info/METADATA,sha256=fouiYjPlS5vhJhisQX7M3qJx3yzyn8owWdpsslFtyAc,17371
+google_genai-0.2.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+google_genai-0.2.1.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-0.2.1.dist-info/RECORD,,