fix: improve query and bot service

core/chat/bot_service.py

@@ -173,10 +173,3 @@ class ChatCompletionService:
         collection = db[self.session_id]  # Replace with your collection name
         result = collection.insert_many(chat_history_json)
         print("Data inserted with record ids", result.inserted_ids)
-
-        chat_history_json = [message.model_dump() for message in chat_history_db]
-
-        db = self.client["bot_database"]  # Replace with your database name
-        collection = db[self.session_id]  # Replace with your collection name
-        result = collection.insert_many(chat_history_json)
-        print("Data inserted with record ids", result.inserted_ids)

db/query/query_user_meta.py

@@ -28,18 +28,34 @@ class UserMetaQuery(BaseQuery):
         return result
 
     def insert_user_meta_entries(self, db, metadata_ids):
-        """Insert new user meta entries."""
+        """Insert new user meta entries if they don't already exist."""
+        # Fetch existing metadata IDs for the user
+        existing_metadata_ids = {
+            entry.metadata_id
+            for entry in db.query(User_Meta).filter_by(user_id=self.user.get("id")).all()
+        }
+
+        # Filter out metadata IDs that already exist
+        new_metadata_ids = [
+            mid for mid in metadata_ids if mid not in existing_metadata_ids
+        ]
+
+        if not new_metadata_ids:
+            return {
+                "message": "No new entries to add; all specified metadata IDs already exist.",
+                "existing_metadata_ids": existing_metadata_ids,
+            }
+
         user_meta_entries = [
             User_Meta(user_id=self.user.get("id"), metadata_id=mid)
-            for mid in metadata_ids
+            for mid in new_metadata_ids
         ]
 
         # Use the method from BaseQuery to insert entries
         self.insert_entries(db, user_meta_entries)
         return {
-            "status":"success",
             "message": "User meta entries added successfully.",
-            "metadata_ids": metadata_ids,  # Include the metadata IDs in the result
+            "metadata_ids": new_metadata_ids,  # Include only new metadata IDs in the result
         }
 
     def update_user_meta_entries(self, db, metadata_ids):