From f328ae4bf9dff216e8a23471cc60b074f0e41e0d Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martinmckeaveney@gmail.com>
Date: Mon, 2 Sep 2024 20:45:32 +0100
Subject: [PATCH] account for arrays

---
 packages/backend-core/src/configs/configs.ts  |  3 +-
 .../portal/settings/ai/ConfigModal.svelte     | 49 ++++++++++++-------
 .../worker/src/api/routes/global/configs.ts   | 20 +++++++-
 3 files changed, 51 insertions(+), 21 deletions(-)

diff --git a/packages/backend-core/src/configs/configs.ts b/packages/backend-core/src/configs/configs.ts
index 6062d3624f..1771716f57 100644
--- a/packages/backend-core/src/configs/configs.ts
+++ b/packages/backend-core/src/configs/configs.ts
@@ -1,5 +1,6 @@
 import {
-  AIConfig, AIInnerConfig,
+  AIConfig,
+  AIInnerConfig,
   Config,
   ConfigType,
   GoogleConfig,
diff --git a/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte b/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte
index e9ec5159f9..749cebeb2c 100644
--- a/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte
+++ b/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte
@@ -23,37 +23,48 @@
   const models = [
     "gpt4o-mini",
   ]
+  
+  const defaultConfig = {
+    active: false,
+    isDefault: false,
+  }
 
-  let formValues = {}
-
+  let aiConfig = defaultConfig
   let validation
 
   $: {
-    const { provider, model, name, apiKey } = formValues
-    validation = provider && model && name && apiKey
+    const { provider, defaultModel, name, apiKey } = aiConfig
+    validation = provider && defaultModel && name && apiKey
   }
 
   function prefillConfig(evt) {
     const provider = evt.detail
     // grab the preset config from the constants for that provider and fill it in
     if (ConfigMap[provider]) {
-      formValues = {
-        ...formValues,
+      aiConfig = {
+        ...aiConfig,
         ...ConfigMap[provider]
       }
     } else {
-      formValues = {
+      aiConfig = {
+        ...aiConfig,
         provider
       }
     }
   }
 
   async function saveConfig() {
-    formValues.type = "ai"
+    const config = {
+      type: "ai",
+      config: [
+        // TODO: include the ones that are already there, or just handle this in the backend
+        aiConfig,
+      ]
+    }
     try {
-      const savedConfig = await API.saveConfig(formValues)
-      formValues._rev = savedConfig._rev
-      formValues._id = savedConfig._id
+      const savedConfig = await API.saveConfig(config)
+      aiConfig._rev = savedConfig._rev
+      aiConfig._id = savedConfig._id
       notifications.success(`Configuration saved`)
     } catch (error) {
       notifications.error(
@@ -92,33 +103,33 @@
     <Label size="M">Provider</Label>
     <Select
       placeholder={null}
-      bind:value={formValues.provider}
+      bind:value={aiConfig.provider}
       options={providers}
       on:change={prefillConfig}
     />
   </div>
   <div class="form-row">
-    <Label size="M">Model</Label>
+    <Label size="M">Default Model</Label>
     <Select
       placeholder={null}
-      bind:value={formValues.model}
+      bind:value={aiConfig.defaultModel}
       options={models}
     />
   </div>
   <div class="form-row">
     <Label size="M">Name</Label>
-    <Input placeholder={"Test 1"} bind:value={formValues.name}/>
+    <Input placeholder={"Test 1"} bind:value={aiConfig.name}/>
   </div>
   <div class="form-row">
     <Label size="M">Base URL</Label>
-    <Input placeholder={"www.google.com"} bind:value={formValues.baseUrl}/>
+    <Input placeholder={"www.google.com"} bind:value={aiConfig.baseUrl}/>
   </div>
   <div class="form-row">
     <Label size="M">API Key</Label>
-    <Input bind:value={formValues.apiKey}/>
+    <Input bind:value={aiConfig.apiKey}/>
   </div>
-  <Toggle text="Active" bind:value={formValues.active}/>
-  <Toggle text="Set as default" bind:value={formValues.isDefault}/>
+  <Toggle text="Active" bind:value={aiConfig.active}/>
+  <Toggle text="Set as default" bind:value={aiConfig.isDefault}/>
 </ModalContent>
 
 <style>
diff --git a/packages/worker/src/api/routes/global/configs.ts b/packages/worker/src/api/routes/global/configs.ts
index fcab2b58e0..b857647652 100644
--- a/packages/worker/src/api/routes/global/configs.ts
+++ b/packages/worker/src/api/routes/global/configs.ts
@@ -65,6 +65,23 @@ function scimValidation() {
   }).unknown(true)
 }
 
+function aiValidation() {
+  // prettier-ignore
+  return Joi.array().items(
+    Joi.object({
+      provider: Joi.string().required(),
+      isDefault: Joi.boolean().required(),
+      name: Joi.string().required(),
+      active: Joi.boolean().required(),
+      baseUrl: Joi.string().optional(),
+      apiKey: Joi.string().required(),
+      // TODO: should be enum
+      defaultModel: Joi.string().optional(),
+
+    })
+  ).required()
+}
+
 function buildConfigSaveValidation() {
   // prettier-ignore
   return auth.joiValidator.body(Joi.object({
@@ -82,7 +99,8 @@ function buildConfigSaveValidation() {
           { is: ConfigType.ACCOUNT, then: Joi.object().unknown(true) },
           { is: ConfigType.GOOGLE, then: googleValidation() },
           { is: ConfigType.OIDC, then: oidcValidation() },
-          { is: ConfigType.SCIM, then: scimValidation() }
+          { is: ConfigType.SCIM, then: scimValidation() },
+          { is: ConfigType.AI, then: aiValidation() }
         ],
       }),
   }).required().unknown(true),