add more providers in api keys form

Agenta-AI · Apr 17, 2024 · 5bbda75 · 5bbda75
1 parent 9073b22
commit 5bbda75
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/agenta-backend/agenta_backend/models/api/evaluation_model.py b/agenta-backend/agenta_backend/models/api/evaluation_model.py
@@ -242,6 +242,12 @@ class LMProvidersEnum(str, Enum):
     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"
+    deepinfra = "DEEPINFRA_API_KEY"
+    alephalpha = "ALEPHALPHA_API_KEY"
+    openrouter = "OPENROUTER_API_KEY"
 
 
 class NewEvaluation(BaseModel):

diff --git a/agenta-web/src/lib/helpers/llmProviders.ts b/agenta-web/src/lib/helpers/llmProviders.ts
@@ -11,13 +11,19 @@ export type LlmProvider = {
 
 export const llmAvailableProviders: LlmProvider[] = [
     {title: "OpenAI", key: "", name: "OPENAI_API_KEY"},
-    {title: "Replicate", key: "", name: "REPLICATE_API_KEY"},
-    {title: "Hugging Face", key: "", name: "HUGGING_FACE_API_KEY"},
+    {title: "Mistral AI", key: "", name: "MISTRAL_API_KEY"},
     {title: "Cohere", key: "", name: "COHERE_API_KEY"},
     {title: "Anthropic", key: "", name: "ANTHROPIC_API_KEY"},
-    {title: "Azure", key: "", name: "AZURE_API_KEY"},
     {title: "TogetherAI", key: "", name: "TOGETHERAI_API_KEY"},
-    {title: "Mistral AI", key: "", name: "MISTRAL_API_KEY"},
+    {title: "Perplexity AI", key: "", name: "PERPLEXITYAI_API_KEY"},
+    {title: "Anyscale", key: "", name: "ANYSCALE_API_KEY"},
+    {title: "Fireworks AI", key: "", name: "FIREWORKS_AI_API_KEY"},
+    {title: "DeepInfra", key: "", name: "DEEPINFRA_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 = () => {