feat(api): add support for saving messages to database

fix(api): change arrowParens prettier option to always
fix(api): update addToCache to include endpointOption and latestMessage
fix(api): update askOpenAI to include endpointOption in abortControllers
fix(client): remove abortKey state and add currentParent state to MessageHandler

api/server/routes/ask/addToCache.js

@@ -1,31 +1,66 @@
 const Keyv = require('keyv');
 const { KeyvFile } = require('keyv-file');
 const crypto = require('crypto');
+const { saveMessage } = require('../../../models');
 
-const addToCache = async ( { conversationId, parentMessageId }) => {
-  const conversationsCache = new Keyv({
-    store: new KeyvFile({ filename: './data/cache.json' })
-  });
+const addToCache = async ({
+  endpointOption,
+  conversationId,
+  userMessage,
+  latestMessage,
+  parentMessageId
+}) => {
+  try {
+    const conversationsCache = new Keyv({
+      store: new KeyvFile({ filename: './data/cache.json' }),
+      namespace: 'chatgpt', // should be 'bing' for bing/sydney
+    });
 
-  let conversation = await conversationsCache.get(conversationId);
-  let isNewConversation = false;
-  if (!conversation) {
-    conversation = {
-      messages: [],
-      createdAt: Date.now()
+    let conversation = await conversationsCache.get(conversationId);
+    // used to generate a title for the conversation if none exists
+    // let isNewConversation = false;
+    if (!conversation) {
+      conversation = {
+        messages: [],
+        createdAt: Date.now()
+      };
+      // isNewConversation = true;
+    }
+
+    // const shouldGenerateTitle = opts.shouldGenerateTitle && isNewConversation;
+
+    const roles = (options) => {
+      const { endpoint } = options;
+      if (endpoint === 'openAI') {
+        return options?.chatGptLabel || 'ChatGPT';
+      } else if (endpoint === 'bingAI') {
+        return options?.jailbreak ? 'Sydney' : 'BingAI';
+      }
     };
-    isNewConversation = true;
+
+    const messageId = crypto.randomUUID();
+
+    let responseMessage = {
+      id: messageId,
+      parentMessageId,
+      role: roles(endpointOption),
+      message: latestMessage
+    };
+
+    await saveMessage({
+      ...responseMessage,
+      conversationId,
+      messageId,
+      sender: responseMessage.role,
+      text: latestMessage
+    });
+
+    conversation.messages.push(userMessage, responseMessage);
+
+    await conversationsCache.set(conversationId, conversation);
+  } catch (error) {
+    console.error('Trouble adding to cache', error);
   }
-
-  // const shouldGenerateTitle = opts.shouldGenerateTitle && isNewConversation;
-
-  const userMessage = {
-    id: crypto.randomUUID(),
-    parentMessageId,
-    role: 'User',
-    message
-  };
-  conversation.messages.push(userMessage);
 };
 
-module.exports = { addToCache };
+module.exports = addToCache;

api/server/routes/ask/askOpenAI.js

@@ -1,6 +1,7 @@
 const express = require('express');
 const crypto = require('crypto');
 const router = express.Router();
+const addToCache = require('./addToCache');
 const { getOpenAIModels } = require('../endpoints');
 const { titleConvo, askClient } = require('../../../app/');
 const { saveMessage, getConvoTitle, saveConvo, updateConvo, getConvo } = require('../../../models');
@@ -8,16 +9,22 @@ const { handleError, sendMessage, createOnProgress, handleText } = require('./ha
 
 const abortControllers = new Map();
 
-router.post('/abort', (req, res) => {
-  const { abortKey, message } = req.body;
+router.post('/abort', async (req, res) => {
+  const { abortKey, latestMessage, parentMessageId } = req.body;
+  console.log(`req.body`, req.body);
   if (!abortControllers.has(abortKey)) {
     return res.status(404).send('Request not found');
   }
+  
+  const { abortController, userMessage, endpointOption } = abortControllers.get(abortKey);
+  if (!endpointOption.endpoint) {
+    endpointOption.endpoint = req.originalUrl.replace('/api/ask/','').split('/abort')[0];
+  }
 
-  const { abortController, userMessage } = abortControllers.get(abortKey);
   abortController.abort();
   abortControllers.delete(abortKey);
-  console.log('Aborted request', abortKey, userMessage);
+  console.log('Aborted request', abortKey, userMessage, endpointOption);
+  await addToCache({ endpointOption, conversationId: abortKey, userMessage, latestMessage, parentMessageId });
   
   res.status(200).send('Aborted');
 });
@@ -118,7 +125,7 @@ const ask = async ({
     const abortController = new AbortController();
     const abortKey = conversationId;
     console.log('conversationId -----> ', conversationId);
-    abortControllers.set(abortKey, { abortController, userMessage });
+    abortControllers.set(abortKey, { abortController, userMessage, endpointOption });
     
     res.on('close', () => {
       abortController.abort();