diff --git a/packages/langchain-agents/tests/test_handler.py b/packages/langchain-agents/tests/test_handler.py index a68c42a4..2790243b 100644 --- a/packages/langchain-agents/tests/test_handler.py +++ b/packages/langchain-agents/tests/test_handler.py @@ -520,8 +520,8 @@ def handoff(_args: dict[str, Any]) -> str: if not chosen: chosen.append("billing") return ( - "Handoff to billing recorded. " - "Finish your own work and provide your final response." + "Handoff to billing recorded. billing takes over when your turn ends. " + "Finish your own part, then end your turn. Do not answer for billing." ) tool = _built_tool( @@ -529,8 +529,8 @@ def handoff(_args: dict[str, Any]) -> str: {"__handoff_billing": handoff}, ) assert await tool() == ( - "Handoff to billing recorded. " - "Finish your own work and provide your final response." + "Handoff to billing recorded. billing takes over when your turn ends. " + "Finish your own part, then end your turn. Do not answer for billing." ) assert chosen == ["billing"]