google-genai 1.38.0__py3-none-any.whl → 1.39.1__py3-none-any.whl

google/genai/types.py

@@ -1021,6 +1021,112 @@ class ExecutableCodeDict(TypedDict, total=False):
 ExecutableCodeOrDict = Union[ExecutableCode, ExecutableCodeDict]
 
 
+class FunctionResponseBlob(_common.BaseModel):
+  """Raw media bytes for function response.
+
+  Text should not be sent as raw bytes, use the FunctionResponse.response
+  field.
+  """
+
+  mime_type: Optional[str] = Field(
+      default=None,
+      description="""Required. The IANA standard MIME type of the source data.""",
+  )
+  data: Optional[bytes] = Field(
+      default=None, description="""Required. Inline media bytes."""
+  )
+
+
+class FunctionResponseBlobDict(TypedDict, total=False):
+  """Raw media bytes for function response.
+
+  Text should not be sent as raw bytes, use the FunctionResponse.response
+  field.
+  """
+
+  mime_type: Optional[str]
+  """Required. The IANA standard MIME type of the source data."""
+
+  data: Optional[bytes]
+  """Required. Inline media bytes."""
+
+
+FunctionResponseBlobOrDict = Union[
+    FunctionResponseBlob, FunctionResponseBlobDict
+]
+
+
+class FunctionResponseFileData(_common.BaseModel):
+  """URI based data for function response."""
+
+  file_uri: Optional[str] = Field(
+      default=None, description="""Required. URI."""
+  )
+  mime_type: Optional[str] = Field(
+      default=None,
+      description="""Required. The IANA standard MIME type of the source data.""",
+  )
+
+
+class FunctionResponseFileDataDict(TypedDict, total=False):
+  """URI based data for function response."""
+
+  file_uri: Optional[str]
+  """Required. URI."""
+
+  mime_type: Optional[str]
+  """Required. The IANA standard MIME type of the source data."""
+
+
+FunctionResponseFileDataOrDict = Union[
+    FunctionResponseFileData, FunctionResponseFileDataDict
+]
+
+
+class FunctionResponsePart(_common.BaseModel):
+  """A datatype containing media that is part of a `FunctionResponse` message.
+
+  A `FunctionResponsePart` consists of data which has an associated datatype. A
+  `FunctionResponsePart` can only contain one of the accepted types in
+  `FunctionResponsePart.data`.
+
+  A `FunctionResponsePart` must have a fixed IANA MIME type identifying the
+  type and subtype of the media if the `inline_data` field is filled with raw
+  bytes.
+  """
+
+  inline_data: Optional[FunctionResponseBlob] = Field(
+      default=None, description="""Optional. Inline media bytes."""
+  )
+  file_data: Optional[FunctionResponseFileData] = Field(
+      default=None, description="""Optional. URI based data."""
+  )
+
+
+class FunctionResponsePartDict(TypedDict, total=False):
+  """A datatype containing media that is part of a `FunctionResponse` message.
+
+  A `FunctionResponsePart` consists of data which has an associated datatype. A
+  `FunctionResponsePart` can only contain one of the accepted types in
+  `FunctionResponsePart.data`.
+
+  A `FunctionResponsePart` must have a fixed IANA MIME type identifying the
+  type and subtype of the media if the `inline_data` field is filled with raw
+  bytes.
+  """
+
+  inline_data: Optional[FunctionResponseBlobDict]
+  """Optional. Inline media bytes."""
+
+  file_data: Optional[FunctionResponseFileDataDict]
+  """Optional. URI based data."""
+
+
+FunctionResponsePartOrDict = Union[
+    FunctionResponsePart, FunctionResponsePartDict
+]
+
+
 class FunctionResponse(_common.BaseModel):
   """A function response."""
 
@@ -1032,6 +1138,11 @@ class FunctionResponse(_common.BaseModel):
       default=None,
       description="""Specifies how the response should be scheduled in the conversation. Only applicable to NON_BLOCKING function calls, is ignored otherwise. Defaults to WHEN_IDLE.""",
   )
+  parts: Optional[list[FunctionResponsePart]] = Field(
+      default=None,
+      description="""List of parts that constitute a function response. Each part may
+      have a different IANA MIME type.""",
+  )
   id: Optional[str] = Field(
       default=None,
       description="""Optional. The id of the function call this response is for. Populated by the client to match the corresponding function call `id`.""",
@@ -1070,6 +1181,10 @@ class FunctionResponseDict(TypedDict, total=False):
   scheduling: Optional[FunctionResponseScheduling]
   """Specifies how the response should be scheduled in the conversation. Only applicable to NON_BLOCKING function calls, is ignored otherwise. Defaults to WHEN_IDLE."""
 
+  parts: Optional[list[FunctionResponsePartDict]]
+  """List of parts that constitute a function response. Each part may
+      have a different IANA MIME type."""
+
   id: Optional[str]
   """Optional. The id of the function call this response is for. Populated by the client to match the corresponding function call `id`."""
 
@@ -1658,7 +1773,7 @@ class Schema(_common.BaseModel):
 
     def convert_schema(schema: Union['Schema', dict[str, Any]]) -> JSONSchema:
       if isinstance(schema, pydantic.BaseModel):
-        schema_dict = schema.model_dump()
+        schema_dict = schema.model_dump(exclude_none=True)
       else:
         schema_dict = schema
       json_schema = JSONSchema()
@@ -1666,7 +1781,13 @@ class Schema(_common.BaseModel):
         if field_value is None:
           continue
         elif field_name == 'nullable':
-          json_schema.type = JSONSchemaType.NULL
+          if json_schema.type is None:
+            json_schema.type = JSONSchemaType.NULL
+          elif isinstance(json_schema.type, JSONSchemaType):
+            current_type: JSONSchemaType = json_schema.type
+            json_schema.type = [current_type, JSONSchemaType.NULL]
+          elif isinstance(json_schema.type, list):
+            json_schema.type.append(JSONSchemaType.NULL)
         elif field_name not in json_schema_field_names:
           continue
         elif field_name == 'type':
@@ -2680,6 +2801,14 @@ class ToolComputerUse(_common.BaseModel):
   environment: Optional[Environment] = Field(
       default=None, description="""Required. The environment being operated."""
   )
+  excluded_predefined_functions: Optional[list[str]] = Field(
+      default=None,
+      description="""By default, predefined functions are included in the final model call.
+    Some of them can be explicitly excluded from being automatically included.
+    This can serve two purposes:
+      1. Using a more restricted / different action space.
+      2. Improving the definitions / instructions of predefined functions.""",
+  )
 
 
 class ToolComputerUseDict(TypedDict, total=False):
@@ -2688,6 +2817,13 @@ class ToolComputerUseDict(TypedDict, total=False):
   environment: Optional[Environment]
   """Required. The environment being operated."""
 
+  excluded_predefined_functions: Optional[list[str]]
+  """By default, predefined functions are included in the final model call.
+    Some of them can be explicitly excluded from being automatically included.
+    This can serve two purposes:
+      1. Using a more restricted / different action space.
+      2. Improving the definitions / instructions of predefined functions."""
+
 
 ToolComputerUseOrDict = Union[ToolComputerUse, ToolComputerUseDict]
 
@@ -6799,7 +6935,7 @@ class _EditImageParameters(_common.BaseModel):
       description="""A text description of the edit to apply to the image.""",
   )
   reference_images: Optional[list[_ReferenceImageAPI]] = Field(
-      default=None, description="""The reference images for Imagen 3 editing."""
+      default=None, description="""The reference images for editing."""
   )
   config: Optional[EditImageConfig] = Field(
       default=None, description="""Configuration for editing."""
@@ -6816,7 +6952,7 @@ class _EditImageParametersDict(TypedDict, total=False):
   """A text description of the edit to apply to the image."""
 
   reference_images: Optional[list[_ReferenceImageAPIDict]]
-  """The reference images for Imagen 3 editing."""
+  """The reference images for editing."""
 
   config: Optional[EditImageConfigDict]
   """Configuration for editing."""
@@ -13150,6 +13286,58 @@ SubjectReferenceImageOrDict = Union[
 ]
 
 
+class ContentReferenceImage(_common.BaseModel):
+  """A content reference image.
+
+  A content reference image represents a subject to reference (ex. person,
+  product, animal) provided by the user. It can optionally be provided in
+  addition to a style reference image (ex. background, style reference).
+  """
+
+  reference_image: Optional[Image] = Field(
+      default=None,
+      description="""The reference image for the editing operation.""",
+  )
+  reference_id: Optional[int] = Field(
+      default=None, description="""The id of the reference image."""
+  )
+  reference_type: Optional[str] = Field(
+      default=None,
+      description="""The type of the reference image. Only set by the SDK.""",
+  )
+
+  @pydantic.model_validator(mode='before')
+  @classmethod
+  def _validate_mask_image_config(self, values: Any) -> Any:
+    if 'reference_type' in values:
+      raise ValueError('Cannot set internal reference_type field directly.')
+    values['reference_type'] = 'REFERENCE_TYPE_CONTENT'
+    return values
+
+
+class ContentReferenceImageDict(TypedDict, total=False):
+  """A content reference image.
+
+  A content reference image represents a subject to reference (ex. person,
+  product, animal) provided by the user. It can optionally be provided in
+  addition to a style reference image (ex. background, style reference).
+  """
+
+  reference_image: Optional[ImageDict]
+  """The reference image for the editing operation."""
+
+  reference_id: Optional[int]
+  """The id of the reference image."""
+
+  reference_type: Optional[str]
+  """The type of the reference image. Only set by the SDK."""
+
+
+ContentReferenceImageOrDict = Union[
+    ContentReferenceImage, ContentReferenceImageDict
+]
+
+
 class LiveServerSetupComplete(_common.BaseModel):
   """Sent in response to a `LiveGenerateContentSetup` message from the client."""
 

google/genai/version.py

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

{google_genai-1.38.0.dist-info → google_genai-1.39.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.38.0
+Version: 1.39.1
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -122,6 +122,87 @@ from google import genai
 client = genai.Client()
 ```
 
+## Close a client
+
+Explicitly close the sync client to ensure that resources, such as the
+ underlying HTTP connections, are properly cleaned up and closed.
+
+```python
+
+from google.genai import Client
+
+client = Client()
+response_1 = client.models.generate_content(
+    model=MODEL_ID,
+    contents='Hello',
+)
+response_2 = client.models.generate_content(
+    model=MODEL_ID,
+    contents='Ask a question',
+)
+# Close the sync client to release resources.
+client.close()
+```
+
+To explicitly close the async client:
+
+```python
+
+from google.genai import Client
+
+aclient = Client(
+    vertexai=True, project='my-project-id', location='us-central1'
+).aio
+response_1 = await aclient.models.generate_content(
+    model=MODEL_ID,
+    contents='Hello',
+)
+response_2 = await aclient.models.generate_content(
+    model=MODEL_ID,
+    contents='Ask a question',
+)
+# Close the async client to release resources.
+await aclient.aclose()
+```
+
+## Client context managers
+
+By using the sync client context manager, it will close the underlying
+ sync client when exiting the with block.
+
+```python
+from google.genai import Client
+
+with Client() as client:
+  response_1 = client.models.generate_content(
+      model=MODEL_ID,
+      contents='Hello',
+  )
+  response_2 = client.models.generate_content(
+      model=MODEL_ID,
+      contents='Ask a question',
+  )
+
+```
+
+By using the async client context manager, it will close the underlying
+ async client when exiting the with block.
+
+```python
+from google.genai import Client
+
+async with Client().aio as aclient:
+  response_1 = await aclient.models.generate_content(
+      model=MODEL_ID,
+      contents='Hello',
+  )
+  response_2 = await aclient.models.generate_content(
+      model=MODEL_ID,
+      contents='Ask a question',
+  )
+
+```
+
 ### API Selection
 
 By default, the SDK uses the beta API endpoints provided by Google to support

google_genai-1.39.1.dist-info/RECORD

@@ -0,0 +1,39 @@
+google/genai/__init__.py,sha256=SKz_9WQKA3R4OpJIDJlgssVfizLNDG2tuWtOD9pxrPE,729
+google/genai/_adapters.py,sha256=Kok38miNYJff2n--l0zEK_hbq0y2rWOH7k75J7SMYbQ,1744
+google/genai/_api_client.py,sha256=MjXU31z7dtbix5hSVM0dkM_TFvpUfonyvRhEdQ-NX-k,62056
+google/genai/_api_module.py,sha256=lj8eUWx8_LBGBz-49qz6_ywWm3GYp3d8Bg5JoOHbtbI,902
+google/genai/_automatic_function_calling_util.py,sha256=xXNkJR-pzSMkeSXMz3Jw-kMHFbTJEiRJ3wocuwtWW4I,11627
+google/genai/_base_transformers.py,sha256=wljA6m4tLl4XLGlBC2DNOls5N9-X9tffBq0M7i8jgpw,1034
+google/genai/_base_url.py,sha256=E5H4dew14Y16qfnB3XRnjSCi19cJVlkaMNoM_8ip-PM,1597
+google/genai/_common.py,sha256=YU5842SoZOW0usJRnz58Wi8hqJ1f-Dd01flGTsqI8-U,20653
+google/genai/_extra_utils.py,sha256=myEPyU_-RXExXtMGitJUKEMWUPiDudScgTN9LqFC8h4,20468
+google/genai/_live_converters.py,sha256=zsYlIE6b7VKuwkW7blgmxyCNqXOuSiDz8H2b1kd3USs,108032
+google/genai/_local_tokenizer_loader.py,sha256=cGN1F0f7hNjRIGCGTLeox7IGAZf_YcvZjSp2rCyhUak,7465
+google/genai/_mcp_utils.py,sha256=HuWJ8FUjquv40Mf_QjcL5r5yXWrS-JjINsjlOSbbyAc,3870
+google/genai/_operations_converters.py,sha256=hPmrlU_yJWT4di2arA0VKaoQIB1MbCPglmAZ4D8M-Ds,8744
+google/genai/_replay_api_client.py,sha256=MmpzqE5AxVeyvCahEnmukYGIZqN8lxS-suSgUszvLSw,22555
+google/genai/_test_api_client.py,sha256=4ruFIy5_1qcbKqqIBu3HSQbpSOBrxiecBtDZaTGFR1s,4797
+google/genai/_tokens_converters.py,sha256=yZx_RXgoYuSC2uyXYw0RXR5LRsykJ_0mxMs67L5piRw,25173
+google/genai/_transformers.py,sha256=-1GIrDNS4fH6Qx2jNm-VhcmlTSjWxanna7N7Tp_ctQ8,40861
+google/genai/batches.py,sha256=x0KyOuj98NABrnZ4SNpGolNIsod6VB92NhGVM1dTnlY,102044
+google/genai/caches.py,sha256=-0TIzBDlVSCZNKR1PQb7q7pNQ0kwnRbb2MQYRwALMAc,67187
+google/genai/chats.py,sha256=pIBw8d13llupLn4a7vP6vnpbzDcvCCrZZ-Q2r8Cvo7g,16652
+google/genai/client.py,sha256=bwKV5gHKpxzmfFTtoudQ_hEz5QfUzKYMJHYT-AnQfNU,13066
+google/genai/errors.py,sha256=zaPEs_GrtZuypvSPnOe32CTHO6nEVtshvc3Av2ug2Ac,5822
+google/genai/files.py,sha256=wNM0k9Hab8I7STThiZtA8BdRDaj6fNIdN7DxMr8zzxc,40211
+google/genai/live.py,sha256=BUyIOt-PON4SuJtKXA6l3ujzA3GUbYWMGSgZvsyKjrg,39674
+google/genai/live_music.py,sha256=3GG9nsto8Vhkohcs-4CPMS4DFp1ZtMuLYzHfvEPYAeg,6971
+google/genai/local_tokenizer.py,sha256=EKZ72cV2Zfutlo_efMOPnLRNZN4WQe57rD3G80cF340,14109
+google/genai/models.py,sha256=FlP9HgR-uaNBXXmDmZ8OQjoTnyT1wmiBhoDGIjKC0jg,266642
+google/genai/operations.py,sha256=mgRuVCoqUf7z0RfB84W9x7S839VCttmIQZkXovHAMrE,17061
+google/genai/pagers.py,sha256=m0SfWWn1EJs2k1On3DZx371qb8g2BRm_188ExsicIRc,7098
+google/genai/py.typed,sha256=RsMFoLwBkAvY05t6izop4UHZtqOPLiKp3GkIEizzmQY,40
+google/genai/tokens.py,sha256=8RbZ0kgvyKT3SwbgIUOHr6TTZL24v4fqYarhlA8r1ac,12503
+google/genai/tunings.py,sha256=_VzlKhlqFQHB7Lt4vU1HfGD11E5iolDfPdWrnigu5pw,62885
+google/genai/types.py,sha256=WfXL6BFOYhuHMLZKoVnPwvNoi0s73ajwtoW-0jK8C88,550882
+google/genai/version.py,sha256=9kaezsq7iGydw527Yusdh0fiol0CTkWV4TaMWjGEbww,627
+google_genai-1.39.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-1.39.1.dist-info/METADATA,sha256=VzVw-MudwcrpSVi0l1SaSda3t1StnVuJ6qxoxm-cdRI,45381
+google_genai-1.39.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+google_genai-1.39.1.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-1.39.1.dist-info/RECORD,,

google_genai-1.38.0.dist-info/RECORD

@@ -1,39 +0,0 @@
-google/genai/__init__.py,sha256=SKz_9WQKA3R4OpJIDJlgssVfizLNDG2tuWtOD9pxrPE,729
-google/genai/_adapters.py,sha256=Kok38miNYJff2n--l0zEK_hbq0y2rWOH7k75J7SMYbQ,1744
-google/genai/_api_client.py,sha256=j3NJI7lfQbe6KqyoC6FvLTnWzzskZJdmjIGGzUslRsY,61331
-google/genai/_api_module.py,sha256=lj8eUWx8_LBGBz-49qz6_ywWm3GYp3d8Bg5JoOHbtbI,902
-google/genai/_automatic_function_calling_util.py,sha256=xXNkJR-pzSMkeSXMz3Jw-kMHFbTJEiRJ3wocuwtWW4I,11627
-google/genai/_base_transformers.py,sha256=wljA6m4tLl4XLGlBC2DNOls5N9-X9tffBq0M7i8jgpw,1034
-google/genai/_base_url.py,sha256=E5H4dew14Y16qfnB3XRnjSCi19cJVlkaMNoM_8ip-PM,1597
-google/genai/_common.py,sha256=VziliVbesmNtke6Fh2wk-kQlCOxEpmkqYb0y4qfKIo4,20074
-google/genai/_extra_utils.py,sha256=myEPyU_-RXExXtMGitJUKEMWUPiDudScgTN9LqFC8h4,20468
-google/genai/_live_converters.py,sha256=5yndouhtCw3WkEIWEcf1k8HywaGrnJpuuTuFKvXaA5M,103950
-google/genai/_local_tokenizer_loader.py,sha256=cGN1F0f7hNjRIGCGTLeox7IGAZf_YcvZjSp2rCyhUak,7465
-google/genai/_mcp_utils.py,sha256=HuWJ8FUjquv40Mf_QjcL5r5yXWrS-JjINsjlOSbbyAc,3870
-google/genai/_operations_converters.py,sha256=I1b46qhGCeB5uHNuyrMxpFqmzWuvSFQ5pS52csd1J4E,9080
-google/genai/_replay_api_client.py,sha256=REKFmX0d8mEgMCf1mhy-bRa-hWOqbqZrBpwpMnzQLFw,22370
-google/genai/_test_api_client.py,sha256=4ruFIy5_1qcbKqqIBu3HSQbpSOBrxiecBtDZaTGFR1s,4797
-google/genai/_tokens_converters.py,sha256=42xx5Inv3YMqEdarb3lmt-w54D2tpb9QEBy0DSUATQk,24963
-google/genai/_transformers.py,sha256=0r3lv77ZcxpkfaGWSw-DMXEGmEbZ4aRfUOrSyqTvnEo,40304
-google/genai/batches.py,sha256=UJ-zc2UXqQfej8bqqtgeNrepc7zDQBzRlmkRf4Wlr4U,102880
-google/genai/caches.py,sha256=OuWOomIf7McQk0_jKivA-aBv__5vDc249rBEPqOh_XE,67637
-google/genai/chats.py,sha256=pIBw8d13llupLn4a7vP6vnpbzDcvCCrZZ-Q2r8Cvo7g,16652
-google/genai/client.py,sha256=wXnfZBSv9p-yKtX_gabUrfBXoYHuqHhzK_VgwRttMgY,10777
-google/genai/errors.py,sha256=zaPEs_GrtZuypvSPnOe32CTHO6nEVtshvc3Av2ug2Ac,5822
-google/genai/files.py,sha256=T8uDgqCQheFwYYYJdN4lax4NOR_q6bJyr0jQf079VeE,40607
-google/genai/live.py,sha256=BUyIOt-PON4SuJtKXA6l3ujzA3GUbYWMGSgZvsyKjrg,39674
-google/genai/live_music.py,sha256=3GG9nsto8Vhkohcs-4CPMS4DFp1ZtMuLYzHfvEPYAeg,6971
-google/genai/local_tokenizer.py,sha256=ZYOdZFS2nJbgDGOMrVzn2hZ_1G8I4_PuVsh6HU8jEgo,14055
-google/genai/models.py,sha256=eyvuAh7Ky-rpWYBFNnXz8cjgGjvP08EYxgHBY8VInBM,268195
-google/genai/operations.py,sha256=B0NSu6XNZwMIBTU7Wz_YU9KY64kJvmu5wY7TThyh2uQ,17509
-google/genai/pagers.py,sha256=m0SfWWn1EJs2k1On3DZx371qb8g2BRm_188ExsicIRc,7098
-google/genai/py.typed,sha256=RsMFoLwBkAvY05t6izop4UHZtqOPLiKp3GkIEizzmQY,40
-google/genai/tokens.py,sha256=8RbZ0kgvyKT3SwbgIUOHr6TTZL24v4fqYarhlA8r1ac,12503
-google/genai/tunings.py,sha256=TVohav5pT5jOj4RgnPdvLI6iDdOVGnPRB8XRBNuyP2E,63513
-google/genai/types.py,sha256=rfC7XHOj0gY623h6o2OYkgG5qwdx3XnIqNbflsKVhiU,544627
-google/genai/version.py,sha256=KuRCODgoTTXMoEmnCuEy6FnYuAvHgdE5EVPAkRRmjYc,627
-google_genai-1.38.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-1.38.0.dist-info/METADATA,sha256=XPd0ingxVwk3XGZkgkU7szy0DonWV6ge4NYAfVKTO0g,43622
-google_genai-1.38.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_genai-1.38.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-1.38.0.dist-info/RECORD,,