rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add API key

Browse Source
This commit is contained in:
oobabooga
2024-02-06 20:00:16 -08:00
parent 2815990589
commit 21fb143718
5 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/additional-headers.js
View File

@ -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;