From f99ada3ae0a66b1d9af73fefabb7582d8d8d9f1f Mon Sep 17 00:00:00 2001 From: Andres Javier Lopez Date: Sat, 14 Mar 2026 13:59:38 -0600 Subject: [PATCH] Add runtime parameter to base implementation --- assistant_core/nodes/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/assistant_core/nodes/base.py b/assistant_core/nodes/base.py index 9b20a3a..68c2889 100644 --- a/assistant_core/nodes/base.py +++ b/assistant_core/nodes/base.py @@ -3,6 +3,7 @@ import abc from langchain_core.runnables import RunnableConfig +from langgraph.runtime import Runtime class BaseNode(abc.ABC): @@ -16,6 +17,8 @@ def __init__(self, name: str = "", *args, **kwargs): self.name = name @abc.abstractmethod - async def __call__(self, state: dict, config: RunnableConfig) -> dict: + async def __call__( + self, state: dict, config: RunnableConfig, runtime: Runtime + ) -> dict: """Executes the node.""" raise NotImplementedError("Subclasses must implement this method.")