mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02:00
71 lines
2.3 KiB
JavaScript
71 lines
2.3 KiB
JavaScript
const express = require('express');
|
|
const router = express.Router();
|
|
const { availableTools } = require('../../app/langchain/tools');
|
|
|
|
const getOpenAIModels = () => {
|
|
let models = ['gpt-4', 'gpt-4-0613', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-0301', 'text-davinci-003' ];
|
|
if (process.env.OPENAI_MODELS) models = String(process.env.OPENAI_MODELS).split(',');
|
|
|
|
return models;
|
|
};
|
|
|
|
const getChatGPTBrowserModels = () => {
|
|
let models = ['text-davinci-002-render-sha', 'gpt-4'];
|
|
if (process.env.CHATGPT_MODELS) models = String(process.env.CHATGPT_MODELS).split(',');
|
|
|
|
return models;
|
|
};
|
|
|
|
const getPluginModels = () => {
|
|
let models = ['gpt-4', 'gpt-4-0613', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-0301'];
|
|
if (process.env.PLUGIN_MODELS) models = String(process.env.PLUGIN_MODELS).split(',');
|
|
|
|
return models;
|
|
};
|
|
|
|
let i = 0;
|
|
router.get('/', async function (req, res) {
|
|
let key, palmUser;
|
|
try {
|
|
key = require('../../data/auth.json');
|
|
} catch (e) {
|
|
if (i === 0) {
|
|
console.log("No 'auth.json' file (service account key) found in /api/data/ for PaLM models");
|
|
i++;
|
|
}
|
|
}
|
|
|
|
if (process.env.PALM_KEY === 'user_provided') {
|
|
palmUser = true;
|
|
if (i <= 1) {
|
|
console.log('User will provide key for PaLM models');
|
|
i++;
|
|
}
|
|
}
|
|
|
|
const google =
|
|
key || palmUser
|
|
? { userProvide: palmUser, availableModels: ['chat-bison', 'text-bison', 'codechat-bison'] }
|
|
: false;
|
|
const azureOpenAI = !!process.env.AZURE_OPENAI_API_KEY;
|
|
const apiKey = process.env.OPENAI_API_KEY || process.env.AZURE_OPENAI_API_KEY;
|
|
const openAI = apiKey
|
|
? { availableModels: getOpenAIModels(), userProvide: apiKey === 'user_provided' }
|
|
: false;
|
|
const gptPlugins = apiKey
|
|
? { availableModels: getPluginModels(), availableTools, availableAgents: ['classic', 'functions'] }
|
|
: false;
|
|
const bingAI = process.env.BINGAI_TOKEN
|
|
? { userProvide: process.env.BINGAI_TOKEN == 'user_provided' }
|
|
: false;
|
|
const chatGPTBrowser = process.env.CHATGPT_TOKEN
|
|
? {
|
|
userProvide: process.env.CHATGPT_TOKEN == 'user_provided',
|
|
availableModels: getChatGPTBrowserModels()
|
|
}
|
|
: false;
|
|
|
|
res.send(JSON.stringify({ azureOpenAI, openAI, google, bingAI, chatGPTBrowser, gptPlugins }));
|
|
});
|
|
|
|
module.exports = { router, getOpenAIModels, getChatGPTBrowserModels };
|