mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-31 04:09:49 +02:00
20 lines
679 B
Python
20 lines
679 B
Python
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"]
|