mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 10:14:26 +00:00
Merge 97fec2a4de
into b82af5b826
This commit is contained in:
commit
5bdc777ced
1 changed files with 2 additions and 2 deletions
|
@ -196,10 +196,10 @@ def get_llm_provider( # noqa: PLR0915
|
|||
elif endpoint == "https://api.ai21.com/studio/v1":
|
||||
custom_llm_provider = "ai21_chat"
|
||||
dynamic_api_key = get_secret_str("AI21_API_KEY")
|
||||
elif endpoint == "https://codestral.mistral.ai/v1":
|
||||
elif endpoint == "codestral.mistral.ai/v1/chat/completions":
|
||||
custom_llm_provider = "codestral"
|
||||
dynamic_api_key = get_secret_str("CODESTRAL_API_KEY")
|
||||
elif endpoint == "https://codestral.mistral.ai/v1":
|
||||
elif endpoint == "codestral.mistral.ai/v1/fim/completions":
|
||||
custom_llm_provider = "text-completion-codestral"
|
||||
dynamic_api_key = get_secret_str("CODESTRAL_API_KEY")
|
||||
elif endpoint == "app.empower.dev/api/v1":
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue