Update modules/database/chat_mongo_db.py

modules/database/chat_mongo_db.py

@@ -1,15 +1,22 @@
-#/modules/database/chat_mongo_db.py
+# /modules/database/chat_mongo_db.py
 from .mongo_db import insert_document, find_documents, get_collection
 from datetime import datetime, timezone
 import logging
-from .database_init import get_mongodb  # Asegúrate de que esta importación esté al principio del archivo
 
 logger = logging.getLogger(__name__)
 COLLECTION_NAME = 'chat_history-v3'
 
-def get_chat_history(username, analysis_type='sidebar', limit=None):
+def get_chat_history(username: str, analysis_type: str = 'sidebar', limit: int = None) -> list:
     """
-    Recupera el historial del chat
+    Recupera el historial del chat.
+    
+    Args:
+        username: Nombre del usuario
+        analysis_type: Tipo de análisis ('sidebar' por defecto)
+        limit: Límite de conversaciones a recuperar
+        
+    Returns:
+        list: Lista de conversaciones con formato
     """
     try:
         query = {
@@ -17,23 +24,51 @@ def get_chat_history(username, analysis_type='sidebar', limit=None):
             "analysis_type": analysis_type
         }
         
-        collection = get_collection(COLLECTION_NAME)  # Usar get_collection en lugar de get_mongodb
+        collection = get_collection(COLLECTION_NAME)
         if collection is None:
             logger.error("No se pudo obtener la colección de chat")
             return []
             
+        # Obtener y formatear conversaciones
         cursor = collection.find(query).sort("timestamp", -1)
         if limit:
             cursor = cursor.limit(limit)
             
-        return list(cursor)
+        conversations = []
+        for chat in cursor:
+            try:
+                formatted_chat = {
+                    'timestamp': chat['timestamp'],
+                    'messages': [
+                        {
+                            'role': msg.get('role', 'unknown'),
+                            'content': msg.get('content', '')
+                        }
+                        for msg in chat.get('messages', [])
+                    ]
+                }
+                conversations.append(formatted_chat)
+            except Exception as e:
+                logger.error(f"Error formateando chat: {str(e)}")
+                continue
+                
+        return conversations
+        
     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'):
+def store_chat_history(username: str, messages: list, analysis_type: str = 'sidebar') -> bool:
     """
-    Guarda el historial del chat
+    Guarda el historial del chat.
+    
+    Args:
+        username: Nombre del usuario
+        messages: Lista de mensajes a guardar
+        analysis_type: Tipo de análisis
+    
+    Returns:
+        bool: True si se guardó correctamente
     """
     try:
         collection = get_collection(COLLECTION_NAME)
@@ -41,10 +76,20 @@ def store_chat_history(username, messages, analysis_type='sidebar'):
             logger.error("No se pudo obtener la colección de chat")
             return False
             
+        # Formatear mensajes antes de guardar
+        formatted_messages = [
+            {
+                'role': msg.get('role', 'unknown'),
+                'content': msg.get('content', ''),
+                'timestamp': datetime.now(timezone.utc).isoformat()
+            }
+            for msg in messages
+        ]
+        
         chat_document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
-            'messages': messages,
+            'messages': formatted_messages,
             'analysis_type': analysis_type
         }
         
@@ -59,7 +104,6 @@ def store_chat_history(username, messages, analysis_type='sidebar'):
     except Exception as e:
         logger.error(f"Error al guardar historial de chat: {str(e)}")
         return False
-
         
         
         #def get_chat_history(username, analysis_type=None, limit=10):