From 5cff35d1c36dc025d5be695d3ea235382d93d131 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 1 May 2025 08:05:38 -0700 Subject: [PATCH] Fix type annotation --- libs/langgraph/langgraph/func/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py index 5992eb5de..120d44205 100644 --- a/libs/langgraph/langgraph/func/__init__.py +++ b/libs/langgraph/langgraph/func/__init__.py @@ -16,6 +16,7 @@ from typing import ( overload, ) +from langgraph.cache.base import BaseCache from langgraph.channels.ephemeral_value import EphemeralValue from langgraph.channels.last_value import LastValue from langgraph.checkpoint.base import BaseCheckpointSaver @@ -318,7 +319,7 @@ class entrypoint: self, checkpointer: Optional[BaseCheckpointSaver] = None, store: Optional[BaseStore] = None, - cache: Optional[CachePolicy] = None, + cache: Optional[BaseCache] = None, config_schema: Optional[type[Any]] = None, ) -> None: """Initialize the entrypoint decorator."""