phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Update inference recorder to handle both Ollama and OpenAI model

Browse source 
o Handle Ollama format where models are nested under response['body']['models']
o Fall back to OpenAI format where models are directly in response['body']

Closes: #3457
Signed-off-by: Derek Higgins <derekh@redhat.com>
This commit is contained in:
Derek Higgins 2025-09-17 10:09:05 +01:00
parent 9acf49753e
commit 2ec323ece1
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/testing/inference_recorder.py
View file

@ -203,6 +203,11 @@ def _model_identifiers_digest(endpoint: str, response: dict[str, Any]) -> str:
- '/v1/models' (OpenAI): response body is: [ { id: ... }, ... ] - '/v1/models' (OpenAI): response body is: [ { id: ... }, ... ]
Returns a list of unique identifiers or None if structure doesn't match. Returns a list of unique identifiers or None if structure doesn't match.
""" """
if "models" in response["body"]:
# ollama
items = response["body"]["models"]
else:
# openai
items = response["body"] items = response["body"]
idents = [m.model if endpoint == "/api/tags" else m.id for m in items] idents = [m.model if endpoint == "/api/tags" else m.id for m in items]
return sorted(set(idents)) return sorted(set(idents))