remove unsupported models

Agenta-AI · Apr 18, 2024 · 3fe5321 · 3fe5321
1 parent c36a463
commit 3fe5321
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/agenta-backend/agenta_backend/models/api/evaluation_model.py b/agenta-backend/agenta_backend/models/api/evaluation_model.py
@@ -234,18 +234,13 @@ class LLMRunRateLimit(BaseModel):
 
 class LMProvidersEnum(str, Enum):
     openai = "OPENAI_API_KEY"
-    replicate = "REPLICATE_API_KEY"
+    mistralai = "MISTRAL_API_KEY"
     cohere = "COHERE_API_KEY"
-    hugging_face = "HUGGING_FACE_API_KEY"
     anthropic = "ANTHROPIC_API_KEY"
-    azure_base = "AZURE_API_BASE"
-    azure_key = "AZURE_API_KEY"
-    togetherai = "TOGETHERAI_API_KEY"
-    mistralai = "MISTRAL_API_KEY"
-    perplexityai = "PERPLEXITYAI_API_KEY"
     anyscale = "ANYSCALE_API_KEY"
-    fireworksai = "FIREWORKS_AI_API_KEY"
+    perplexityai = "PERPLEXITYAI_API_KEY"
     deepinfra = "DEEPINFRA_API_KEY"
+    togetherai = "TOGETHERAI_API_KEY"
     alephalpha = "ALEPHALPHA_API_KEY"
     openrouter = "OPENROUTER_API_KEY"
 

diff --git a/agenta-web/src/lib/helpers/llmProviders.ts b/agenta-web/src/lib/helpers/llmProviders.ts
@@ -20,9 +20,6 @@ export const llmAvailableProviders: LlmProvider[] = [
     {title: "TogetherAI", key: "", name: "TOGETHERAI_API_KEY"},
     {title: "Aleph Alpha", key: "", name: "ALEPHALPHA_API_KEY"},
     {title: "OpenRouter", key: "", name: "OPENROUTER_API_KEY"},
-    {title: "Replicate", key: "", name: "REPLICATE_API_KEY"},
-    {title: "Hugging Face", key: "", name: "HUGGING_FACE_API_KEY"},
-    {title: "Azure", key: "", name: "AZURE_API_KEY"},
 ]
 
 export const getApikeys = () => {