google-genai 0.6.0__tar.gz → 0.7.0__tar.gz

{google_genai-0.6.0/google_genai.egg-info → google_genai-0.7.0}/PKG-INFO

@@ -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.6.0 → google_genai-0.7.0}/README.md

@@ -55,12 +55,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
@@ -99,7 +99,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,
@@ -113,7 +113,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
@@ -204,10 +240,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",
             ),
@@ -235,10 +271,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:
@@ -338,6 +373,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
@@ -406,10 +492,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
@@ -417,7 +503,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)
 ```
@@ -429,7 +515,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)
 ```
@@ -439,7 +525,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)
 ```
@@ -449,7 +535,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)
 ```
@@ -458,7 +544,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),
 )
 
@@ -467,16 +553,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,
@@ -575,7 +661,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)
 ```
 
@@ -651,7 +737,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(
@@ -667,7 +753,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
 
@@ -732,7 +818,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)
@@ -792,43 +878,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
 
@@ -904,12 +953,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()
@@ -920,14 +969,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])