require('dotenv').config();
const Keyv = require('keyv');
const { Configuration, OpenAIApi } = require('openai');
const messageStore = new Keyv(process.env.MONGODB_URI, { namespace: 'chatgpt' });

const ask = async (question, progressCallback, convo) => {
  const { ChatGPTAPI } = await import('chatgpt');
  const api = new ChatGPTAPI({ apiKey: process.env.OPENAI_KEY, messageStore });
  let options = {
    onProgress: async (partialRes) => {
      if (partialRes.text.length > 0) {
        await progressCallback(partialRes);
      }
    }
  };

  if (!!convo.parentMessageId && !!convo.conversationId) {
    options = { ...options, ...convo };
  }

  const res = await api.sendMessage(question, options);
  return res;
};

const titleConversation = async (message, response) => {
  const configuration = new Configuration({
    apiKey: process.env.OPENAI_KEY
  });
  const openai = new OpenAIApi(configuration);
  const completion = await openai.createCompletion({
    model: 'text-davinci-002',
    prompt: `Make a short title (goal: 5 words or less) in title case summarizing this conversation:\nuser:"${message}"\nGPT:"${response}"\nTitle: `
  });
  console.log(completion.data.choices[0].text);
  return completion.data.choices[0].text.replace(/\n/g, '');
};

module.exports = { ask, titleConversation };