This commit is contained in:
jhpiedrahitao 2025-05-06 11:55:53 -05:00
parent d42a9ea53d
commit f592408978

View file

@ -24,13 +24,14 @@ def skip_if_model_doesnt_support_completion(client_with_models, model_id):
providers = {p.provider_id: p for p in client_with_models.providers.list()}
provider = providers[provider_id]
if (
provider.provider_type
in (
"remote::openai",
"remote::anthropic",
"remote::gemini",
"remote::groq",
"remote::sambanova",
provider.provider_type
in (
"remote::openai",
"remote::anthropic",
"remote::gemini",
"remote::groq",
"remote::sambanova",
)
or "openai-compat" in provider.provider_type
):
pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support completion")