From c9a00287ee5734acca5fd185fc7a6ce3ccedc17b Mon Sep 17 00:00:00 2001 From: Chris Trevino Date: Thu, 18 Apr 2024 15:53:48 -0700 Subject: [PATCH] use TypedDict from typing_extensions (#160) --- graphrag/llm/types/llm_io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graphrag/llm/types/llm_io.py b/graphrag/llm/types/llm_io.py index be80a54d96..52bf4b9e82 100644 --- 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