From be4b6aaceb47a1b6048e9c6b4cd9edf007b090d6 Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martinmckeaveney@gmail.com>
Date: Tue, 3 Sep 2024 19:19:33 +0100
Subject: [PATCH] adding config for azure openai and more customisation for
 custom ai models

---
 .../builder/portal/settings/ai/ConfigModal.svelte  | 14 +++++++++-----
 .../pages/builder/portal/settings/ai/constants.ts  | 11 +++++++++++
 2 files changed, 20 insertions(+), 5 deletions(-)

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 747a070c6b..cac8f74786 100644
--- a/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte
+++ b/packages/builder/src/pages/builder/portal/settings/ai/ConfigModal.svelte
@@ -63,11 +63,15 @@
   </div>
   <div class="form-row">
     <Label size="M">Default Model</Label>
-    <Select
-      placeholder={config.provider ? "Choose an option" : "Select a provider first"}
-      bind:value={config.defaultModel}
-      options={config.provider ? Providers[config.provider].models : []}
-    />
+    {#if config.provider !== Providers.Custom.name}
+      <Select
+        placeholder={config.provider ? "Choose an option" : "Select a provider first"}
+        bind:value={config.defaultModel}
+        options={config.provider ? Providers[config.provider].models : []}
+      />
+    {:else}
+      <Input bind:value={config.defaultModel} />
+    {/if}
   </div>
   <div class="form-row">
     <Label size="M">Base URL</Label>
diff --git a/packages/builder/src/pages/builder/portal/settings/ai/constants.ts b/packages/builder/src/pages/builder/portal/settings/ai/constants.ts
index 9c82f714a4..d0db0d0d9b 100644
--- a/packages/builder/src/pages/builder/portal/settings/ai/constants.ts
+++ b/packages/builder/src/pages/builder/portal/settings/ai/constants.ts
@@ -22,6 +22,17 @@ export const Providers = {
   TogetherAI: {
     name: "Together AI",
     // TODO: too many - probably need to use an autocomplete for this
+    models: [
+      "gpt-4o-mini",
+      "gpt-4o",
+      "gpt-3.5-turbo",
+      "chatgpt-4o-latest",
+      "gpt-4-turbo",
+      "gpt-4",
+    ]
+  },
+  AzureOpenAI: {
+    name: "Azure Open AI",
     models: ["whatever"]
   },
   Custom: {