From c69724e1da38abb01447c78a5984d2fd3cbf5078 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Tue, 2 Jan 2024 10:28:34 +0200
Subject: [PATCH] Fix GUI Kobold

---
 public/script.js               | 3 ++-
 public/scripts/kai-settings.js | 2 --
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/public/script.js b/public/script.js
index d80bb98d8..a437e6423 100644
--- a/public/script.js
+++ b/public/script.js
@@ -2891,7 +2891,7 @@ export async function generateRaw(prompt, api, instructOverride) {
         case 'kobold':
         case 'koboldhorde':
             if (preset_settings === 'gui') {
-                generateData = { prompt: prompt, gui_settings: true, max_length: amount_gen, max_context_length: max_context };
+                generateData = { prompt: prompt, gui_settings: true, max_length: amount_gen, max_context_length: max_context, api_server };
             } else {
                 const isHorde = api === 'koboldhorde';
                 const koboldSettings = koboldai_settings[koboldai_setting_names[preset_settings]];
@@ -3724,6 +3724,7 @@ async function Generate(type, { automatic_trigger, force_name2, quiet_prompt, qu
             gui_settings: true,
             max_length: maxLength,
             max_context_length: max_context,
+            api_server,
         };
 
         if (preset_settings != 'gui') {
diff --git a/public/scripts/kai-settings.js b/public/scripts/kai-settings.js
index d88a6ee87..f2f295dc3 100644
--- a/public/scripts/kai-settings.js
+++ b/public/scripts/kai-settings.js
@@ -4,7 +4,6 @@ import {
     getStoppingStrings,
     substituteParams,
     api_server,
-    main_api,
 } from '../script.js';
 
 import {
@@ -142,7 +141,6 @@ export function getKoboldGenerationData(finalPrompt, settings, maxLength, maxCon
         sampler_seed: kai_settings.seed >= 0 ? kai_settings.seed : undefined,
 
         api_server,
-        main_api,
     };
     return generate_data;
 }