phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 17:29:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

mv sdk tests

Browse source
This commit is contained in:
Xi Yan 2024-12-16 11:26:00 -08:00
parent 3cf4fdb740
commit 546fd84177
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/client-sdk/inference/test_inference.py
View file

@ -10,7 +10,11 @@ from llama_stack_client.lib.inference.event_logger import EventLogger
def test_text_chat_completion(llama_stack_client): def test_text_chat_completion(llama_stack_client):
# non-streaming # non-streaming
available_models = [model.identifier for model in llama_stack_client.models.list()] available_models = [
model.identifier
for model in llama_stack_client.models.list()
if model.identifier.startswith("meta-llama")
]
assert len(available_models) > 0 assert len(available_models) > 0
model_id = available_models[0] model_id = available_models[0]
response = llama_stack_client.inference.chat_completion( response = llama_stack_client.inference.chat_completion(