2023-12-06 19:36:57 -05:00
|
|
|
const { availableTools } = require('~/app/clients/tools');
|
|
|
|
const { addOpenAPISpecs } = require('~/app/clients/tools/util/addOpenAPISpecs');
|
2023-12-10 14:54:13 -05:00
|
|
|
const { openAIApiKey, azureOpenAIApiKey, useAzurePlugins, userProvidedOpenAI, googleKey } =
|
2023-12-06 19:36:57 -05:00
|
|
|
require('./EndpointService').config;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Load async endpoints and return a configuration object
|
|
|
|
*/
|
|
|
|
async function loadAsyncEndpoints() {
|
|
|
|
let i = 0;
|
2023-12-10 14:54:13 -05:00
|
|
|
let key, googleUserProvides;
|
2023-12-06 19:36:57 -05:00
|
|
|
try {
|
|
|
|
key = require('~/data/auth.json');
|
|
|
|
} catch (e) {
|
|
|
|
if (i === 0) {
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-12-10 14:54:13 -05:00
|
|
|
if (googleKey === 'user_provided') {
|
|
|
|
googleUserProvides = true;
|
2023-12-06 19:36:57 -05:00
|
|
|
if (i <= 1) {
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const tools = await addOpenAPISpecs(availableTools);
|
|
|
|
function transformToolsToMap(tools) {
|
|
|
|
return tools.reduce((map, obj) => {
|
|
|
|
map[obj.pluginKey] = obj.name;
|
|
|
|
return map;
|
|
|
|
}, {});
|
|
|
|
}
|
|
|
|
const plugins = transformToolsToMap(tools);
|
|
|
|
|
2023-12-10 14:54:13 -05:00
|
|
|
const google = key || googleUserProvides ? { userProvide: googleUserProvides } : false;
|
2023-12-06 19:36:57 -05:00
|
|
|
|
|
|
|
const gptPlugins =
|
|
|
|
openAIApiKey || azureOpenAIApiKey
|
|
|
|
? {
|
|
|
|
plugins,
|
|
|
|
availableAgents: ['classic', 'functions'],
|
|
|
|
userProvide: userProvidedOpenAI,
|
|
|
|
azure: useAzurePlugins,
|
|
|
|
}
|
|
|
|
: false;
|
|
|
|
|
|
|
|
return { google, gptPlugins };
|
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = loadAsyncEndpoints;
|