From 8ad46284b9b12af43d542f4751975c4d5911e1ef Mon Sep 17 00:00:00 2001
From: hs_junxiang <jimmy871117@gmail.com>
Date: Mon, 11 Sep 2023 15:29:48 +0800
Subject: Fix: `TypedDict` from `typing` error #886

Ref: https://github.com/python/typeshed/issues/3500
---
 g4f/typing.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'g4f/typing.py')

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
+]
-- 
cgit v1.2.3