aiter, anext not exist error (#127)

* aiter, anext not exist error

* Un-skip test, format with black.

* Fix aiter & anext implementations.

---------

Co-authored-by: Mark Daoust <[email protected]>

Author: Andy963

google/generativeai/types/generation_types.py

@@ -27,6 +27,20 @@
     "GenerateContentResponse",
 ]
 
+if sys.version_info < (3, 10):
+
+    def aiter(obj):
+        return obj.__aiter__()
+
+    async def anext(obj, default=None):
+        try:
+            return await obj.__anext__()
+        except StopAsyncIteration:
+            if default is not None:
+                return default
+            else:
+                raise
+
 
 class BlockedPromptException(Exception):
     pass

tests/test_generative_models_async.py

@@ -83,10 +83,6 @@ async def test_basic(self):
 
         self.assertEqual(response.text, "world!")
 
-    @unittest.skipIf(
-        sys.version_info.major == 3 and sys.version_info.minor < 10,
-        "streaming async requires python 3.10+",
-    )
     async def test_streaming(self):
         # Generate text from text prompt
         model = generative_models.GenerativeModel(model_name="gemini-m")