diff --git a/tests/test_dispatcher.py b/tests/test_dispatcher.py index cca39d3..b4b21be 100644 --- a/tests/test_dispatcher.py +++ b/tests/test_dispatcher.py @@ -232,7 +232,8 @@ class TestBatchDecision: results = dispatcher.dispatch_pending(tasks) assert len(results) == 3 assert results[0]["level"] == DispatchLevel.FULL_AGENT - assert results[1]["level"] == DispatchLevel.SUB_AGENT + assert results[1]["level"] == DispatchLevel.FULL_AGENT + assert results[1]["agent_id"] == "pangtong-fujunshi" # fallback assert results[2]["level"] == DispatchLevel.ESCALATE def test_message_building(self, dispatcher):