remi/Perplexica
1
0
Fork 0
mirror of https://github.com/ItzCrazyKns/Perplexica.git synced 2025-09-20 16:21:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add optional env var system

Browse Source
This commit is contained in:
chand1012
2025-03-20 21:55:14 -04:00
parent 45df9dc5bf
commit 8e413796d6
1 changed files with 41 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
src/lib/config.ts
View File

@@ -45,32 +45,62 @@ const loadConfig = () =>
fs.readFileSync(path.join(process.cwd(), `${configFileName}`), 'utf-8'), fs.readFileSync(path.join(process.cwd(), `${configFileName}`), 'utf-8'),
) as any as Config; ) as any as Config;
const getEnvVar = (key: string): string | undefined => {
return process.env[key];
};
const getConfigValue = (path: string[], defaultValue: string): string => {
// Convert path to environment variable name (e.g., ['MODELS', 'GROQ', 'API_KEY'] -> 'GROQ_API_KEY')
const envKey = path.slice(1).join('_').toUpperCase();
const envValue = getEnvVar(envKey);
if (envValue !== undefined) {
return envValue;
}
// Fall back to config.toml
let value: any = loadConfig();
for (const key of path) {
value = value[key];
if (value === undefined) {
return defaultValue;
}
}
return value;
};
export const getSimilarityMeasure = () => export const getSimilarityMeasure = () =>
loadConfig().GENERAL.SIMILARITY_MEASURE; getConfigValue(['GENERAL', 'SIMILARITY_MEASURE'], 'cosine');
export const getKeepAlive = () => loadConfig().GENERAL.KEEP_ALIVE; export const getKeepAlive = () =>
getConfigValue(['GENERAL', 'KEEP_ALIVE'], '30s');
export const getOpenaiApiKey = () => loadConfig().MODELS.OPENAI.API_KEY; export const getOpenaiApiKey = () =>
getConfigValue(['MODELS', 'OPENAI', 'API_KEY'], '');
export const getGroqApiKey = () => loadConfig().MODELS.GROQ.API_KEY; export const getGroqApiKey = () =>
getConfigValue(['MODELS', 'GROQ', 'API_KEY'], '');
export const getAnthropicApiKey = () => loadConfig().MODELS.ANTHROPIC.API_KEY; export const getAnthropicApiKey = () =>
getConfigValue(['MODELS', 'ANTHROPIC', 'API_KEY'], '');
export const getGeminiApiKey = () => loadConfig().MODELS.GEMINI.API_KEY; export const getGeminiApiKey = () =>
getConfigValue(['MODELS', 'GEMINI', 'API_KEY'], '');
export const getSearxngApiEndpoint = () => export const getSearxngApiEndpoint = () =>
process.env.SEARXNG_API_URL || loadConfig().API_ENDPOINTS.SEARXNG; getConfigValue(['API_ENDPOINTS', 'SEARXNG'], '');
export const getOllamaApiEndpoint = () => loadConfig().MODELS.OLLAMA.API_URL; export const getOllamaApiEndpoint = () =>
getConfigValue(['MODELS', 'OLLAMA', 'API_URL'], 'http://localhost:11434');
export const getCustomOpenaiApiKey = () => export const getCustomOpenaiApiKey = () =>
loadConfig().MODELS.CUSTOM_OPENAI.API_KEY; getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'API_KEY'], '');
export const getCustomOpenaiApiUrl = () => export const getCustomOpenaiApiUrl = () =>
loadConfig().MODELS.CUSTOM_OPENAI.API_URL; getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'API_URL'], '');
export const getCustomOpenaiModelName = () => export const getCustomOpenaiModelName = () =>
loadConfig().MODELS.CUSTOM_OPENAI.MODEL_NAME; getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'MODEL_NAME'], '');
const mergeConfigs = (current: any, update: any): any => { const mergeConfigs = (current: any, update: any): any => {
if (update === null || update === undefined) { if (update === null || update === undefined) {