google-genai 1.33.0__py3-none-any.whl → 1.34.0__py3-none-any.whl

google/genai/types.py

@@ -885,6 +885,41 @@ class FileDataDict(TypedDict, total=False):
 FileDataOrDict = Union[FileData, FileDataDict]
 
 
+class FunctionCall(_common.BaseModel):
+  """A function call."""
+
+  id: Optional[str] = Field(
+      default=None,
+      description="""The unique id of the function call. If populated, the client to execute the
+   `function_call` and return the response with the matching `id`.""",
+  )
+  args: Optional[dict[str, Any]] = Field(
+      default=None,
+      description="""Optional. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details.""",
+  )
+  name: Optional[str] = Field(
+      default=None,
+      description="""Required. The name of the function to call. Matches [FunctionDeclaration.name].""",
+  )
+
+
+class FunctionCallDict(TypedDict, total=False):
+  """A function call."""
+
+  id: Optional[str]
+  """The unique id of the function call. If populated, the client to execute the
+   `function_call` and return the response with the matching `id`."""
+
+  args: Optional[dict[str, Any]]
+  """Optional. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details."""
+
+  name: Optional[str]
+  """Required. The name of the function to call. Matches [FunctionDeclaration.name]."""
+
+
+FunctionCallOrDict = Union[FunctionCall, FunctionCallDict]
+
+
 class CodeExecutionResult(_common.BaseModel):
   """Result of executing the [ExecutableCode].
 
@@ -953,41 +988,6 @@ class ExecutableCodeDict(TypedDict, total=False):
 ExecutableCodeOrDict = Union[ExecutableCode, ExecutableCodeDict]
 
 
-class FunctionCall(_common.BaseModel):
-  """A function call."""
-
-  id: Optional[str] = Field(
-      default=None,
-      description="""The unique id of the function call. If populated, the client to execute the
-   `function_call` and return the response with the matching `id`.""",
-  )
-  args: Optional[dict[str, Any]] = Field(
-      default=None,
-      description="""Optional. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details.""",
-  )
-  name: Optional[str] = Field(
-      default=None,
-      description="""Required. The name of the function to call. Matches [FunctionDeclaration.name].""",
-  )
-
-
-class FunctionCallDict(TypedDict, total=False):
-  """A function call."""
-
-  id: Optional[str]
-  """The unique id of the function call. If populated, the client to execute the
-   `function_call` and return the response with the matching `id`."""
-
-  args: Optional[dict[str, Any]]
-  """Optional. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details."""
-
-  name: Optional[str]
-  """Required. The name of the function to call. Matches [FunctionDeclaration.name]."""
-
-
-FunctionCallOrDict = Union[FunctionCall, FunctionCallDict]
-
-
 class FunctionResponse(_common.BaseModel):
   """A function response."""
 
@@ -1075,6 +1075,12 @@ class Part(_common.BaseModel):
       default=None,
       description="""An opaque signature for the thought so it can be reused in subsequent requests.""",
   )
+  function_call: Optional[FunctionCall] = Field(
+      default=None,
+      description="""A predicted [FunctionCall] returned from the model that contains a string
+      representing the [FunctionDeclaration.name] and a structured JSON object
+      containing the parameters and their values.""",
+  )
   code_execution_result: Optional[CodeExecutionResult] = Field(
       default=None,
       description="""Optional. Result of executing the [ExecutableCode].""",
@@ -1083,10 +1089,6 @@ class Part(_common.BaseModel):
       default=None,
       description="""Optional. Code generated by the model that is meant to be executed.""",
   )
-  function_call: Optional[FunctionCall] = Field(
-      default=None,
-      description="""Optional. A predicted [FunctionCall] returned from the model that contains a string representing the [FunctionDeclaration.name] with the parameters and their values.""",
-  )
   function_response: Optional[FunctionResponse] = Field(
       default=None,
       description="""Optional. The result output of a [FunctionCall] that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing any output from the function call. It is used as context to the model.""",
@@ -1181,15 +1183,17 @@ class PartDict(TypedDict, total=False):
   thought_signature: Optional[bytes]
   """An opaque signature for the thought so it can be reused in subsequent requests."""
 
+  function_call: Optional[FunctionCallDict]
+  """A predicted [FunctionCall] returned from the model that contains a string
+      representing the [FunctionDeclaration.name] and a structured JSON object
+      containing the parameters and their values."""
+
   code_execution_result: Optional[CodeExecutionResultDict]
   """Optional. Result of executing the [ExecutableCode]."""
 
   executable_code: Optional[ExecutableCodeDict]
   """Optional. Code generated by the model that is meant to be executed."""
 
-  function_call: Optional[FunctionCallDict]
-  """Optional. A predicted [FunctionCall] returned from the model that contains a string representing the [FunctionDeclaration.name] with the parameters and their values."""
-
   function_response: Optional[FunctionResponseDict]
   """Optional. The result output of a [FunctionCall] that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing any output from the function call. It is used as context to the model."""
 
@@ -11435,6 +11439,70 @@ class InlinedResponseDict(TypedDict, total=False):
 InlinedResponseOrDict = Union[InlinedResponse, InlinedResponseDict]
 
 
+class SingleEmbedContentResponse(_common.BaseModel):
+  """Config for `response` parameter."""
+
+  embedding: Optional[ContentEmbedding] = Field(
+      default=None,
+      description="""The response to the request.
+      """,
+  )
+  token_count: Optional[int] = Field(
+      default=None,
+      description="""The error encountered while processing the request.
+      """,
+  )
+
+
+class SingleEmbedContentResponseDict(TypedDict, total=False):
+  """Config for `response` parameter."""
+
+  embedding: Optional[ContentEmbeddingDict]
+  """The response to the request.
+      """
+
+  token_count: Optional[int]
+  """The error encountered while processing the request.
+      """
+
+
+SingleEmbedContentResponseOrDict = Union[
+    SingleEmbedContentResponse, SingleEmbedContentResponseDict
+]
+
+
+class InlinedEmbedContentResponse(_common.BaseModel):
+  """Config for `inlined_embedding_responses` parameter."""
+
+  response: Optional[SingleEmbedContentResponse] = Field(
+      default=None,
+      description="""The response to the request.
+      """,
+  )
+  error: Optional[JobError] = Field(
+      default=None,
+      description="""The error encountered while processing the request.
+      """,
+  )
+
+
+class InlinedEmbedContentResponseDict(TypedDict, total=False):
+  """Config for `inlined_embedding_responses` parameter."""
+
+  response: Optional[SingleEmbedContentResponseDict]
+  """The response to the request.
+      """
+
+  error: Optional[JobErrorDict]
+  """The error encountered while processing the request.
+      """
+
+
+InlinedEmbedContentResponseOrDict = Union[
+    InlinedEmbedContentResponse, InlinedEmbedContentResponseDict
+]
+
+
 class BatchJobDestination(_common.BaseModel):
   """Config for `des` parameter."""
 
@@ -11470,6 +11538,15 @@ class BatchJobDestination(_common.BaseModel):
       the input requests.
       """,
   )
+  inlined_embed_content_responses: Optional[
+      list[InlinedEmbedContentResponse]
+  ] = Field(
+      default=None,
+      description="""The responses to the requests in the batch. Returned when the batch was
+      built using inlined requests. The responses will be in the same order as
+      the input requests.
+      """,
+  )
 
 
 class BatchJobDestinationDict(TypedDict, total=False):
@@ -11502,6 +11579,14 @@ class BatchJobDestinationDict(TypedDict, total=False):
       the input requests.
       """
 
+  inlined_embed_content_responses: Optional[
+      list[InlinedEmbedContentResponseDict]
+  ]
+  """The responses to the requests in the batch. Returned when the batch was
+      built using inlined requests. The responses will be in the same order as
+      the input requests.
+      """
+
 
 BatchJobDestinationOrDict = Union[BatchJobDestination, BatchJobDestinationDict]
 
@@ -11662,6 +11747,13 @@ class BatchJob(_common.BaseModel):
       """,
   )
 
+  @property
+  def done(self) -> bool:
+    """Returns True if the batch job has ended."""
+    if self.state is None:
+      return False
+    return self.state.name in JOB_STATES_ENDED
+
 
 class BatchJobDict(TypedDict, total=False):
   """Config for batches.create return value."""
@@ -11712,6 +11804,138 @@ class BatchJobDict(TypedDict, total=False):
 BatchJobOrDict = Union[BatchJob, BatchJobDict]
 
 
+class EmbedContentBatch(_common.BaseModel):
+  """Parameters for the embed_content method."""
+
+  contents: Optional[ContentListUnion] = Field(
+      default=None,
+      description="""The content to embed. Only the `parts.text` fields will be counted.
+      """,
+  )
+  config: Optional[EmbedContentConfig] = Field(
+      default=None,
+      description="""Configuration that contains optional parameters.
+      """,
+  )
+
+
+class EmbedContentBatchDict(TypedDict, total=False):
+  """Parameters for the embed_content method."""
+
+  contents: Optional[ContentListUnionDict]
+  """The content to embed. Only the `parts.text` fields will be counted.
+      """
+
+  config: Optional[EmbedContentConfigDict]
+  """Configuration that contains optional parameters.
+      """
+
+
+EmbedContentBatchOrDict = Union[EmbedContentBatch, EmbedContentBatchDict]
+
+
+class EmbeddingsBatchJobSource(_common.BaseModel):
+
+  file_name: Optional[str] = Field(
+      default=None,
+      description="""The Gemini Developer API's file resource name of the input data
+      (e.g. "files/12345").
+      """,
+  )
+  inlined_requests: Optional[EmbedContentBatch] = Field(
+      default=None,
+      description="""The Gemini Developer API's inlined input data to run batch job.
+      """,
+  )
+
+
+class EmbeddingsBatchJobSourceDict(TypedDict, total=False):
+
+  file_name: Optional[str]
+  """The Gemini Developer API's file resource name of the input data
+      (e.g. "files/12345").
+      """
+
+  inlined_requests: Optional[EmbedContentBatchDict]
+  """The Gemini Developer API's inlined input data to run batch job.
+      """
+
+
+EmbeddingsBatchJobSourceOrDict = Union[
+    EmbeddingsBatchJobSource, EmbeddingsBatchJobSourceDict
+]
+
+
+class CreateEmbeddingsBatchJobConfig(_common.BaseModel):
+  """Config for optional parameters."""
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+  display_name: Optional[str] = Field(
+      default=None,
+      description="""The user-defined name of this BatchJob.
+      """,
+  )
+
+
+class CreateEmbeddingsBatchJobConfigDict(TypedDict, total=False):
+  """Config for optional parameters."""
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+  display_name: Optional[str]
+  """The user-defined name of this BatchJob.
+      """
+
+
+CreateEmbeddingsBatchJobConfigOrDict = Union[
+    CreateEmbeddingsBatchJobConfig, CreateEmbeddingsBatchJobConfigDict
+]
+
+
+class _CreateEmbeddingsBatchJobParameters(_common.BaseModel):
+  """Config for batches.create parameters."""
+
+  model: Optional[str] = Field(
+      default=None,
+      description="""The name of the model to produces the predictions via the BatchJob.
+      """,
+  )
+  src: Optional[EmbeddingsBatchJobSource] = Field(
+      default=None,
+      description="""input data to run batch job".
+      """,
+  )
+  config: Optional[CreateEmbeddingsBatchJobConfig] = Field(
+      default=None,
+      description="""Optional parameters for creating a BatchJob.
+      """,
+  )
+
+
+class _CreateEmbeddingsBatchJobParametersDict(TypedDict, total=False):
+  """Config for batches.create parameters."""
+
+  model: Optional[str]
+  """The name of the model to produces the predictions via the BatchJob.
+      """
+
+  src: Optional[EmbeddingsBatchJobSourceDict]
+  """input data to run batch job".
+      """
+
+  config: Optional[CreateEmbeddingsBatchJobConfigDict]
+  """Optional parameters for creating a BatchJob.
+      """
+
+
+_CreateEmbeddingsBatchJobParametersOrDict = Union[
+    _CreateEmbeddingsBatchJobParameters, _CreateEmbeddingsBatchJobParametersDict
+]
+
+
 class GetBatchJobConfig(_common.BaseModel):
   """Optional parameters."""
 

google/genai/version.py

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

{google_genai-1.33.0.dist-info → google_genai-1.34.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.33.0
+Version: 1.34.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -30,6 +30,9 @@ Requires-Dist: websockets<15.1.0,>=13.0.0
 Requires-Dist: typing-extensions<5.0.0,>=4.11.0
 Provides-Extra: aiohttp
 Requires-Dist: aiohttp<4.0.0; extra == "aiohttp"
+Provides-Extra: local-tokenizer
+Requires-Dist: sentencepiece>=0.2.0; extra == "local-tokenizer"
+Requires-Dist: protobuf; extra == "local-tokenizer"
 Dynamic: license-file
 
 # Google Gen AI SDK
@@ -1063,6 +1066,20 @@ response = await client.aio.models.count_tokens(
 print(response)
 ```
 
+#### Local Count Tokens
+
+```python
+tokenizer = genai.LocalTokenizer(model_name='gemini-2.0-flash-001')
+result = tokenizer.count_tokens("What is your name?")
+```
+
+#### Local Compute Tokens
+
+```python
+tokenizer = genai.LocalTokenizer(model_name='gemini-2.0-flash-001')
+result = tokenizer.compute_tokens("What is your name?")
+```
+
 ### Embed Content
 
 ```python

{google_genai-1.33.0.dist-info → google_genai-1.34.0.dist-info}/RECORD

@@ -5,35 +5,35 @@ google/genai/_api_module.py,sha256=lj8eUWx8_LBGBz-49qz6_ywWm3GYp3d8Bg5JoOHbtbI,9
 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=SmBlz7AQZbKbT8KE5vsvK5Iz1OaRwHF6J51KDJBbjMo,19936
-google/genai/_extra_utils.py,sha256=6mxUnbwKkrAJ9zR8JP4nCKuA0F1BFqAc2t5mzdAONxk,20603
-google/genai/_live_converters.py,sha256=brba9Wk_ljc8auy8w0n49dZXYHTBYhegKQcnjr89QgM,101812
+google/genai/_common.py,sha256=VziliVbesmNtke6Fh2wk-kQlCOxEpmkqYb0y4qfKIo4,20074
+google/genai/_extra_utils.py,sha256=myEPyU_-RXExXtMGitJUKEMWUPiDudScgTN9LqFC8h4,20468
+google/genai/_live_converters.py,sha256=OBKnJD7zgrfG6TOVSXoo9TkVHwXvLzudod0U40MqtZw,103192
 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=hXs5NuCpcZU-e6rfeWBStZmF33_2vn1dAo3NeTA8P2o,22413
 google/genai/_test_api_client.py,sha256=4ruFIy5_1qcbKqqIBu3HSQbpSOBrxiecBtDZaTGFR1s,4797
-google/genai/_tokens_converters.py,sha256=jPHlskl303A63uEshdwLeSSOls9kuNO1WMY5NAYF5U0,24382
-google/genai/_transformers.py,sha256=Kfo6LP3HoCco2E9EQYnRyElgOSBXZ_SKI-Spx3AwUWw,38917
-google/genai/batches.py,sha256=1YIkibVzEX0p_va2kfNL3tc-X7Id13whWuhKto0mch8,87774
-google/genai/caches.py,sha256=6Y3D2qSwlvX8CunOiJRcRWtaiFP0r-9oMcyqnegC7Sc,66451
+google/genai/_tokens_converters.py,sha256=42xx5Inv3YMqEdarb3lmt-w54D2tpb9QEBy0DSUATQk,24963
+google/genai/_transformers.py,sha256=rtaCagL4htyu8MzuxqQJ_xnJqRwwuQKR7CWXYpydBZ4,40260
+google/genai/batches.py,sha256=UJ-zc2UXqQfej8bqqtgeNrepc7zDQBzRlmkRf4Wlr4U,102880
+google/genai/caches.py,sha256=0LVzGcF1t5tHq5QUZzstEGfOqbUPV3B8eywknqFWKD8,67650
 google/genai/chats.py,sha256=pIBw8d13llupLn4a7vP6vnpbzDcvCCrZZ-Q2r8Cvo7g,16652
 google/genai/client.py,sha256=wXnfZBSv9p-yKtX_gabUrfBXoYHuqHhzK_VgwRttMgY,10777
 google/genai/errors.py,sha256=GlEvypbRgF3h5BxlocmWVf-S9kzERA_lwGLCMyAvpcY,5714
 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=HpQk8toJFGvRozdyboOhdAFDOFAiZ07IdXeGiR25nSM,13252
-google/genai/models.py,sha256=RxGIm7dmgE4I-4yru835VrJ3MuN5mz-OaRcTrFfefKI,264975
+google/genai/local_tokenizer.py,sha256=ZYOdZFS2nJbgDGOMrVzn2hZ_1G8I4_PuVsh6HU8jEgo,14055
+google/genai/models.py,sha256=84KEXFWSVaE9vLgPwUWZCE-MXldBZUlHAiI3WBoW7qM,267291
 google/genai/operations.py,sha256=Ccm4EW13cT_EtCq8Fa3n8jlQySyy9KqJ98ZK4pF0DDA,12815
 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=vVV9NvrtzoWDLzcTLSrgD5lzFeSRJdCB1QtfzL3yj3c,63275
-google/genai/types.py,sha256=UcWcyrKlKgbfumHluov2G8UsOoq6bnWDKQJIfKifuv8,531058
-google/genai/version.py,sha256=t_FYlWqCnBvBCPieE9c3vw6-ven4MXsOqgsxhMq9U0c,627
-google_genai-1.33.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-1.33.0.dist-info/METADATA,sha256=C4_2IQgzIZ77LF1Vpl2VKdoL8Xx2KCy3ni8COdjvalI,43080
-google_genai-1.33.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_genai-1.33.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-1.33.0.dist-info/RECORD,,
+google/genai/types.py,sha256=X_EJw9XImkNUV-kei6sX_EyamVh3ZcllCyhDY6oGIKg,537135
+google/genai/version.py,sha256=pq67EsG5-MVW2erGgNTP2s1tkENaUdCa412taBfzfhM,627
+google_genai-1.34.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-1.34.0.dist-info/METADATA,sha256=hfQsErEKbSrvHuHNO9w49pDdJnDcjIdj5mDKNAst85o,43556
+google_genai-1.34.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+google_genai-1.34.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-1.34.0.dist-info/RECORD,,