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()} providers = {p.provider_id: p for p in client_with_models.providers.list()}
provider = providers[provider_id] provider = providers[provider_id]
if ( if (
provider.provider_type provider.provider_type
in ( in (
"remote::openai", "remote::openai",
"remote::anthropic", "remote::anthropic",
"remote::gemini", "remote::gemini",
"remote::groq", "remote::groq",
"remote::sambanova", "remote::sambanova",
)
or "openai-compat" in provider.provider_type or "openai-compat" in provider.provider_type
): ):
pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support completion") pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support completion")