Add get_store function (#3112)

langchain-ai · Jan 21, 2025 · 6cbc7e8 · 6cbc7e8
2 parents ac64f50 + b2213e5
commit 6cbc7e8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/libs/langgraph/langgraph/utils/config.py b/libs/langgraph/langgraph/utils/config.py
@@ -23,9 +23,11 @@
     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]