google-genai 0.6.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

google/genai/version.py

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

{google_genai-0.6.0.dist-info → google_genai-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: google-genai
-Version: 0.6.0
+Version: 0.7.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -21,7 +21,6 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: google-auth<3.0.0dev,>=2.14.1
-Requires-Dist: pillow<12.0.0,>=10.0.0
 Requires-Dist: pydantic<3.0.0dev,>=2.0.0
 Requires-Dist: requests<3.0.0dev,>=2.28.1
 Requires-Dist: websockets<15.0dev,>=13.0
@@ -83,12 +82,12 @@ The `client.models` modules exposes model inferencing and model getters.
 
 ```python
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp", contents="What is your name?"
+    model="gemini-2.0-flash-exp", contents="why is the sky blue?"
 )
 print(response.text)
 ```
 
-#### with uploaded file (Google AI only)
+#### with uploaded file (Gemini API only)
 download the file in console.
 
 ```cmd
@@ -127,7 +126,7 @@ dictionaries. You can get the type from `google.genai.types`.
 ```python
 response = client.models.generate_content(
     model="gemini-2.0-flash-exp",
-    contents=types.Part.from_text("Why is the sky blue?"),
+    contents=types.Part.from_text(text="Why is the sky blue?"),
     config=types.GenerateContentConfig(
         temperature=0,
         top_p=0.95,
@@ -141,7 +140,43 @@ response = client.models.generate_content(
     ),
 )
 
-response
+print(response.text)
+```
+
+### Thinking
+
+The Gemini 2.0 Flash Thinking model is an experimental model that could return
+"thoughts" as part of its response.
+
+#### Gemini Developer API
+
+Thinking config is only available in v1alpha for Gemini AI API.
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-thinking-exp',
+    contents='What is the sum of natural numbers from 1 to 100?',
+    config=types.GenerateContentConfig(
+        thinking_config=types.ThinkingConfig(include_thoughts=True),
+        http_options=types.HttpOptions(api_version='v1alpha'),
+    )
+)
+for part in response.candidates[0].content.parts:
+    print(part)
+```
+
+#### Vertex AI API
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-thinking-exp-01-21',
+    contents='What is the sum of natural numbers from 1 to 100?',
+    config=types.GenerateContentConfig(
+        thinking_config=types.ThinkingConfig(include_thoughts=True),
+    )
+)
+for part in response.candidates[0].content.parts:
+    print(part)
 ```
 
 ### List Base Models
@@ -232,10 +267,10 @@ Then you will receive a function call part in the response.
 function = types.FunctionDeclaration(
     name="get_current_weather",
     description="Get the current weather in a given location",
-    parameters=types.FunctionParameters(
+    parameters=types.Schema(
         type="OBJECT",
         properties={
-            "location": types.ParameterType(
+            "location": types.Schema(
                 type="STRING",
                 description="The city and state, e.g. San Francisco, CA",
             ),
@@ -263,10 +298,9 @@ The following example shows how to do it for a simple function invocation.
 ```python
 user_prompt_content = types.Content(
     role="user",
-    parts=[types.Part.from_text("What is the weather like in Boston?")],
+    parts=[types.Part.from_text(text="What is the weather like in Boston?")],
 )
-function_call_content = response.candidates[0].content
-function_call_part = function_call_content.parts[0]
+function_call_part = response.function_calls[0]
 
 
 try:
@@ -366,6 +400,57 @@ response = client.models.generate_content(
 print(response.text)
 ```
 
+### Enum Response Schema
+
+#### Text Response
+
+You can set response_mime_type to 'text/x.enum' to return one of those enum
+values as the response.
+
+```python
+class InstrumentEnum(Enum):
+  PERCUSSION = 'Percussion'
+  STRING = 'String'
+  WOODWIND = 'Woodwind'
+  BRASS = 'Brass'
+  KEYBOARD = 'Keyboard'
+
+response = client.models.generate_content(
+      model='gemini-2.0-flash-exp',
+      contents='What instrument plays multiple notes at once?',
+      config={
+          'response_mime_type': 'text/x.enum',
+          'response_schema': InstrumentEnum,
+      },
+  )
+print(response.text)
+```
+
+#### JSON Response
+
+You can also set response_mime_type to 'application/json', the response will be identical but in quotes.
+
+```python
+from enum import Enum
+
+class InstrumentEnum(Enum):
+  PERCUSSION = 'Percussion'
+  STRING = 'String'
+  WOODWIND = 'Woodwind'
+  BRASS = 'Brass'
+  KEYBOARD = 'Keyboard'
+
+response = client.models.generate_content(
+      model='gemini-2.0-flash-exp',
+      contents='What instrument plays multiple notes at once?',
+      config={
+          'response_mime_type': 'application/json',
+          'response_schema': InstrumentEnum,
+      },
+  )
+print(response.text)
+```
+
 ### Streaming
 
 #### Streaming for text content
@@ -434,10 +519,10 @@ print(response.text)
 ### Streaming
 
 ```python
-async for response in client.aio.models.generate_content_stream(
+async for chunk in await client.aio.models.generate_content_stream(
     model="gemini-2.0-flash-exp", contents="Tell me a story in 300 words."
 ):
-    print(response.text, end="")
+    print(chunk.text, end="")
 ```
 
 ### Count Tokens and Compute Tokens
@@ -445,7 +530,7 @@ async for response in client.aio.models.generate_content_stream(
 ```python
 response = client.models.count_tokens(
     model="gemini-2.0-flash-exp",
-    contents="What is your name?",
+    contents="why is the sky blue?",
 )
 print(response)
 ```
@@ -457,7 +542,7 @@ Compute tokens is only supported in Vertex AI.
 ```python
 response = client.models.compute_tokens(
     model="gemini-2.0-flash-exp",
-    contents="What is your name?",
+    contents="why is the sky blue?",
 )
 print(response)
 ```
@@ -467,7 +552,7 @@ print(response)
 ```python
 response = await client.aio.models.count_tokens(
     model="gemini-2.0-flash-exp",
-    contents="What is your name?",
+    contents="why is the sky blue?",
 )
 print(response)
 ```
@@ -477,7 +562,7 @@ print(response)
 ```python
 response = client.models.embed_content(
     model="text-embedding-004",
-    contents="What is your name?",
+    contents="why is the sky blue?",
 )
 print(response)
 ```
@@ -486,7 +571,7 @@ print(response)
 # multiple contents with config
 response = client.models.embed_content(
     model="text-embedding-004",
-    contents=["What is your name?", "What is your age?"],
+    contents=["why is the sky blue?", "What is your age?"],
     config=types.EmbedContentConfig(output_dimensionality=10),
 )
 
@@ -495,16 +580,16 @@ print(response)
 
 ### Imagen
 
-#### Generate Image
+#### Generate Images
 
-Support for generate image in Gemini Developer API is behind an allowlist
+Support for generate images in Gemini Developer API is behind an allowlist
 
 ```python
 # Generate Image
-response1 = client.models.generate_image(
-    model="imagen-3.0-generate-001",
+response1 = client.models.generate_images(
+    model="imagen-3.0-generate-002",
     prompt="An umbrella in the foreground, and a rainy night sky in the background",
-    config=types.GenerateImageConfig(
+    config=types.GenerateImagesConfig(
         negative_prompt="human",
         number_of_images=1,
         include_rai_reason=True,
@@ -603,7 +688,7 @@ print(response.text)
 
 ```python
 chat = client.aio.chats.create(model="gemini-2.0-flash-exp")
-async for chunk in chat.send_message_stream("tell me a story"):
+async for chunk in await chat.send_message_stream("tell me a story"):
     print(chunk.text)
 ```
 
@@ -679,7 +764,7 @@ cached_content = client.caches.create(
 cached_content = client.caches.get(name=cached_content.name)
 ```
 
-### Generate Content
+### Generate Content with Caches
 
 ```python
 response = client.models.generate_content(
@@ -695,7 +780,7 @@ print(response.text)
 ## Tunings
 
 `client.tunings` contains tuning job APIs and supports supervised fine
-tuning through `tune` and distillation through `distill`
+tuning through `tune`.
 
 ### Tune
 
@@ -760,7 +845,7 @@ while tuning_job.state in running_states:
 ```python
 response = client.models.generate_content(
     model=tuning_job.tuned_model.endpoint,
-    contents="What is your name?",
+    contents="why is the sky blue?",
 )
 
 print(response.text)
@@ -820,43 +905,6 @@ model = client.models.update(
 print(model)
 ```
 
-### Distillation
-
-Only supported in Vertex AI. Requires allowlist.
-
-```python
-distillation_job = client.tunings.distill(
-    student_model="gemma-2b-1.1-it",
-    teacher_model="gemini-1.5-pro-002",
-    training_dataset=genai.types.DistillationDataset(
-        gcs_uri="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl",
-    ),
-    config=genai.types.CreateDistillationJobConfig(
-        epoch_count=1,
-        pipeline_root_directory=("gs://my-bucket"),
-    ),
-)
-print(distillation_job)
-```
-
-```python
-completed_states = set(
-    [
-        "JOB_STATE_SUCCEEDED",
-        "JOB_STATE_FAILED",
-        "JOB_STATE_CANCELLED",
-        "JOB_STATE_PAUSED",
-    ]
-)
-
-while distillation_job.state not in completed_states:
-    print(distillation_job.state)
-    distillation_job = client.tunings.get(name=distillation_job.name)
-    time.sleep(10)
-
-print(distillation_job)
-```
-
 
 ### List Tuning Jobs
 
@@ -932,12 +980,12 @@ job
 ### List
 
 ```python
-for job in client.batches.list(config=types.ListBatchJobConfig(page_size=10)):
+for job in client.batches.list(config=types.ListBatchJobsConfig(page_size=10)):
     print(job)
 ```
 
 ```python
-pager = client.batches.list(config=types.ListBatchJobConfig(page_size=10))
+pager = client.batches.list(config=types.ListBatchJobsConfig(page_size=10))
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -948,14 +996,14 @@ print(pager[0])
 
 ```python
 async for job in await client.aio.batches.list(
-    config=types.ListBatchJobConfig(page_size=10)
+    config=types.ListBatchJobsConfig(page_size=10)
 ):
     print(job)
 ```
 
 ```python
 async_pager = await client.aio.batches.list(
-    config=types.ListBatchJobConfig(page_size=10)
+    config=types.ListBatchJobsConfig(page_size=10)
 )
 print(async_pager.page_size)
 print(async_pager[0])

google_genai-0.7.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+google/genai/__init__.py,sha256=IYw-PcsdgjSpS1mU_ZcYkTfPocsJ4aVmrDxP7vX7c6Y,709
+google/genai/_api_client.py,sha256=JPCynlSCc_yPEV4-rdV2-BPv-OHoBvotxXbI0VLA3Q4,22882
+google/genai/_api_module.py,sha256=9bxmtcSTpT8Ht6VwJyw7fQqiR0jJXz7350dWGl-bC5E,780
+google/genai/_automatic_function_calling_util.py,sha256=sEaDAeHjv-H71o1L3_P8sqOslK4TK0Rybn4WPymeEBk,10665
+google/genai/_common.py,sha256=aAtTQeGYn8WeiQD2lbjYiHZXWzBdj59wKiqAyQeaByc,8610
+google/genai/_extra_utils.py,sha256=y-6Jr2GN2BKZV67I6fTgDtwfsOTQs7QlLDBQdmW_jKk,11258
+google/genai/_replay_api_client.py,sha256=8KMjwjah52Mgf0rURK-CO10MqtULh8vkRSOT4-AqPpU,14722
+google/genai/_test_api_client.py,sha256=2PvDcW3h01U4UOSoj7TUo6TwdBHSEN_lO2tXjBoh5Fw,4765
+google/genai/_transformers.py,sha256=AiSVoQML3MK6AP5xTStIiJUOlrZO4m_qBULOjgdZHC0,21963
+google/genai/batches.py,sha256=jv8pW_g_cZee6ol5ER5bQRUkXsj4IUcZC5cMo-YAnt0,38033
+google/genai/caches.py,sha256=jsiclHO71kIa2CNrds3O8PL2fCNr_dlhUSPjhiRsjNE,53152
+google/genai/chats.py,sha256=GyufXQPtyP_v4L3943xaKXMpo1Us9sBTdMSTUV4P6s8,7827
+google/genai/client.py,sha256=MTZ3DOXk1_xgljaHlvF16jr_SKVPRfU8lZ1eH_dfDeQ,9334
+google/genai/errors.py,sha256=DtpDZT5UDqumk2cTRUlg3k4ypmO_0tkMNzJgA3qzCmc,3666
+google/genai/files.py,sha256=FQLGv-at7T_AZiVkhTsreIaLK_U1acufx-zt8coOjkc,41706
+google/genai/live.py,sha256=wxz8ebqcPR6JJs39OOVz8zPzfAf31Zol7sGE7byQEyI,23302
+google/genai/models.py,sha256=mitGmy1myd6-zmmV8BEDkdc88cbp8ZF9SyZTNeWnBfk,166628
+google/genai/pagers.py,sha256=hSHd-gLvEzYWwK85i8EcFNWUMKtszUs7Nw2r3L7d6_U,6686
+google/genai/tunings.py,sha256=iJJYn1O_wjFKTIL8VS2zpIRqpfCNRrO2REP2ztgFW6M,39144
+google/genai/types.py,sha256=vOP1JkluVGFhmNFBJk6Z1BoOv4zzwysjIUiU7GFPXwg,273187
+google/genai/version.py,sha256=nKmMytxBOALBA2ILryXEQSQpZZJxtfZwKkJcznog9Lw,626
+google_genai-0.7.0.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-0.7.0.dist-info/METADATA,sha256=eUcx3JIC0OOIh3_hT-pfkYvkyP2ZWrUXU5JoCZqTDwg,23948
+google_genai-0.7.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+google_genai-0.7.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-0.7.0.dist-info/RECORD,,

google_genai-0.6.0.dist-info/RECORD

@@ -1,25 +0,0 @@
-google/genai/__init__.py,sha256=IYw-PcsdgjSpS1mU_ZcYkTfPocsJ4aVmrDxP7vX7c6Y,709
-google/genai/_api_client.py,sha256=_Ta8Sjkg5hyO9UspTboevtk19-h7x1DBD940reMVM5c,23341
-google/genai/_automatic_function_calling_util.py,sha256=qbMCO8x6THe1O7Bn-L97rlbDSYJoX_gUfztvKHh-u6E,10078
-google/genai/_common.py,sha256=xJadfXcUP2ry348wz6Pd7v6abTi_5KlCieJre2nZUi8,8724
-google/genai/_extra_utils.py,sha256=QhmZV-vfKfNQ2KMUVdAD-Le37qzLD4dSLl953zyOvCk,11101
-google/genai/_replay_api_client.py,sha256=xD6NAxy-8vJQS8FtcWf4dpVMjW90V5xaDO9-ndzY8VU,14722
-google/genai/_test_api_client.py,sha256=p771T27icmzENxKtyNDwPG1sTI7jaoJNFPwlwq9GK6o,4759
-google/genai/_transformers.py,sha256=pMMq6IJTLh_QKHi63XTYKNbWOPJ2mpXLB2Q6v78eQsk,19268
-google/genai/batches.py,sha256=1TYMFgpe6-jz1WG-PzP2aj0p_Pvb0y81HFfnrJ8Vj7g,37504
-google/genai/caches.py,sha256=yXk1apId5SWzfZ5yi6ejonfT7AWiYgzb3FsRxAXFN5s,53998
-google/genai/chats.py,sha256=GobIFlez3eTRWWDtUycnubrMz0hB3v3gvDVSdMFJTNc,7642
-google/genai/client.py,sha256=AM7yAthSW5Ajnig2BfwCNPExjTTtrZR0b0pgyThnDBo,9497
-google/genai/errors.py,sha256=DtpDZT5UDqumk2cTRUlg3k4ypmO_0tkMNzJgA3qzCmc,3666
-google/genai/files.py,sha256=ySacctcUekVTHMHKWepEfge-KgVMvJlek6UIxn7qHSw,42537
-google/genai/live.py,sha256=vWAzuyAbK63n1Vr-iRNMuDq3McjgmSXY5ToO2bzdp28,23106
-google/genai/models.py,sha256=2Aj98r1B5cxryAxBRm1iUhpZ3px0c26ZhyH_0h9NQp4,160762
-google/genai/pagers.py,sha256=hSHd-gLvEzYWwK85i8EcFNWUMKtszUs7Nw2r3L7d6_U,6686
-google/genai/tunings.py,sha256=IddtUlYqPZmaicPoO4yhCPefop_FhJiKl9l4dyPEeVE,49396
-google/genai/types.py,sha256=wFhtYoto-ZfcqqRyu9ZGOQss9wBwPYTdcLuCjnil_kY,274625
-google/genai/version.py,sha256=GkH2NQgVNItrcxGOfKBijKTmc_gqY3cPzldxCE11Jm4,626
-google_genai-0.6.0.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-0.6.0.dist-info/METADATA,sha256=AiUA_1JhtT5M6vqc6mzs0i6qI1RJFOccuUJH-ZlX_z8,22743
-google_genai-0.6.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-google_genai-0.6.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-0.6.0.dist-info/RECORD,,