Skip to content

Commit 1aad067

Browse files
committed
Update type to be able to use LLMMessage in a Pydantic model (#338)
1 parent 4f6895b commit 1aad067

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/neo4j_graphrag/types.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@
1616

1717
import warnings
1818
from enum import Enum
19-
from typing import Any, Callable, Literal, Optional, TypedDict, Union
19+
from typing import Any, Callable, Literal, Optional, Union
20+
from typing_extensions import TypedDict
2021

2122
import neo4j
2223
from pydantic import (

0 commit comments

Comments
 (0)