DESKTOP-I1T6TF3\Q
72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import type { EngineConfig, TtsConfig } from "@infiplot/types";
|
|
|
|
function readVar(name: string): string {
|
|
const v = process.env[name];
|
|
if (!v) throw new Error(`Missing required environment variable: ${name}`);
|
|
return v;
|
|
}
|
|
|
|
function readOptionalVar(name: string): string | undefined {
|
|
const v = process.env[name];
|
|
return v && v.length > 0 ? v : undefined;
|
|
}
|
|
|
|
function loadTtsConfig(): TtsConfig | undefined {
|
|
const baseUrl = readOptionalVar("TTS_BASE_URL");
|
|
const apiKey = readOptionalVar("TTS_API_KEY");
|
|
const speechModel = readOptionalVar("TTS_SPEECH_MODEL");
|
|
|
|
// Missing any → TTS disabled (game runs silently).
|
|
if (!baseUrl || !apiKey || !speechModel) return undefined;
|
|
|
|
return { baseUrl, apiKey, speechModel };
|
|
}
|
|
|
|
export function loadEngineConfig(headers?: Headers): EngineConfig {
|
|
const config: EngineConfig = {
|
|
text: {
|
|
baseUrl: readVar("TEXT_BASE_URL"),
|
|
apiKey: readVar("TEXT_API_KEY"),
|
|
model: readVar("TEXT_MODEL"),
|
|
},
|
|
image: {
|
|
baseUrl: readVar("IMAGE_BASE_URL"),
|
|
apiKey: readVar("IMAGE_API_KEY"),
|
|
model: readVar("IMAGE_MODEL"),
|
|
},
|
|
vision: {
|
|
baseUrl: readVar("VISION_BASE_URL"),
|
|
apiKey: readVar("VISION_API_KEY"),
|
|
model: readVar("VISION_MODEL"),
|
|
},
|
|
tts: loadTtsConfig(),
|
|
mockImage: readOptionalVar("MOCK_IMAGE") === "true",
|
|
};
|
|
|
|
const byoHeader = headers?.get("x-byo-api");
|
|
if (byoHeader) {
|
|
try {
|
|
const byo = JSON.parse(byoHeader);
|
|
if (byo.llm?.enabled) {
|
|
if (byo.llm.endpoint) config.text.baseUrl = byo.llm.endpoint;
|
|
if (byo.llm.apiKey) config.text.apiKey = byo.llm.apiKey;
|
|
if (byo.llm.model) config.text.model = byo.llm.model;
|
|
|
|
// Also override vision if llm is enabled
|
|
if (byo.llm.endpoint) config.vision.baseUrl = byo.llm.endpoint;
|
|
if (byo.llm.apiKey) config.vision.apiKey = byo.llm.apiKey;
|
|
if (byo.llm.model) config.vision.model = byo.llm.model;
|
|
}
|
|
if (byo.painter?.enabled) {
|
|
if (byo.painter.endpoint) config.image.baseUrl = byo.painter.endpoint;
|
|
if (byo.painter.apiKey) config.image.apiKey = byo.painter.apiKey;
|
|
if (byo.painter.model) config.image.model = byo.painter.model;
|
|
}
|
|
} catch (e) {
|
|
console.error("Failed to parse x-byo-api header in loadEngineConfig:", e);
|
|
}
|
|
}
|
|
|
|
return config;
|
|
}
|