This commit is contained in:
Matthew Farrellee 2025-07-24 16:41:17 -04:00 committed by GitHub
commit 0ff0658484
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 33 additions and 0 deletions

View file

@ -17,6 +17,7 @@ class GeminiInferenceAdapter(LiteLLMOpenAIMixin):
MODEL_ENTRIES,
api_key_from_config=config.api_key,
provider_data_api_key_field="gemini_api_key",
litellm_provider_name="gemini",
)
self.config = config

View file

@ -36,6 +36,7 @@ class GroqInferenceAdapter(LiteLLMOpenAIMixin):
model_entries=MODEL_ENTRIES,
api_key_from_config=config.api_key,
provider_data_api_key_field="groq_api_key",
litellm_provider_name="groq",
)
self.config = config

View file

@ -184,6 +184,7 @@ class SambaNovaInferenceAdapter(LiteLLMOpenAIMixin):
model_entries=MODEL_ENTRIES,
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",
litellm_provider_name="sambanova",
)
def _get_api_key(self) -> str: