getting the keys in backend

Agenta-AI · Apr 19, 2024 · db0ce7f · db0ce7f
1 parent 85cf465
commit db0ce7f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/agenta-backend/agenta_backend/models/api/evaluation_model.py b/agenta-backend/agenta_backend/models/api/evaluation_model.py
@@ -149,7 +149,8 @@ class HumanEvaluationScenario(BaseModel):
 class HumanEvaluationScenarioUpdate(BaseModel):
     vote: Optional[str]
     score: Optional[Union[str, int]]
-    correct_answer: Optional[str]  # will be used when running custom code evaluation
+    # will be used when running custom code evaluation
+    correct_answer: Optional[str]
     outputs: Optional[List[HumanEvaluationScenarioOutput]]
     inputs: Optional[List[HumanEvaluationScenarioInput]]
     is_pinned: Optional[bool]
@@ -171,7 +172,8 @@ class EvaluationScenario(BaseModel):
 class EvaluationScenarioUpdate(BaseModel):
     vote: Optional[str]
     score: Optional[Any]
-    correct_answer: Optional[str]  # will be used when running custom code evaluation
+    # will be used when running custom code evaluation
+    correct_answer: Optional[str]
     outputs: Optional[List[EvaluationScenarioOutput]]
     inputs: Optional[List[EvaluationScenarioInput]]
     is_pinned: Optional[bool]
@@ -243,6 +245,7 @@ class LMProvidersEnum(str, Enum):
     togetherai = "TOGETHERAI_API_KEY"
     alephalpha = "ALEPHALPHA_API_KEY"
     openrouter = "OPENROUTER_API_KEY"
+    groq = "GROQ_API_KEY"
 
 
 class NewEvaluation(BaseModel):

diff --git a/agenta-backend/agenta_backend/routers/app_router.py b/agenta-backend/agenta_backend/routers/app_router.py
@@ -577,6 +577,7 @@ async def create_app_and_variant_from_template(
                 "TOGETHERAI_API_KEY",
                 "ALEPHALPHA_API_KEY",
                 "OPENROUTER_API_KEY",
+                "GROQ_API_KEY",
             ]
             missing_keys = [
                 key for key in supported_llm_prodviders_keys if not os.environ.get(key)