google-genai 1.0.0rc0__py3-none-any.whl → 1.1.0__py3-none-any.whl

{google_genai-1.0.0rc0.dist-info → google_genai-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: google-genai
-Version: 1.0.0rc0
+Version: 1.1.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -34,7 +34,7 @@ Requires-Dist: websockets<15.0dev,>=13.0
 
 -----
 
-Google Gen AI Python SDK provides an interface for developers to integrate Google's generative models into their Python applications. It supports the [Gemini Developer API](https://ai.google.dev/gemini-api/docs) and [Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/overview) APIs. This is an early release. API is subject to change. Please do not use this SDK in production environments at this stage.
+Google Gen AI Python SDK provides an interface for developers to integrate Google's generative models into their Python applications. It supports the [Gemini Developer API](https://ai.google.dev/gemini-api/docs) and [Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/overview) APIs.
 
 ## Installation
 
@@ -56,16 +56,33 @@ different services ([Gemini Developer API](https://ai.google.dev/gemini-api/docs
 
 ```python
 # Only run this block for Gemini Developer API
-client = genai.Client(api_key="GEMINI_API_KEY")
+client = genai.Client(api_key='GEMINI_API_KEY')
 ```
 
 ```python
 # Only run this block for Vertex AI API
 client = genai.Client(
-    vertexai=True, project="your-project-id", location="us-central1"
+    vertexai=True, project='your-project-id', location='us-central1'
 )
 ```
 
+To set the API version use `http_options`. For example, to set the API version
+to `v1` for Vertex AI:
+
+```python
+client = genai.Client(
+    vertexai=True, project='your-project-id', location='us-central1',
+    http_options={'api_version': 'v1'}
+)
+```
+
+To set the API version to `v1alpha` for the Gemini API:
+
+```python
+client = genai.Client(api_key='GEMINI_API_KEY',
+                      http_options={'api_version': 'v1alpha'})
+```
+
 ## Types
 
 Parameter types can be specified as either dictionaries(`TypedDict`) or
@@ -82,7 +99,7 @@ The `client.models` modules exposes model inferencing and model getters.
 
 ```python
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp", contents="why is the sky blue?"
+    model='gemini-2.0-flash-001', contents='why is the sky blue?'
 )
 print(response.text)
 ```
@@ -97,22 +114,53 @@ download the file in console.
 python code.
 
 ```python
-file = client.files.upload(path="a11.txt")
+file = client.files.upload(path='a11.txt')
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents=["Could you summarize this file?", file]
+    model='gemini-2.0-flash-001',
+    contents=['Could you summarize this file?', file]
 )
 print(response.text)
 ```
 
+#### How to structure `contents`
+There are several ways to structure the `contents` in your request.
+
+Provide a single string as shown in the text example above:
+
+```python
+contents='Can you recommend some things to do in Boston and New York in the winter?'
+```
+
+Provide a single `Content` instance with multiple `Part` instances:
+
+```python
+contents=types.Content(parts=[
+    types.Part.from_text(text='Can you recommend some things to do in Boston in the winter?'),
+    types.Part.from_text(text='Can you recommend some things to do in New York in the winter?')
+], role='user')
+```
+
+When sending more than one input type, provide a list with multiple `Content`
+instances:
+
+```python
+contents=[
+    'What is this a picture of?',
+    types.Part.from_uri(
+        file_uri='gs://generativeai-downloads/images/scones.jpg',
+        mime_type='image/jpeg',
+    ),
+],
+```
+
 ### System Instructions and Other Configs
 
 ```python
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="high",
+    model='gemini-2.0-flash-001',
+    contents='high',
     config=types.GenerateContentConfig(
-        system_instruction="I say high, you say low",
+        system_instruction='I say high, you say low',
         temperature=0.3,
     ),
 )
@@ -126,8 +174,8 @@ 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(text="Why is the sky blue?"),
+    model='gemini-2.0-flash-001',
+    contents=types.Part.from_text(text='Why is the sky blue?'),
     config=types.GenerateContentConfig(
         temperature=0,
         top_p=0.95,
@@ -135,7 +183,7 @@ response = client.models.generate_content(
         candidate_count=1,
         seed=5,
         max_output_tokens=100,
-        stop_sequences=["STOP!"],
+        stop_sequences=['STOP!'],
         presence_penalty=0.0,
         frequency_penalty=0.0,
     ),
@@ -154,7 +202,7 @@ for model in client.models.list():
 ```
 
 ```python
-pager = client.models.list(config={"page_size": 10})
+pager = client.models.list(config={'page_size': 10})
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -169,7 +217,7 @@ async for job in await client.aio.models.list():
 ```
 
 ```python
-async_pager = await client.aio.models.list(config={"page_size": 10})
+async_pager = await client.aio.models.list(config={'page_size': 10})
 print(async_pager.page_size)
 print(async_pager[0])
 await async_pager.next_page()
@@ -180,13 +228,13 @@ print(async_pager[0])
 
 ```python
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="Say something bad.",
+    model='gemini-2.0-flash-001',
+    contents='Say something bad.',
     config=types.GenerateContentConfig(
         safety_settings=[
             types.SafetySetting(
-                category="HARM_CATEGORY_HATE_SPEECH",
-                threshold="BLOCK_ONLY_HIGH",
+                category='HARM_CATEGORY_HATE_SPEECH',
+                threshold='BLOCK_ONLY_HIGH',
             )
         ]
     ),
@@ -208,12 +256,12 @@ def get_current_weather(location: str) -> str:
     Args:
       location: The city and state, e.g. San Francisco, CA
     """
-    return "sunny"
+    return 'sunny'
 
 
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="What is the weather like in Boston?",
+    model='gemini-2.0-flash-001',
+    contents='What is the weather like in Boston?',
     config=types.GenerateContentConfig(tools=[get_current_weather]),
 )
 
@@ -230,25 +278,25 @@ Then you will receive a function call part in the response.
 
 ```python
 function = types.FunctionDeclaration(
-    name="get_current_weather",
-    description="Get the current weather in a given location",
+    name='get_current_weather',
+    description='Get the current weather in a given location',
     parameters=types.Schema(
-        type="OBJECT",
+        type='OBJECT',
         properties={
-            "location": types.Schema(
-                type="STRING",
-                description="The city and state, e.g. San Francisco, CA",
+            'location': types.Schema(
+                type='STRING',
+                description='The city and state, e.g. San Francisco, CA',
             ),
         },
-        required=["location"],
+        required=['location'],
     ),
 )
 
 tool = types.Tool(function_declarations=[function])
 
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="What is the weather like in Boston?",
+    model='gemini-2.0-flash-001',
+    contents='What is the weather like in Boston?',
     config=types.GenerateContentConfig(tools=[tool]),
 )
 
@@ -262,33 +310,34 @@ 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(text="What is the weather like in Boston?")],
+    role='user',
+    parts=[types.Part.from_text(text='What is the weather like in Boston?')],
 )
 function_call_part = response.function_calls[0]
+function_call_content = response.candidates[0].content
 
 
 try:
     function_result = get_current_weather(
         **function_call_part.function_call.args
     )
-    function_response = {"result": function_result}
+    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 = {'error': str(e)}
 
 
 function_response_part = types.Part.from_function_response(
-    name=function_call_part.function_call.name,
+    name=function_call_part.name,
     response=function_response,
 )
 function_response_content = types.Content(
-    role="tool", parts=[function_response_part]
+    role='tool', parts=[function_response_part]
 )
 
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
+    model='gemini-2.0-flash-001',
     contents=[
         user_prompt_content,
         function_call_content,
@@ -302,6 +351,66 @@ response = client.models.generate_content(
 print(response.text)
 ```
 
+#### Function calling with `ANY` tools config mode
+
+If you configure function calling mode to be `ANY`, then the model will always
+return function call parts. If you also pass a python function as a tool, by
+default the SDK will perform automatic function calling until the remote calls exceed the
+maximum remote call for automatic function calling (default to 10 times).
+
+If you'd like to disable automatic function calling in `ANY` mode:
+
+```python
+def get_current_weather(location: str) -> str:
+    """Returns the current weather.
+
+    Args:
+      location: The city and state, e.g. San Francisco, CA
+    """
+    return "sunny"
+
+response = client.models.generate_content(
+    model="gemini-2.0-flash-001",
+    contents="What is the weather like in Boston?",
+    config=types.GenerateContentConfig(
+        tools=[get_current_weather],
+        automatic_function_calling=types.AutomaticFunctionCallingConfig(
+            disable=True
+        ),
+        tool_config=types.ToolConfig(
+            function_calling_config=types.FunctionCallingConfig(mode='ANY')
+        ),
+    ),
+)
+```
+
+If you'd like to set `x` number of automatic function call turns, you can
+configure the maximum remote calls to be `x + 1`.
+Assuming you prefer `1` turn for automatic function calling.
+
+```python
+def get_current_weather(location: str) -> str:
+    """Returns the current weather.
+
+    Args:
+      location: The city and state, e.g. San Francisco, CA
+    """
+    return "sunny"
+
+response = client.models.generate_content(
+    model="gemini-2.0-flash-001",
+    contents="What is the weather like in Boston?",
+    config=types.GenerateContentConfig(
+        tools=[get_current_weather],
+        automatic_function_calling=types.AutomaticFunctionCallingConfig(
+            maximum_remote_calls=2
+        ),
+        tool_config=types.ToolConfig(
+            function_calling_config=types.FunctionCallingConfig(mode='ANY')
+        ),
+    ),
+)
+```
 ### JSON Response Schema
 
 #### Pydantic Model Schema support
@@ -323,10 +432,10 @@ class CountryInfo(BaseModel):
 
 
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="Give me information for the United States.",
+    model='gemini-2.0-flash-001',
+    contents='Give me information for the United States.',
     config=types.GenerateContentConfig(
-        response_mime_type="application/json",
+        response_mime_type='application/json',
         response_schema=CountryInfo,
     ),
 )
@@ -335,30 +444,30 @@ print(response.text)
 
 ```python
 response = client.models.generate_content(
-    model="gemini-2.0-flash-exp",
-    contents="Give me information for the United States.",
+    model='gemini-2.0-flash-001',
+    contents='Give me information for the United States.',
     config=types.GenerateContentConfig(
-        response_mime_type="application/json",
+        response_mime_type='application/json',
         response_schema={
-            "required": [
-                "name",
-                "population",
-                "capital",
-                "continent",
-                "gdp",
-                "official_language",
-                "total_area_sq_mi",
+            'required': [
+                'name',
+                'population',
+                'capital',
+                'continent',
+                'gdp',
+                'official_language',
+                'total_area_sq_mi',
             ],
-            "properties": {
-                "name": {"type": "STRING"},
-                "population": {"type": "INTEGER"},
-                "capital": {"type": "STRING"},
-                "continent": {"type": "STRING"},
-                "gdp": {"type": "INTEGER"},
-                "official_language": {"type": "STRING"},
-                "total_area_sq_mi": {"type": "INTEGER"},
+            'properties': {
+                'name': {'type': 'STRING'},
+                'population': {'type': 'INTEGER'},
+                'capital': {'type': 'STRING'},
+                'continent': {'type': 'STRING'},
+                'gdp': {'type': 'INTEGER'},
+                'official_language': {'type': 'STRING'},
+                'total_area_sq_mi': {'type': 'INTEGER'},
             },
-            "type": "OBJECT",
+            'type': 'OBJECT',
         },
     ),
 )
@@ -381,7 +490,7 @@ class InstrumentEnum(Enum):
   KEYBOARD = 'Keyboard'
 
 response = client.models.generate_content(
-      model='gemini-2.0-flash-exp',
+      model='gemini-2.0-flash-001',
       contents='What instrument plays multiple notes at once?',
       config={
           'response_mime_type': 'text/x.enum',
@@ -406,7 +515,7 @@ class InstrumentEnum(Enum):
   KEYBOARD = 'Keyboard'
 
 response = client.models.generate_content(
-      model='gemini-2.0-flash-exp',
+      model='gemini-2.0-flash-001',
       contents='What instrument plays multiple notes at once?',
       config={
           'response_mime_type': 'application/json',
@@ -422,9 +531,9 @@ print(response.text)
 
 ```python
 for chunk in client.models.generate_content_stream(
-    model="gemini-2.0-flash-exp", contents="Tell me a story in 300 words."
+    model='gemini-2.0-flash-001', contents='Tell me a story in 300 words.'
 ):
-    print(chunk.text, end="")
+    print(chunk.text, end='')
 ```
 
 #### Streaming for image content
@@ -434,35 +543,35 @@ you can use the `from_uri` class method to create a `Part` object.
 
 ```python
 for chunk in client.models.generate_content_stream(
-    model="gemini-2.0-flash-exp",
+    model='gemini-2.0-flash-001',
     contents=[
-        "What is this image about?",
+        'What is this image about?',
         types.Part.from_uri(
-            file_uri="gs://generativeai-downloads/images/scones.jpg",
-            mime_type="image/jpeg",
+            file_uri='gs://generativeai-downloads/images/scones.jpg',
+            mime_type='image/jpeg',
         ),
     ],
 ):
-    print(chunk.text, end="")
+    print(chunk.text, end='')
 ```
 
 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:
+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-2.0-flash-exp",
+    model='gemini-2.0-flash-001',
     contents=[
-        "What is this image about?",
+        'What is this image about?',
         types.Part.from_bytes(data=image_bytes, mime_type=YOUR_IMAGE_MIME_TYPE),
     ],
 ):
-    print(chunk.text, end="")
+    print(chunk.text, end='')
 ```
 
 ### Async
@@ -475,7 +584,7 @@ of `client.models.generate_content`
 
 ```python
 response = await client.aio.models.generate_content(
-    model="gemini-2.0-flash-exp", contents="Tell me a story in 300 words."
+    model='gemini-2.0-flash-001', contents='Tell me a story in 300 words.'
 )
 
 print(response.text)
@@ -485,17 +594,17 @@ print(response.text)
 
 ```python
 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."
+    model='gemini-2.0-flash-001', contents='Tell me a story in 300 words.'
 ):
-    print(chunk.text, end="")
+    print(chunk.text, end='')
 ```
 
 ### Count Tokens and Compute Tokens
 
 ```python
 response = client.models.count_tokens(
-    model="gemini-2.0-flash-exp",
-    contents="why is the sky blue?",
+    model='gemini-2.0-flash-001',
+    contents='why is the sky blue?',
 )
 print(response)
 ```
@@ -506,8 +615,8 @@ Compute tokens is only supported in Vertex AI.
 
 ```python
 response = client.models.compute_tokens(
-    model="gemini-2.0-flash-exp",
-    contents="why is the sky blue?",
+    model='gemini-2.0-flash-001',
+    contents='why is the sky blue?',
 )
 print(response)
 ```
@@ -516,8 +625,8 @@ print(response)
 
 ```python
 response = await client.aio.models.count_tokens(
-    model="gemini-2.0-flash-exp",
-    contents="why is the sky blue?",
+    model='gemini-2.0-flash-001',
+    contents='why is the sky blue?',
 )
 print(response)
 ```
@@ -526,8 +635,8 @@ print(response)
 
 ```python
 response = client.models.embed_content(
-    model="text-embedding-004",
-    contents="why is the sky blue?",
+    model='text-embedding-004',
+    contents='why is the sky blue?',
 )
 print(response)
 ```
@@ -535,8 +644,8 @@ print(response)
 ```python
 # multiple contents with config
 response = client.models.embed_content(
-    model="text-embedding-004",
-    contents=["why is the sky blue?", "What is your age?"],
+    model='text-embedding-004',
+    contents=['why is the sky blue?', 'What is your age?'],
     config=types.EmbedContentConfig(output_dimensionality=10),
 )
 
@@ -552,13 +661,13 @@ Support for generate images in Gemini Developer API is behind an allowlist
 ```python
 # Generate Image
 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",
+    model='imagen-3.0-generate-002',
+    prompt='An umbrella in the foreground, and a rainy night sky in the background',
     config=types.GenerateImagesConfig(
-        negative_prompt="human",
+        negative_prompt='human',
         number_of_images=1,
         include_rai_reason=True,
-        output_mime_type="image/jpeg",
+        output_mime_type='image/jpeg',
     ),
 )
 response1.generated_images[0].image.show()
@@ -571,12 +680,12 @@ Upscale image is only supported in Vertex AI.
 ```python
 # Upscale the generated image from above
 response2 = client.models.upscale_image(
-    model="imagen-3.0-generate-001",
+    model='imagen-3.0-generate-001',
     image=response1.generated_images[0].image,
-    upscale_factor="x2",
+    upscale_factor='x2',
     config=types.UpscaleImageConfig(
         include_rai_reason=True,
-        output_mime_type="image/jpeg",
+        output_mime_type='image/jpeg',
     ),
 )
 response2.generated_images[0].image.show()
@@ -601,21 +710,21 @@ raw_ref_image = RawReferenceImage(
 mask_ref_image = MaskReferenceImage(
     reference_id=2,
     config=types.MaskReferenceConfig(
-        mask_mode="MASK_MODE_BACKGROUND",
+        mask_mode='MASK_MODE_BACKGROUND',
         mask_dilation=0,
     ),
 )
 
 response3 = client.models.edit_image(
-    model="imagen-3.0-capability-001",
-    prompt="Sunlight and clear sky",
+    model='imagen-3.0-capability-001',
+    prompt='Sunlight and clear sky',
     reference_images=[raw_ref_image, mask_ref_image],
     config=types.EditImageConfig(
-        edit_mode="EDIT_MODE_INPAINT_INSERTION",
+        edit_mode='EDIT_MODE_INPAINT_INSERTION',
         number_of_images=1,
-        negative_prompt="human",
+        negative_prompt='human',
         include_rai_reason=True,
-        output_mime_type="image/jpeg",
+        output_mime_type='image/jpeg',
     ),
 )
 response3.generated_images[0].image.show()
@@ -628,32 +737,32 @@ Create a chat session to start a multi-turn conversations with the model.
 ### Send Message
 
 ```python
-chat = client.chats.create(model="gemini-2.0-flash-exp")
-response = chat.send_message("tell me a story")
+chat = client.chats.create(model='gemini-2.0-flash-001')
+response = chat.send_message('tell me a story')
 print(response.text)
 ```
 
 ### Streaming
 
 ```python
-chat = client.chats.create(model="gemini-2.0-flash-exp")
-for chunk in chat.send_message_stream("tell me a story"):
+chat = client.chats.create(model='gemini-2.0-flash-001')
+for chunk in chat.send_message_stream('tell me a story'):
     print(chunk.text)
 ```
 
 ### Async
 
 ```python
-chat = client.aio.chats.create(model="gemini-2.0-flash-exp")
-response = await chat.send_message("tell me a story")
+chat = client.aio.chats.create(model='gemini-2.0-flash-001')
+response = await chat.send_message('tell me a story')
 print(response.text)
 ```
 
 ### Async Streaming
 
 ```python
-chat = client.aio.chats.create(model="gemini-2.0-flash-exp")
-async for chunk in await chat.send_message_stream("tell me a story"):
+chat = client.aio.chats.create(model='gemini-2.0-flash-001')
+async for chunk in await chat.send_message_stream('tell me a story'):
     print(chunk.text)
 ```
 
@@ -669,8 +778,8 @@ Files are only supported in Gemini Developer API.
 ### Upload
 
 ```python
-file1 = client.files.upload(path="2312.11805v3.pdf")
-file2 = client.files.upload(path="2403.05530.pdf")
+file1 = client.files.upload(path='2312.11805v3.pdf')
+file2 = client.files.upload(path='2403.05530.pdf')
 
 print(file1)
 print(file2)
@@ -679,7 +788,7 @@ print(file2)
 ### Delete
 
 ```python
-file3 = client.files.upload(path="2312.11805v3.pdf")
+file3 = client.files.upload(path='2312.11805v3.pdf')
 
 client.files.delete(name=file3.name)
 ```
@@ -693,32 +802,32 @@ client.files.delete(name=file3.name)
 ```python
 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",
+        'gs://cloud-samples-data/generative-ai/pdf/2312.11805v3.pdf',
+        'gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf',
     ]
 else:
     file_uris = [file1.uri, file2.uri]
 
 cached_content = client.caches.create(
-    model="gemini-1.5-pro-002",
+    model='gemini-1.5-pro-002',
     config=types.CreateCachedContentConfig(
         contents=[
             types.Content(
-                role="user",
+                role='user',
                 parts=[
                     types.Part.from_uri(
-                        file_uri=file_uris[0], mime_type="application/pdf"
+                        file_uri=file_uris[0], mime_type='application/pdf'
                     ),
                     types.Part.from_uri(
                         file_uri=file_uris[1],
-                        mime_type="application/pdf",
+                        mime_type='application/pdf',
                     ),
                 ],
             )
         ],
-        system_instruction="What is the sum of the two pdfs?",
-        display_name="test cache",
-        ttl="3600s",
+        system_instruction='What is the sum of the two pdfs?',
+        display_name='test cache',
+        ttl='3600s',
     ),
 )
 ```
@@ -733,8 +842,8 @@ cached_content = client.caches.get(name=cached_content.name)
 
 ```python
 response = client.models.generate_content(
-    model="gemini-1.5-pro-002",
-    contents="Summarize the pdfs",
+    model='gemini-1.5-pro-002',
+    contents='Summarize the pdfs',
     config=types.GenerateContentConfig(
         cached_content=cached_content.name,
     ),
@@ -754,17 +863,17 @@ tuning through `tune`.
 
 ```python
 if client.vertexai:
-    model = "gemini-1.5-pro-002"
+    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",
+        gcs_uri='gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl',
     )
 else:
-    model = "models/gemini-1.0-pro-001"
+    model = 'models/gemini-1.0-pro-001'
     training_dataset = types.TuningDataset(
         examples=[
             types.TuningExample(
-                text_input=f"Input text {i}",
-                output=f"Output text {i}",
+                text_input=f'Input text {i}',
+                output=f'Output text {i}',
             )
             for i in range(5)
         ],
@@ -776,7 +885,7 @@ tuning_job = client.tunings.tune(
     base_model=model,
     training_dataset=training_dataset,
     config=types.CreateTuningJobConfig(
-        epoch_count=1, tuned_model_display_name="test_dataset_examples model"
+        epoch_count=1, tuned_model_display_name='test_dataset_examples model'
     ),
 )
 print(tuning_job)
@@ -794,8 +903,8 @@ import time
 
 running_states = set(
     [
-        "JOB_STATE_PENDING",
-        "JOB_STATE_RUNNING",
+        'JOB_STATE_PENDING',
+        'JOB_STATE_RUNNING',
     ]
 )
 
@@ -810,7 +919,7 @@ while tuning_job.state in running_states:
 ```python
 response = client.models.generate_content(
     model=tuning_job.tuned_model.endpoint,
-    contents="why is the sky blue?",
+    contents='why is the sky blue?',
 )
 
 print(response.text)
@@ -828,12 +937,12 @@ print(tuned_model)
 To retrieve base models, see [list base models](#list-base-models).
 
 ```python
-for model in client.models.list(config={"page_size": 10, "query_base": False}):
+for model in client.models.list(config={'page_size': 10, 'query_base': False}):
     print(model)
 ```
 
 ```python
-pager = client.models.list(config={"page_size": 10, "query_base": False})
+pager = client.models.list(config={'page_size': 10, 'query_base': False})
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -843,12 +952,12 @@ print(pager[0])
 #### Async
 
 ```python
-async for job in await client.aio.models.list(config={"page_size": 10, "query_base": False}):
+async for job in await client.aio.models.list(config={'page_size': 10, 'query_base': False}):
     print(job)
 ```
 
 ```python
-async_pager = await client.aio.models.list(config={"page_size": 10, "query_base": False})
+async_pager = await client.aio.models.list(config={'page_size': 10, 'query_base': False})
 print(async_pager.page_size)
 print(async_pager[0])
 await async_pager.next_page()
@@ -863,7 +972,7 @@ model = pager[0]
 model = client.models.update(
     model=model.name,
     config=types.UpdateModelConfig(
-        display_name="my tuned model", description="my tuned model description"
+        display_name='my tuned model', description='my tuned model description'
     ),
 )
 
@@ -874,12 +983,12 @@ print(model)
 ### List Tuning Jobs
 
 ```python
-for job in client.tunings.list(config={"page_size": 10}):
+for job in client.tunings.list(config={'page_size': 10}):
     print(job)
 ```
 
 ```python
-pager = client.tunings.list(config={"page_size": 10})
+pager = client.tunings.list(config={'page_size': 10})
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -889,12 +998,12 @@ print(pager[0])
 #### Async
 
 ```python
-async for job in await client.aio.tunings.list(config={"page_size": 10}):
+async for job in await client.aio.tunings.list(config={'page_size': 10}):
     print(job)
 ```
 
 ```python
-async_pager = await client.aio.tunings.list(config={"page_size": 10})
+async_pager = await client.aio.tunings.list(config={'page_size': 10})
 print(async_pager.page_size)
 print(async_pager[0])
 await async_pager.next_page()
@@ -910,8 +1019,8 @@ Only supported in Vertex AI.
 ```python
 # 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://my-project.my-dataset.my-table",
+    model='gemini-1.5-flash-002',
+    src='bq://my-project.my-dataset.my-table',
 )
 
 job
@@ -927,10 +1036,10 @@ job.state
 ```python
 completed_states = set(
     [
-        "JOB_STATE_SUCCEEDED",
-        "JOB_STATE_FAILED",
-        "JOB_STATE_CANCELLED",
-        "JOB_STATE_PAUSED",
+        'JOB_STATE_SUCCEEDED',
+        'JOB_STATE_FAILED',
+        'JOB_STATE_CANCELLED',
+        'JOB_STATE_PAUSED',
     ]
 )