From 4f1dbaa34b7b2d6e86b3914ba778d9e703a2315b Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Wed, 8 Nov 2023 17:09:33 +0200
Subject: [PATCH] Remove /v1 from the ooba API URLs

---
 server.js | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/server.js b/server.js
index 02aa7a1bd..676731425 100644
--- a/server.js
+++ b/server.js
@@ -496,8 +496,8 @@ app.post("/api/textgenerationwebui/status", jsonParser, async function (request,
 
         console.log('Trying to connect to API:', request.body);
 
-        // Convert to string + remove trailing slash
-        const baseUrl = String(request.body.api_server).replace(/\/$/, '');
+        // Convert to string + remove trailing slash + /v1 suffix
+        const baseUrl = String(request.body.api_server).replace(/\/$/, '').replace(/\/v1$/, '');
 
         const args = {
             headers: { "Content-Type": "application/json" },
@@ -567,8 +567,8 @@ app.post("/api/textgenerationwebui/generate", jsonParser, async function (reques
             controller.abort();
         });
 
-        // Convert to string + remove trailing slash
-        let url = String(baseUrl).replace(/\/$/, '');
+        // Convert to string + remove trailing slash + /v1 suffix
+        let url = String(baseUrl).replace(/\/$/, '').replace(/\/v1$/, '');
 
         if (request.body.legacy_api) {
             url += "/v1/generate";
@@ -3356,7 +3356,8 @@ app.post("/tokenize_via_api", jsonParser, async function (request, response) {
 
             setAdditionalHeaders(request, args, null);
 
-            let url = String(baseUrl).replace(/\/$/, '');
+            // Convert to string + remove trailing slash + /v1 suffix
+            let url = String(baseUrl).replace(/\/$/, '').replace(/\/v1$/, '');
 
             if (legacyApi) {
                 url += '/v1/token-count';