use TypedDict from typing_extensions (#160)

microsoft · Apr 18, 2024 · c9a0028 · c9a0028
1 parent 8b0755d
commit c9a0028
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphrag/llm/types/llm_io.py b/graphrag/llm/types/llm_io.py
@@ -3,9 +3,9 @@
 """LLM Types."""
 
 from dataclasses import dataclass, field
-from typing import Generic, TypedDict, TypeVar
+from typing import Generic, TypeVar
 
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, TypedDict
 
 from .llm_callbacks import IsResponseValidFn