phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Move runpod provider to the correct directory

Browse source 
Also cleanup the test code to avoid skipping tests. Let failures be
known and public.
This commit is contained in:
Ashwin Bharambe 2025-01-23 12:25:12 -08:00
parent 22dc684da6
commit d78027f3b5
6 changed files with 7 additions and 102 deletions
Download patch file Download diff file Expand all files Collapse all files

27
llama_stack/providers/tests/inference/test_vision_inference.py
View file

@ -51,20 +51,6 @@ class TestVisionModelInference:
self, inference_model, inference_stack, image, expected_strings
):
inference_impl, _ = inference_stack
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"inline::meta-reference",
"remote::together",
"remote::fireworks",
"remote::ollama",
"remote::vllm",
"remote::sambanova",
):
pytest.skip(
"Other inference providers don't support vision chat completion() yet"
)
response = await inference_impl.chat_completion(
model_id=inference_model,
messages=[
@ -92,19 +78,6 @@ class TestVisionModelInference:
):
inference_impl, _ = inference_stack
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"inline::meta-reference",
"remote::together",
"remote::fireworks",
"remote::ollama",
"remote::vllm",
"remote::sambanova",
):
pytest.skip(
"Other inference providers don't support vision chat completion() yet"
)
images = [
ImageContentItem(
image=dict(