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 @@
1const { TEXTGEN_TYPES, OPENROUTER_HEADERS } = require('./constants');1const { TEXTGEN_TYPES, OPENROUTER_HEADERS, FEATHERLESS_HEADERS } = require('./constants');
2const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');2const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
3const { getConfigValue } = require('./util');3const { getConfigValue } = require('./util');
44
@@ -154,10 +154,9 @@ function getKoboldCppHeaders(directories) {
154 */154 */
155function getFeatherlessHeaders(directories) {155function getFeatherlessHeaders(directories) {
156 const apiKey = readSecret(directories, SECRET_KEYS.FEATHERLESS);156 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 }) : {};
161}160}
162161
163/**162/**
src/constants.js+6 -0
@@ -348,6 +348,11 @@ const OPENROUTER_HEADERS = {
348 'X-Title': 'SillyTavern',348 'X-Title': 'SillyTavern',
349};349};
350350
351const FEATHERLESS_HEADERS = {
352 'HTTP-Referer': 'https://sillytavern.app',
353 'X-Title': 'SillyTavern',
354};
355
351const OPENROUTER_KEYS = [356const OPENROUTER_KEYS = [
352 'max_tokens',357 'max_tokens',
353 'temperature',358 'temperature',
@@ -432,4 +437,5 @@ module.exports = {
432 OPENROUTER_KEYS,437 OPENROUTER_KEYS,
433 VLLM_KEYS,438 VLLM_KEYS,
434 FEATHERLESS_KEYS,439 FEATHERLESS_KEYS,
440 FEATHERLESS_HEADERS,
435};441};