60 lines
1.7 KiB
TypeScript
60 lines
1.7 KiB
TypeScript
/* eslint-disable unicorn/prefer-module */
|
|
import fs from "node:fs";
|
|
import path from "node:path";
|
|
import toml from "@iarna/toml";
|
|
|
|
const configFileName = "config.toml";
|
|
|
|
interface Config {
|
|
GENERAL: {
|
|
PORT: number;
|
|
SIMILARITY_MEASURE: string;
|
|
};
|
|
API_KEYS: {
|
|
OPENAI: string;
|
|
GROQ: string;
|
|
};
|
|
API_ENDPOINTS: {
|
|
SEARXNG: string;
|
|
OLLAMA: string;
|
|
};
|
|
}
|
|
|
|
type RecursivePartial<T> = {
|
|
[P in keyof T]?: RecursivePartial<T[P]>;
|
|
};
|
|
|
|
const loadConfig = () =>
|
|
toml.parse(fs.readFileSync(path.join(__dirname, `../${configFileName}`), "utf8")) as unknown as Config;
|
|
|
|
export const getPort = () => loadConfig().GENERAL.PORT;
|
|
|
|
export const getSimilarityMeasure = () => loadConfig().GENERAL.SIMILARITY_MEASURE;
|
|
|
|
export const getOpenaiApiKey = () => loadConfig().API_KEYS.OPENAI;
|
|
|
|
export const getGroqApiKey = () => loadConfig().API_KEYS.GROQ;
|
|
|
|
export const getSearxngApiEndpoint = () => loadConfig().API_ENDPOINTS.SEARXNG;
|
|
|
|
export const getOllamaApiEndpoint = () => loadConfig().API_ENDPOINTS.OLLAMA;
|
|
|
|
export const updateConfig = (config: RecursivePartial<Config>) => {
|
|
const currentConfig = loadConfig();
|
|
|
|
for (const key in currentConfig) {
|
|
if (!config[key]) config[key] = {};
|
|
|
|
if (typeof currentConfig[key] === "object" && currentConfig[key] !== null) {
|
|
for (const nestedKey in currentConfig[key]) {
|
|
if (!config[key][nestedKey] && currentConfig[key][nestedKey] && config[key][nestedKey] !== "") {
|
|
config[key][nestedKey] = currentConfig[key][nestedKey];
|
|
}
|
|
}
|
|
} else if (currentConfig[key] && config[key] !== "") {
|
|
config[key] = currentConfig[key];
|
|
}
|
|
}
|
|
|
|
fs.writeFileSync(path.join(__dirname, `../${configFileName}`), toml.stringify(config));
|
|
};
|