Update modules/database/chat_mongo_db.py

modules/database/chat_mongo_db.py

@@ -7,15 +7,40 @@ from .database_init import get_mongodb  # Asegúrate de que esta importación es
 logger = logging.getLogger(__name__)
 COLLECTION_NAME = 'chat_history-v3'
 
+def get_chat_history(username, analysis_type='sidebar', limit=None):
+    """
+    Recupera el historial del chat
+    """
+    try:
+        query = {
+            "username": username,
+            "analysis_type": analysis_type
+        }
+        
+        collection = get_collection(COLLECTION_NAME)  # Usar get_collection en lugar de get_mongodb
+        if collection is None:
+            logger.error("No se pudo obtener la colección de chat")
+            return []
+            
+        cursor = collection.find(query).sort("timestamp", -1)
+        if limit:
+            cursor = cursor.limit(limit)
+            
+        return list(cursor)
+    except Exception as e:
+        logger.error(f"Error al recuperar historial de chat: {str(e)}")
+        return []
+
 def store_chat_history(username, messages, analysis_type='sidebar'):
     """
     Guarda el historial del chat
-    Args:
-        username: nombre del usuario
-        messages: lista de mensajes
-        analysis_type: tipo de chat ('sidebar' por defecto)
     """
     try:
+        collection = get_collection(COLLECTION_NAME)
+        if collection is None:
+            logger.error("No se pudo obtener la colección de chat")
+            return False
+            
         chat_document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
@@ -23,46 +48,21 @@ def store_chat_history(username, messages, analysis_type='sidebar'):
             'analysis_type': analysis_type
         }
         
-        result = insert_document(COLLECTION_NAME, chat_document)
-        if result:
-            logger.info(f"Historial de chat guardado con ID: {result} para el usuario: {username}")
+        result = collection.insert_one(chat_document)
+        if result.inserted_id:
+            logger.info(f"Historial de chat guardado con ID: {result.inserted_id} para el usuario: {username}")
             return True
+            
+        logger.error("No se pudo insertar el documento")
         return False
+        
     except Exception as e:
         logger.error(f"Error al guardar historial de chat: {str(e)}")
         return False
 
-def get_chat_history(username, analysis_type='sidebar', limit=None):
-    """
-    Recupera el historial del chat
-    Args:
-        username: nombre del usuario
-        analysis_type: tipo de chat ('sidebar' por defecto)
-        limit: límite de mensajes a recuperar
-    Returns:
-        list: Lista de conversaciones
-    """
-    try:
-        query = {
-            "username": username,
-            "analysis_type": analysis_type
-        }
         
-        db = get_mongodb()
-        if not db:
-            raise Exception("No se pudo conectar a la base de datos")
-            
-        collection = db[COLLECTION_NAME]
         
-        cursor = collection.find(query).sort("timestamp", -1)
-        if limit:
-            cursor = cursor.limit(limit)
-            
-        return list(cursor)
-    except Exception as e:
-        logger.error(f"Error al recuperar historial de chat: {str(e)}")
-        return []
-#def get_chat_history(username, analysis_type=None, limit=10):
+        #def get_chat_history(username, analysis_type=None, limit=10):
 #    query = {"username": username}
 #    if analysis_type:
 #        query["analysis_type"] = analysis_type