github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-01-29 21:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve AsyncCallbackManager (#2410)

Browse Source
This commit is contained in:
Ankush Gola
2023-04-05 09:31:42 +02:00
committed by GitHub
parent af7f20fa42
commit 4d730a9bbc
2 changed files with 76 additions and 143 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/unit_tests/callbacks/test_callback_manager.py
View File

@@ -176,5 +176,6 @@ async def test_async_callback_manager_sync_handler() -> None:
"""Test the AsyncCallbackManager."""
handler1 = FakeCallbackHandler(always_verbose_=True)
handler2 = FakeAsyncCallbackHandler()
manager = AsyncCallbackManager([handler1, handler2])
await _test_callback_manager_async(manager, handler1, handler2)
handler3 = FakeAsyncCallbackHandler(always_verbose_=True)
manager = AsyncCallbackManager([handler1, handler2, handler3])
await _test_callback_manager_async(manager, handler1, handler2, handler3)