WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-29 19:29:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

merge callbacks in ensure_config

Browse Source
This commit is contained in:
Chester Curme
2026-04-05 20:15:56 -04:00
parent 0c52212098
commit 442c6d90b0
1 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/_internal/_config.py
+31
View File
@@ -303,6 +303,37 @@ def ensure_config(*configs: RunnableConfig | None) -> RunnableConfig:
if _is_not_empty(v) and k in CONFIG_KEYS:
if k == CONF:
empty[k] = cast(dict, v).copy()
elif k == "callbacks":
base_callbacks = empty.get("callbacks")
if base_callbacks is None:
if isinstance(v, list):
empty["callbacks"] = v.copy()
elif isinstance(v, BaseCallbackManager):
empty["callbacks"] = v.copy()
else:
empty["callbacks"] = v
elif isinstance(v, list):
if isinstance(base_callbacks, list):
empty["callbacks"] = base_callbacks + v
elif isinstance(base_callbacks, BaseCallbackManager):
mngr = base_callbacks.copy()
for callback in v:
mngr.add_handler(callback, inherit=True)
empty["callbacks"] = mngr
else:
empty["callbacks"] = v
elif isinstance(v, BaseCallbackManager):
if isinstance(base_callbacks, list):
mngr = v.copy()
for callback in base_callbacks:
mngr.add_handler(callback, inherit=True)
empty["callbacks"] = mngr
elif isinstance(base_callbacks, BaseCallbackManager):
empty["callbacks"] = base_callbacks.merge(v)
else:
empty["callbacks"] = v
else:
empty["callbacks"] = v
else:
empty[k] = v # type: ignore[literal-required]
for k, v in config.items():