mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-23 00:49:42 +00:00
feat(tests): record responses for evals and telemetry tests
This commit is contained in:
parent
81c7d6fa2e
commit
77bdd585e2
13 changed files with 881 additions and 41 deletions
|
|
@ -345,7 +345,7 @@ def test_inference_store(compat_client, client_with_models, text_model_id, strea
|
|||
response_id = response.id
|
||||
content = response.choices[0].message.content
|
||||
|
||||
responses = client.chat.completions.list()
|
||||
responses = client.chat.completions.list(limit=1000)
|
||||
assert response_id in [r.id for r in responses.data]
|
||||
|
||||
retrieved_response = client.chat.completions.retrieve(response_id)
|
||||
|
|
@ -410,7 +410,7 @@ def test_inference_store_tool_calls(compat_client, client_with_models, text_mode
|
|||
response_id = response.id
|
||||
content = response.choices[0].message.content
|
||||
|
||||
responses = client.chat.completions.list()
|
||||
responses = client.chat.completions.list(limit=1000)
|
||||
assert response_id in [r.id for r in responses.data]
|
||||
|
||||
retrieved_response = client.chat.completions.retrieve(response_id)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue