mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-19 01:40:15 +01:00
0bae503a0a
* refactor: move endpoint services to own directory * refactor: make endpointconfig handling more concise, separate logic, and cache result for subsequent serving * refactor: ModelController gets same treatment as EndpointController, draft OverrideController * wip: flesh out override controller more to return real value * refactor: client/api changes in anticipation of override
32 lines
1.2 KiB
JavaScript
32 lines
1.2 KiB
JavaScript
const {
|
|
getOpenAIModels,
|
|
getChatGPTBrowserModels,
|
|
getAnthropicModels,
|
|
} = require('~/server/services/ModelService');
|
|
const { EModelEndpoint } = require('~/server/routes/endpoints/schemas');
|
|
const { useAzurePlugins } = require('~/server/services/Config/EndpointService').config;
|
|
|
|
const fitlerAssistantModels = (str) => {
|
|
return /gpt-4|gpt-3\\.5/i.test(str) && !/vision|instruct/i.test(str);
|
|
};
|
|
|
|
async function loadDefaultModels() {
|
|
const openAI = await getOpenAIModels();
|
|
const anthropic = getAnthropicModels();
|
|
const chatGPTBrowser = getChatGPTBrowserModels();
|
|
const azureOpenAI = await getOpenAIModels({ azure: true });
|
|
const gptPlugins = await getOpenAIModels({ azure: useAzurePlugins, plugins: true });
|
|
|
|
return {
|
|
[EModelEndpoint.openAI]: openAI,
|
|
[EModelEndpoint.azureOpenAI]: azureOpenAI,
|
|
[EModelEndpoint.assistant]: openAI.filter(fitlerAssistantModels),
|
|
[EModelEndpoint.google]: ['chat-bison', 'text-bison', 'codechat-bison'],
|
|
[EModelEndpoint.bingAI]: ['BingAI', 'Sydney'],
|
|
[EModelEndpoint.chatGPTBrowser]: chatGPTBrowser,
|
|
[EModelEndpoint.gptPlugins]: gptPlugins,
|
|
[EModelEndpoint.anthropic]: anthropic,
|
|
};
|
|
}
|
|
|
|
module.exports = loadDefaultModels;
|