Format updated generation_types and test using black.

Unity-google-new · Nov 26, 2024 · 7bf8083 · 7bf8083
1 parent acae3a2
commit 7bf8083
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/google/generativeai/types/generation_types.py b/google/generativeai/types/generation_types.py
@@ -544,7 +544,7 @@ def prompt_feedback(self):
     @property
     def usage_metadata(self):
         return self._result.usage_metadata
-    
+
     @property
     def model_version(self):
         return self._result.model_version

diff --git a/tests/test_generation.py b/tests/test_generation.py
@@ -493,7 +493,7 @@ def test_join_chunks(self):
             prompt_token_count=5
         )
 
-        chunks[-1].model_version = 'gemini-1.5-flash-002'
+        chunks[-1].model_version = "gemini-1.5-flash-002"
 
         result = generation_types._join_chunks(chunks)
 
@@ -511,7 +511,7 @@ def test_join_chunks(self):
                     ],
                 },
                 "usage_metadata": {"prompt_token_count": 5},
-                "model_version": 'gemini-1.5-flash-002'
+                "model_version": "gemini-1.5-flash-002",
             },
         )