diff --git a/public/index.html b/public/index.html
index 3d317cb69..85a44a8e7 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1916,6 +1916,15 @@
Make sure you run it with --api
flag
+
API key
+
+
+
+
+
+ For privacy reasons, your API key will be hidden after you reload the page.
+
Server URL
Example: http://127.0.0.1:5000
diff --git a/public/script.js b/public/script.js
index de1172b7a..3b28344dc 100644
--- a/public/script.js
+++ b/public/script.js
@@ -8541,6 +8541,11 @@ jQuery(async function () {
await writeSecret(SECRET_KEYS.TOGETHERAI, togetherKey);
}
+ const oobaKey = String($('#api_key_ooba').val()).trim();
+ if (oobaKey.length) {
+ await writeSecret(SECRET_KEYS.OOBA, oobaKey);
+ }
+
validateTextGenUrl();
startStatusLoading();
main_api = 'textgenerationwebui';
diff --git a/public/scripts/secrets.js b/public/scripts/secrets.js
index 1c1154180..2e96136d3 100644
--- a/public/scripts/secrets.js
+++ b/public/scripts/secrets.js
@@ -17,6 +17,7 @@ export const SECRET_KEYS = {
MISTRALAI: 'api_key_mistralai',
TOGETHERAI: 'api_key_togetherai',
CUSTOM: 'api_key_custom',
+ OOBA: 'api_key_ooba'
};
const INPUT_MAP = {
@@ -35,6 +36,7 @@ const INPUT_MAP = {
[SECRET_KEYS.MISTRALAI]: '#api_key_mistralai',
[SECRET_KEYS.CUSTOM]: '#api_key_custom',
[SECRET_KEYS.TOGETHERAI]: '#api_key_togetherai',
+ [SECRET_KEYS.OOBA]: '#api_key_ooba',
};
async function clearSecret() {
diff --git a/src/additional-headers.js b/src/additional-headers.js
index de8b0d91a..bab8e900d 100644
--- a/src/additional-headers.js
+++ b/src/additional-headers.js
@@ -37,6 +37,15 @@ function getTabbyHeaders() {
}) : {};
}
+function getOobaHeaders() {
+ const apiKey = readSecret(SECRET_KEYS.OOBA);
+ console.log(apiKey);
+
+ return apiKey ? ({
+ 'Authorization': `Bearer ${apiKey}`,
+ }) : {};
+}
+
function getOverrideHeaders(urlHost) {
const requestOverrides = getConfigValue('requestOverrides', []);
const overrideHeaders = requestOverrides?.find((e) => e.hosts?.includes(urlHost))?.headers;
@@ -69,6 +78,9 @@ function setAdditionalHeaders(request, args, server) {
case TEXTGEN_TYPES.TOGETHERAI:
headers = getTogetherAIHeaders();
break;
+ case TEXTGEN_TYPES.OOBA:
+ headers = getOobaHeaders();
+ break;
default:
headers = server ? getOverrideHeaders((new URL(server))?.host) : {};
break;
diff --git a/src/endpoints/secrets.js b/src/endpoints/secrets.js
index d7b44ed1e..3236dc3a9 100644
--- a/src/endpoints/secrets.js
+++ b/src/endpoints/secrets.js
@@ -29,6 +29,7 @@ const SECRET_KEYS = {
TOGETHERAI: 'api_key_togetherai',
MISTRALAI: 'api_key_mistralai',
CUSTOM: 'api_key_custom',
+ OOBA: 'api_key_ooba'
};
/**