From a729f6575f02f4c04e52f95a7c8f9ed1dae48e56 Mon Sep 17 00:00:00 2001 From: r3v5 Date: Wed, 23 Jul 2025 17:46:09 +0100 Subject: [PATCH] fix: fixed test_access_control.py unit test --- tests/unit/server/test_access_control.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/unit/server/test_access_control.py b/tests/unit/server/test_access_control.py index fb9c6f95e..d6a420c13 100644 --- a/tests/unit/server/test_access_control.py +++ b/tests/unit/server/test_access_control.py @@ -285,9 +285,15 @@ async def test_access_policy(mock_get_authenticated_user, test_setup_with_access "projects": ["foo", "bar"], }, ) - await routing_table.register_model("model-1", provider_id="test_provider") - await routing_table.register_model("model-2", provider_id="test_provider") - await routing_table.register_model("model-3", provider_id="test_provider") + await routing_table.register_model( + "model-1", provider_model_id="test_provider/model-1", provider_id="test_provider" + ) + await routing_table.register_model( + "model-2", provider_model_id="test_provider/model-2", provider_id="test_provider" + ) + await routing_table.register_model( + "model-3", provider_model_id="test_provider/model-3", provider_id="test_provider" + ) model = await routing_table.get_model("model-1") assert model.identifier == "model-1" model = await routing_table.get_model("model-2")