mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-13 13:17:52 +02:00
Pass tags when configuring async callback manager
This commit is contained in:
@@ -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"),
|
||||
)
|
||||
|
||||
|
||||
@@ -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"]
|
||||
Reference in New Issue
Block a user