mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-05 10:23:44 +00:00
Merge branch 'main' into fix/list-command-show-builtin-distros
This commit is contained in:
commit
e5e5e3b5fe
344 changed files with 50428 additions and 80186 deletions
|
|
@ -166,6 +166,14 @@ async def test_models_routing_table(cached_disk_dist_registry):
|
|||
assert "test_provider/test-model" in openai_model_ids
|
||||
assert "test_provider/test-model-2" in openai_model_ids
|
||||
|
||||
# Verify custom_metadata is populated with Llama Stack-specific data
|
||||
for openai_model in openai_models.data:
|
||||
assert openai_model.custom_metadata is not None
|
||||
assert "model_type" in openai_model.custom_metadata
|
||||
assert "provider_id" in openai_model.custom_metadata
|
||||
assert "provider_resource_id" in openai_model.custom_metadata
|
||||
assert openai_model.custom_metadata["provider_id"] == "test_provider"
|
||||
|
||||
# Test get_object_by_identifier
|
||||
model = await table.get_object_by_identifier("model", "test_provider/test-model")
|
||||
assert model is not None
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue