Merge pull request #14 from markmcd/pyink

Run `pyink` over the Python sample code.

Author: pcoet

python/cache.py

@@ -19,6 +19,7 @@
 
 
 class UnitTests(absltest.TestCase):
+
     def test_cache_create(self):
         # [START cache_create]
         from google import genai
@@ -32,17 +33,15 @@ def test_cache_create(self):
             model=model_name,
             config=types.CreateCachedContentConfig(
                 contents=[document],
-                system_instruction="You are an expert analyzing transcripts."
-            )
+                system_instruction="You are an expert analyzing transcripts.",
+            ),
         )
         print(cache)
 
         response = client.models.generate_content(
             model=model_name,
             contents="Please summarize this transcript",
-            config=types.GenerateContentConfig(
-                cached_content=cache.name
-            )
+            config=types.GenerateContentConfig(cached_content=cache.name),
         )
         print(response.text)
         # [END cache_create]
@@ -61,8 +60,8 @@ def test_cache_create_from_name(self):
             model=model_name,
             config=types.CreateCachedContentConfig(
                 contents=[document],
-                system_instruction="You are an expert analyzing transcripts."
-            )
+                system_instruction="You are an expert analyzing transcripts.",
+            ),
         )
         cache_name = cache.name  # Save the name for later
 
@@ -71,9 +70,7 @@ def test_cache_create_from_name(self):
         response = client.models.generate_content(
             model=model_name,
             contents="Find a lighthearted moment from this transcript",
-            config=types.GenerateContentConfig(
-                cached_content=cache.name
-            )
+            config=types.GenerateContentConfig(cached_content=cache.name),
         )
         print(response.text)
         # [END cache_create_from_name]
@@ -91,9 +88,7 @@ def test_cache_create_from_chat(self):
         # Create a chat session with the given system instruction.
         chat = client.chats.create(
             model=model_name,
-            config=types.GenerateContentConfig(
-                system_instruction=system_instruction
-            )
+            config=types.GenerateContentConfig(system_instruction=system_instruction),
         )
         document = client.files.upload(file=media / "a11.txt")
 
@@ -110,16 +105,14 @@ def test_cache_create_from_chat(self):
         cache = client.caches.create(
             model=model_name,
             config={
-                'contents': chat.get_history(),
-                'system_instruction': system_instruction
-            }
+                "contents": chat.get_history(),
+                "system_instruction": system_instruction,
+            },
         )
         # Continue the conversation using the cached content.
         chat = client.chats.create(
             model=model_name,
-            config=types.GenerateContentConfig(
-                cached_content=cache.name
-            )
+            config=types.GenerateContentConfig(cached_content=cache.name),
         )
         response = chat.send_message(
             message="I didn't understand that last part, could you explain it in simpler language?"
@@ -139,9 +132,9 @@ def test_cache_delete(self):
         cache = client.caches.create(
             model=model_name,
             config={
-                'contents': [document],
-                'system_instruction': "You are an expert analyzing transcripts.",
-            }
+                "contents": [document],
+                "system_instruction": "You are an expert analyzing transcripts.",
+            },
         )
         client.caches.delete(name=cache.name)
         # [END cache_delete]
@@ -157,9 +150,9 @@ def test_cache_get(self):
         cache = client.caches.create(
             model=model_name,
             config={
-                'contents': [document],
-                'system_instruction': "You are an expert analyzing transcripts.",
-            }
+                "contents": [document],
+                "system_instruction": "You are an expert analyzing transcripts.",
+            },
         )
         print(client.caches.get(name=cache.name))
         # [END cache_get]
@@ -176,9 +169,9 @@ def test_cache_list(self):
         cache = client.caches.create(
             model=model_name,
             config={
-                'contents': [document],
-                'system_instruction': "You are an expert analyzing transcripts.",
-            }
+                "contents": [document],
+                "system_instruction": "You are an expert analyzing transcripts.",
+            },
         )
         print("My caches:")
         for c in client.caches.list():
@@ -199,31 +192,31 @@ def test_cache_update(self):
         cache = client.caches.create(
             model=model_name,
             config={
-                'contents': [document],
-                'system_instruction': "You are an expert analyzing transcripts.",
-            }
+                "contents": [document],
+                "system_instruction": "You are an expert analyzing transcripts.",
+            },
         )
 
         # Update the cache's time-to-live (ttl)
         ttl = f"{int(datetime.timedelta(hours=2).total_seconds())}s"
         client.caches.update(
-            name=cache.name,
-            config=types.UpdateCachedContentConfig(
-                ttl=ttl
-            )
+            name=cache.name, config=types.UpdateCachedContentConfig(ttl=ttl)
         )
         print(f"After update:\n {cache}")
 
         # Alternatively, update the expire_time directly
         # Update the expire_time directly in valid RFC 3339 format (UTC with a "Z" suffix)
         expire_time = (
-            datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(minutes=15)
-        ).isoformat().replace('+00:00', 'Z')
+            (
+                datetime.datetime.now(datetime.timezone.utc)
+                + datetime.timedelta(minutes=15)
+            )
+            .isoformat()
+            .replace("+00:00", "Z")
+        )
         client.caches.update(
             name=cache.name,
-            config=types.UpdateCachedContentConfig(
-                expire_time=expire_time
-            )
+            config=types.UpdateCachedContentConfig(expire_time=expire_time),
         )
         # [END cache_update]
         client.caches.delete(name=cache.name)

python/chat.py

@@ -19,6 +19,7 @@
 
 
 class UnitTests(absltest.TestCase):
+
     def test_chat(self):
         # [START chat]
         from google import genai
@@ -30,8 +31,15 @@ def test_chat(self):
             model="gemini-2.0-flash",
             history=[
                 types.Content(role="user", parts=[types.Part(text="Hello")]),
-                types.Content(role="model", parts=[types.Part(text="Great to meet you. What would you like to know?")])
-            ]
+                types.Content(
+                    role="model",
+                    parts=[
+                        types.Part(
+                            text="Great to meet you. What would you like to know?"
+                        )
+                    ],
+                ),
+            ],
         )
         response = chat.send_message(message="I have 2 dogs in my house.")
         print(response.text)
@@ -49,8 +57,15 @@ def test_chat_streaming(self):
             model="gemini-2.0-flash",
             history=[
                 types.Content(role="user", parts=[types.Part(text="Hello")]),
-                types.Content(role="model", parts=[types.Part(text="Great to meet you. What would you like to know?")])
-            ]
+                types.Content(
+                    role="model",
+                    parts=[
+                        types.Part(
+                            text="Great to meet you. What would you like to know?"
+                        )
+                    ],
+                ),
+            ],
         )
         response = chat.send_message_stream(message="I have 2 dogs in my house.")
         for chunk in response:
@@ -72,15 +87,20 @@ def test_chat_streaming_with_images(self):
         client = genai.Client()
         chat = client.chats.create(model="gemini-2.0-flash")
 
-        response = chat.send_message_stream(message="Hello, I'm interested in learning about musical instruments. Can I show you one?")
+        response = chat.send_message_stream(
+            message="Hello, I'm interested in learning about musical instruments. Can I show you one?"
+        )
         for chunk in response:
             print(chunk.text)
             print("_" * 80)
 
         # Upload image file locally
         image_file = client.files.upload(file=media / "organ.jpg")
         response = chat.send_message_stream(
-            message=["What family of instruments does this instrument belong to?", image_file]
+            message=[
+                "What family of instruments does this instrument belong to?",
+                image_file,
+            ]
         )
         for chunk in response:
             print(chunk.text)

python/code_execution.py

@@ -14,11 +14,14 @@
 # limitations under the License.
 from absl.testing import absltest
 
+
 class UnitTests(absltest.TestCase):
+
     def test_code_execution_basic(self):
         # [START code_execution_basic]
         from google import genai
         from google.genai import types
+
         client = genai.Client()
         response = client.models.generate_content(
             model="gemini-2.0-flash",
@@ -28,7 +31,7 @@ def test_code_execution_basic(self):
             ),
             config=types.GenerateContentConfig(
                 tools=[types.Tool(code_execution=types.ToolCodeExecution())],
-            )
+            ),
         )
         # Each part may contain text, executable code, or an execution result.
         for part in response.candidates[0].content.parts:
@@ -71,7 +74,6 @@ def test_code_execution_basic(self):
         # print(f'{primes=}')
         # print(f'{sum(primes)=}')
 
-
         # primes=[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229]
         # sum(primes)=5117
         # [END code_execution_basic_return]
@@ -80,6 +82,7 @@ def test_code_execution_request_override(self):
         # [START code_execution_request_override]
         from google import genai
         from google.genai import types
+
         client = genai.Client()
         response = client.models.generate_content(
             model="gemini-2.0-flash",
@@ -89,7 +92,7 @@ def test_code_execution_request_override(self):
             ),
             config=types.GenerateContentConfig(
                 tools=[types.Tool(code_execution=types.ToolCodeExecution())],
-            )
+            ),
         )
         print(response.executable_code)
         print(response.code_execution_result)
@@ -125,12 +128,13 @@ def test_code_execution_chat(self):
         # [START code_execution_chat]
         from google import genai
         from google.genai import types
+
         client = genai.Client()
         chat = client.chats.create(
             model="gemini-2.0-flash",
             config=types.GenerateContentConfig(
                 tools=[types.Tool(code_execution=types.ToolCodeExecution())],
-            )
+            ),
         )
         # First, a simple chat message.
         response = chat.send_message(message='Can you print "Hello world!"?')

python/configure_model_parameters.py

@@ -14,7 +14,9 @@
 # limitations under the License.
 from absl.testing import absltest
 
+
 class UnitTests(absltest.TestCase):
+
     def test_configure_model(self):
         # [START configure_model_parameters]
         from google import genai
@@ -28,11 +30,12 @@ def test_configure_model(self):
                 candidate_count=1,
                 stop_sequences=["x"],
                 max_output_tokens=20,
-                temperature=1.0
-            )
+                temperature=1.0,
+            ),
         )
         print(response.text)
         # [END configure_model_parameters]
 
+
 if __name__ == "__main__":
     absltest.main()