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;
|
|
|
|
|
const bingAI = !!process.env.BINGAI_TOKEN;
|
|
|
|
|
const chatGPTBrowser = process.env.CHATGPT_TOKEN ? { 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 };
|