Expose AI Assistant to bootstrap endpoint

ds-wizard · Oct 22, 2024 · d724c2a · d724c2a
1 parent cf78417
commit d724c2a
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigDTO.hs b/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigDTO.hs
@@ -21,6 +21,7 @@ data ClientConfigDTO = ClientConfigDTO
  , locales :: [ClientConfigLocaleDTO]
  , owl :: TenantConfigOwl
  , admin :: ClientConfigAdminDTO
+ , aiAssistant :: ClientConfigAiAssistantDTO
  , signalBridge :: ClientConfigSignalBridgeDTO
  , modules :: [ClientConfigModuleDTO]
  }
@@ -81,6 +82,11 @@ data ClientConfigAdminDTO = ClientConfigAdminDTO
  }
  deriving (Generic, Eq, Show)
 
+data ClientConfigAiAssistantDTO = ClientConfigAiAssistantDTO
+ { enabled :: Bool
+ }
+ deriving (Generic, Eq, Show)
+
 data ClientConfigSignalBridgeDTO = ClientConfigSignalBridgeDTO
  { webSocketUrl :: Maybe String
  }

diff --git a/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigJM.hs b/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigJM.hs
@@ -63,6 +63,12 @@ instance FromJSON ClientConfigAdminDTO where
 instance ToJSON ClientConfigAdminDTO where
  toJSON = genericToJSON jsonOptions
 
+instance FromJSON ClientConfigAiAssistantDTO where
+ parseJSON = genericParseJSON jsonOptions
+
+instance ToJSON ClientConfigAiAssistantDTO where
+ toJSON = genericToJSON jsonOptions
+
 instance FromJSON ClientConfigSignalBridgeDTO where
  parseJSON = genericParseJSON jsonOptions
 

diff --git a/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigSM.hs b/wizard-server/src/Wizard/Api/Resource/Config/ClientConfigSM.hs
@@ -45,6 +45,9 @@ instance ToSchema ClientConfigLocaleDTO where
 instance ToSchema ClientConfigAdminDTO where
  declareNamedSchema = toSwagger (toClientConfigAdminDTO S.defaultAdmin defaultTenant)
 
+instance ToSchema ClientConfigAiAssistantDTO where
+ declareNamedSchema = toSwagger (toClientConfigAiAssistantDTO defaultTenant)
+
 instance ToSchema ClientConfigSignalBridgeDTO where
  declareNamedSchema = toSwagger (toClientConfigSignalBridgeDTO defaultTenant)
 

diff --git a/wizard-server/src/Wizard/Service/Config/Client/ClientConfigMapper.hs b/wizard-server/src/Wizard/Service/Config/Client/ClientConfigMapper.hs
@@ -1,6 +1,7 @@
 module Wizard.Service.Config.Client.ClientConfigMapper where
 
 import Data.Maybe
+import qualified Data.UUID as U
 
 import Shared.Common.Model.Config.ServerConfig
 import Shared.Common.Model.Config.SimpleFeature
@@ -27,6 +28,7 @@ toClientConfigDTO serverConfig tenantConfig mUserProfile tenant locales =
  , locales = fmap toClientConfigLocaleDTO locales
  , owl = tenantConfig.owl
  , admin = toClientConfigAdminDTO serverConfig.admin tenant
+ , aiAssistant = toClientConfigAiAssistantDTO tenant
  , signalBridge = toClientConfigSignalBridgeDTO tenant
  , modules =
  if serverConfig.admin.enabled
@@ -111,6 +113,12 @@ toClientConfigAdminDTO :: ServerConfigAdmin -> Tenant -> ClientConfigAdminDTO
 toClientConfigAdminDTO serverConfig tenant =
  ClientConfigAdminDTO {enabled = serverConfig.enabled, clientUrl = tenant.adminClientUrl}
 
+toClientConfigAiAssistantDTO :: Tenant -> ClientConfigAiAssistantDTO
+toClientConfigAiAssistantDTO tenant =
+ ClientConfigAiAssistantDTO
+ { enabled = not (U.toString tenant.uuid == "731ccc18-b833-4927-9458-2dea090e26cc" || U.toString tenant.uuid == "61a65ac9-e055-49eb-89ae-f4a4b6410208")
+ }
+
 toClientConfigSignalBridgeDTO :: Tenant -> ClientConfigSignalBridgeDTO
 toClientConfigSignalBridgeDTO tenant =
  ClientConfigSignalBridgeDTO {webSocketUrl = tenant.signalBridgeUrl}