From bf1467132360bfca5e2a8c92481531faddc67644 Mon Sep 17 00:00:00 2001
From: 732642856 <732642856@qq.com>
Date: Sun, 19 Jul 2026 10:24:39 +0800
Subject: [PATCH] test: align consult agent stream contract

---
 tests/test_agent_chat_contract.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_agent_chat_contract.py b/tests/test_agent_chat_contract.py
index 0a15ee59..bb5480a5 100644
--- a/tests/test_agent_chat_contract.py
+++ b/tests/test_agent_chat_contract.py
@@ -75,7 +75,7 @@ def test_multi_model_route_and_persistence_contract() -> None:
     assert "reserveConsultationModel(" in consult_route
     assert "resolveLanguageModel," in consult_route
     assert 'runCreditRpc(accounting, "begin_consultation_credit"' in consult_route
-    assert "getJyotishAgent(selectedModel).stream" in consult_route
+    assert "getJyotishAgent(selectedModel, workflowContext).stream" in consult_route
     assert "modelSelection.usageModelId" in consult_route
     assert 'if (!model) return { status: "unavailable" }' in selection
     assert 'reservation: await reserveCredit()' in selection