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

View file

@ -137,26 +137,18 @@ interface ProviderSettings {
}
enum Providers {
Anthropic = "Anthropic",
Anyscale = "Anyscale",
OpenAI = "OpenAI",
Azure = "Azure",
Azure_AI_Studio = "Azure AI Studio",
Bedrock = "Amazon Bedrock",
Codestral = "Codestral API (MistralAI)",
Databricks = "Databricks",
Anthropic = "Anthropic",
Google_AI_Studio = "Google AI Studio",
Groq = "Groq",
MistralAI = "MistralAI",
Ollama = "Ollama",
OpenAI = "OpenAI",
OpenAI_Compatible = "OpenAI-Compatible Endpoints (vLLM etc.)",
OpenRouter = "OpenRouter",
PerplexityAI = "PerplexityAI",
TogetherAI = "TogetherAI",
Bedrock = "Amazon Bedrock",
OpenAI_Compatible = "OpenAI-Compatible Endpoints (Groq, Together AI, Mistral AI, etc.)",
Vertex_AI = "Vertex AI (Anthropic, Gemini, etc.)",
Databricks = "Databricks",
Ollama = "Ollama",
}
const provider_map: Record<string, string> = {
OpenAI: "openai",
Azure: "azure",
@ -166,13 +158,6 @@ const provider_map: Record<string, string> = {
Bedrock: "bedrock",
OpenAI_Compatible: "openai",
Vertex_AI: "vertex_ai",
TogetherAI: "together_ai",
MistralAI: "mistral",
Codestral: "codestral",
PerplexityAI:"perplexity",
OpenRouter: "openrouter",
Anyscale: "anyscale",
Groq: "groq",
Databricks: "databricks",
Ollama: "ollama",
@ -1739,7 +1724,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
tooltip="Actual model name used for making litellm.completion() call."
className="mb-0"
>
{ (selectedProvider === Providers.Azure) || (selectedProvider === Providers.OpenAI_Compatible) || (selectedProvider === Providers.Ollama) || (selectedProvider === Providers.TogetherAI) ? (
{ (selectedProvider === Providers.Azure) || (selectedProvider === Providers.OpenAI_Compatible) || (selectedProvider === Providers.Ollama) ? (
<TextInput placeholder={getPlaceholder(selectedProvider.toString())} />
) : providerModels.length > 0 ? (
<MultiSelect value={providerModels}>