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

Temporary Commit at 1/5/2025, 10:46:53 PM

Browse source 
Summary:
### THIS DIFF

### PLAN

### CONTEXT

### DESIGN

Test Plan:
# Test Setup

**Type checker and check that the build compiles**

**Unit Tests**

**E2E Tests**

// Screenshots and videos
| Before | After |
|--|
| … | … |

# Monitoring Plan.
This commit is contained in:
Sixian Yi 2025-01-05 22:46:53 -08:00
parent ab47d4a71e
commit 8c5b328566
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/tests/inference/ci_test.py
View file

@ -14,6 +14,11 @@ INFERENCE_PROVIDER_ENV_KEY = {
"together": "TOGETHER_API_KEY",
}
TEST_MODELS = {
"text": "meta-llama/Llama-3.1-8B-Instruct",
"vision": "meta-llama/Llama-3.2-11B-Vision-Instruct",
}
# Model category and the keywords of the corresponding functionality tests
CATEGORY_FUNCTIONALITY_TESTS = {
"text": ["streaming", "tool_calling", "structured_output"],
@ -35,6 +40,7 @@ def generate_pytest_args(category, provider, env_key):
"-s",
"-k",
provider,
"--inference-model={model_name}".format(model_name=TEST_MODELS[category]),
]
if env_key is not None:
pytest_args.extend(