diff --git a/agenta-backend/agenta_backend/models/api/evaluation_model.py b/agenta-backend/agenta_backend/models/api/evaluation_model.py index b09e12681e..c9954479ae 100644 --- a/agenta-backend/agenta_backend/models/api/evaluation_model.py +++ b/agenta-backend/agenta_backend/models/api/evaluation_model.py @@ -241,6 +241,7 @@ class LMProvidersEnum(str, Enum): azure_base = "AZURE_API_BASE" azure_key = "AZURE_API_KEY" togetherai = "TOGETHERAI_API_KEY" + mistralai = "MISTRALAI_API_KEY" class NewEvaluation(BaseModel): diff --git a/agenta-web/src/lib/Types.ts b/agenta-web/src/lib/Types.ts index 42a0ef4c8a..d2e94c7b0b 100644 --- a/agenta-web/src/lib/Types.ts +++ b/agenta-web/src/lib/Types.ts @@ -261,6 +261,7 @@ export interface LlmProvidersKeys { AZURE_API_KEY: string AZURE_API_BASE: string TOGETHERAI_API_KEY: string + MISTRALAI_API_KEY: string } export interface AppTemplate { diff --git a/agenta-web/src/lib/helpers/llmProviders.ts b/agenta-web/src/lib/helpers/llmProviders.ts index 21a8b33e6b..aac6d4e383 100644 --- a/agenta-web/src/lib/helpers/llmProviders.ts +++ b/agenta-web/src/lib/helpers/llmProviders.ts @@ -17,6 +17,7 @@ export const llmAvailableProviders: LlmProvider[] = [ {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: "MISTRALAI_API_KEY"}, ] export const getApikeys = () => {