LibreChat/app/chatgpt.js

24 lines
695 B
JavaScript
Raw Normal View History

2023-02-04 20:09:02 -05:00
require('dotenv').config();
const Keyv = require('keyv');
const messageStore = new Keyv(process.env.MONGODB_URI, { namespace: 'chatgpt' });
2023-02-04 20:09:02 -05:00
const ask = async (question, progressCallback, convo) => {
const { ChatGPTAPI } = await import('chatgpt');
const api = new ChatGPTAPI({ apiKey: process.env.OPENAI_KEY, messageStore });
let options = {
2023-02-06 13:28:21 -05:00
onProgress: async (partialRes) => {
if (partialRes.text.length > 0) {
2023-02-06 13:28:21 -05:00
await progressCallback(partialRes);
}
}
};
if (!!convo.parentMessageId && !!convo.conversationId) {
options = { ...options, ...convo };
}
const res = await api.sendMessage(question, options);
return res;
};
2023-02-04 20:09:02 -05:00
module.exports = { ask };