wagesj45/Sortana
wagesj45/Sortana
1
0
Fork 0
Code Issues Pull requests Projects Releases 7 Packages Wiki Activity Actions

Add optional OpenAI auth headers

Browse source
This commit is contained in:
Jordan Wages 2026-01-30 02:54:19 -06:00
commit 1680ad6c30
8 changed files with 112 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

28
modules/AiClassifier.js
View file

@ -40,6 +40,9 @@ let gTemplateText = "";
let gAiParams = Object.assign({}, DEFAULT_AI_PARAMS);
let gModel = "";
let gApiKey = "";
let gOpenaiOrganization = "";
let gOpenaiProject = "";
let gCache = new Map();
let gCacheLoaded = false;
@ -223,6 +226,15 @@ async function setConfig(config = {}) {
if (typeof config.model === "string") {
gModel = config.model.trim();
}
if (typeof config.apiKey === "string") {
gApiKey = config.apiKey.trim();
}
if (typeof config.openaiOrganization === "string") {
gOpenaiOrganization = config.openaiOrganization.trim();
}
if (typeof config.openaiProject === "string") {
gOpenaiProject = config.openaiProject.trim();
}
if (typeof config.debugLogging === "boolean") {
setDebug(config.debugLogging);
}
@ -241,6 +253,20 @@ async function setConfig(config = {}) {
aiLog(`[AiClassifier] Template set to ${gTemplateName}`, {debug: true});
}
function buildAuthHeaders() {
const headers = {};
if (gApiKey) {
headers.Authorization = `Bearer ${gApiKey}`;
}
if (gOpenaiOrganization) {
headers["OpenAI-Organization"] = gOpenaiOrganization;
}
if (gOpenaiProject) {
headers["OpenAI-Project"] = gOpenaiProject;
}
return headers;
}
function buildSystemPrompt() {
return SYSTEM_PREFIX + (gCustomSystemPrompt || DEFAULT_CUSTOM_SYSTEM_PROMPT) + SYSTEM_SUFFIX;
}
@ -453,7 +479,7 @@ async function classifyText(text, criterion, cacheKey = null) {
try {
const response = await fetch(gEndpoint, {
method: "POST",
headers: { "Content-Type": "application/json" },
headers: { "Content-Type": "application/json", ...buildAuthHeaders() },
body: payload,
});