Rename modules/database/chat_db.py to modules/database/chat_mongo_db.py

modules/database/{chat_db.py → chat_mongo_db.py}

@@ -1,45 +1,45 @@
-#/modules/database/chat_db.py
-from .mongo_db import insert_document, find_documents
-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 store_chat_history(username, messages, analysis_type):
-    chat_document = {
-        'username': username,
-        'timestamp': datetime.now(timezone.utc).isoformat(),
-        'messages': messages,
-        '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}")
-        return True
-    return False
-
-def get_chat_history(username, analysis_type, limit=None):
-    query = {"username": username}
-    if analysis_type:
-        query["analysis_type"] = analysis_type
-
-    db = get_mongodb()
-    collection = db['chat_history-v3']
-    cursor = collection.find(query).sort("timestamp", -1)
-    if limit:
-        cursor = cursor.limit(limit)
-
-    return list(cursor)
-
-#def get_chat_history(username, analysis_type=None, limit=10):
-#    query = {"username": username}
-#    if analysis_type:
-#        query["analysis_type"] = analysis_type
-
-#    return find_documents(COLLECTION_NAME, query, sort=[("timestamp", -1)], limit=limit)
-
+#/modules/database/chat_mongo_db.py
+from .mongo_db import insert_document, find_documents, get_collections
+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 store_chat_history(username, messages, analysis_type):
+    chat_document = {
+        'username': username,
+        'timestamp': datetime.now(timezone.utc).isoformat(),
+        'messages': messages,
+        '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}")
+        return True
+    return False
+
+def get_chat_history(username, analysis_type, limit=None):
+    query = {"username": username}
+    if analysis_type:
+        query["analysis_type"] = analysis_type
+
+    db = get_mongodb()
+    collection = db['chat_history-v3']
+    cursor = collection.find(query).sort("timestamp", -1)
+    if limit:
+        cursor = cursor.limit(limit)
+
+    return list(cursor)
+
+#def get_chat_history(username, analysis_type=None, limit=10):
+#    query = {"username": username}
+#    if analysis_type:
+#        query["analysis_type"] = analysis_type
+
+#    return find_documents(COLLECTION_NAME, query, sort=[("timestamp", -1)], limit=limit)
+
 # Agregar funciones para actualizar y eliminar chat si es necesario