From b2213e523efb2e8ee78a742e9685a11c3a34b2f1 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 20 Jan 2025 16:03:27 -0800 Subject: [PATCH] Add get_store function --- libs/langgraph/langgraph/utils/config.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/utils/config.py b/libs/langgraph/langgraph/utils/config.py index 309c6d6be..55c89b3ba 100644 --- a/libs/langgraph/langgraph/utils/config.py +++ b/libs/langgraph/langgraph/utils/config.py @@ -23,9 +23,11 @@ from langgraph.constants import ( CONFIG_KEY_CHECKPOINT_ID, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_CHECKPOINT_NS, + CONFIG_KEY_STORE, NS_END, NS_SEP, ) +from langgraph.store.base import BaseStore def recast_checkpoint_ns(ns: str) -> str: @@ -332,4 +334,9 @@ def get_config() -> RunnableConfig: if var_config := var_child_runnable_config.get(): return var_config else: - raise RuntimeError("Called get_configurable outside of a runnable context") + raise RuntimeError("Called get_config outside of a runnable context") + + +def get_store() -> BaseStore: + config = get_config() + return config[CONF][CONFIG_KEY_STORE]