diff --git a/src/additional-headers.js b/src/additional-headers.js index e77507b0e..5ac879411 100644 --- a/src/additional-headers.js +++ b/src/additional-headers.js @@ -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 { getConfigValue } = require('./util'); @@ -154,10 +154,9 @@ function getKoboldCppHeaders(directories) { */ function getFeatherlessHeaders(directories) { const apiKey = readSecret(directories, SECRET_KEYS.FEATHERLESS); + const baseHeaders = { ...FEATHERLESS_HEADERS }; - return apiKey ? ({ - 'Authorization': `Bearer ${apiKey}`, - }) : {}; + return apiKey ? Object.assign(baseHeaders, { 'Authorization': `Bearer ${apiKey}` }) : baseHeaders; } /** diff --git a/src/constants.js b/src/constants.js index 7968b4748..6fd64ae24 100644 --- a/src/constants.js +++ b/src/constants.js @@ -348,6 +348,11 @@ const OPENROUTER_HEADERS = { 'X-Title': 'SillyTavern', }; +const FEATHERLESS_HEADERS = { + 'HTTP-Referer': 'https://sillytavern.app', + 'X-Title': 'SillyTavern', +}; + const OPENROUTER_KEYS = [ 'max_tokens', 'temperature', @@ -432,4 +437,5 @@ module.exports = { OPENROUTER_KEYS, VLLM_KEYS, FEATHERLESS_KEYS, + FEATHERLESS_HEADERS, };