mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-22 22:39:41 +00:00
fix: fixed test_access_control.py unit test
This commit is contained in:
parent
e1ed152779
commit
a729f6575f
1 changed files with 9 additions and 3 deletions
|
|
@ -285,9 +285,15 @@ async def test_access_policy(mock_get_authenticated_user, test_setup_with_access
|
||||||
"projects": ["foo", "bar"],
|
"projects": ["foo", "bar"],
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
await routing_table.register_model("model-1", provider_id="test_provider")
|
await routing_table.register_model(
|
||||||
await routing_table.register_model("model-2", provider_id="test_provider")
|
"model-1", provider_model_id="test_provider/model-1", provider_id="test_provider"
|
||||||
await routing_table.register_model("model-3", 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")
|
model = await routing_table.get_model("model-1")
|
||||||
assert model.identifier == "model-1"
|
assert model.identifier == "model-1"
|
||||||
model = await routing_table.get_model("model-2")
|
model = await routing_table.get_model("model-2")
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue