Include additional headers for all supported Text Completion types.

This commit is contained in:
Cohee
2024-03-14 00:48:08 +02:00
parent acf36b6107
commit bd223486de
3 changed files with 19 additions and 2 deletions

View File

@ -124,10 +124,23 @@ function setAdditionalHeaders(request, args, server) {
headers = getKoboldCppHeaders();
break;
default:
headers = server ? getOverrideHeaders((new URL(server))?.host) : {};
headers = {};
break;
}
if (typeof server === 'string' && server.length > 0) {
try {
const url = new URL(server);
const overrideHeaders = getOverrideHeaders(url.host);
if (overrideHeaders && Object.keys(overrideHeaders).length > 0) {
Object.assign(headers, overrideHeaders);
}
} catch {
// Do nothing
}
}
Object.assign(args.headers, headers);
}