2023-04-05 16:15:46 +08:00
|
|
|
const express = require('express');
|
|
|
|
|
const router = express.Router();
|
|
|
|
|
|
2023-04-05 21:21:02 +08:00
|
|
|
const getOpenAIModels = () => {
|
|
|
|
|
let models = ['gpt-4', 'text-davinci-003', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0301'];
|
|
|
|
|
if (process.env.OPENAI_MODELS) models = String(process.env.OPENAI_MODELS).split(',');
|
|
|
|
|
|
|
|
|
|
return models;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
const getChatGPTBrowserModels = () => {
|
|
|
|
|
let models = ['text-davinci-002-render-sha', 'text-davinci-002-render-paid', 'gpt-4'];
|
|
|
|
|
if (process.env.CHATGPT_MODELS) models = String(process.env.CHATGPT_MODELS).split(',');
|
|
|
|
|
|
|
|
|
|
return models;
|
|
|
|
|
};
|
|
|
|
|
|
2023-04-05 16:15:46 +08:00
|
|
|
router.get('/', function (req, res) {
|
|
|
|
|
const azureOpenAI = !!process.env.AZURE_OPENAI_KEY;
|
2023-04-05 21:21:02 +08:00
|
|
|
const openAI = process.env.OPENAI_KEY ? { availableModels: getOpenAIModels() } : false;
|
2023-04-10 00:41:34 +08:00
|
|
|
const bingAI = process.env.BINGAI_TOKEN
|
|
|
|
|
? { userProvide: process.env.BINGAI_TOKEN == 'user_provide' }
|
|
|
|
|
: false;
|
|
|
|
|
const chatGPTBrowser = process.env.CHATGPT_TOKEN
|
|
|
|
|
? {
|
|
|
|
|
userProvide: process.env.CHATGPT_TOKEN == 'user_provide',
|
|
|
|
|
availableModels: getChatGPTBrowserModels()
|
|
|
|
|
}
|
|
|
|
|
: false;
|
2023-04-05 16:15:46 +08:00
|
|
|
|
|
|
|
|
res.send(JSON.stringify({ azureOpenAI, openAI, bingAI, chatGPTBrowser }));
|
|
|
|
|
});
|
|
|
|
|
|
2023-04-05 21:21:02 +08:00
|
|
|
module.exports = { router, getOpenAIModels, getChatGPTBrowserModels };
|