phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-02 09:14:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix intergration tests

Browse source
This commit is contained in:
Honglin Cao 2025-03-11 11:18:50 -04:00
parent d1f67d90ca
commit a454b53bda
5 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/integration/fixtures/common.py
View file

@ -34,6 +34,7 @@ def provider_data():
"GEMINI_API_KEY": "gemini_api_key",
"OPENAI_API_KEY": "openai_api_key",
"TOGETHER_API_KEY": "together_api_key",
"CENTML_API_KEY": "centml_api_key",
"ANTHROPIC_API_KEY": "anthropic_api_key",
"GROQ_API_KEY": "groq_api_key",
"WOLFRAM_ALPHA_API_KEY": "wolfram_alpha_api_key",

2
tests/integration/inference/test_text_inference.py
View file

@ -12,7 +12,7 @@ from llama_stack.models.llama.sku_list import resolve_model
from ..test_cases.test_case import TestCase
PROVIDER_LOGPROBS_TOP_K = {"remote::together", "remote::fireworks", "remote::vllm"}
PROVIDER_LOGPROBS_TOP_K = {"remote::together", "remote::fireworks", "remote::vllm", "remote::centml"}
def skip_if_model_doesnt_support_completion(client_with_models, model_id):