🛠️ fix: Update Conversation and Message Models to Return Objects Inst…

…ead of Using Lean() (danny-avila#3230)
qingfengfenga · Jun 29, 2024 · a87d4e0 · a87d4e0
1 parent a2fd975
commit a87d4e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/api/models/Conversation.js b/api/models/Conversation.js
@@ -27,10 +27,12 @@ module.exports = {
         update.conversationId = newConversationId;
       }
 
-      return await Conversation.findOneAndUpdate({ conversationId: conversationId, user }, update, {
+      const conversation = await Conversation.findOneAndUpdate({ conversationId, user }, update, {
         new: true,
         upsert: true,
-      }).lean();
+      });
+
+      return conversation.toObject();
     } catch (error) {
       logger.error('[saveConvo] Error saving conversation', error);
       return { message: 'Error saving conversation' };

diff --git a/api/models/Message.js b/api/models/Message.js
@@ -58,10 +58,12 @@ module.exports = {
         update.files = files;
       }
 
-      return await Message.findOneAndUpdate({ messageId }, update, {
+      const message = await Message.findOneAndUpdate({ messageId }, update, {
         upsert: true,
         new: true,
-      }).lean();
+      });
+
+      return message.toObject();
     } catch (err) {
       logger.error('Error saving message:', err);
       throw new Error('Failed to save message.');