mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-27 06:28:50 +00:00
Merge bf63470c22
into 632cf9eb72
This commit is contained in:
commit
0ff0658484
4 changed files with 33 additions and 0 deletions
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
|
@ -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:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue