WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-13 21:27:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Set store & stream writer

Browse Source
This commit is contained in:
William Fu-Hinthorn
2025-03-26 15:17:31 -07:00
parent 7021ce3742
commit 933619d751
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/config.py
+22
View File
@@ -118,6 +118,17 @@ def get_store() -> BaseStore:
return config[CONF][CONFIG_KEY_STORE]
def set_store(store: BaseStore) -> None:
"""Set LangGraph store in context."""
var_config = var_child_runnable_config.get()
if not var_config:
var_config = {CONF: {CONFIG_KEY_STORE: store}}
var_child_runnable_config.set(var_config)
else:
var_config[CONF][CONFIG_KEY_STORE] = store
var_child_runnable_config.set(var_config)
def get_stream_writer() -> StreamWriter:
"""Access LangGraph [StreamWriter][langgraph.types.StreamWriter] from inside a graph node or entrypoint task at runtime.
@@ -183,3 +194,14 @@ def get_stream_writer() -> StreamWriter:
"""
config = get_config()
return config[CONF].get(CONFIG_KEY_STREAM_WRITER, _no_op_stream_writer)
def set_stream_writer(stream_writer: StreamWriter) -> None:
"""Set LangGraph stream writer in context."""
var_config = var_child_runnable_config.get()
if not var_config:
var_config = {CONF: {CONFIG_KEY_STREAM_WRITER: stream_writer}}
var_child_runnable_config.set(var_config)
else:
var_config[CONF][CONFIG_KEY_STREAM_WRITER] = stream_writer
var_child_runnable_config.set(var_config)