fix: treat instances with __aiter__ as async iterators

promplate · May 15, 2024 · ff8bfaa · ff8bfaa
1 parent e057123
commit ff8bfaa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/promplate/chain/utils.py b/python/promplate/chain/utils.py
@@ -1,4 +1,4 @@
-from inspect import Parameter, isasyncgen, isawaitable, signature
+from inspect import Parameter, isawaitable, signature
 from itertools import accumulate
 from typing import AsyncIterable, Awaitable, Callable, Iterable, TypeVar, cast
 
@@ -37,8 +37,8 @@ async def async_accumulate(async_iterable: AsyncIterable[str]):
 
 
 def accumulate_any(any_iterable: Iterable[str] | AsyncIterable[str]):
-    if isasyncgen(any_iterable):
-        return async_accumulate(any_iterable)
+    if "__aiter__" in dir(any_iterable):
+        return async_accumulate(any_iterable)  # type: ignore
 
     async def _():
         for i in accumulate(cast(Iterable[str], any_iterable)):

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promplate"
-version = "0.3.4.4"
+version = "0.3.4.5"
 description = "Prompt engineering framework for humans"
 homepage = "https://promplate.dev/"
 documentation = "https://docs.py.promplate.dev/"