mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 17:00:15 +01:00
* fix: endpoint will not be select if disabled * feat: order and disable endpoints with ENDPOINTS env var * chore: remove console.log
95 lines
2.3 KiB
JavaScript
95 lines
2.3 KiB
JavaScript
const { EModelEndpoint } = require('~/server/routes/endpoints/schemas');
|
|
const { availableTools } = require('~/app/clients/tools');
|
|
const { addOpenAPISpecs } = require('~/app/clients/tools/util/addOpenAPISpecs');
|
|
const {
|
|
openAIApiKey,
|
|
azureOpenAIApiKey,
|
|
useAzurePlugins,
|
|
userProvidedOpenAI,
|
|
palmKey,
|
|
openAI,
|
|
// assistant,
|
|
azureOpenAI,
|
|
bingAI,
|
|
chatGPTBrowser,
|
|
anthropic,
|
|
} = require('~/server/services/EndpointService').config;
|
|
|
|
let i = 0;
|
|
async function endpointController(req, res) {
|
|
let key, palmUser;
|
|
try {
|
|
key = require('~/data/auth.json');
|
|
} catch (e) {
|
|
if (i === 0) {
|
|
i++;
|
|
}
|
|
}
|
|
|
|
if (palmKey === 'user_provided') {
|
|
palmUser = true;
|
|
if (i <= 1) {
|
|
i++;
|
|
}
|
|
}
|
|
|
|
const tools = await addOpenAPISpecs(availableTools);
|
|
function transformToolsToMap(tools) {
|
|
return tools.reduce((map, obj) => {
|
|
map[obj.pluginKey] = obj.name;
|
|
return map;
|
|
}, {});
|
|
}
|
|
const plugins = transformToolsToMap(tools);
|
|
|
|
const google = key || palmUser ? { userProvide: palmUser } : false;
|
|
|
|
const gptPlugins =
|
|
openAIApiKey || azureOpenAIApiKey
|
|
? {
|
|
plugins,
|
|
availableAgents: ['classic', 'functions'],
|
|
userProvide: userProvidedOpenAI,
|
|
azure: useAzurePlugins,
|
|
}
|
|
: false;
|
|
|
|
let enabledEndpoints = [
|
|
EModelEndpoint.openAI,
|
|
EModelEndpoint.azureOpenAI,
|
|
EModelEndpoint.google,
|
|
EModelEndpoint.bingAI,
|
|
EModelEndpoint.chatGPTBrowser,
|
|
EModelEndpoint.gptPlugins,
|
|
EModelEndpoint.anthropic,
|
|
];
|
|
|
|
const endpointsEnv = process.env.ENDPOINTS || '';
|
|
if (endpointsEnv) {
|
|
enabledEndpoints = endpointsEnv
|
|
.split(',')
|
|
.filter((endpoint) => endpoint?.trim())
|
|
.map((endpoint) => endpoint.trim());
|
|
}
|
|
|
|
const endpointConfig = {
|
|
[EModelEndpoint.openAI]: openAI,
|
|
[EModelEndpoint.azureOpenAI]: azureOpenAI,
|
|
[EModelEndpoint.google]: google,
|
|
[EModelEndpoint.bingAI]: bingAI,
|
|
[EModelEndpoint.chatGPTBrowser]: chatGPTBrowser,
|
|
[EModelEndpoint.gptPlugins]: gptPlugins,
|
|
[EModelEndpoint.anthropic]: anthropic,
|
|
};
|
|
|
|
const orderedAndFilteredEndpoints = enabledEndpoints.reduce((config, key, index) => {
|
|
if (endpointConfig[key]) {
|
|
config[key] = { ...(endpointConfig[key] ?? {}), order: index };
|
|
}
|
|
return config;
|
|
}, {});
|
|
|
|
res.send(JSON.stringify(orderedAndFilteredEndpoints));
|
|
}
|
|
|
|
module.exports = endpointController;
|