Config

cohere-ai · Jul 15, 2024 · 0cf3abb · 0cf3abb
1 parent b04b9d2
commit 0cf3abb
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 22 deletions.
diff --git a/src/backend/config/auth.py b/src/backend/config/auth.py
@@ -14,7 +14,7 @@
 # Ex: [BasicAuthentication]
 ENABLED_AUTH_STRATEGIES = []
 if "pytest" in sys.modules or SKIP_AUTH == "true":
-    ENABLED_AUTH_STRATEGIES = []
+    ENABLED_AUTH_STRATEGIES = [OpenIDConnect]
 
 # Define the mapping from Auth strategy name to class obj - does not need to be manually modified.
 # During runtime, this will create an instance of each enabled strategy class.

diff --git a/src/backend/config/deployments.py b/src/backend/config/deployments.py
@@ -44,27 +44,27 @@ class ModelDeploymentName(StrEnum):
         is_available=SingleContainerDeployment.is_available(),
         env_vars=SC_ENV_VARS,
     ),
-    ModelDeploymentName.SageMaker: Deployment(
-        name=ModelDeploymentName.SageMaker,
-        deployment_class=SageMakerDeployment,
-        models=SageMakerDeployment.list_models(),
-        is_available=SageMakerDeployment.is_available(),
-        env_vars=SAGE_MAKER_ENV_VARS,
-    ),
-    ModelDeploymentName.Azure: Deployment(
-        name=ModelDeploymentName.Azure,
-        deployment_class=AzureDeployment,
-        models=AzureDeployment.list_models(),
-        is_available=AzureDeployment.is_available(),
-        env_vars=AZURE_ENV_VARS,
-    ),
-    ModelDeploymentName.Bedrock: Deployment(
-        name=ModelDeploymentName.Bedrock,
-        deployment_class=BedrockDeployment,
-        models=BedrockDeployment.list_models(),
-        is_available=BedrockDeployment.is_available(),
-        env_vars=BEDROCK_ENV_VARS,
-    ),
+    # ModelDeploymentName.SageMaker: Deployment(
+    #     name=ModelDeploymentName.SageMaker,
+    #     deployment_class=SageMakerDeployment,
+    #     models=SageMakerDeployment.list_models(),
+    #     is_available=SageMakerDeployment.is_available(),
+    #     env_vars=SAGE_MAKER_ENV_VARS,
+    # ),
+    # ModelDeploymentName.Azure: Deployment(
+    #     name=ModelDeploymentName.Azure,
+    #     deployment_class=AzureDeployment,
+    #     models=AzureDeployment.list_models(),
+    #     is_available=AzureDeployment.is_available(),
+    #     env_vars=AZURE_ENV_VARS,
+    # ),
+    # ModelDeploymentName.Bedrock: Deployment(
+    #     name=ModelDeploymentName.Bedrock,
+    #     deployment_class=BedrockDeployment,
+    #     models=BedrockDeployment.list_models(),
+    #     is_available=BedrockDeployment.is_available(),
+    #     env_vars=BEDROCK_ENV_VARS,
+    # ),
 }