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

Pass app headers to featherless

Browse Source
This commit is contained in:
Cohee
2024-08-01 23:03:32 +03:00
parent 8660c96412
commit 9cd59d2e4b
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/additional-headers.js
View File

@ -1,4 +1,4 @@
const { TEXTGEN_TYPES, OPENROUTER_HEADERS } = require('./constants'); const { TEXTGEN_TYPES, OPENROUTER_HEADERS, FEATHERLESS_HEADERS } = require('./constants');
const { SECRET_KEYS, readSecret } = require('./endpoints/secrets'); const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
const { getConfigValue } = require('./util'); const { getConfigValue } = require('./util');
@ -154,10 +154,9 @@ function getKoboldCppHeaders(directories) {
*/ */
function getFeatherlessHeaders(directories) { function getFeatherlessHeaders(directories) {
const apiKey = readSecret(directories, SECRET_KEYS.FEATHERLESS); const apiKey = readSecret(directories, SECRET_KEYS.FEATHERLESS);
const baseHeaders = { ...FEATHERLESS_HEADERS };
return apiKey ? ({ return apiKey ? Object.assign(baseHeaders, { 'Authorization': `Bearer ${apiKey}` }) : baseHeaders;
'Authorization': `Bearer ${apiKey}`,
}) : {};
} }
/** /**

6
src/constants.js
View File

@ -348,6 +348,11 @@ const OPENROUTER_HEADERS = {
'X-Title': 'SillyTavern', 'X-Title': 'SillyTavern',
}; };
const FEATHERLESS_HEADERS = {
'HTTP-Referer': 'https://sillytavern.app',
'X-Title': 'SillyTavern',
};
const OPENROUTER_KEYS = [ const OPENROUTER_KEYS = [
'max_tokens', 'max_tokens',
'temperature', 'temperature',
@ -432,4 +437,5 @@ module.exports = {
OPENROUTER_KEYS, OPENROUTER_KEYS,
VLLM_KEYS, VLLM_KEYS,
FEATHERLESS_KEYS, FEATHERLESS_KEYS,
FEATHERLESS_HEADERS,
}; };