google-genai 1.23.0__py3-none-any.whl → 1.24.0__py3-none-any.whl

google/genai/_api_client.py

@@ -74,8 +74,6 @@ try:
 except ImportError:
   pass
 
-# internal comment
-
 
 if TYPE_CHECKING:
   from multidict import CIMultiDictProxy
@@ -338,9 +336,11 @@ class HttpResponse:
 
 # Default retry options.
 # The config is based on https://cloud.google.com/storage/docs/retry-strategy.
-_RETRY_ATTEMPTS = 3
+# By default, the client will retry 4 times with approximately 1.0, 2.0, 4.0,
+# 8.0 seconds between each attempt.
+_RETRY_ATTEMPTS = 5  # including the initial call.
 _RETRY_INITIAL_DELAY = 1.0  # seconds
-_RETRY_MAX_DELAY = 120.0  # seconds
+_RETRY_MAX_DELAY = 60.0  # seconds
 _RETRY_EXP_BASE = 2
 _RETRY_JITTER = 1
 _RETRY_HTTP_STATUS_CODES = (
@@ -364,14 +364,13 @@ def _retry_args(options: Optional[HttpRetryOptions]) -> dict[str, Any]:
     The arguments passed to the tenacity.(Async)Retrying constructor.
   """
   if options is None:
-    return {'stop': tenacity.stop_after_attempt(1)}
+    return {'stop': tenacity.stop_after_attempt(1), 'reraise': True}
 
   stop = tenacity.stop_after_attempt(options.attempts or _RETRY_ATTEMPTS)
   retriable_codes = options.http_status_codes or _RETRY_HTTP_STATUS_CODES
-  retry = tenacity.retry_if_result(
-      lambda response: response.status_code in retriable_codes,
+  retry = tenacity.retry_if_exception(
+      lambda e: isinstance(e, errors.APIError) and e.code in retriable_codes,
   )
-  retry_error_callback = lambda retry_state: retry_state.outcome.result()
   wait = tenacity.wait_exponential_jitter(
       initial=options.initial_delay or _RETRY_INITIAL_DELAY,
       max=options.max_delay or _RETRY_MAX_DELAY,
@@ -381,7 +380,7 @@ def _retry_args(options: Optional[HttpRetryOptions]) -> dict[str, Any]:
   return {
       'stop': stop,
       'retry': retry,
-      'retry_error_callback': retry_error_callback,
+      'reraise': True,
       'wait': wait,
   }
 
@@ -569,18 +568,16 @@ class BaseApiClient:
     )
     self._httpx_client = SyncHttpxClient(**client_args)
     self._async_httpx_client = AsyncHttpxClient(**async_client_args)
-    if has_aiohttp:
+    if self._use_aiohttp():
       # Do it once at the genai.Client level. Share among all requests.
       self._async_client_session_request_args = self._ensure_aiohttp_ssl_ctx(
           self._http_options
-      ) 
-    self._websocket_ssl_ctx = self._ensure_websocket_ssl_ctx(
-        self._http_options
-    )
+      )
+    self._websocket_ssl_ctx = self._ensure_websocket_ssl_ctx(self._http_options)
 
     retry_kwargs = _retry_args(self._http_options.retry_options)
-    self._retry = tenacity.Retrying(**retry_kwargs, reraise=True)
-    self._async_retry = tenacity.AsyncRetrying(**retry_kwargs, reraise=True)
+    self._retry = tenacity.Retrying(**retry_kwargs)
+    self._async_retry = tenacity.AsyncRetrying(**retry_kwargs)
 
   @staticmethod
   def _ensure_httpx_ssl_ctx(
@@ -706,7 +703,6 @@ class BaseApiClient:
 
     return _maybe_set(async_args, ctx)
 
-
   @staticmethod
   def _ensure_websocket_ssl_ctx(options: HttpOptions) -> dict[str, Any]:
     """Ensures the SSL context is present in the async client args.
@@ -762,6 +758,14 @@ class BaseApiClient:
 
     return _maybe_set(async_args, ctx)
 
+  def _use_aiohttp(self) -> bool:
+    # If the instantiator has passed a custom transport, they want httpx not
+    # aiohttp.
+    return (
+        has_aiohttp
+        and (self._http_options.async_client_args or {}).get('transport')
+        is None
+    )
 
   def _websocket_base_url(self) -> str:
     url_parts = urlparse(self._http_options.base_url)
@@ -975,7 +979,7 @@ class BaseApiClient:
           data = http_request.data
 
     if stream:
-      if has_aiohttp:
+      if self._use_aiohttp():
         session = aiohttp.ClientSession(
             headers=http_request.headers,
             trust_env=True,
@@ -1007,7 +1011,7 @@ class BaseApiClient:
         await errors.APIError.raise_for_async_response(client_response)
         return HttpResponse(client_response.headers, client_response)
     else:
-      if has_aiohttp:
+      if self._use_aiohttp():
         async with aiohttp.ClientSession(
             headers=http_request.headers,
             trust_env=True,
@@ -1061,11 +1065,10 @@ class BaseApiClient:
         http_method, path, request_dict, http_options
     )
     response = self._request(http_request, stream=False)
-    response_body = response.response_stream[0] if response.response_stream else ''
-    return SdkHttpResponse(
-        headers=response.headers, body=response_body
+    response_body = (
+        response.response_stream[0] if response.response_stream else ''
     )
-
+    return SdkHttpResponse(headers=response.headers, body=response_body)
 
   def request_streamed(
       self,
@@ -1080,7 +1083,9 @@ class BaseApiClient:
 
     session_response = self._request(http_request, stream=True)
     for chunk in session_response.segments():
-      yield SdkHttpResponse(headers=session_response.headers, body=json.dumps(chunk))
+      yield SdkHttpResponse(
+          headers=session_response.headers, body=json.dumps(chunk)
+      )
 
   async def async_request(
       self,
@@ -1095,10 +1100,7 @@ class BaseApiClient:
 
     result = await self._async_request(http_request=http_request, stream=False)
     response_body = result.response_stream[0] if result.response_stream else ''
-    return SdkHttpResponse(
-        headers=result.headers, body=response_body
-    )
-
+    return SdkHttpResponse(headers=result.headers, body=response_body)
 
   async def async_request_streamed(
       self,
@@ -1324,7 +1326,7 @@ class BaseApiClient:
     """
     offset = 0
     # Upload the file in chunks
-    if has_aiohttp:  # pylint: disable=g-import-not-at-top
+    if self._use_aiohttp():  # pylint: disable=g-import-not-at-top
       async with aiohttp.ClientSession(
           headers=self._http_options.headers,
           trust_env=True,
@@ -1507,7 +1509,7 @@ class BaseApiClient:
       else:
         data = http_request.data
 
-    if has_aiohttp:
+    if self._use_aiohttp():
       async with aiohttp.ClientSession(
           headers=http_request.headers,
           trust_env=True,

google/genai/_live_converters.py

@@ -877,6 +877,9 @@ def _Tool_to_mldev(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -942,6 +945,9 @@ def _Tool_to_vertex(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -1649,16 +1655,16 @@ def _LiveSendRealtimeInputParameters_to_vertex(
     setv(to_object, ['mediaChunks'], t.t_blobs(getv(from_object, ['media'])))
 
   if getv(from_object, ['audio']) is not None:
-    raise ValueError('audio parameter is not supported in Vertex AI.')
+    setv(to_object, ['audio'], t.t_audio_blob(getv(from_object, ['audio'])))
 
   if getv(from_object, ['audio_stream_end']) is not None:
     setv(to_object, ['audioStreamEnd'], getv(from_object, ['audio_stream_end']))
 
   if getv(from_object, ['video']) is not None:
-    raise ValueError('video parameter is not supported in Vertex AI.')
+    setv(to_object, ['video'], t.t_image_blob(getv(from_object, ['video'])))
 
   if getv(from_object, ['text']) is not None:
-    raise ValueError('text parameter is not supported in Vertex AI.')
+    setv(to_object, ['text'], getv(from_object, ['text']))
 
   if getv(from_object, ['activity_start']) is not None:
     setv(
@@ -1935,7 +1941,7 @@ def _LiveClientRealtimeInput_to_vertex(
     setv(to_object, ['mediaChunks'], getv(from_object, ['media_chunks']))
 
   if getv(from_object, ['audio']) is not None:
-    raise ValueError('audio parameter is not supported in Vertex AI.')
+    setv(to_object, ['audio'], getv(from_object, ['audio']))
 
   if getv(from_object, ['audio_stream_end']) is not None:
     raise ValueError(
@@ -1943,10 +1949,10 @@ def _LiveClientRealtimeInput_to_vertex(
     )
 
   if getv(from_object, ['video']) is not None:
-    raise ValueError('video parameter is not supported in Vertex AI.')
+    setv(to_object, ['video'], getv(from_object, ['video']))
 
   if getv(from_object, ['text']) is not None:
-    raise ValueError('text parameter is not supported in Vertex AI.')
+    setv(to_object, ['text'], getv(from_object, ['text']))
 
   if getv(from_object, ['activity_start']) is not None:
     setv(
@@ -2467,6 +2473,8 @@ def _LiveServerSetupComplete_from_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sessionId']) is not None:
+    setv(to_object, ['session_id'], getv(from_object, ['sessionId']))
 
   return to_object
 

google/genai/_tokens_converters.py

@@ -877,6 +877,9 @@ def _Tool_to_mldev(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -942,6 +945,9 @@ def _Tool_to_vertex(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 

google/genai/batches.py

@@ -515,6 +515,9 @@ def _Tool_to_mldev(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -1288,6 +1291,25 @@ def _ListBatchJobsParameters_to_mldev(
   return to_object
 
 
+def _DeleteBatchJobParameters_to_mldev(
+    api_client: BaseApiClient,
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['name']) is not None:
+    setv(
+        to_object,
+        ['_url', 'name'],
+        t.t_batch_job_name(api_client, getv(from_object, ['name'])),
+    )
+
+  if getv(from_object, ['config']) is not None:
+    setv(to_object, ['config'], getv(from_object, ['config']))
+
+  return to_object
+
+
 def _VideoMetadata_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1796,6 +1818,9 @@ def _Tool_to_vertex(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -3036,6 +3061,9 @@ def _Tool_from_mldev(
   if getv(from_object, ['codeExecution']) is not None:
     setv(to_object, ['code_execution'], getv(from_object, ['codeExecution']))
 
+  if getv(from_object, ['computerUse']) is not None:
+    setv(to_object, ['computer_use'], getv(from_object, ['computerUse']))
+
   return to_object
 
 
@@ -3697,6 +3725,27 @@ def _ListBatchJobsResponse_from_mldev(
   return to_object
 
 
+def _DeleteResourceJob_from_mldev(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['name'], getv(from_object, ['name']))
+
+  if getv(from_object, ['done']) is not None:
+    setv(to_object, ['done'], getv(from_object, ['done']))
+
+  if getv(from_object, ['error']) is not None:
+    setv(
+        to_object,
+        ['error'],
+        _JobError_from_mldev(getv(from_object, ['error']), to_object),
+    )
+
+  return to_object
+
+
 def _JobError_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -4220,6 +4269,9 @@ def _Tool_from_vertex(
   if getv(from_object, ['codeExecution']) is not None:
     setv(to_object, ['code_execution'], getv(from_object, ['codeExecution']))
 
+  if getv(from_object, ['computerUse']) is not None:
+    setv(to_object, ['computer_use'], getv(from_object, ['computerUse']))
+
   return to_object
 
 
@@ -4937,7 +4989,7 @@ class Batches(_api_module.BaseModule):
       name (str): A fully-qualified BatchJob resource name or ID.
         Example: "projects/.../locations/.../batchPredictionJobs/456" or "456"
           when project and location are initialized in the Vertex AI client. Or
-          "files/abc" using the Gemini Developer AI client.
+          "batches/abc" using the Gemini Developer AI client.
 
     Returns:
       A BatchJob object that contains details about the batch job.
@@ -5022,7 +5074,7 @@ class Batches(_api_module.BaseModule):
       name (str): A fully-qualified BatchJob resource name or ID.
         Example: "projects/.../locations/.../batchPredictionJobs/456" or "456"
           when project and location are initialized in the Vertex AI client. Or
-          "files/abc" using the Gemini Developer AI client.
+          "batches/abc" using the Gemini Developer AI client.
 
     Usage:
 
@@ -5161,9 +5213,8 @@ class Batches(_api_module.BaseModule):
     )
 
     request_url_dict: Optional[dict[str, str]]
-    if not self._api_client.vertexai:
-      raise ValueError('This method is only supported in the Vertex AI client.')
-    else:
+
+    if self._api_client.vertexai:
       request_dict = _DeleteBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
@@ -5172,7 +5223,15 @@ class Batches(_api_module.BaseModule):
         path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
       else:
         path = 'batchPredictionJobs/{name}'
-
+    else:
+      request_dict = _DeleteBatchJobParameters_to_mldev(
+          self._api_client, parameter_model
+      )
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batches/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batches/{name}'
     query_params = request_dict.get('_query')
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
@@ -5198,6 +5257,9 @@ class Batches(_api_module.BaseModule):
     if self._api_client.vertexai:
       response_dict = _DeleteResourceJob_from_vertex(response_dict)
 
+    else:
+      response_dict = _DeleteResourceJob_from_mldev(response_dict)
+
     return_value = types.DeleteResourceJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
@@ -5354,7 +5416,7 @@ class AsyncBatches(_api_module.BaseModule):
       name (str): A fully-qualified BatchJob resource name or ID.
         Example: "projects/.../locations/.../batchPredictionJobs/456" or "456"
           when project and location are initialized in the Vertex AI client. Or
-          "files/abc" using the Gemini Developer AI client.
+          "batches/abc" using the Gemini Developer AI client.
 
     Returns:
       A BatchJob object that contains details about the batch job.
@@ -5441,7 +5503,7 @@ class AsyncBatches(_api_module.BaseModule):
       name (str): A fully-qualified BatchJob resource name or ID.
         Example: "projects/.../locations/.../batchPredictionJobs/456" or "456"
           when project and location are initialized in the Vertex AI client. Or
-          "files/abc" using the Gemini Developer AI client.
+          "batches/abc" using the Gemini Developer AI client.
 
     Usage:
 
@@ -5582,9 +5644,8 @@ class AsyncBatches(_api_module.BaseModule):
     )
 
     request_url_dict: Optional[dict[str, str]]
-    if not self._api_client.vertexai:
-      raise ValueError('This method is only supported in the Vertex AI client.')
-    else:
+
+    if self._api_client.vertexai:
       request_dict = _DeleteBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
@@ -5593,7 +5654,15 @@ class AsyncBatches(_api_module.BaseModule):
         path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
       else:
         path = 'batchPredictionJobs/{name}'
-
+    else:
+      request_dict = _DeleteBatchJobParameters_to_mldev(
+          self._api_client, parameter_model
+      )
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batches/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batches/{name}'
     query_params = request_dict.get('_query')
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
@@ -5619,6 +5688,9 @@ class AsyncBatches(_api_module.BaseModule):
     if self._api_client.vertexai:
       response_dict = _DeleteResourceJob_from_vertex(response_dict)
 
+    else:
+      response_dict = _DeleteResourceJob_from_mldev(response_dict)
+
     return_value = types.DeleteResourceJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )

google/genai/caches.py

@@ -395,6 +395,9 @@ def _Tool_to_mldev(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -1057,6 +1060,9 @@ def _Tool_to_vertex(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 

google/genai/models.py

@@ -517,6 +517,9 @@ def _Tool_to_mldev(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 
@@ -1958,6 +1961,9 @@ def _Tool_to_vertex(
   if getv(from_object, ['code_execution']) is not None:
     setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
 
+  if getv(from_object, ['computer_use']) is not None:
+    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+
   return to_object
 
 

google/genai/tunings.py

@@ -521,6 +521,12 @@ def _TuningJob_from_mldev(
   if getv(from_object, ['pipelineJob']) is not None:
     setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
 
+  if getv(from_object, ['satisfiesPzi']) is not None:
+    setv(to_object, ['satisfies_pzi'], getv(from_object, ['satisfiesPzi']))
+
+  if getv(from_object, ['satisfiesPzs']) is not None:
+    setv(to_object, ['satisfies_pzs'], getv(from_object, ['satisfiesPzs']))
+
   if getv(from_object, ['serviceAccount']) is not None:
     setv(to_object, ['service_account'], getv(from_object, ['serviceAccount']))
 
@@ -700,6 +706,12 @@ def _TuningJob_from_vertex(
   if getv(from_object, ['pipelineJob']) is not None:
     setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
 
+  if getv(from_object, ['satisfiesPzi']) is not None:
+    setv(to_object, ['satisfies_pzi'], getv(from_object, ['satisfiesPzi']))
+
+  if getv(from_object, ['satisfiesPzs']) is not None:
+    setv(to_object, ['satisfies_pzs'], getv(from_object, ['satisfiesPzs']))
+
   if getv(from_object, ['serviceAccount']) is not None:
     setv(to_object, ['service_account'], getv(from_object, ['serviceAccount']))
 

google/genai/types.py

@@ -138,6 +138,18 @@ class HarmCategory(_common.CaseInSensitiveEnum):
   """The harm category is sexually explicit content."""
   HARM_CATEGORY_CIVIC_INTEGRITY = 'HARM_CATEGORY_CIVIC_INTEGRITY'
   """Deprecated: Election filter is not longer supported. The harm category is civic integrity."""
+  HARM_CATEGORY_IMAGE_HATE = 'HARM_CATEGORY_IMAGE_HATE'
+  """The harm category is image hate."""
+  HARM_CATEGORY_IMAGE_DANGEROUS_CONTENT = (
+      'HARM_CATEGORY_IMAGE_DANGEROUS_CONTENT'
+  )
+  """The harm category is image dangerous content."""
+  HARM_CATEGORY_IMAGE_HARASSMENT = 'HARM_CATEGORY_IMAGE_HARASSMENT'
+  """The harm category is image harassment."""
+  HARM_CATEGORY_IMAGE_SEXUALLY_EXPLICIT = (
+      'HARM_CATEGORY_IMAGE_SEXUALLY_EXPLICIT'
+  )
+  """The harm category is image sexually explicit content."""
 
 
 class HarmBlockMethod(_common.CaseInSensitiveEnum):
@@ -199,6 +211,37 @@ class AuthType(_common.CaseInSensitiveEnum):
   """OpenID Connect (OIDC) Auth."""
 
 
+class ApiSpec(_common.CaseInSensitiveEnum):
+  """The API spec that the external API implements."""
+
+  API_SPEC_UNSPECIFIED = 'API_SPEC_UNSPECIFIED'
+  """Unspecified API spec. This value should not be used."""
+  SIMPLE_SEARCH = 'SIMPLE_SEARCH'
+  """Simple search API spec."""
+  ELASTIC_SEARCH = 'ELASTIC_SEARCH'
+  """Elastic search API spec."""
+
+
+class Environment(_common.CaseInSensitiveEnum):
+  """Required. The environment being operated."""
+
+  ENVIRONMENT_UNSPECIFIED = 'ENVIRONMENT_UNSPECIFIED'
+  """Defaults to browser."""
+  ENVIRONMENT_BROWSER = 'ENVIRONMENT_BROWSER'
+  """Operates in a web browser."""
+
+
+class UrlRetrievalStatus(_common.CaseInSensitiveEnum):
+  """Status of the url retrieval."""
+
+  URL_RETRIEVAL_STATUS_UNSPECIFIED = 'URL_RETRIEVAL_STATUS_UNSPECIFIED'
+  """Default value. This value is unused"""
+  URL_RETRIEVAL_STATUS_SUCCESS = 'URL_RETRIEVAL_STATUS_SUCCESS'
+  """Url retrieval is successful."""
+  URL_RETRIEVAL_STATUS_ERROR = 'URL_RETRIEVAL_STATUS_ERROR'
+  """Url retrieval is failed due to error."""
+
+
 class FinishReason(_common.CaseInSensitiveEnum):
   """Output only. The reason why the model stopped generating tokens.
 
@@ -276,6 +319,8 @@ class BlockedReason(_common.CaseInSensitiveEnum):
   """Candidates blocked due to the terms which are included from the terminology blocklist."""
   PROHIBITED_CONTENT = 'PROHIBITED_CONTENT'
   """Candidates blocked due to prohibited content."""
+  IMAGE_SAFETY = 'IMAGE_SAFETY'
+  """Candidates blocked due to unsafe image generation content."""
 
 
 class TrafficType(_common.CaseInSensitiveEnum):
@@ -411,17 +456,6 @@ class FunctionCallingConfigMode(_common.CaseInSensitiveEnum):
   """Model will not predict any function calls. Model behavior is same as when not passing any function declarations."""
 
 
-class UrlRetrievalStatus(_common.CaseInSensitiveEnum):
-  """Status of the url retrieval."""
-
-  URL_RETRIEVAL_STATUS_UNSPECIFIED = 'URL_RETRIEVAL_STATUS_UNSPECIFIED'
-  """Default value. This value is unused"""
-  URL_RETRIEVAL_STATUS_SUCCESS = 'URL_RETRIEVAL_STATUS_SUCCESS'
-  """Url retrieval is successful."""
-  URL_RETRIEVAL_STATUS_ERROR = 'URL_RETRIEVAL_STATUS_ERROR'
-  """Url retrieval is failed due to error."""
-
-
 class SafetyFilterLevel(_common.CaseInSensitiveEnum):
   """Enum that controls the safety filter level for objectionable content."""
 
@@ -743,7 +777,8 @@ FileDataOrDict = Union[FileData, FileDataDict]
 class CodeExecutionResult(_common.BaseModel):
   """Result of executing the [ExecutableCode].
 
-  Always follows a `part` containing the [ExecutableCode].
+  Only generated when using the [CodeExecution] tool, and always follows a
+  `part` containing the [ExecutableCode].
   """
 
   outcome: Optional[Outcome] = Field(
@@ -758,7 +793,8 @@ class CodeExecutionResult(_common.BaseModel):
 class CodeExecutionResultDict(TypedDict, total=False):
   """Result of executing the [ExecutableCode].
 
-  Always follows a `part` containing the [ExecutableCode].
+  Only generated when using the [CodeExecution] tool, and always follows a
+  `part` containing the [ExecutableCode].
   """
 
   outcome: Optional[Outcome]
@@ -774,8 +810,9 @@ CodeExecutionResultOrDict = Union[CodeExecutionResult, CodeExecutionResultDict]
 class ExecutableCode(_common.BaseModel):
   """Code generated by the model that is meant to be executed, and the result returned to the model.
 
-  Generated when using the [FunctionDeclaration] tool and
-  [FunctionCallingConfig] mode is set to [Mode.CODE].
+  Generated when using the [CodeExecution] tool, in which the code will be
+  automatically executed, and a corresponding [CodeExecutionResult] will also be
+  generated.
   """
 
   code: Optional[str] = Field(
@@ -790,8 +827,9 @@ class ExecutableCode(_common.BaseModel):
 class ExecutableCodeDict(TypedDict, total=False):
   """Code generated by the model that is meant to be executed, and the result returned to the model.
 
-  Generated when using the [FunctionDeclaration] tool and
-  [FunctionCallingConfig] mode is set to [Mode.CODE].
+  Generated when using the [CodeExecution] tool, in which the code will be
+  automatically executed, and a corresponding [CodeExecutionResult] will also be
+  generated.
   """
 
   code: Optional[str]
@@ -2473,6 +2511,168 @@ class UrlContextDict(TypedDict, total=False):
 UrlContextOrDict = Union[UrlContext, UrlContextDict]
 
 
+class ApiAuthApiKeyConfig(_common.BaseModel):
+  """The API secret."""
+
+  api_key_secret_version: Optional[str] = Field(
+      default=None,
+      description="""Required. The SecretManager secret version resource name storing API key. e.g. projects/{project}/secrets/{secret}/versions/{version}""",
+  )
+  api_key_string: Optional[str] = Field(
+      default=None,
+      description="""The API key string. Either this or `api_key_secret_version` must be set.""",
+  )
+
+
+class ApiAuthApiKeyConfigDict(TypedDict, total=False):
+  """The API secret."""
+
+  api_key_secret_version: Optional[str]
+  """Required. The SecretManager secret version resource name storing API key. e.g. projects/{project}/secrets/{secret}/versions/{version}"""
+
+  api_key_string: Optional[str]
+  """The API key string. Either this or `api_key_secret_version` must be set."""
+
+
+ApiAuthApiKeyConfigOrDict = Union[ApiAuthApiKeyConfig, ApiAuthApiKeyConfigDict]
+
+
+class ApiAuth(_common.BaseModel):
+  """The generic reusable api auth config.
+
+  Deprecated. Please use AuthConfig (google/cloud/aiplatform/master/auth.proto)
+  instead.
+  """
+
+  api_key_config: Optional[ApiAuthApiKeyConfig] = Field(
+      default=None, description="""The API secret."""
+  )
+
+
+class ApiAuthDict(TypedDict, total=False):
+  """The generic reusable api auth config.
+
+  Deprecated. Please use AuthConfig (google/cloud/aiplatform/master/auth.proto)
+  instead.
+  """
+
+  api_key_config: Optional[ApiAuthApiKeyConfigDict]
+  """The API secret."""
+
+
+ApiAuthOrDict = Union[ApiAuth, ApiAuthDict]
+
+
+class ExternalApiElasticSearchParams(_common.BaseModel):
+  """The search parameters to use for the ELASTIC_SEARCH spec."""
+
+  index: Optional[str] = Field(
+      default=None, description="""The ElasticSearch index to use."""
+  )
+  num_hits: Optional[int] = Field(
+      default=None,
+      description="""Optional. Number of hits (chunks) to request. When specified, it is passed to Elasticsearch as the `num_hits` param.""",
+  )
+  search_template: Optional[str] = Field(
+      default=None, description="""The ElasticSearch search template to use."""
+  )
+
+
+class ExternalApiElasticSearchParamsDict(TypedDict, total=False):
+  """The search parameters to use for the ELASTIC_SEARCH spec."""
+
+  index: Optional[str]
+  """The ElasticSearch index to use."""
+
+  num_hits: Optional[int]
+  """Optional. Number of hits (chunks) to request. When specified, it is passed to Elasticsearch as the `num_hits` param."""
+
+  search_template: Optional[str]
+  """The ElasticSearch search template to use."""
+
+
+ExternalApiElasticSearchParamsOrDict = Union[
+    ExternalApiElasticSearchParams, ExternalApiElasticSearchParamsDict
+]
+
+
+class ExternalApiSimpleSearchParams(_common.BaseModel):
+  """The search parameters to use for SIMPLE_SEARCH spec."""
+
+  pass
+
+
+class ExternalApiSimpleSearchParamsDict(TypedDict, total=False):
+  """The search parameters to use for SIMPLE_SEARCH spec."""
+
+  pass
+
+
+ExternalApiSimpleSearchParamsOrDict = Union[
+    ExternalApiSimpleSearchParams, ExternalApiSimpleSearchParamsDict
+]
+
+
+class ExternalApi(_common.BaseModel):
+  """Retrieve from data source powered by external API for grounding.
+
+  The external API is not owned by Google, but need to follow the pre-defined
+  API spec.
+  """
+
+  api_auth: Optional[ApiAuth] = Field(
+      default=None,
+      description="""The authentication config to access the API. Deprecated. Please use auth_config instead.""",
+  )
+  api_spec: Optional[ApiSpec] = Field(
+      default=None,
+      description="""The API spec that the external API implements.""",
+  )
+  auth_config: Optional[AuthConfig] = Field(
+      default=None,
+      description="""The authentication config to access the API.""",
+  )
+  elastic_search_params: Optional[ExternalApiElasticSearchParams] = Field(
+      default=None, description="""Parameters for the elastic search API."""
+  )
+  endpoint: Optional[str] = Field(
+      default=None,
+      description="""The endpoint of the external API. The system will call the API at this endpoint to retrieve the data for grounding. Example: https://acme.com:443/search""",
+  )
+  simple_search_params: Optional[ExternalApiSimpleSearchParams] = Field(
+      default=None, description="""Parameters for the simple search API."""
+  )
+
+
+class ExternalApiDict(TypedDict, total=False):
+  """Retrieve from data source powered by external API for grounding.
+
+  The external API is not owned by Google, but need to follow the pre-defined
+  API spec.
+  """
+
+  api_auth: Optional[ApiAuthDict]
+  """The authentication config to access the API. Deprecated. Please use auth_config instead."""
+
+  api_spec: Optional[ApiSpec]
+  """The API spec that the external API implements."""
+
+  auth_config: Optional[AuthConfigDict]
+  """The authentication config to access the API."""
+
+  elastic_search_params: Optional[ExternalApiElasticSearchParamsDict]
+  """Parameters for the elastic search API."""
+
+  endpoint: Optional[str]
+  """The endpoint of the external API. The system will call the API at this endpoint to retrieve the data for grounding. Example: https://acme.com:443/search"""
+
+  simple_search_params: Optional[ExternalApiSimpleSearchParamsDict]
+  """Parameters for the simple search API."""
+
+
+ExternalApiOrDict = Union[ExternalApi, ExternalApiDict]
+
+
 class VertexAISearchDataStoreSpec(_common.BaseModel):
   """Define data stores within engine to filter on in a search call and configurations for those data stores.
 
@@ -2815,6 +3015,10 @@ class Retrieval(_common.BaseModel):
       default=None,
       description="""Optional. Deprecated. This option is no longer supported.""",
   )
+  external_api: Optional[ExternalApi] = Field(
+      default=None,
+      description="""Use data source powered by external API for grounding.""",
+  )
   vertex_ai_search: Optional[VertexAISearch] = Field(
       default=None,
       description="""Set to use data source powered by Vertex AI Search.""",
@@ -2831,6 +3035,9 @@ class RetrievalDict(TypedDict, total=False):
   disable_attribution: Optional[bool]
   """Optional. Deprecated. This option is no longer supported."""
 
+  external_api: Optional[ExternalApiDict]
+  """Use data source powered by external API for grounding."""
+
   vertex_ai_search: Optional[VertexAISearchDict]
   """Set to use data source powered by Vertex AI Search."""
 
@@ -2864,6 +3071,24 @@ class ToolCodeExecutionDict(TypedDict, total=False):
 ToolCodeExecutionOrDict = Union[ToolCodeExecution, ToolCodeExecutionDict]
 
 
+class ToolComputerUse(_common.BaseModel):
+  """Tool to support computer use."""
+
+  environment: Optional[Environment] = Field(
+      default=None, description="""Required. The environment being operated."""
+  )
+
+
+class ToolComputerUseDict(TypedDict, total=False):
+  """Tool to support computer use."""
+
+  environment: Optional[Environment]
+  """Required. The environment being operated."""
+
+
+ToolComputerUseOrDict = Union[ToolComputerUse, ToolComputerUseDict]
+
+
 class Tool(_common.BaseModel):
   """Tool details of a tool that the model may use to generate a response."""
 
@@ -2902,6 +3127,10 @@ class Tool(_common.BaseModel):
       default=None,
       description="""Optional. CodeExecution tool type. Enables the model to execute code as part of generation.""",
   )
+  computer_use: Optional[ToolComputerUse] = Field(
+      default=None,
+      description="""Optional. Tool to support the model interacting directly with the computer. If enabled, it automatically populates computer-use specific Function Declarations.""",
+  )
 
 
 class ToolDict(TypedDict, total=False):
@@ -2934,6 +3163,9 @@ class ToolDict(TypedDict, total=False):
   code_execution: Optional[ToolCodeExecutionDict]
   """Optional. CodeExecution tool type. Enables the model to execute code as part of generation."""
 
+  computer_use: Optional[ToolComputerUseDict]
+  """Optional. Tool to support the model interacting directly with the computer. If enabled, it automatically populates computer-use specific Function Declarations."""
+
 
 ToolOrDict = Union[Tool, ToolDict]
 if _is_mcp_imported:
@@ -4341,7 +4573,7 @@ class GroundingSupport(_common.BaseModel):
 
   confidence_scores: Optional[list[float]] = Field(
       default=None,
-      description="""Confidence score of the support references. Ranges from 0 to 1. 1 is the most confident. This list must have the same size as the grounding_chunk_indices.""",
+      description="""Confidence score of the support references. Ranges from 0 to 1. 1 is the most confident. For Gemini 2.0 and before, this list must have the same size as the grounding_chunk_indices. For Gemini 2.5 and after, this list will be empty and should be ignored.""",
   )
   grounding_chunk_indices: Optional[list[int]] = Field(
       default=None,
@@ -4357,7 +4589,7 @@ class GroundingSupportDict(TypedDict, total=False):
   """Grounding support."""
 
   confidence_scores: Optional[list[float]]
-  """Confidence score of the support references. Ranges from 0 to 1. 1 is the most confident. This list must have the same size as the grounding_chunk_indices."""
+  """Confidence score of the support references. Ranges from 0 to 1. 1 is the most confident. For Gemini 2.0 and before, this list must have the same size as the grounding_chunk_indices. For Gemini 2.5 and after, this list will be empty and should be ignored."""
 
   grounding_chunk_indices: Optional[list[int]]
   """A list of indices (into 'grounding_chunk') specifying the citations associated with the claim. For instance [1,3,4] means that grounding_chunk[1], grounding_chunk[3], grounding_chunk[4] are the retrieved content attributed to the claim."""
@@ -4554,6 +4786,10 @@ class SafetyRating(_common.BaseModel):
   category: Optional[HarmCategory] = Field(
       default=None, description="""Output only. Harm category."""
   )
+  overwritten_threshold: Optional[HarmBlockThreshold] = Field(
+      default=None,
+      description="""Output only. The overwritten threshold for the safety category of Gemini 2.0 image out. If minors are detected in the output image, the threshold of each safety category will be overwritten if user sets a lower threshold.""",
+  )
   probability: Optional[HarmProbability] = Field(
       default=None,
       description="""Output only. Harm probability levels in the content.""",
@@ -4579,6 +4815,9 @@ class SafetyRatingDict(TypedDict, total=False):
   category: Optional[HarmCategory]
   """Output only. Harm category."""
 
+  overwritten_threshold: Optional[HarmBlockThreshold]
+  """Output only. The overwritten threshold for the safety category of Gemini 2.0 image out. If minors are detected in the output image, the threshold of each safety category will be overwritten if user sets a lower threshold."""
+
   probability: Optional[HarmProbability]
   """Output only. Harm probability levels in the content."""
 
@@ -6866,6 +7105,10 @@ class GenerationConfig(_common.BaseModel):
       default=None,
       description="""Optional. Number of candidates to generate.""",
   )
+  enable_affective_dialog: Optional[bool] = Field(
+      default=None,
+      description="""Optional. If enabled, the model will detect emotions and adapt its responses accordingly.""",
+  )
   frequency_penalty: Optional[float] = Field(
       default=None, description="""Optional. Frequency penalties."""
   )
@@ -6883,6 +7126,10 @@ class GenerationConfig(_common.BaseModel):
   presence_penalty: Optional[float] = Field(
       default=None, description="""Optional. Positive penalties."""
   )
+  response_json_schema: Optional[Any] = Field(
+      default=None,
+      description="""Optional. Output schema of the generated response. This is an alternative to `response_schema` that accepts [JSON Schema](https://json-schema.org/). If set, `response_schema` must be omitted, but `response_mime_type` is required. While the full JSON Schema may be sent, not all features are supported. Specifically, only the following properties are supported: - `$id` - `$defs` - `$ref` - `$anchor` - `type` - `format` - `title` - `description` - `enum` (for strings and numbers) - `items` - `prefixItems` - `minItems` - `maxItems` - `minimum` - `maximum` - `anyOf` - `oneOf` (interpreted the same as `anyOf`) - `properties` - `additionalProperties` - `required` The non-standard `propertyOrdering` property may also be set. Cyclic references are unrolled to a limited degree and, as such, may only be used within non-required properties. (Nullable properties are not sufficient.) If `$ref` is set on a sub-schema, no other properties, except for than those starting as a `$`, may be set.""",
+  )
   response_logprobs: Optional[bool] = Field(
       default=None,
       description="""Optional. If true, export the logprobs results in response.""",
@@ -6898,10 +7145,6 @@ class GenerationConfig(_common.BaseModel):
       default=None,
       description="""Optional. The `Schema` object allows the definition of input and output data types. These types can be objects, but also primitives and arrays. Represents a select subset of an [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema). If set, a compatible response_mime_type must also be set. Compatible mimetypes: `application/json`: Schema for JSON response.""",
   )
-  response_json_schema: Optional[Any] = Field(
-      default=None,
-      description="""Optional. Output schema of the generated response. This is an alternative to `response_schema` that accepts [JSON Schema](https://json-schema.org/). If set, `response_schema` must be omitted, but `response_mime_type` is required. While the full JSON Schema may be sent, not all features are supported. Specifically, only the following properties are supported: - `$id` - `$defs` - `$ref` - `$anchor` - `type` - `format` - `title` - `description` - `enum` (for strings and numbers) - `items` - `prefixItems` - `minItems` - `maxItems` - `minimum` - `maximum` - `anyOf` - `oneOf` (interpreted the same as `anyOf`) - `properties` - `additionalProperties` - `required` The non-standard `propertyOrdering` property may also be set. Cyclic references are unrolled to a limited degree and, as such, may only be used within non-required properties. (Nullable properties are not sufficient.) If `$ref` is set on a sub-schema, no other properties, except for than those starting as a `$`, may be set.""",
-  )
   routing_config: Optional[GenerationConfigRoutingConfig] = Field(
       default=None, description="""Optional. Routing configuration."""
   )
@@ -6942,6 +7185,9 @@ class GenerationConfigDict(TypedDict, total=False):
   candidate_count: Optional[int]
   """Optional. Number of candidates to generate."""
 
+  enable_affective_dialog: Optional[bool]
+  """Optional. If enabled, the model will detect emotions and adapt its responses accordingly."""
+
   frequency_penalty: Optional[float]
   """Optional. Frequency penalties."""
 
@@ -6957,6 +7203,9 @@ class GenerationConfigDict(TypedDict, total=False):
   presence_penalty: Optional[float]
   """Optional. Positive penalties."""
 
+  response_json_schema: Optional[Any]
+  """Optional. Output schema of the generated response. This is an alternative to `response_schema` that accepts [JSON Schema](https://json-schema.org/). If set, `response_schema` must be omitted, but `response_mime_type` is required. While the full JSON Schema may be sent, not all features are supported. Specifically, only the following properties are supported: - `$id` - `$defs` - `$ref` - `$anchor` - `type` - `format` - `title` - `description` - `enum` (for strings and numbers) - `items` - `prefixItems` - `minItems` - `maxItems` - `minimum` - `maximum` - `anyOf` - `oneOf` (interpreted the same as `anyOf`) - `properties` - `additionalProperties` - `required` The non-standard `propertyOrdering` property may also be set. Cyclic references are unrolled to a limited degree and, as such, may only be used within non-required properties. (Nullable properties are not sufficient.) If `$ref` is set on a sub-schema, no other properties, except for than those starting as a `$`, may be set."""
+
   response_logprobs: Optional[bool]
   """Optional. If true, export the logprobs results in response."""
 
@@ -6969,9 +7218,6 @@ class GenerationConfigDict(TypedDict, total=False):
   response_schema: Optional[SchemaDict]
   """Optional. The `Schema` object allows the definition of input and output data types. These types can be objects, but also primitives and arrays. Represents a select subset of an [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema). If set, a compatible response_mime_type must also be set. Compatible mimetypes: `application/json`: Schema for JSON response."""
 
-  response_json_schema: Optional[Any]
-  """Optional. Output schema of the generated response. This is an alternative to `response_schema` that accepts [JSON Schema](https://json-schema.org/). If set, `response_schema` must be omitted, but `response_mime_type` is required. While the full JSON Schema may be sent, not all features are supported. Specifically, only the following properties are supported: - `$id` - `$defs` - `$ref` - `$anchor` - `type` - `format` - `title` - `description` - `enum` (for strings and numbers) - `items` - `prefixItems` - `minItems` - `maxItems` - `minimum` - `maximum` - `anyOf` - `oneOf` (interpreted the same as `anyOf`) - `properties` - `additionalProperties` - `required` The non-standard `propertyOrdering` property may also be set. Cyclic references are unrolled to a limited degree and, as such, may only be used within non-required properties. (Nullable properties are not sufficient.) If `$ref` is set on a sub-schema, no other properties, except for than those starting as a `$`, may be set."""
-
   routing_config: Optional[GenerationConfigRoutingConfigDict]
   """Optional. Routing configuration."""
 
@@ -7780,7 +8026,7 @@ class SupervisedHyperParameters(_common.BaseModel):
   )
   learning_rate_multiplier: Optional[float] = Field(
       default=None,
-      description="""Optional. Multiplier for adjusting the default learning rate.""",
+      description="""Optional. Multiplier for adjusting the default learning rate. Mutually exclusive with `learning_rate`.""",
   )
 
 
@@ -7794,7 +8040,7 @@ class SupervisedHyperParametersDict(TypedDict, total=False):
   """Optional. Number of complete passes the model makes over the entire training dataset during training."""
 
   learning_rate_multiplier: Optional[float]
-  """Optional. Multiplier for adjusting the default learning rate."""
+  """Optional. Multiplier for adjusting the default learning rate. Mutually exclusive with `learning_rate`."""
 
 
 SupervisedHyperParametersOrDict = Union[
@@ -7814,11 +8060,11 @@ class SupervisedTuningSpec(_common.BaseModel):
   )
   training_dataset_uri: Optional[str] = Field(
       default=None,
-      description="""Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
+      description="""Required. Training dataset used for tuning. The dataset can be specified as either a Cloud Storage path to a JSONL file or as the resource name of a Vertex Multimodal Dataset.""",
   )
   validation_dataset_uri: Optional[str] = Field(
       default=None,
-      description="""Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.""",
+      description="""Optional. Validation dataset used for tuning. The dataset can be specified as either a Cloud Storage path to a JSONL file or as the resource name of a Vertex Multimodal Dataset.""",
   )
 
 
@@ -7832,10 +8078,10 @@ class SupervisedTuningSpecDict(TypedDict, total=False):
   """Optional. Hyperparameters for SFT."""
 
   training_dataset_uri: Optional[str]
-  """Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
+  """Required. Training dataset used for tuning. The dataset can be specified as either a Cloud Storage path to a JSONL file or as the resource name of a Vertex Multimodal Dataset."""
 
   validation_dataset_uri: Optional[str]
-  """Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file."""
+  """Optional. Validation dataset used for tuning. The dataset can be specified as either a Cloud Storage path to a JSONL file or as the resource name of a Vertex Multimodal Dataset."""
 
 
 SupervisedTuningSpecOrDict = Union[
@@ -8504,6 +8750,12 @@ class TuningJob(_common.BaseModel):
       default=None,
       description="""Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`.""",
   )
+  satisfies_pzi: Optional[bool] = Field(
+      default=None, description="""Output only. Reserved for future use."""
+  )
+  satisfies_pzs: Optional[bool] = Field(
+      default=None, description="""Output only. Reserved for future use."""
+  )
   service_account: Optional[str] = Field(
       default=None,
       description="""The service account that the tuningJob workload runs as. If not specified, the Vertex AI Secure Fine-Tuned Service Agent in the project will be used. See https://cloud.google.com/iam/docs/service-agents#vertex-ai-secure-fine-tuning-service-agent Users starting the pipeline must have the `iam.serviceAccounts.actAs` permission on this service account.""",
@@ -8581,6 +8833,12 @@ class TuningJobDict(TypedDict, total=False):
   pipeline_job: Optional[str]
   """Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`."""
 
+  satisfies_pzi: Optional[bool]
+  """Output only. Reserved for future use."""
+
+  satisfies_pzs: Optional[bool]
+  """Output only. Reserved for future use."""
+
   service_account: Optional[str]
   """The service account that the tuningJob workload runs as. If not specified, the Vertex AI Secure Fine-Tuned Service Agent in the project will be used. See https://cloud.google.com/iam/docs/service-agents#vertex-ai-secure-fine-tuning-service-agent Users starting the pipeline must have the `iam.serviceAccounts.actAs` permission on this service account."""
 
@@ -11139,13 +11397,16 @@ SubjectReferenceImageOrDict = Union[
 class LiveServerSetupComplete(_common.BaseModel):
   """Sent in response to a `LiveGenerateContentSetup` message from the client."""
 
-  pass
+  session_id: Optional[str] = Field(
+      default=None, description="""The session id of the live session."""
+  )
 
 
 class LiveServerSetupCompleteDict(TypedDict, total=False):
   """Sent in response to a `LiveGenerateContentSetup` message from the client."""
 
-  pass
+  session_id: Optional[str]
+  """The session id of the live session."""
 
 
 LiveServerSetupCompleteOrDict = Union[

google/genai/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 #
 
-__version__ = '1.23.0' # x-release-please-version
+__version__ = '1.24.0' # x-release-please-version

{google_genai-1.23.0.dist-info → google_genai-1.24.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.23.0
+Version: 1.24.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -1132,9 +1132,9 @@ response3.generated_images[0].image.show()
 
 ### Veo
 
-#### Generate Videos
+Support for generating videos is considered public preview
 
-Support for generate videos in Vertex and Gemini Developer API is behind an allowlist
+#### Generate Videos (Text to Video)
 
 ```python
 from google.genai import types
@@ -1145,7 +1145,6 @@ operation = client.models.generate_videos(
     prompt='A neon hologram of a cat driving at top speed',
     config=types.GenerateVideosConfig(
         number_of_videos=1,
-        fps=24,
         duration_seconds=5,
         enhance_prompt=True,
     ),
@@ -1156,7 +1155,73 @@ while not operation.done:
     time.sleep(20)
     operation = client.operations.get(operation)
 
-video = operation.result.generated_videos[0].video
+video = operation.response.generated_videos[0].video
+video.show()
+```
+
+#### Generate Videos (Image to Video)
+
+```python
+from google.genai import types
+
+# Read local image (uses mimetypes.guess_type to infer mime type)
+image = types.Image.from_file("local/path/file.png")
+
+# Create operation
+operation = client.models.generate_videos(
+    model='veo-2.0-generate-001',
+    # Prompt is optional if image is provided
+    prompt='Night sky',
+    image=image,
+    config=types.GenerateVideosConfig(
+        number_of_videos=1,
+        duration_seconds=5,
+        enhance_prompt=True,
+        # Can also pass an Image into last_frame for frame interpolation
+    ),
+)
+
+# Poll operation
+while not operation.done:
+    time.sleep(20)
+    operation = client.operations.get(operation)
+
+video = operation.response.generated_videos[0].video
+video.show()
+```
+
+#### Generate Videos (Video to Video)
+
+Currently, only Vertex supports Video to Video generation (Video extension).
+
+```python
+from google.genai import types
+
+# Read local video (uses mimetypes.guess_type to infer mime type)
+video = types.Video.from_file("local/path/video.mp4")
+
+# Create operation
+operation = client.models.generate_videos(
+    model='veo-2.0-generate-001',
+    # Prompt is optional if Video is provided
+    prompt='Night sky',
+    # Input video must be in GCS
+    video=types.Video(
+        uri="gs://bucket-name/inputs/videos/cat_driving.mp4",
+    ),
+    config=types.GenerateVideosConfig(
+        number_of_videos=1,
+        duration_seconds=5,
+        enhance_prompt=True,
+    ),
+)
+
+# Poll operation
+while not operation.done:
+    time.sleep(20)
+    operation = client.operations.get(operation)
+
+video = operation.response.generated_videos[0].video
 video.show()
 ```
 

{google_genai-1.23.0.dist-info → google_genai-1.24.0.dist-info}/RECORD

@@ -1,35 +1,35 @@
 google/genai/__init__.py,sha256=SYTxz3Ho06pP2TBlvDU0FkUJz8ytbR3MgEpS9HvVYq4,709
 google/genai/_adapters.py,sha256=Kok38miNYJff2n--l0zEK_hbq0y2rWOH7k75J7SMYbQ,1744
-google/genai/_api_client.py,sha256=bbuu6LUCEf6MH-DzhEg-w7iv84SqRy5vIluXDKOn1UU,52447
+google/genai/_api_client.py,sha256=ISgrtT7-XVOdyv3Dtwz2Eve9w3C3RO4ylEY1bND8ElY,52757
 google/genai/_api_module.py,sha256=lj8eUWx8_LBGBz-49qz6_ywWm3GYp3d8Bg5JoOHbtbI,902
 google/genai/_automatic_function_calling_util.py,sha256=IJkPq2fT9pYxYm5Pbu5-e0nBoZKoZla7yT4_txWRKLs,10324
 google/genai/_base_url.py,sha256=E5H4dew14Y16qfnB3XRnjSCi19cJVlkaMNoM_8ip-PM,1597
 google/genai/_common.py,sha256=CuG7wvroFu9kV0Im-asXJGmfbaFCAjCocZyRxYDss5g,19849
 google/genai/_extra_utils.py,sha256=GoIh_SxB5nf_0sjErYkjm8wdg93sfemLKrU_5QhlcBo,20562
-google/genai/_live_converters.py,sha256=lbpN5E8DjL74a2XLcgND4pzs3bHDeVpUtPDN8s7IZCU,108613
+google/genai/_live_converters.py,sha256=mkxudlXXXAxuK6eWMj_-jVLQs9eOr4EX0GYCBvhIZAw,108962
 google/genai/_mcp_utils.py,sha256=khECx-DMuHemKzOQQ3msWp7FivPeEOnl3n1lvWc_b5o,3833
 google/genai/_replay_api_client.py,sha256=0IXroewCZbyZ01qN3ZQwW5OUZRMzCrf2-3nyfTerqNY,21363
 google/genai/_test_api_client.py,sha256=4ruFIy5_1qcbKqqIBu3HSQbpSOBrxiecBtDZaTGFR1s,4797
-google/genai/_tokens_converters.py,sha256=LxLZGgSWn6VC2-1R-4ArcuaPK45svg5P-bpC1LjBb3w,48901
+google/genai/_tokens_converters.py,sha256=jQlwRZU4mlIBv6i8Lu62e6u9dvgNtOnko5WEYnjkU_A,49159
 google/genai/_transformers.py,sha256=Nd_h_zYVOkWN1XTSnM9r3IR7R_qg3tPe2mE9jozaX-Q,36935
-google/genai/batches.py,sha256=ZpF3qgJUkPcpWjbaJtESiQq5qiZ0obsJl77yCtgZknA,170525
-google/genai/caches.py,sha256=Ckqqi0osoxNmkqYGvBGpvrZUx19D6hmuwS9UXcnGWmk,65911
+google/genai/batches.py,sha256=qnyWQ1RfAdSVQgV7vGor9BIWmRwUilJ0wfqOvxUwpq8,172745
+google/genai/caches.py,sha256=q8N3YIpgJYWrLWSy-zUZ6w9CPRjH113E8ff4MyykYA8,66169
 google/genai/chats.py,sha256=0QdOUeWEYDQgAWBy1f7a3z3yY9S8tXSowUzNrzazzj4,16651
 google/genai/client.py,sha256=wXnfZBSv9p-yKtX_gabUrfBXoYHuqHhzK_VgwRttMgY,10777
 google/genai/errors.py,sha256=UebysH1cDeIdtp1O06CW7lQjnNrWtuqZTeVgEJylX48,5589
 google/genai/files.py,sha256=MCcHRXiMFKjbnt78yVPejszEgGz_MHRXfJyDi5-07Gs,39655
 google/genai/live.py,sha256=8Vts1bX8ZWOa0PiS3zmYhG1QMhLtBPC_ATLTSAIrazs,38945
 google/genai/live_music.py,sha256=3GG9nsto8Vhkohcs-4CPMS4DFp1ZtMuLYzHfvEPYAeg,6971
-google/genai/models.py,sha256=7gBXwg6_RdW_HRD_iRFIAdqu0bkUkW0fAp_PMEPVwoU,239309
+google/genai/models.py,sha256=DQx31b43CxVg2YCWHVGGKBzcuB-AfBApmPm9zPVLq6o,239567
 google/genai/operations.py,sha256=99zs__fTWsyQu7rMGmBI1_4tuAJxLR0g3yp7ymsUsBA,19609
 google/genai/pagers.py,sha256=nyVYxp92rS-UaewO_oBgP593knofeLU6yOn6RolNoGQ,6797
 google/genai/py.typed,sha256=RsMFoLwBkAvY05t6izop4UHZtqOPLiKp3GkIEizzmQY,40
 google/genai/tokens.py,sha256=oew9I0fGuGLlXMdCn2Ot9Lv_herHMnvZOYjjdOz2CAM,12324
-google/genai/tunings.py,sha256=u6Hcfkc-nUYB_spSfY98h7EIiSz0dH3wVH0b2feMh1M,49072
-google/genai/types.py,sha256=LGLNmXkmnCLU-narShhG_1rtglfARNm6pvvZOiO5_lA,460023
-google/genai/version.py,sha256=Snuhu088Fxuibck04VaPaZ69LvJ5pKNdicb9oZYwcrY,627
-google_genai-1.23.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-1.23.0.dist-info/METADATA,sha256=Yylx0cTXrq87K2-VIbM9e1RT80LHMcvUC5ZB8Y0YTqc,38902
-google_genai-1.23.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_genai-1.23.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-1.23.0.dist-info/RECORD,,
+google/genai/tunings.py,sha256=fSiD4RQZI8YsFaPbBoXOIoSDA0BIuJwGg2Ac_uy5_kM,49596
+google/genai/types.py,sha256=dj1Va75V8kLkrtFPIiIYXQNukE5xuogoJmqo0nu0-N4,469923
+google/genai/version.py,sha256=3V_J-TpVrcSp70_3LzpCOt3SQpDmFAXnGSiq5eVjSrU,627
+google_genai-1.24.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-1.24.0.dist-info/METADATA,sha256=yxyVgjMGvqGuGrRM8HO9fO35MTuewHnGMQYohChHULs,40539
+google_genai-1.24.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+google_genai-1.24.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-1.24.0.dist-info/RECORD,,