google-genai 1.39.1__tar.gz → 1.40.0__tar.gz

{google_genai-1.39.1/google_genai.egg-info → google_genai-1.40.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.39.1
+Version: 1.40.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/_live_converters.py

@@ -291,6 +291,42 @@ def _Blob_to_vertex(
   return to_object
 
 
+def _ComputerUse_to_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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
+def _ComputerUse_to_vertex(
+    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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
 def _Content_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -3189,42 +3225,6 @@ def _SpeechConfig_to_vertex(
   return to_object
 
 
-def _ToolComputerUse_to_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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
-def _ToolComputerUse_to_vertex(
-    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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
 def _Tool_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -3278,9 +3278,7 @@ def _Tool_to_mldev(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_mldev(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:
@@ -3352,9 +3350,7 @@ def _Tool_to_vertex(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_vertex(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/_tokens_converters.py

@@ -106,6 +106,24 @@ def _Blob_to_mldev(
   return to_object
 
 
+def _ComputerUse_to_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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
 def _Content_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -743,24 +761,6 @@ def _SpeechConfig_to_mldev(
   return to_object
 
 
-def _ToolComputerUse_to_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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
 def _Tool_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -814,9 +814,7 @@ def _Tool_to_mldev(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_mldev(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/batches.py

@@ -467,6 +467,24 @@ def _CitationMetadata_from_mldev(
   return to_object
 
 
+def _ComputerUse_to_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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
 def _ContentEmbedding_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1131,6 +1149,13 @@ def _GenerateContentConfig_to_mldev(
         ),
     )
 
+  if getv(from_object, ['image_config']) is not None:
+    setv(
+        to_object,
+        ['imageConfig'],
+        _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
+    )
+
   return to_object
 
 
@@ -1238,6 +1263,17 @@ def _GoogleSearch_to_mldev(
   return to_object
 
 
+def _ImageConfig_to_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, ['aspect_ratio']) is not None:
+    setv(to_object, ['aspectRatio'], getv(from_object, ['aspect_ratio']))
+
+  return to_object
+
+
 def _InlinedEmbedContentResponse_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1877,24 +1913,6 @@ def _ThinkingConfig_to_mldev(
   return to_object
 
 
-def _ToolComputerUse_to_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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
 def _ToolConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1974,9 +1992,7 @@ def _Tool_to_mldev(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_mldev(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/caches.py

@@ -181,6 +181,42 @@ def _CachedContent_from_vertex(
   return to_object
 
 
+def _ComputerUse_to_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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
+def _ComputerUse_to_vertex(
+    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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
 def _Content_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1135,42 +1171,6 @@ def _RetrievalConfig_to_vertex(
   return to_object
 
 
-def _ToolComputerUse_to_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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
-def _ToolComputerUse_to_vertex(
-    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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
 def _ToolConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1276,9 +1276,7 @@ def _Tool_to_mldev(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_mldev(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:
@@ -1350,9 +1348,7 @@ def _Tool_to_vertex(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_vertex(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/live.py

@@ -82,9 +82,15 @@ _FUNCTION_RESPONSE_REQUIRES_ID = (
 class AsyncSession:
   """[Preview] AsyncSession."""
 
-  def __init__(self, api_client: BaseApiClient, websocket: ClientConnection):
+  def __init__(
+      self,
+      api_client: BaseApiClient,
+      websocket: ClientConnection,
+      session_id: Optional[str] = None,
+  ):
     self._api_client = api_client
     self._ws = websocket
+    self.session_id = session_id
 
   async def send(
       self,
@@ -1054,11 +1060,34 @@ class AsyncLive(_api_module.BaseModule):
       await ws.send(request)
       try:
         # websockets 14.0+
-        logger.info(await ws.recv(decode=False))
+        raw_response = await ws.recv(decode=False)
       except TypeError:
-        logger.info(await ws.recv())
+        raw_response = await ws.recv()  # type: ignore[assignment]
+      if raw_response:
+        try:
+          response = json.loads(raw_response)
+        except json.decoder.JSONDecodeError:
+          raise ValueError(f'Failed to parse response: {raw_response!r}')
+      else:
+        response = {}
 
-      yield AsyncSession(api_client=self._api_client, websocket=ws)
+      if self._api_client.vertexai:
+        response_dict = live_converters._LiveServerMessage_from_vertex(response)
+      else:
+        response_dict = live_converters._LiveServerMessage_from_mldev(response)
+
+      setup_response = types.LiveServerMessage._from_response(
+          response=response_dict, kwargs=parameter_model.model_dump()
+      )
+      if setup_response.setup_complete:
+        session_id = setup_response.setup_complete.session_id
+      else:
+        session_id = None
+      yield AsyncSession(
+          api_client=self._api_client,
+          websocket=ws,
+          session_id=session_id,
+      )
 
 
 async def _t_live_connect_config(

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/models.py

@@ -366,6 +366,42 @@ def _ComputeTokensResponse_from_vertex(
   return to_object
 
 
+def _ComputerUse_to_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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
+def _ComputerUse_to_vertex(
+    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, ['environment']) is not None:
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  return to_object
+
+
 def _ContentEmbeddingStatistics_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1602,6 +1638,13 @@ def _GenerateContentConfig_to_mldev(
         ),
     )
 
+  if getv(from_object, ['image_config']) is not None:
+    setv(
+        to_object,
+        ['imageConfig'],
+        _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
+    )
+
   return to_object
 
 
@@ -1774,6 +1817,13 @@ def _GenerateContentConfig_to_vertex(
         ),
     )
 
+  if getv(from_object, ['image_config']) is not None:
+    setv(
+        to_object,
+        ['imageConfig'],
+        _ImageConfig_to_vertex(getv(from_object, ['image_config']), to_object),
+    )
+
   return to_object
 
 
@@ -2960,6 +3010,28 @@ def _GoogleSearch_to_vertex(
   return to_object
 
 
+def _ImageConfig_to_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, ['aspect_ratio']) is not None:
+    setv(to_object, ['aspectRatio'], getv(from_object, ['aspect_ratio']))
+
+  return to_object
+
+
+def _ImageConfig_to_vertex(
+    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, ['aspect_ratio']) is not None:
+    setv(to_object, ['aspectRatio'], getv(from_object, ['aspect_ratio']))
+
+  return to_object
+
+
 def _Image_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -4467,42 +4539,6 @@ def _ThinkingConfig_to_vertex(
   return to_object
 
 
-def _ToolComputerUse_to_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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
-def _ToolComputerUse_to_vertex(
-    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, ['environment']) is not None:
-    setv(to_object, ['environment'], getv(from_object, ['environment']))
-
-  if getv(from_object, ['excluded_predefined_functions']) is not None:
-    setv(
-        to_object,
-        ['excludedPredefinedFunctions'],
-        getv(from_object, ['excluded_predefined_functions']),
-    )
-
-  return to_object
-
-
 def _ToolConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -4608,9 +4644,7 @@ def _Tool_to_mldev(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_mldev(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:
@@ -4682,9 +4716,7 @@ def _Tool_to_vertex(
     setv(
         to_object,
         ['computerUse'],
-        _ToolComputerUse_to_vertex(
-            getv(from_object, ['computer_use']), to_object
-        ),
+        _ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
     )
 
   if getv(from_object, ['code_execution']) is not None:

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/types.py

@@ -285,6 +285,8 @@ class FinishReason(_common.CaseInSensitiveEnum):
   """Token generation stopped because generated images have safety violations."""
   UNEXPECTED_TOOL_CALL = 'UNEXPECTED_TOOL_CALL'
   """The tool call generated by the model is invalid."""
+  IMAGE_PROHIBITED_CONTENT = 'IMAGE_PROHIBITED_CONTENT'
+  """Image generation stopped because the generated images have prohibited content."""
 
 
 class HarmProbability(_common.CaseInSensitiveEnum):
@@ -2795,7 +2797,7 @@ class UrlContextDict(TypedDict, total=False):
 UrlContextOrDict = Union[UrlContext, UrlContextDict]
 
 
-class ToolComputerUse(_common.BaseModel):
+class ComputerUse(_common.BaseModel):
   """Tool to support computer use."""
 
   environment: Optional[Environment] = Field(
@@ -2811,7 +2813,7 @@ class ToolComputerUse(_common.BaseModel):
   )
 
 
-class ToolComputerUseDict(TypedDict, total=False):
+class ComputerUseDict(TypedDict, total=False):
   """Tool to support computer use."""
 
   environment: Optional[Environment]
@@ -2825,7 +2827,7 @@ class ToolComputerUseDict(TypedDict, total=False):
       2. Improving the definitions / instructions of predefined functions."""
 
 
-ToolComputerUseOrDict = Union[ToolComputerUse, ToolComputerUseDict]
+ComputerUseOrDict = Union[ComputerUse, ComputerUseDict]
 
 
 class ApiAuthApiKeyConfig(_common.BaseModel):
@@ -3422,7 +3424,7 @@ class Tool(_common.BaseModel):
       default=None,
       description="""Optional. Tool to support URL context retrieval.""",
   )
-  computer_use: Optional[ToolComputerUse] = Field(
+  computer_use: Optional[ComputerUse] = Field(
       default=None,
       description="""Optional. Tool to support the model interacting directly with the
       computer. If enabled, it automatically populates computer-use specific
@@ -3461,7 +3463,7 @@ class ToolDict(TypedDict, total=False):
   url_context: Optional[UrlContextDict]
   """Optional. Tool to support URL context retrieval."""
 
-  computer_use: Optional[ToolComputerUseDict]
+  computer_use: Optional[ComputerUseDict]
   """Optional. Tool to support the model interacting directly with the
       computer. If enabled, it automatically populates computer-use specific
       Function Declarations."""
@@ -3826,6 +3828,27 @@ class ThinkingConfigDict(TypedDict, total=False):
 ThinkingConfigOrDict = Union[ThinkingConfig, ThinkingConfigDict]
 
 
+class ImageConfig(_common.BaseModel):
+  """The image generation configuration to be used in GenerateContentConfig."""
+
+  aspect_ratio: Optional[str] = Field(
+      default=None,
+      description="""Aspect ratio of the generated images. Supported values are
+      "1:1", "2:3", "3:2", "3:4", "4:3", "9:16", "16:9", and "21:9".""",
+  )
+
+
+class ImageConfigDict(TypedDict, total=False):
+  """The image generation configuration to be used in GenerateContentConfig."""
+
+  aspect_ratio: Optional[str]
+  """Aspect ratio of the generated images. Supported values are
+      "1:1", "2:3", "3:2", "3:4", "4:3", "9:16", "16:9", and "21:9"."""
+
+
+ImageConfigOrDict = Union[ImageConfig, ImageConfigDict]
+
+
 class FileStatus(_common.BaseModel):
   """Status of a File that uses a common error model."""
 
@@ -4263,6 +4286,11 @@ class GenerateContentConfig(_common.BaseModel):
       description="""The thinking features configuration.
       """,
   )
+  image_config: Optional[ImageConfig] = Field(
+      default=None,
+      description="""The image generation configuration.
+      """,
+  )
 
   @pydantic.field_validator('response_schema', mode='before')
   @classmethod
@@ -4278,6 +4306,13 @@ class GenerateContentConfig(_common.BaseModel):
       return Enum('PlaceholderLiteralEnum', {s: s for s in enum_vals})
     return value
 
+  @pydantic.field_validator('image_config', mode='before')
+  @classmethod
+  def _check_image_config_type(cls, value: Any) -> Any:
+    if not isinstance(value, ImageConfig):
+      raise ValueError('image_config must be an instance of ImageConfig.')
+    return value
+
 
 class GenerateContentConfigDict(TypedDict, total=False):
   """Optional model configuration parameters.
@@ -4452,6 +4487,10 @@ class GenerateContentConfigDict(TypedDict, total=False):
   """The thinking features configuration.
       """
 
+  image_config: Optional[ImageConfigDict]
+  """The image generation configuration.
+      """
+
 
 GenerateContentConfigOrDict = Union[
     GenerateContentConfig, GenerateContentConfigDict

{google_genai-1.39.1 → google_genai-1.40.0}/google/genai/version.py

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

{google_genai-1.39.1 → google_genai-1.40.0/google_genai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.39.1
+Version: 1.40.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0

{google_genai-1.39.1 → google_genai-1.40.0}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel", "twine>=6.1.0", "packaging>=24.2", "pkginfo>=
 
 [project]
 name = "google-genai"
-version = "1.39.1"
+version = "1.40.0"
 description = "GenAI Python SDK"
 readme = "README.md"
 license = {text = "Apache-2.0"}