Merge pull request #13 from pcoet/cleanup

Cleanup

Author: markmcd

python/README.md

@@ -0,0 +1,12 @@
+# Python examples
+
+This directory contains examples of working with the Gemini API using the
+[Google Gen AI SDK for Python](https://googleapis.github.io/python-genai/).
+
+## Install dependencies
+
+    pip install absl-py google-genai Pillow
+
+## Run a test file
+
+    python <filename>.py

python/files.py

@@ -17,11 +17,11 @@
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
 
-
 class UnitTests(absltest.TestCase):
     def test_files_create_text(self):
         # [START files_create_text]
         from google import genai
+
         client = genai.Client()
         myfile = client.files.upload(file=media / "poem.txt")
         print(f"{myfile=}")
@@ -36,6 +36,7 @@ def test_files_create_text(self):
     def test_files_create_image(self):
         # [START files_create_image]
         from google import genai
+
         client = genai.Client()
         myfile = client.files.upload(file=media / "Cajun_instruments.jpg")
         print(f"{myfile=}")
@@ -50,6 +51,7 @@ def test_files_create_image(self):
     def test_files_create_audio(self):
         # [START files_create_audio]
         from google import genai
+
         client = genai.Client()
         myfile = client.files.upload(file=media / "sample.mp3")
         print(f"{myfile=}")
@@ -87,6 +89,7 @@ def test_files_create_video(self):
     def test_files_create_pdf(self):
         # [START files_create_pdf]
         from google import genai
+
         client = genai.Client()
         sample_pdf = client.files.upload(file=media / "test.pdf")
         response = client.models.generate_content(
@@ -118,6 +121,7 @@ def test_files_create_from_IO(self):
     def test_files_list(self):
         # [START files_list]
         from google import genai
+
         client = genai.Client()
         print("My files:")
         for f in client.files.list():
@@ -127,6 +131,7 @@ def test_files_list(self):
     def test_files_get(self):
         # [START files_get]
         from google import genai
+
         client = genai.Client()
         myfile = client.files.upload(file=media / "poem.txt")
         file_name = myfile.name
@@ -139,7 +144,7 @@ def test_files_get(self):
     def test_files_delete(self):
         # [START files_delete]
         from google import genai
-        from google.api_core import exceptions
+
         client = genai.Client()
         myfile = client.files.upload(file=media / "poem.txt")
 
@@ -150,10 +155,10 @@ def test_files_delete(self):
                 model="gemini-2.0-flash",
                 contents=[myfile, "Describe this file."]
             )
+            print(result)
         except genai.errors.ClientError:
             pass
         # [END files_delete]
 
-
 if __name__ == "__main__":
     absltest.main()

python/models.py

@@ -18,6 +18,7 @@ class UnitTests(absltest.TestCase):
     def test_models_list(self):
         # [START models_list]
         from google import genai
+
         client = genai.Client()
 
         print("List of models that support generateContent:\n")
@@ -36,6 +37,7 @@ def test_models_list(self):
     def test_models_get(self):
         # [START models_get]
         from google import genai
+        
         client = genai.Client()
         model_info = client.models.get(model="gemini-2.0-flash")
         print(model_info)

python/safety_settings.py

@@ -19,6 +19,7 @@ def test_safety_settings(self):
         # [START safety_settings]
         from google import genai
         from google.genai import types
+
         client = genai.Client()
         unsafe_prompt = (
             "I support Martians Soccer Club and I think Jupiterians Football Club sucks! "
@@ -45,6 +46,7 @@ def test_safety_settings_multi(self):
         # [START safety_settings_multi]
         from google import genai
         from google.genai import types
+        
         client = genai.Client()
         unsafe_prompt = (
             "I support Martians Soccer Club and I think Jupiterians Football Club sucks! "

python/text_generation.py

@@ -12,13 +12,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from absl.testing import absltest
 
 import pathlib
+from absl.testing import absltest
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
 
-
 class UnitTests(absltest.TestCase):
     def test_text_gen_text_only_prompt(self):
         # [START text_gen_text_only_prompt]
@@ -49,7 +48,6 @@ def test_text_gen_text_only_prompt_streaming(self):
     def test_text_gen_multimodal_one_image_prompt(self):
         # [START text_gen_multimodal_one_image_prompt]
         from google import genai
-
         import PIL.Image
 
         client = genai.Client()
@@ -64,7 +62,6 @@ def test_text_gen_multimodal_one_image_prompt(self):
     def test_text_gen_multimodal_one_image_prompt_streaming(self):
         # [START text_gen_multimodal_one_image_prompt_streaming]
         from google import genai
-
         import PIL.Image
 
         client = genai.Client()
@@ -81,7 +78,6 @@ def test_text_gen_multimodal_one_image_prompt_streaming(self):
     def test_text_gen_multimodal_multi_image_prompt(self):
         # [START text_gen_multimodal_multi_image_prompt]
         from google import genai
-
         import PIL.Image
 
         client = genai.Client()
@@ -97,7 +93,6 @@ def test_text_gen_multimodal_multi_image_prompt(self):
     def test_text_gen_multimodal_multi_image_prompt_streaming(self):
         # [START text_gen_multimodal_multi_image_prompt_streaming]
         from google import genai
-
         import PIL.Image
 
         client = genai.Client()
@@ -143,7 +138,6 @@ def test_text_gen_multimodal_audio_streaming(self):
     def test_text_gen_multimodal_video_prompt(self):
         # [START text_gen_multimodal_video_prompt]
         from google import genai
-
         import time
 
         client = genai.Client()
@@ -167,7 +161,6 @@ def test_text_gen_multimodal_video_prompt(self):
     def test_text_gen_multimodal_video_prompt_streaming(self):
         # [START text_gen_multimodal_video_prompt_streaming]
         from google import genai
-
         import time
 
         client = genai.Client()
@@ -219,6 +212,5 @@ def test_text_gen_multimodal_pdf_streaming(self):
             print("_" * 80)
         # [END text_gen_multimodal_pdf_streaming]
 
-
 if __name__ == "__main__":
     absltest.main()