github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-19 17:45:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Mason Daugherty 2025-07-21 21:18:59 -04:00
parent cd14cc5014
commit 0e0b1f39ca
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/core/tests/unit_tests/callbacks/test_async_callback_manager.py
View File

@ -172,7 +172,7 @@ async def test_shielded_callback_context_preservation() -> None:
def __init__(self) -> None: def __init__(self) -> None:
self.run_inline = False self.run_inline = False
self.context_values = [] self.context_values: list[str] = []
@override @override
async def on_llm_end(self, response: Any, **kwargs: Any) -> None: async def on_llm_end(self, response: Any, **kwargs: Any) -> None:
@ -198,7 +198,7 @@ async def test_shielded_callback_context_preservation() -> None:
chain_run_manager = await manager.on_chain_start({}, {"test": "input"}) chain_run_manager = await manager.on_chain_start({}, {"test": "input"})
# Test LLM end callback (which is shielded) # Test LLM end callback (which is shielded)
await llm_run_manager.on_llm_end({"response": "test"}) await llm_run_manager.on_llm_end({"response": "test"}) # type: ignore[arg-type]
# Test Chain end callback (which is shielded) # Test Chain end callback (which is shielded)
await chain_run_manager.on_chain_end({"output": "test"}) await chain_run_manager.on_chain_end({"output": "test"})