84 lines
2.1 KiB
JavaScript
84 lines
2.1 KiB
JavaScript
const { TEXTGEN_TYPES } = require('./constants');
|
|
const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
|
|
const { getConfigValue } = require('./util');
|
|
|
|
function getMancerHeaders() {
|
|
const apiKey = readSecret(SECRET_KEYS.MANCER);
|
|
|
|
return apiKey ? ({
|
|
'X-API-KEY': apiKey,
|
|
'Authorization': `Bearer ${apiKey}`,
|
|
}) : {};
|
|
}
|
|
|
|
function getTogetherAIHeaders() {
|
|
const apiKey = readSecret(SECRET_KEYS.TOGETHERAI);
|
|
|
|
return apiKey ? ({
|
|
'Authorization': `Bearer ${apiKey}`,
|
|
}) : {};
|
|
}
|
|
|
|
function getAphroditeHeaders() {
|
|
const apiKey = readSecret(SECRET_KEYS.APHRODITE);
|
|
|
|
return apiKey ? ({
|
|
'X-API-KEY': apiKey,
|
|
'Authorization': `Bearer ${apiKey}`,
|
|
}) : {};
|
|
}
|
|
|
|
function getTabbyHeaders() {
|
|
const apiKey = readSecret(SECRET_KEYS.TABBY);
|
|
|
|
return apiKey ? ({
|
|
'x-api-key': apiKey,
|
|
'Authorization': `Bearer ${apiKey}`,
|
|
}) : {};
|
|
}
|
|
|
|
function getOverrideHeaders(urlHost) {
|
|
const requestOverrides = getConfigValue('requestOverrides', []);
|
|
const overrideHeaders = requestOverrides?.find((e) => e.hosts?.includes(urlHost))?.headers;
|
|
if (overrideHeaders && urlHost) {
|
|
return overrideHeaders;
|
|
} else {
|
|
return {};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Sets additional headers for the request.
|
|
* @param {object} request Original request body
|
|
* @param {object} args New request arguments
|
|
* @param {string|null} server API server for new request
|
|
*/
|
|
function setAdditionalHeaders(request, args, server) {
|
|
let headers;
|
|
|
|
switch (request.body.api_type) {
|
|
case TEXTGEN_TYPES.MANCER:
|
|
headers = getMancerHeaders();
|
|
break;
|
|
case TEXTGEN_TYPES.APHRODITE:
|
|
headers = getAphroditeHeaders();
|
|
break;
|
|
case TEXTGEN_TYPES.TABBY:
|
|
headers = getTabbyHeaders();
|
|
break;
|
|
case TEXTGEN_TYPES.TOGETHERAI:
|
|
headers = getTogetherAIHeaders();
|
|
break;
|
|
default:
|
|
headers = server ? getOverrideHeaders((new URL(server))?.host) : {};
|
|
break;
|
|
}
|
|
|
|
Object.assign(args.headers, headers);
|
|
}
|
|
|
|
module.exports = {
|
|
getOverrideHeaders,
|
|
setAdditionalHeaders,
|
|
};
|