mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-30 00:04:19 +00:00
Merge 0a8c992708
into 0e287763cd
This commit is contained in:
commit
f52e15cde2
@ -13,3 +13,24 @@ def test_remove_handler() -> None:
|
||||
manager = BaseCallbackManager([handler1], inheritable_handlers=[handler2])
|
||||
manager.remove_handler(handler1)
|
||||
manager.remove_handler(handler2)
|
||||
|
||||
|
||||
def test_merge_preserves_handler_distinction() -> None:
|
||||
"""Test that merging managers preserves the distinction between handlers.
|
||||
|
||||
This test verifies the correct behavior of the BaseCallbackManager.merge()
|
||||
method. When two managers are merged, their handlers and
|
||||
inheritable_handlers should be combined independently.
|
||||
"""
|
||||
h1 = BaseCallbackHandler()
|
||||
h2 = BaseCallbackHandler()
|
||||
ih1 = BaseCallbackHandler()
|
||||
ih2 = BaseCallbackHandler()
|
||||
|
||||
m1 = BaseCallbackManager(handlers=[h1], inheritable_handlers=[ih1])
|
||||
m2 = BaseCallbackManager(handlers=[h2], inheritable_handlers=[ih2])
|
||||
|
||||
merged = m1.merge(m2)
|
||||
|
||||
assert set(merged.handlers) == {h1, h2}
|
||||
assert set(merged.inheritable_handlers) == {ih1, ih2}
|
||||
|
Loading…
Reference in New Issue
Block a user