Add sample test case for asyncio file uploads

Author: shilpakancharla

google/generativeai/files.py

@@ -28,7 +28,8 @@
 
 from google.generativeai.client import get_default_file_client
 
-__all__ = ["upload_file", "get_file", "list_files", "delete_file"]
+__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

@@ -12,11 +12,14 @@
 # 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.
+import unittest
 from absl.testing import absltest
 
 import google
 import google.generativeai as genai
 import pathlib
+import tempfile
+import asyncio
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
 
@@ -127,5 +130,28 @@ def test_files_delete(self):
         # [END files_delete]
 
 
+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')
+            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)))
+
+
 if __name__ == "__main__":
     absltest.main()