Pass tags when configuring async callback manager

This commit is contained in:
Nuno Campos
2025-06-03 16:17:17 -07:00
parent b00ea605e2
commit 5ab5da767b
2 changed files with 20 additions and 1 deletions
+1 -1
View File
@@ -263,7 +263,7 @@ def get_async_callback_manager_for_config(
# otherwise create a new manager
return AsyncCallbackManager.configure(
inheritable_callbacks=config.get("callbacks"),
inheritable_tags=config.get("tags"),
inheritable_tags=all_tags,
inheritable_metadata=config.get("metadata"),
)
+19
View File
@@ -0,0 +1,19 @@
import pytest
from langchain_core.callbacks import AsyncCallbackManager
from langgraph.utils.config import get_async_callback_manager_for_config
pytestmark = pytest.mark.anyio
def test_new_async_manager_includes_tags() -> None:
config = {"callbacks": None}
manager = get_async_callback_manager_for_config(config, tags=["x", "y"])
assert isinstance(manager, AsyncCallbackManager)
assert manager.inheritable_tags == ["x", "y"]
def test_new_async_manager_merges_tags_with_config() -> None:
config = {"callbacks": None, "tags": ["a"]}
manager = get_async_callback_manager_for_config(config, tags=["b"])
assert manager.inheritable_tags == ["a", "b"]