fix types

acheong08 · Feb 15, 2023 · 8f31f82 · 8f31f82
1 parent 359dd99
commit 8f31f82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/EdgeGPT.py b/src/EdgeGPT.py
@@ -152,7 +152,7 @@ def __init__(self, conversation: Conversation) -> None:
             conversation_id=conversation.struct["conversationId"],
         )
 
-    async def ask_stream(self, prompt: str) -> Generator[str]:
+    async def ask_stream(self, prompt: str) -> Generator[str, None, None]:
         """
         Ask a question to the bot
         """
@@ -220,7 +220,7 @@ async def ask(self, prompt: str) -> dict:
             if final:
                 return response
 
-    async def ask_stream(self, prompt: str) -> Generator[str]:
+    async def ask_stream(self, prompt: str) -> Generator[str, None, None]:
         """
         Ask a question to the bot
         """