update api_key_from_config initialization

This commit is contained in:
jhpiedrahitao 2025-07-11 15:07:22 -05:00
parent d739fe77a9
commit bc65d5c9ef
2 changed files with 1 additions and 2 deletions

View file

@ -182,7 +182,7 @@ class SambaNovaInferenceAdapter(LiteLLMOpenAIMixin):
LiteLLMOpenAIMixin.__init__( LiteLLMOpenAIMixin.__init__(
self, self,
model_entries=MODEL_ENTRIES, model_entries=MODEL_ENTRIES,
api_key_from_config=self.config.api_key, api_key_from_config=self.config.api_key.get_secret_value() if self.config.api_key else None,
provider_data_api_key_field="sambanova_api_key", provider_data_api_key_field="sambanova_api_key",
) )

View file

@ -71,7 +71,6 @@ def skip_if_model_doesnt_support_openai_chat_completion(client_with_models, mode
"remote::cerebras", "remote::cerebras",
"remote::databricks", "remote::databricks",
"remote::runpod", "remote::runpod",
"remote::sambanova",
"remote::tgi", "remote::tgi",
): ):
pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support OpenAI chat completions.") pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support OpenAI chat completions.")