fix: functions must use AsyncGenerator type or no type

Author: AlmogBaku

.github/workflows/publish.yaml

@@ -4,7 +4,6 @@ on:
   push:
     paths:
       - '**.py'
-      - '!**/tests/**'
   workflow_dispatch: ~
 
 permissions:

tests/example.py

@@ -3,7 +3,7 @@
 import openai
 import asyncio
 from openai_streaming import process_response
-from typing import Generator, AsyncGenerator
+from typing import AsyncGenerator
 from openai_streaming import openai_streaming_function
 
 # Initialize API key

tests/test_with_functions.py

@@ -5,10 +5,10 @@
 import openai
 from unittest.mock import patch
 from openai_streaming import process_response, openai_streaming_function
-from typing import Generator, AsyncGenerator
+from typing import AsyncGenerator
 
 
-async def content_handler(content: Generator[str, None, None]):
+async def content_handler(content: AsyncGenerator[str, None]):
     async for token in content:
         print(token, end="")