Pass app headers to featherless

9cd59d2e4b8eca509f8635f691b15c51f1eaaf9b

Cohee <18619528+Cohee1207@users.noreply.github.com>

2 files changed, +9 -4Ignore whitespace
src/additional-headers.js+3 -4
@@ -1,4 +1,4 @@
11const { TEXTGEN_TYPES, OPENROUTER_HEADERS, FEATHERLESS_HEADERS } = require('./constants');
22const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
33const { getConfigValue } = require('./util');
44
@@ -154,10 +154,9 @@ function getKoboldCppHeaders(directories) {
154154 */
155155function getFeatherlessHeaders(directories) {
156156 const apiKey = readSecret(directories, SECRET_KEYS.FEATHERLESS);
157+ const baseHeaders = { ...FEATHERLESS_HEADERS };
157158
158- return apiKey ? ({
159+ return apiKey ? Object.assign(baseHeaders, { 'Authorization': `Bearer ${apiKey}` }) : baseHeaders;
159- 'Authorization': `Bearer ${apiKey}`,
160- }) : {};
161160}
162161
163162/**
src/constants.js+6 -0
@@ -348,6 +348,11 @@ const OPENROUTER_HEADERS = {
348348 'X-Title': 'SillyTavern',
349349};
350350
351+const FEATHERLESS_HEADERS = {
352+ 'HTTP-Referer': 'https://sillytavern.app',
353+ 'X-Title': 'SillyTavern',
354+};
355+
351356const OPENROUTER_KEYS = [
352357 'max_tokens',
353358 'temperature',
@@ -432,4 +437,5 @@ module.exports = {
432437 OPENROUTER_KEYS,
433438 VLLM_KEYS,
434439 FEATHERLESS_KEYS,
440+ FEATHERLESS_HEADERS,
435441};