Fix: TypedDict from typing error #886

Ref: python/typeshed#3500
unique0311 · Sep 11, 2023 · b42b6fe · b42b6fe
1 parent 7c58dc0
commit b42b6fe
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/g4f/typing.py b/g4f/typing.py
@@ -1,4 +1,10 @@
-from typing import Any, AsyncGenerator, Generator, NewType, Tuple, TypedDict, Union
+import sys
+from typing import Any, AsyncGenerator, Generator, NewType, Tuple, Union
+
+if sys.version_info >= (3, 8):
+    from typing import TypedDict
+else:
+    from typing_extensions import TypedDict
 
 SHA256 = NewType('sha_256_hash', str)
 CreateResult = Generator[str, None, None]
@@ -11,4 +17,4 @@
     'TypedDict',
     'SHA256',
     'CreateResult',
-]
+]
diff --git a/requirements.txt b/requirements.txt
@@ -9,3 +9,4 @@ js2py
 quickjs
 flask
 flask-cors
+typing-extensions