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 = () => {
|
2023-05-16 14:30:24 -04:00
|
|
|
let models = ['text-davinci-002-render-sha', 'gpt-4'];
|
2023-04-05 21:21:02 +08:00
|
|
|
if (process.env.CHATGPT_MODELS) models = String(process.env.CHATGPT_MODELS).split(',');
|
|
|
|
|
|
|
|
|
|
return models;
|
|
|
|
|
};
|
|
|
|
|
|
2023-05-13 16:29:06 -04:00
|
|
|
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 =
|
2023-05-18 11:09:31 -07:00
|
|
|
key || palmUser
|
|
|
|
|
? { userProvide: palmUser, availableModels: ['chat-bison', 'text-bison'] }
|
|
|
|
|
: false;
|
2023-04-05 16:15:46 +08:00
|
|
|
const azureOpenAI = !!process.env.AZURE_OPENAI_KEY;
|
2023-05-17 21:58:56 -04:00
|
|
|
const apiKey = process.env.OPENAI_KEY || process.env.AZURE_OPENAI_API_KEY;
|
2023-05-18 11:09:31 -07:00
|
|
|
const openAI = apiKey
|
|
|
|
|
? { availableModels: getOpenAIModels(), userProvide: apiKey === 'user_provided' }
|
|
|
|
|
: false;
|
2023-04-10 00:41:34 +08:00
|
|
|
const bingAI = process.env.BINGAI_TOKEN
|
2023-04-11 21:50:53 -04:00
|
|
|
? { userProvide: process.env.BINGAI_TOKEN == 'user_provided' }
|
2023-04-10 00:41:34 +08:00
|
|
|
: false;
|
|
|
|
|
const chatGPTBrowser = process.env.CHATGPT_TOKEN
|
|
|
|
|
? {
|
2023-04-11 21:50:53 -04:00
|
|
|
userProvide: process.env.CHATGPT_TOKEN == 'user_provided',
|
2023-04-11 10:28:11 -04:00
|
|
|
availableModels: getChatGPTBrowserModels()
|
|
|
|
|
}
|
2023-04-10 00:41:34 +08:00
|
|
|
: false;
|
2023-04-05 16:15:46 +08:00
|
|
|
|
2023-05-13 16:29:06 -04:00
|
|
|
res.send(JSON.stringify({ azureOpenAI, openAI, google, bingAI, chatGPTBrowser }));
|
2023-04-05 16:15:46 +08:00
|
|
|
});
|
|
|
|
|
|
2023-05-18 11:09:31 -07:00
|
|
|
module.exports = { router, getOpenAIModels, getChatGPTBrowserModels };
|