summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--g4f/typing.py10
-rw-r--r--requirements.txt1
2 files changed, 9 insertions, 2 deletions
diff --git a/g4f/typing.py b/g4f/typing.py
index 02386037..5f63c222 100644
--- 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 @@ __all__ = [
'TypedDict',
'SHA256',
'CreateResult',
-] \ No newline at end of file
+]
diff --git a/requirements.txt b/requirements.txt
index 76431391..7ee6e37c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,3 +9,4 @@ js2py
quickjs
flask
flask-cors
+typing-extensions