Revert "[Ui] add together AI, Mistral, PerplexityAI, OpenRouter models on Admin UI "

This commit is contained in:
Ishaan Jaff 2024-07-20 19:04:22 -07:00 committed by GitHub
parent 28bb2919b6
commit 9a545c1ff8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 26 deletions

View file

@ -79,10 +79,7 @@ def set_client(litellm_router_instance: LitellmRouter, model: dict):
default_api_key = None
if custom_llm_provider in litellm.openai_compatible_providers:
_, custom_llm_provider, api_key, api_base = litellm.get_llm_provider(
model=model_name,
custom_llm_provider=custom_llm_provider,
api_key=litellm_params.get("api_key"),
api_base=litellm_params.get("api_base"),
model=model_name
)
default_api_base = api_base
default_api_key = api_key