phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-11 19:56:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: update tests for OpenAI-style models endpoint

Browse source 
The llama-stack-client now uses /v1/openai/v1/models which returns OpenAI-compatible model objects with 'id' and 'custom_metadata' fields instead of the Resource-style 'identifier' field. Updated api_recorder to handle the new endpoint and modified tests to access model metadata appropriately. Deleted stale model recordings for re-recording.
This commit is contained in:
Ashwin Bharambe 2025-11-03 16:16:16 -08:00
parent 4a5ef65286
commit 809dae01c2
24 changed files with 823 additions and 6697 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/integration/inference/test_openai_completion.py
View file

@ -27,9 +27,11 @@ def _normalize_text(text: str) -> str:
def provider_from_model(client_with_models, model_id):
models = {m.identifier: m for m in client_with_models.models.list()}
models.update({m.provider_resource_id: m for m in client_with_models.models.list()})
provider_id = models[model_id].provider_id
models = {m.id: m for m in client_with_models.models.list()}
models.update(
{m.custom_metadata["provider_resource_id"]: m for m in client_with_models.models.list() if m.custom_metadata}
)
provider_id = models[model_id].custom_metadata["provider_id"]
providers = {p.provider_id: p for p in client_with_models.providers.list()}
return providers[provider_id]