From d54af7fa3131fb0cdfd4c0ce70308fcc7efb001f Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Tue, 9 Apr 2024 14:55:27 +0800
Subject: [PATCH] updated swagger json link fix

---
 colab.ipynb  | 2 +-
 klite.embd   | 5 +++--
 koboldcpp.py | 2 +-
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/colab.ipynb b/colab.ipynb
index cc9d608a1..e29081a7b 100644
--- a/colab.ipynb
+++ b/colab.ipynb
@@ -58,7 +58,7 @@
         "VCommand = \"\"\r\n",
         "#@markdown <hr>\r\n",
         "LoadImgModel = False #@param {type:\"boolean\"}\r\n",
-        "ImgModel = \"https://huggingface.co/koboldcpp/imgmodel/resolve/main/imgmodel_older_q4_0.gguf\" #@param [\"https://huggingface.co/koboldcpp/imgmodel/resolve/main/imgmodel_older_q4_0.gguf\"]{allow-input: true}\r\n",
+        "ImgModel = \"https://huggingface.co/koboldcpp/imgmodel/resolve/main/imgmodel_ftuned_q4_0.gguf\" #@param [\"https://huggingface.co/koboldcpp/imgmodel/resolve/main/imgmodel_ftuned_q4_0.gguf\"]{allow-input: true}\r\n",
         "SCommand = \"\"\r\n",
         "\r\n",
         "import os\r\n",
diff --git a/klite.embd b/klite.embd
index c33ec11c6..352a71625 100644
--- a/klite.embd
+++ b/klite.embd
@@ -7,7 +7,7 @@ Just copy this single static HTML file anywhere and open it in a browser, or fro
 Please go to https://github.com/LostRuins/lite.koboldai.net for updates on Kobold Lite.
 If you are submitting a pull request for Lite, PLEASE use the above repo, not the KoboldCpp one.
 Kobold Lite is under the AGPL v3.0 License unless otherwise exempted. Please do not remove this line.
-Current version: 128
+Current version: 129
 -Concedo
 -->
 
@@ -10343,7 +10343,7 @@ Current version: 128
 						}
 					];
 				}
-				else if(mdlname=="gemini-pro" || mdlname=="gemini-1.5-pro-latest")
+				else if(mdlname=="gemini-pro" || mdlname=="gemini-1.5-pro-latest" || mdlname=="gemini-ultra")
 				{
 					if(localsettings.opmode==1)
 					{
@@ -14036,6 +14036,7 @@ Current version: 128
 					<option value="text-bison-001" selected="selected">text-bison-001</option>
 					<option value="gemini-pro">gemini-pro</option>
 					<option value="gemini-1.5-pro-latest">gemini-1.5-pro-latest</option>
+					<option value="gemini-ultra">gemini-ultra</option>
 				</select>
 				<span class="color_green" style="font-weight: bold;">Please input Gemini or PaLM API Key.</span><br><br>
 				<input class="form-control" type="password" id="custom_palm_key" placeholder="PaLM/Gemini API Key" value="" onfocus="focus_api_keys()" onblur="blur_api_keys()"><br>
diff --git a/koboldcpp.py b/koboldcpp.py
index 4108c3a5f..5754ef662 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -1115,7 +1115,7 @@ Enter Prompt:<br>
            response_body = (json.dumps([]).encode())
 
 
-        elif self.path=="/api" or self.path=="/docs":
+        elif self.path=="/api" or self.path=="/docs" or self.path.startswith(('/api/?json=','/api?json=','/docs/?json=','/docs?json=')):
             content_type = 'text/html'
             if self.embedded_kcpp_docs is None:
                 response_body = (f"KoboldCpp API is running!\n\nAPI usage reference can be found at the wiki: https://github.com/LostRuins/koboldcpp/wiki").encode()