google-genai 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

google/genai/__init__.py

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

google/genai/_api_client.py

@@ -51,7 +51,7 @@ class HttpOptions(TypedDict):
 def _append_library_version_headers(headers: dict[str, str]) -> None:
   """Appends the telemetry header to the headers dict."""
   # TODO: Automate revisions to the SDK library version.
-  library_label = f'google-genai-sdk/0.2.0'
+  library_label = f'google-genai-sdk/0.2.2'
   language_label = 'gl-python/' + sys.version.split()[0]
   version_header_value = f'{library_label} {language_label}'
   if (
@@ -200,7 +200,7 @@ class ApiClient:
       _append_library_version_headers(self._http_options['headers'])
 
   def _websocket_base_url(self):
-    url_parts = urllib.parse.urlparse(self._http_options['base_url'])
+    url_parts = urlparse(self._http_options['base_url'])
     return url_parts._replace(scheme='wss').geturl()
 
   def _build_request(

google/genai/errors.py

@@ -29,29 +29,49 @@ class APIError(Exception):
   code: int
   response: requests.Response
 
-  message: str = ''
-  status: str = 'UNKNOWN'
-  details: Optional[Any] = None
+  status: Optional[str] = None
+  message: Optional[str] = None
+  response: Optional[Any] = None
 
   def __init__(
       self, code: int, response: Union[requests.Response, 'ReplayResponse']
   ):
-    self.code = code
     self.response = response
 
     if isinstance(response, requests.Response):
       try:
-        raw_error = response.json().get('error', {})
+        # do not do any extra muanipulation on the response.
+        # return the raw response json as is.
+        response_json = response.json()
       except requests.exceptions.JSONDecodeError:
-        raw_error = {'message': response.text, 'status': response.reason}
+        response_json = {
+            'message': response.text,
+            'status': response.reason,
+        }
     else:
-      raw_error = response.body_segments[0].get('error', {})
+      response_json = response.body_segments[0].get('error', {})
+
+    self.details = response_json
+    self.message = self._get_message(response_json)
+    self.status = self._get_status(response_json)
+    self.code = code if code else self._get_code(response_json)
+
+    super().__init__(f'{self.code} {self.status}. {self.details}')
+
+  def _get_status(self, response_json):
+    return response_json.get(
+        'status', response_json.get('error', {}).get('status', None)
+    )
 
-    self.message = raw_error.get('message', '')
-    self.status = raw_error.get('status', 'UNKNOWN')
-    self.details = raw_error.get('details', None)
+  def _get_message(self, response_json):
+    return response_json.get(
+        'message', response_json.get('error', {}).get('message', None)
+    )
 
-    super().__init__(f'{self.code} {self.status}. {self.message}')
+  def _get_code(self, response_json):
+    return response_json.get(
+        'code', response_json.get('error', {}).get('code', None)
+    )
 
   def _to_replay_record(self):
     """Returns a dictionary representation of the error for replay recording.

google/genai/types.py

@@ -1280,7 +1280,7 @@ class AutomaticFunctionCallingConfig(_common.BaseModel):
       """,
   )
   maximum_remote_calls: Optional[int] = Field(
-      default=None,
+      default=10,
       description="""If automatic function calling is enabled,
       maximum number of remote calls for automatic function calling.
       This number should be a positive integer.

{google_genai-0.2.0.dist-info → google_genai-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-genai
-Version: 0.2.0
+Version: 0.2.2
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -159,6 +159,14 @@ response = client.models.generate_content(
 response.text
 ```
 
+#### Manually declare and invoke a function for function calling
+
+If you don't want to use the automatic function support, you can manually
+declare the function and invoke it.
+
+The following example shows how to declare a function and pass it as a tool.
+Then you will receive a function call part in the response.
+
 ``` python
 function = dict(
     name="get_current_weather",
@@ -187,15 +195,24 @@ response = client.models.generate_content(
 response.candidates[0].content.parts[0].function_call
 ```
 
+After you receive the function call part from model, you can invoke the function
+and get the function response. And then you can pass the function response to
+the model.
+The following example shows how to do it for a simple function invocation.
+
 ``` python
 function_call_part = response.candidates[0].content.parts[0]
 
-function_response = get_current_weather(**function_call_part.function_call.args)
+try:
+  function_result = get_current_weather(**function_call_part.function_call.args)
+  function_response = {'result': function_result}
+except Exception as e:  # instead of raising the exception, you can let the model handle it
+  function_response = {'error': str(e)}
 
 
 function_response_part = types.Part.from_function_response(
     name=function_call_part.function_call.name,
-    response={'result': function_response}
+    response=function_response,
 )
 
 response = client.models.generate_content(
@@ -273,6 +290,8 @@ print(response.text)
 
 ### Streaming
 
+#### Streaming for text content
+
 ``` python
 for chunk in client.models.generate_content_stream(
     model='gemini-2.0-flash-exp', contents='Tell me a story in 300 words.'
@@ -280,6 +299,47 @@ for chunk in client.models.generate_content_stream(
   print(chunk.text)
 ```
 
+#### Streaming for image content
+
+If your image is stored in Google Cloud Storage, you can use the `from_uri`
+class method to create a Part object.
+
+``` python
+for chunk in client.models.generate_content_stream(
+    model='gemini-1.5-flash',
+    contents=[
+      'What is this image about?',
+      types.Part.from_uri(
+        file_uri='gs://generativeai-downloads/images/scones.jpg',
+        mime_type='image/jpeg'
+      )
+    ],
+):
+  print(chunk.text)
+```
+
+If your image is stored in your local file system, you can read it in as bytes
+data and use the `from_bytes` class method to create a Part object.
+
+``` python
+YOUR_IMAGE_PATH = 'your_image_path'
+YOUR_IMAGE_MIME_TYPE = 'your_image_mime_type'
+with open(YOUR_IMAGE_PATH, 'rb') as f:
+  image_bytes = f.read()
+
+for chunk in client.models.generate_content_stream(
+    model='gemini-1.5-flash',
+    contents=[
+      'What is this image about?',
+      types.Part.from_bytes(
+        data=image_bytes,
+        mime_type=YOUR_IMAGE_MIME_TYPE
+      )
+    ],
+):
+  print(chunk.text)
+```
+
 ### Async
 
 `client.aio` exposes all the analogous `async` methods that are

{google_genai-0.2.0.dist-info → google_genai-0.2.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-google/genai/__init__.py,sha256=bO4TBLSOack_93tDti_USNTkE8cvkLUn45TH-WS-HOE,674
-google/genai/_api_client.py,sha256=74qm-UTxDQ3-KCb61eqD3SAM379MrsQw8ly6GBLoDB8,15963
+google/genai/__init__.py,sha256=R7sy9MQmlItVERIQEWt25bApyNXxDIkYP4nl3-EtX50,674
+google/genai/_api_client.py,sha256=naJy-6OjgjUNlciZNxDconmSOFhewD_0QsggNY1aCik,15950
 google/genai/_automatic_function_calling_util.py,sha256=E25_66RH3DbDIucq7x-93XWPPBwB9FnzwD1NCGyPrjM,10242
 google/genai/_common.py,sha256=Yj5cBkq5QRNFSBqvpB949Rjo7cbIhdtKp5dJxMW_I6I,7971
 google/genai/_extra_utils.py,sha256=GQZnraFCrMffqrBEpurdcBmgrltRsnYgMizt-Ok6xX8,11098
@@ -10,15 +10,15 @@ google/genai/batches.py,sha256=Wi4Kptampp2WepAqv_AawwNCR6MKVhLKmzJdYXDQ_aE,37148
 google/genai/caches.py,sha256=LJm2raykec7_iCHsVbEtX4v942mR-OSQvxTVKcBN2RA,53434
 google/genai/chats.py,sha256=x-vCXrsxZ8kdEZ_0ZDfrBQnQ9urCr42x3urP0OXHyTo,5688
 google/genai/client.py,sha256=HH_lYnjPOwW-4Vgynyw4K8cwurT2g578Dc51H_uk7GY,9244
-google/genai/errors.py,sha256=TrlUk1jz7r1aN1lrL3FZZ30LU4iMfSonm1ZwEAk07k4,3048
+google/genai/errors.py,sha256=ZqJvfuJ7oS334WBrq3otzdZfmhEgcM1OBZhHccYzDok,3665
 google/genai/files.py,sha256=dn3q8P9aTN9OG3PtA4AYDs9hF6Uk-jkMjgAW7dSlt_4,35573
 google/genai/live.py,sha256=T-pOtq7k43wE2VjQzqLrx-kqhotS66I2PY_NHBdv9G8,22056
 google/genai/models.py,sha256=t5XgwlgkNrQKb6eww0oBGzjMiMQaj-BQedc8lVdJHz4,154834
 google/genai/pagers.py,sha256=hSHd-gLvEzYWwK85i8EcFNWUMKtszUs7Nw2r3L7d6_U,6686
 google/genai/tunings.py,sha256=tFTSEaECKZ6xeYcxUTIKUmXqPoDymYP3eyTcEKjnPa4,49010
-google/genai/types.py,sha256=mIjtCSXbp6CRL5iEhtdxczoMTtyQ1EKYpBlzLvGIedY,263841
-google_genai-0.2.0.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-0.2.0.dist-info/METADATA,sha256=qhc4AtoMxFa_-BIEzNTw36s-cE1imA3zOoILYIpL7as,17371
-google_genai-0.2.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-google_genai-0.2.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-0.2.0.dist-info/RECORD,,
+google/genai/types.py,sha256=JC7CBQVRzVwImsT03t6Qv_vMYq8V58z3SF-rzvrUJHc,263839
+google_genai-0.2.2.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-0.2.2.dist-info/METADATA,sha256=Q1eNLWWM0fqko3S5gfudpPYrlmnebdnXXLOJGjAtmG0,19175
+google_genai-0.2.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+google_genai-0.2.2.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-0.2.2.dist-info/RECORD,,