fix: customGpt persistence blindspot, also remove free gpt

api/models/Conversation.js

@@ -75,17 +75,30 @@ module.exports = {
   },
   // getConvos: async () => await Conversation.find({}).sort({ created: -1 }).exec(),
   getConvos: async (pageNumber = 1, pageSize = 12) => {
-    const skip = (pageNumber - 1) * pageSize;
-    // const limit = pageNumber * pageSize;
+    try {
+      const skip = (pageNumber - 1) * pageSize;
+      // const limit = pageNumber * pageSize;
 
-    const conversations = await Conversation.find({})
-      .sort({ created: -1 })
-      .skip(skip)
-      // .limit(limit)
-      .limit(pageSize)
-      .exec();
+      const conversations = await Conversation.find({})
+        .sort({ created: -1 })
+        .skip(skip)
+        // .limit(limit)
+        .limit(pageSize)
+        .exec();
 
-    return conversations;
+      return conversations;
+    } catch (error) {
+      console.log(error);
+      return { message: 'Error getting conversations' };
+    }
+  },
+  getConvo: async (conversationId) => {
+    try {
+      return await Conversation.findOne({ conversationId }).exec();
+    } catch (error) {
+      console.log(error);
+      return { message: 'Error getting single conversation' };
+    }
   },
   deleteConvos: async (filter) => {
     let deleteCount = await Conversation.deleteMany(filter).exec();

api/models/index.js

@@ -1,10 +1,11 @@
 const { saveMessage, deleteMessages } = require('./Message');
 const { getCustomGpts, updateCustomGpt, updateByLabel, deleteCustomGpts } = require('./CustomGpt');
-const { saveConvo } = require('./Conversation');
+const { getConvo, saveConvo } = require('./Conversation');
 
 module.exports = {
   saveMessage,
   deleteMessages,
+  getConvo,
   saveConvo,
   getCustomGpts,
   updateCustomGpt,

api/server/routes/ask.js

@@ -9,14 +9,13 @@ const {
   customClient,
   detectCode
 } = require('../../app/');
-const { saveMessage, deleteMessages, saveConvo } = require('../../models');
+const { getConvo, saveMessage, deleteMessages, saveConvo } = require('../../models');
 const { handleError, sendMessage } = require('./handlers');
 
 router.use('/bing', askBing);
 
 router.post('/', async (req, res) => {
-  const { model, text, parentMessageId, conversationId, chatGptLabel, promptPrefix } =
-    req.body;
+  let { model, text, parentMessageId, conversationId, chatGptLabel, promptPrefix } = req.body;
   if (!text.trim().includes(' ') && text.length < 5) {
     return handleError(res, 'Prompt empty or too short');
   }
@@ -43,6 +42,15 @@ router.post('/', async (req, res) => {
     client = browserClient;
   }
 
+  if (model === 'chatgptCustom' && !chatGptLabel && conversationId) {
+    const convo = await getConvo({ conversationId });
+    if (convo) {
+      console.log('found convo for custom gpt', { convo })
+      chatGptLabel = convo.chatGptLabel;
+      promptPrefix = convo.promptPrefix;
+    }
+  }
+
   res.writeHead(200, {
     Connection: 'keep-alive',
     'Content-Type': 'text/event-stream',