fix: change content fn signature to asyncgen

AlmogBaku · Sep 27, 2023 · 5f05d46 · 5f05d46
1 parent 6c65c28
commit 5f05d46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openai_streaming/stream_processing.py b/openai_streaming/stream_processing.py
@@ -100,7 +100,7 @@ def preprocess(self, key, current_dict):
 
 async def process_response(
         response: Union[Iterator[OpenAIObject], List[OpenAIObject]],
-        content_func: Optional[Callable[[Generator[str, None, None]], Awaitable[None]]] = None,
+        content_func: Optional[Callable[[AsyncGenerator[str, None]], Awaitable[None]]] = None,
         funcs: Optional[List[Callable[[], Awaitable[None]]]] = None,
         self: Optional = None
 ) -> Tuple[Set[str], Dict[str, Any]]: