diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index f127859ff..e871815ea 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -227,9 +227,7 @@ class Pregel( class Config: arbitrary_types_allowed = True - def with_config( - self, config: RunnableConfig | None = None, **kwargs: Any - ) -> Runnable[Dict[All, Any] | Any, Dict[All, Any] | Any]: + def with_config(self, config: RunnableConfig | None = None, **kwargs: Any) -> Self: return self.copy( update={"config": cast(RunnableConfig, {**(config or {}), **kwargs})} )