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

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

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: google-genai
-Version: 0.0.1
+Version: 0.2.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
-Project-URL: Homepage, https://github.com/googlapis/genai-python
+Project-URL: Homepage, https://github.com/googleapis/python-genai
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Internet
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
@@ -27,6 +28,13 @@ Requires-Dist: websockets<15.0dev,>=13.0
 
 # Google Gen AI SDK
 
+[![PyPI version](https://img.shields.io/pypi/v/google-genai.svg)](https://pypi.org/project/google-genai/)
+
+--------
+**Documentation:** https://googleapis.github.io/python-genai/
+
+-----
+
 ## Imports
 
 ``` python
@@ -66,7 +74,7 @@ The `client.models` modules exposes model inferencing and model getters.
 
 ``` python
 response = client.models.generate_content(
-    model='gemini-1.5-pro-002', contents='What is your name?'
+    model='gemini-2.0-flash-exp', contents='What is your name?'
 )
 print(response.text)
 ```
@@ -75,7 +83,7 @@ print(response.text)
 
 ``` python
 response = client.models.generate_content(
-    model='gemini-1.5-pro-002',
+    model='gemini-2.0-flash-exp',
     contents='high',
     config=types.GenerateContentConfig(
         system_instruction='I say high, you say low',
@@ -92,7 +100,7 @@ dictionaries. You can get the type from `google.genai.types`.
 
 ``` python
 response = client.models.generate_content(
-    model='gemini-1.5-flash-002',
+    model='gemini-2.0-flash-exp',
     contents=types.Part.from_text('Why is sky blue?'),
     config=types.GenerateContentConfig(
         temperature=0,
@@ -114,7 +122,7 @@ response
 
 ``` python
 response = client.models.generate_content(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='Say something bad.',
     config=types.GenerateContentConfig(
         safety_settings= [types.SafetySetting(
@@ -143,7 +151,7 @@ def get_current_weather(location: str,) -> int:
   return 'sunny'
 
 response = client.models.generate_content(
-    model='gemini-1.5-pro-002',
+    model='gemini-2.0-flash-exp',
     contents="What is the weather like in Boston?",
     config=types.GenerateContentConfig(tools=[get_current_weather],)
 )
@@ -171,7 +179,7 @@ tool = types.Tool(function_declarations=[function])
 
 
 response = client.models.generate_content(
-    model='gemini-1.5-pro-002',
+    model='gemini-2.0-flash-exp',
     contents="What is the weather like in Boston?",
     config=types.GenerateContentConfig(tools=[tool],)
 )
@@ -191,7 +199,7 @@ function_response_part = types.Part.from_function_response(
 )
 
 response = client.models.generate_content(
-    model='gemini-1.5-pro-002',
+    model='gemini-2.0-flash-exp',
     contents=[
         types.Part.from_text("What is the weather like in Boston?"),
         function_call_part,
@@ -221,7 +229,7 @@ class CountryInfo(BaseModel):
 
 
 response = client.models.generate_content(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='Give me information of the United States.',
     config=types.GenerateContentConfig(
         response_mime_type= 'application/json',
@@ -233,7 +241,7 @@ print(response.text)
 
 ``` python
 response = client.models.generate_content(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='Give me information of the United States.',
     config={
         'response_mime_type': 'application/json',
@@ -267,7 +275,7 @@ print(response.text)
 
 ``` python
 for chunk in client.models.generate_content_stream(
-    model='gemini-1.5-flash', contents='Tell me a story in 300 words.'
+    model='gemini-2.0-flash-exp', contents='Tell me a story in 300 words.'
 ):
   print(chunk.text)
 ```
@@ -282,7 +290,7 @@ of `client.models.generate_content`
 
 ``` python
 request = await client.aio.models.generate_content(
-    model='gemini-1.5-flash', contents='Tell me a story in 300 words.'
+    model='gemini-2.0-flash-exp', contents='Tell me a story in 300 words.'
 )
 
 print(response.text)
@@ -292,7 +300,7 @@ print(response.text)
 
 ``` python
 async for response in client.aio.models.generate_content_stream(
-    model='gemini-1.5-flash', contents='Tell me a story in 300 words.'
+    model='gemini-2.0-flash-exp', contents='Tell me a story in 300 words.'
 ):
   print(response.text)
 ```
@@ -301,7 +309,7 @@ async for response in client.aio.models.generate_content_stream(
 
 ``` python
 response = client.models.count_tokens(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='What is your name?',
 )
 print(response)
@@ -313,7 +321,7 @@ Compute tokens is not supported by Google AI.
 
 ``` python
 response = client.models.compute_tokens(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='What is your name?',
 )
 print(response)
@@ -323,7 +331,7 @@ print(response)
 
 ``` python
 response = await client.aio.models.count_tokens(
-    model='gemini-1.5-flash',
+    model='gemini-2.0-flash-exp',
     contents='What is your name?',
 )
 print(response)
@@ -360,14 +368,12 @@ Support for generate image in Google AI is behind an allowlist
 # Generate Image
 response1 = client.models.generate_image(
     model='imagen-3.0-generate-001',
-    prompt='Robot holding a red skateboard',
+    prompt='An umbrella in the foreground, and a rainy night sky in the background',
     config=types.GenerateImageConfig(
+        negative_prompt= "human",
         number_of_images= 1,
-        person_generation= "ALLOW_ADULT",
-        safety_filter_level= "BLOCK_LOW_AND_ABOVE",
         include_rai_reason= True,
-        add_watermark= True,
-        aspect_ratio= "4:3"
+        output_mime_type= "image/jpeg"
     )
 )
 response1.generated_images[0].image.show()
@@ -375,9 +381,10 @@ response1.generated_images[0].image.show()
 
 #### Upscale Image
 
+Upscale image is not supported in Google AI.
+
 ``` python
-# Upscale the generated image from the cell above
-from google.genai.types import Image
+# Upscale the generated image from above
 response2 = client.models.upscale_image(
     model='imagen-3.0-generate-001',
     image=response1.generated_images[0].image,
@@ -388,21 +395,36 @@ response2.generated_images[0].image.show()
 
 #### Edit Image
 
+Edit image is not supported in Google AI.
+
 ``` python
-# Edit the generated image from the first cell
-from google.genai.types import Image
+# Edit the generated image from above
+from google.genai.types import RawReferenceImage, MaskReferenceImage
+raw_ref_image = RawReferenceImage(
+    reference_id=1,
+    reference_image=response1.generated_images[0].image,
+)
+
+# Model computes a mask of the background
+mask_ref_image = MaskReferenceImage(
+    reference_id=2,
+    config=types.MaskReferenceConfig(
+        mask_mode='MASK_MODE_BACKGROUND',
+        mask_dilation=0,
+    ),
+)
+
 response3 = client.models.edit_image(
-    model='imagegeneration@006',
-    prompt='holding yellow surfboard',
-    image=response1.generated_images[0].image,
+    model='imagen-3.0-capability-preview-0930',
+    prompt='Sunlight and clear sky',
+    reference_images=[raw_ref_image, mask_ref_image],
     config=types.EditImageConfig(
-        edit_mode="inpainting-insert",
-        mask_type="semantic",
-        segmentation_classes=[156],
+        edit_mode= 'EDIT_MODE_INPAINT_INSERTION',
         number_of_images= 1,
+        negative_prompt= 'human',
         include_rai_reason= True,
-        product_position="reposition"
-    )
+        output_mime_type= 'image/jpeg',
+    ),
 )
 response3.generated_images[0].image.show()
 ```
@@ -439,7 +461,7 @@ client.files.delete(name=file3.name)
 ### Create
 
 ``` python
-if client._api_client.vertexai:
+if client.vertexai:
   file_uris = [
       'gs://cloud-samples-data/generative-ai/pdf/2312.11805v3.pdf',
       'gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf'
@@ -489,7 +511,7 @@ client.models.generate_content(
 ## Tunings
 
 `client.tunings` contains tuning job APIs and supports supervised fine
-tuning through `tune` and distiallation through `distill`
+tuning through `tune` and distillation through `distill`
 
 ### Tune
 
@@ -497,7 +519,7 @@ tuning through `tune` and distiallation through `distill`
 -   Google AI supports tuning from inline examples
 
 ``` python
-if client._api_client.vertexai:
+if client.vertexai:
   model = 'gemini-1.5-pro-002'
   training_dataset=types.TuningDataset(
         gcs_uri='gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl',
@@ -576,9 +598,9 @@ for model in client.models.list(config={'page_size': 10}):
 ``` python
 pager = client.models.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -591,15 +613,15 @@ async for job in await client.aio.models.list(config={'page_size': 10}):
 ``` python
 async_pager = await client.aio.models.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ### Update Tuned Model
 
 ``` python
-model = pager.page[0]
+model = pager[0]
 
 model = client.models.update(
     model=model.name,
@@ -624,7 +646,7 @@ distillation_job = client.tunings.distill(
     config=genai.types.CreateDistillationJobConfig(
         epoch_count=1,
         pipeline_root_directory=(
-            "gs://vertex-sdk-dev-staging-us-central1/tmp/distillation_pipeline_root"
+            "gs://my-bucket"
         ),
     ),
 )
@@ -659,9 +681,9 @@ for job in client.tunings.list(config={'page_size': 10}):
 ``` python
 pager = client.tunings.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -674,9 +696,9 @@ async for job in await client.aio.tunings.list(config={'page_size': 10}):
 ``` python
 async_pager = await client.aio.tunings.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ## Batch Prediction
@@ -689,12 +711,7 @@ Only supported in Vertex AI.
 # Specify model and source file only, destination and job display name will be auto-populated
 job = client.batches.create(
     model='gemini-1.5-flash-002',
-    src='bq://vertex-sdk-dev.unified_genai_tests_batches.generate_content_requests',
-    # # optionally specify destination and display_name by yourself
-    # config = {
-    #     'dest': 'bq://vertex-sdk-dev.unified_genai_tests_batches.generate_content_responses',
-    #     'display_name': 'create_batch_job_demo'
-    # }
+    src='bq://my-project.my-dataset.my-table',
 )
 
 job
@@ -733,9 +750,9 @@ for job in client.batches.list(config={'page_size': 10}):
 ``` python
 pager = client.batches.list(config={'page_size': 10})
 print(pager.page_size)
-print(pager.page[0])
+print(pager[0])
 pager.next_page()
-print(pager.page[0])
+print(pager[0])
 ```
 
 #### Async
@@ -746,11 +763,11 @@ async for job in await client.aio.batches.list(config={'page_size': 10}):
 ```
 
 ``` python
-async_pager = await client.aio.tunings.list(config={'page_size': 10})
+async_pager = await client.aio.batches.list(config={'page_size': 10})
 print(async_pager.page_size)
-print(async_pager.page[0])
+print(async_pager[0])
 await async_pager.next_page()
-print(async_pager.page[0])
+print(async_pager[0])
 ```
 
 ### Delete

google_genai-0.2.0.dist-info/RECORD

@@ -0,0 +1,24 @@
+google/genai/__init__.py,sha256=bO4TBLSOack_93tDti_USNTkE8cvkLUn45TH-WS-HOE,674
+google/genai/_api_client.py,sha256=74qm-UTxDQ3-KCb61eqD3SAM379MrsQw8ly6GBLoDB8,15963
+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
+google/genai/_replay_api_client.py,sha256=QPNg4SBpOLS58bx-kuJQngxy1tbjMpCpJzmImCwYePA,16226
+google/genai/_test_api_client.py,sha256=p771T27icmzENxKtyNDwPG1sTI7jaoJNFPwlwq9GK6o,4759
+google/genai/_transformers.py,sha256=_zWNr7zFTrUFniECYaZUn0n4TdioLpj783l3-z1XvIE,13443
+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/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-0.0.1.dist-info/RECORD

@@ -1,24 +0,0 @@
-google/genai/__init__.py,sha256=49gQv3uugzoY9BDHbhMKfHzYfn5CcttAm47Ecp6gvQs,651
-google/genai/_api_client.py,sha256=gzgQ7-Zu48trXpcr0WR6OqBk-BzujsFkNKBdifjS3U8,15637
-google/genai/_automatic_function_calling_util.py,sha256=Q7_wty4RNBoYBT2aHWf2EdJSJkqG965nri8F5NCLx20,11578
-google/genai/_common.py,sha256=Yj5cBkq5QRNFSBqvpB949Rjo7cbIhdtKp5dJxMW_I6I,7971
-google/genai/_extra_utils.py,sha256=TT3YieI_SxSWYyLJwhe7i33262vf2W0YTatBSF0dPHg,10684
-google/genai/_replay_api_client.py,sha256=QPNg4SBpOLS58bx-kuJQngxy1tbjMpCpJzmImCwYePA,16226
-google/genai/_test_api_client.py,sha256=p771T27icmzENxKtyNDwPG1sTI7jaoJNFPwlwq9GK6o,4759
-google/genai/_transformers.py,sha256=CYyQibFoW3ZZAOXrIb_BPZulLcmJvuBNJcCp3m3sGeQ,13475
-google/genai/batches.py,sha256=wpdyI5IFXDBLPzVkkEAcElPZEt76lWBkf7fNpqG9bb8,30406
-google/genai/caches.py,sha256=h4MX19ZUC7CAd7DY91FWPjUbTQHG-3-Oi4eDU-5nn40,53424
-google/genai/chats.py,sha256=80GWA431KehYTfAggMh6W3jHdDizbjk0RfO8MXilwDM,5342
-google/genai/client.py,sha256=HH_lYnjPOwW-4Vgynyw4K8cwurT2g578Dc51H_uk7GY,9244
-google/genai/errors.py,sha256=TrlUk1jz7r1aN1lrL3FZZ30LU4iMfSonm1ZwEAk07k4,3048
-google/genai/files.py,sha256=9B1Xsb2R7jBAU3Ot-1Z4YJjC9gQEywzAfLT24yHN4EI,35567
-google/genai/live.py,sha256=htpzNj8L8nionvbzQJLNZbwD4_1UQHTBGkgQVHXRkQo,20835
-google/genai/models.py,sha256=sQwTPqjxiv4jhP-Rn6nBlAyGBP_gK6lAUcVL56DwcwE,151681
-google/genai/pagers.py,sha256=hSHd-gLvEzYWwK85i8EcFNWUMKtszUs7Nw2r3L7d6_U,6686
-google/genai/tunings.py,sha256=srGxr4KlRxZ01lPT5ohtrNxqlA9I6RCbCLcxcfATY4s,39452
-google/genai/types.py,sha256=Fm-Y8CTYF6BY3ZE0EgzOCJ6C1SuygqT6Kfu4_Obw5Ko,250959
-google_genai-0.0.1.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-0.0.1.dist-info/METADATA,sha256=TgTn66RX2xEykoOUkwVsSncghI8TGug0GhHe5QfHDhE,17151
-google_genai-0.0.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-google_genai-0.0.1.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-0.0.1.dist-info/RECORD,,