From 11ae529edc895a3200b9baa7a85bcb40b3155db4 Mon Sep 17 00:00:00 2001 From: estelle Date: Wed, 14 May 2025 16:43:33 +0200 Subject: [PATCH] Update type to be able to use LLMMessage in a Pydantic model --- src/neo4j_graphrag/types.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/neo4j_graphrag/types.py b/src/neo4j_graphrag/types.py index 1c0b74542..d6b969a51 100644 --- a/src/neo4j_graphrag/types.py +++ b/src/neo4j_graphrag/types.py @@ -16,7 +16,8 @@ import warnings from enum import Enum -from typing import Any, Callable, Literal, Optional, TypedDict, Union +from typing import Any, Callable, Literal, Optional, Union +from typing_extensions import TypedDict import neo4j from pydantic import (