From 0cf3abb80e04ac9df6eedfc184c802b5bbdfdab9 Mon Sep 17 00:00:00 2001 From: BeatrixCohere <128378696+BeatrixCohere@users.noreply.github.com> Date: Mon, 15 Jul 2024 14:29:23 +0100 Subject: [PATCH] Config --- src/backend/config/auth.py | 2 +- src/backend/config/deployments.py | 42 +++++++++++++++---------------- 2 files changed, 22 insertions(+), 22 deletions(-) diff --git a/src/backend/config/auth.py b/src/backend/config/auth.py index 3c4f931815..30350c1e54 100644 --- 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 index 5c12b99c21..4a17d7eaaf 100644 --- 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, + # ), }