mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-19 01:40:15 +01:00
55 lines
1.6 KiB
JavaScript
55 lines
1.6 KiB
JavaScript
|
|
const { CacheKeys, EModelEndpoint } = require('librechat-data-provider');
|
||
|
|
const { isUserProvided, extractEnvVariable } = require('~/server/utils');
|
||
|
|
const loadCustomConfig = require('./loadCustomConfig');
|
||
|
|
const { getLogStores } = require('~/cache');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Load config endpoints from the cached configuration object
|
||
|
|
* @function loadConfigEndpoints */
|
||
|
|
async function loadConfigEndpoints() {
|
||
|
|
const cache = getLogStores(CacheKeys.CONFIG_STORE);
|
||
|
|
let customConfig = await cache.get(CacheKeys.CUSTOM_CONFIG);
|
||
|
|
|
||
|
|
if (!customConfig) {
|
||
|
|
customConfig = await loadCustomConfig();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!customConfig) {
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
|
||
|
|
const { endpoints = {} } = customConfig ?? {};
|
||
|
|
const endpointsConfig = {};
|
||
|
|
|
||
|
|
if (Array.isArray(endpoints[EModelEndpoint.custom])) {
|
||
|
|
const customEndpoints = endpoints[EModelEndpoint.custom].filter(
|
||
|
|
(endpoint) =>
|
||
|
|
endpoint.baseURL &&
|
||
|
|
endpoint.apiKey &&
|
||
|
|
endpoint.name &&
|
||
|
|
endpoint.models &&
|
||
|
|
(endpoint.models.fetch || endpoint.models.default),
|
||
|
|
);
|
||
|
|
|
||
|
|
for (let i = 0; i < customEndpoints.length; i++) {
|
||
|
|
const endpoint = customEndpoints[i];
|
||
|
|
const { baseURL, apiKey, name, iconURL, modelDisplayLabel } = endpoint;
|
||
|
|
|
||
|
|
const resolvedApiKey = extractEnvVariable(apiKey);
|
||
|
|
const resolvedBaseURL = extractEnvVariable(baseURL);
|
||
|
|
|
||
|
|
endpointsConfig[name] = {
|
||
|
|
type: EModelEndpoint.custom,
|
||
|
|
userProvide: isUserProvided(resolvedApiKey),
|
||
|
|
userProvideURL: isUserProvided(resolvedBaseURL),
|
||
|
|
modelDisplayLabel,
|
||
|
|
iconURL,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return endpointsConfig;
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = loadConfigEndpoints;
|