fde5b5e318
- Add database initialization scripts - Add configuration files - Add documentation - Add public assets - Add source code structure - Update README
92 lines
2.3 KiB
TypeScript
92 lines
2.3 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
import toml from '@iarna/toml';
|
|
|
|
const configFileName = 'config.toml';
|
|
|
|
interface Config {
|
|
GENERAL: {
|
|
PORT: number;
|
|
SIMILARITY_MEASURE: string;
|
|
KEEP_ALIVE: string;
|
|
};
|
|
API_KEYS: {
|
|
OPENAI: string;
|
|
GROQ: string;
|
|
ANTHROPIC: string;
|
|
GEMINI: 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}`), 'utf-8'),
|
|
) as any as Config;
|
|
|
|
export const getPort = () => loadConfig().GENERAL.PORT;
|
|
|
|
export const getSimilarityMeasure = () =>
|
|
loadConfig().GENERAL.SIMILARITY_MEASURE;
|
|
|
|
export const getKeepAlive = () => loadConfig().GENERAL.KEEP_ALIVE;
|
|
|
|
export const getOpenaiApiKey = () => loadConfig().API_KEYS.OPENAI;
|
|
|
|
export const getGroqApiKey = () => loadConfig().API_KEYS.GROQ;
|
|
|
|
export const getAnthropicApiKey = () => loadConfig().API_KEYS.ANTHROPIC;
|
|
|
|
export const getGeminiApiKey = () => loadConfig().API_KEYS.GEMINI;
|
|
|
|
export const getSearxngApiEndpoint = () =>
|
|
process.env.SEARXNG_API_URL || 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),
|
|
);
|
|
};
|
|
|
|
export const config = {
|
|
ollama: {
|
|
url: process.env.OLLAMA_URL || 'http://localhost:11434',
|
|
model: process.env.OLLAMA_MODEL || 'mistral',
|
|
options: {
|
|
temperature: 0.1,
|
|
top_p: 0.9,
|
|
timeout: 30000 // 30 seconds timeout
|
|
}
|
|
},
|
|
// ... other config
|
|
};
|