LibreChat/app/chatgpt-client.js

51 lines
1.4 KiB
JavaScript
Raw Normal View History

require('dotenv').config();
const { KeyvFile } = require('keyv-file');
const proxyOptions = {
// Warning: This will expose your access token to a third party. Consider the risks before using this.
reverseProxyUrl: 'https://chatgpt.duti.tech/api/conversation',
// Access token from https://chat.openai.com/api/auth/session
accessToken: process.env.CHATGPT_TOKEN
};
const davinciOptions = {
modelOptions: {
model: 'text-davinci-003'
},
debug: false
};
const askClient = async ({ model, text, progressCallback, convo }) => {
const davinciClient = (await import('@waylaidwanderer/chatgpt-api')).default;
const { ChatGPTBrowserClient } = await import('@waylaidwanderer/chatgpt-api');
const clientOptions = model === 'chatgpt' ? proxyOptions : davinciOptions;
const modelClient = model === 'chatgpt' ? ChatGPTBrowserClient : davinciClient;
const store = {
store: new KeyvFile({ filename: 'cache.json' })
};
const params =
model === 'chatgpt'
? [clientOptions, store]
: [
process.env.OPENAI_KEY,
clientOptions,
store
];
const client = new modelClient(...params);
let options = {
onProgress: async (partialRes) => await progressCallback(partialRes)
};
if (!!convo.parentMessageId && !!convo.conversationId) {
options = { ...options, ...convo };
}
const res = await client.sendMessage(text, options);
return res;
};
module.exports = { askClient };