format

Change-Id: Id63b0ab5ce27e632ef57b1156f12907428003c8a

Author: MarkDaoust

google/generativeai/files.py

@@ -28,8 +28,16 @@
 
 from google.generativeai.client import get_default_file_client
 
-__all__ = ["upload_file", "get_file", "list_files", "delete_file",
-           "upload_file_async", "get_file_async", "list_files_async", "delete_file_async"]
+__all__ = [
+    "upload_file",
+    "get_file",
+    "list_files",
+    "delete_file",
+    "upload_file_async",
+    "get_file_async",
+    "list_files_async",
+    "delete_file_async",
+]
 
 mimetypes.add_type("image/webp", ".webp")
 

samples/files.py

@@ -133,24 +133,25 @@ def test_files_delete(self):
 class AsyncTests(absltest.TestCase, unittest.IsolatedAsyncioTestCase):
     async def test_upload_file_async(self):
         import google.generativeai.files as files
+
         tempdir = pathlib.Path(tempfile.mkdtemp())
         results = []
+
         async def create_and_upload_file(n: int):
             fname = tempdir / str(n)
             fname.write_text(str(n))
-            file_obj = await files.upload_file_async(fname, mime_type='text/plain')
+            file_obj = await files.upload_file_async(fname, mime_type="text/plain")
             results.append(file_obj)
 
         tasks = []
         for n in range(5):
             tasks.append(asyncio.create_task(create_and_upload_file(n)))
-        
+
         for task in tasks:
             await task
 
         self.assertLen(results, 5)
-        self.assertEqual(sorted(int(f.display_name) for f in results),
-                          list(range(5)))
+        self.assertEqual(sorted(int(f.display_name) for f in results), list(range(5)))
 
 
 if __name__ == "__main__":