new update

db/mongoDB.py

@@ -1,31 +1,83 @@
-from pymongo import MongoClient, errors
-import config
+# db/mongoDB.py
+
 import logging
+from typing import Optional
+from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase, AsyncIOMotorCollection
+from pymongo.errors import ConnectionFailure, ConfigurationError
+import config # Import từ file config tập trung
+
 logger = logging.getLogger(__name__)
 
+class MongoDatabase:
+    """
+    Một lớp singleton để quản lý kết nối và các collection của MongoDB.
+    Điều này đảm bảo chúng ta chỉ có một kết nối duy nhất trong toàn bộ ứng dụng.
+    """
+    client: Optional[AsyncIOMotorClient] = None
+    db: Optional[AsyncIOMotorDatabase] = None
+
+    # Khai báo các collection bạn sẽ sử dụng
+    users: Optional[AsyncIOMotorCollection] = None
+    token_blacklist: Optional[AsyncIOMotorCollection] = None
+    conversations: Optional[AsyncIOMotorCollection] = None
+
+# Tạo một instance duy nhất để import và sử dụng trong toàn bộ ứng dụng
+mongo_db = MongoDatabase()
+
+async def connect_to_mongo():
+    """
+    Hàm khởi tạo kết nối đến MongoDB Atlas và gán vào object mongo_db.
+    Hàm này sẽ được gọi từ lifespan của FastAPI.
+    """
+    if mongo_db.client:
+        logger.info("MongoDB connection already established.")
+        return
+
+    logger.info(f"🔸 Connecting to MongoDB Atlas...")
+
+    if not config.MONGO_URI or not config.MONGO_DB_NAME:
+        logger.error("❌ MONGO_URI hoặc MONGO_DB_NAME chưa được thiết lập trong biến môi trường.")
+        raise ConfigurationError("MONGO_URI and MONGO_DB_NAME must be set.")
+
+    try:
+        # 1. Khởi tạo client bất đồng bộ
+        mongo_db.client = AsyncIOMotorClient(
+            config.MONGO_URI,
+            serverSelectionTimeoutMS=5000 # Thời gian chờ kết nối là 5 giây
+        )
+
+        # 2. Kiểm tra kết nối một cách rõ ràng
+        await mongo_db.client.admin.command('ping')
 
-try:
-    # Kết nối MongoDB
-    client = MongoClient(config.MONGODB_CLOUD_URI, serverSelectionTimeoutMS=30000)
+        # 3. Gán các đối tượng database và collection
+        mongo_db.db = mongo_db.client[config.MONGO_DB_NAME]
+        mongo_db.user = mongo_db.db["users"]
+        mongo_db.token_blacklist = mongo_db.db["token_blacklist"]
+        mongo_db.conversations = mongo_db.db["conversations"]
 
-    # Trigger kết nối thử
-    client.server_info()  # Gây lỗi nếu không kết nối được
+        # 4. Tạo TTL index một cách an toàn
+        # Lấy danh sách index hiện có
+        index_info = await mongo_db.token_blacklist.index_information()
+        if "expires_at_1" not in index_info:
+            # Chỉ tạo index nếu nó chưa tồn tại
+            await mongo_db.token_blacklist.create_index("expires_at", expireAfterSeconds=0)
+            logger.info("🔸 Successfully created TTL index for 'expires_at' in 'token_blacklist'.")
 
-    db = client[config.DB_NAME]
-    user_collection = db["users"]
-    blacklist_collection = db["token_blacklist"]
-    conversations_collection = db["conversations"]
+        logger.info("✅ MongoDB connection successful and collections are ready.")
 
-    # ⚠️ Tạo TTL index để MongoDB tự động xoá token khi tới hạn
-    # Chỉ tạo index nếu chưa tồn tại
-    if "expires_at_1" not in blacklist_collection.index_information():
-        blacklist_collection.create_index("expires_at", expireAfterSeconds=0)
-        logger.info("🔸Đã tạo TTL index cho 'expires_at' trong 'token_blacklist'.")
+    except ConnectionFailure as e:
+        logger.error(f"❌ Failed to connect to MongoDB: Connection Failure. Check your URI and network access rules in Atlas. Error: {e}", exc_info=True)
+        raise e
+    except ConfigurationError as e:
+        logger.error(f"❌ Failed to connect to MongoDB: Configuration Error. Check your connection string format. Error: {e}", exc_info=True)
+        raise e
+    except Exception as e:
+        logger.error(f"❌ An unexpected error occurred while connecting to MongoDB: {e}", exc_info=True)
+        raise e
 
-    logger.info("🔸Đã kết nối tới MongoDB Cloud thành công!")
 
-except errors.ServerSelectionTimeoutError as e:
-    logger.error("🔸Không thể kết nối tới MongoDB Cloud:")
-    logger.error(f"🔸Error:{e}")
-    user_collection = None
-    blacklist_collection = None
+async def close_mongo_connection():
+    """Hàm đóng kết nối MongoDB khi ứng dụng tắt."""
+    if mongo_db.client:
+        mongo_db.client.close()
+        logger.info("✅ MongoDB connection closed.")

dependencies.py

@@ -3,7 +3,8 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from jose import jwt, JWTError, ExpiredSignatureError
 import os
 from dotenv import load_dotenv
-from db.mongoDB import user_collection, blacklist_collection
+# from db.mongoDB import user_collection, blacklist_collection
+from db.mongoDB import mongo_db
 import torch
 import rag_components
 from schemas.chat import AppState
@@ -69,7 +70,7 @@ async def initialize_api_components(app_state: AppState):
     app_state.dict = load_legal_dictionary(config.LEGAL_DIC_FOLDER+ "/legal_terms.json")
     app_state.weaviateDB = connect_to_weaviate(run_diagnostics=False)
     # --- Kiểm tra kết nối tới MongoDB ---
-    if user_collection is  None or app_state.weaviateDB is None:
+    if mongo_db.users is  None or app_state.weaviateDB is None:
         logger.error("🔸Lỗi kết nối tới MongoDB hoặc Weaviate.")
         raise HTTPException(status_code=500, detail="Lỗi kết nối tới database.")
 
@@ -188,7 +189,7 @@ async def get_current_user(
     # 1. Kiểm tra token trong blacklist
     try:
         logger.info("GET_CURRENT_USER: Đang kiểm tra blacklist...")
-        is_blacklisted = blacklist_collection.find_one({"token": token_to_verify})
+        is_blacklisted = await mongo_db.token_blacklist.find_one({"token": token_to_verify})
         if is_blacklisted:
             logger.error(f"GET_CURRENT_USER: *** TOKEN TRONG BLACKLIST - RAISING 401 ***")
             raise HTTPException( # Sử dụng credentials_exception hoặc một cái cụ thể hơn
@@ -265,7 +266,7 @@ async def get_current_user(
     user_data: Optional[dict] = None # Khởi tạo để tránh UnboundLocalError
     try:
         logger.info(f"GET_CURRENT_USER: Đang tìm user trong DB: {email.lower()}") # email đã được validate là str
-        user_data = user_collection.find_one({"email": email.lower()}, {"password": 0, "_id": 0})
+        user_data =await mongo_db.users.find_one({"email": email.lower()}, {"password": 0, "_id": 0})
         # print(user_data) # Bỏ print trong production
 
         if user_data is None:

main.py

@@ -15,6 +15,7 @@ import os
 import traceback
 from core.logging_config import setup_logging
 setup_logging()
+from db.mongoDB import connect_to_mongo, close_mongo_connection
 
 logger = logging.getLogger(__name__)
 
@@ -22,6 +23,10 @@ logger = logging.getLogger(__name__)
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     logger.info("✅ [Lifespan] STARTING UP...")
+
+    await connect_to_mongo()
+
+
     current_app_state_instance = AppState()
     initialization_successful = False
     try:
@@ -32,6 +37,7 @@ async def lifespan(app: FastAPI):
         logger.info("✅ [Lifespan] SUCCESSFULLY set app.state.app_state.")
         yield
         logger.info("✅ [Lifespan] SHUTTING DOWN (after yield)...")
+        await close_mongo_connection()
     except Exception as e:
         logger.error(f"❌ [Lifespan] FATAL ERROR DURING STARTUP: {type(e).__name__} - {e}")
         logger.error(traceback.format_exc())

requirements.txt

@@ -37,4 +37,5 @@ itsdangerous
 Authlib[fastapi]
 httpx
 
-gunicorn
+gunicorn
+motor

routers/chat.py

@@ -9,7 +9,7 @@ import logging
 import uuid
 from redis.asyncio import Redis
 from datetime import datetime, timezone
-from db.mongoDB import conversations_collection
+from db.mongoDB import mongo_db
 from fastapi.responses import StreamingResponse
 from schemas.chat import Message,ConversationResponse
 from typing import List
@@ -135,7 +135,7 @@ async def delete_chat(chat_id: str, request: Request, user: UserOut = Depends(ge
     # Xóa chat
     delete_chat_from_redis(redis, chat_id)
     # Xóa hội thoại trong MongoDB
-    result =  conversations_collection.delete_one({"conversation_id": chat_id, "user_id": user.email})
+    result = await mongo_db.conversations.delete_one({"conversation_id": chat_id, "user_id": user.email})
     if result.deleted_count == 0:
         raise HTTPException(status_code=404, detail="Chat not found in MongoDB")
 
@@ -147,7 +147,7 @@ async def delete_chat(chat_id: str, request: Request, user: UserOut = Depends(ge
 async def get_conversations(user: UserOut = Depends(get_current_user)):
     try:
         logger.info(f"Attempting to get conversations for user: {user.email}")
-        db_conversations_cursor = conversations_collection.find({"user_id": user.email})
+        db_conversations_cursor = await mongo_db.conversations.find({"user_id": user.email})
 
         response_list = []
 
@@ -182,7 +182,7 @@ async def load_conversation_and_sync_redis(
 
     # 1. Kiểm tra hội thoại trong MongoDB
     try:
-        conversation_doc = conversations_collection.find_one(
+        conversation_doc =await mongo_db.conversations.find_one(
             {"conversation_id": chat_id, "user_id": current_user.email}
         )
         if not conversation_doc:

routers/user.py

@@ -12,7 +12,7 @@ from starlette.config import Config
 from authlib.integrations.starlette_client import OAuth
 import os
 import config
-from db.mongoDB import user_collection
+from db.mongoDB import mongo_db
 import uuid
 from passlib.context import CryptContext
 from datetime import datetime, timedelta, timezone
@@ -304,7 +304,7 @@ async def auth_google_callback(request: Request):
 
 
     # Kiểm tra xem user đã tồn tại trong DB chưa
-    db_user = user_collection.find_one({"email": user_email})
+    db_user = await mongo_db.users.find_one({"email": user_email})
 
     if not db_user:
         placeholder_password = f"google-oauth2|{uuid.uuid4()}"
@@ -314,7 +314,7 @@ async def auth_google_callback(request: Request):
 
 
         # Tạo user mới với thông tin từ Google
-        user_collection.insert_one({
+        await mongo_db.users.insert_one({
             "email": user_email,
             "username": username,
             "password": hashed_password,  # Mật khẩu tạm thời, sẽ không dùng đến
@@ -324,7 +324,7 @@ async def auth_google_callback(request: Request):
         })
 
         # Lấy lại user vừa tạo để đảm bảo có _id và các trường khác
-        db_user = user_collection.find_one({"email": user_email})
+        db_user =await mongo_db.users.find_one({"email": user_email})
         if not db_user: # Kiểm tra lại sau khi insert
              raise HTTPException(status_code=500, detail="Could not create and retrieve new user account.")
 
@@ -362,7 +362,7 @@ async def exchange_google_code_for_token(request: Request,response: Response, co
     redis_client.delete(redis_key) # Dùng một lần
     user_email = user_email_bytes.decode()
 
-    user_collection.update_one({
+    await mongo_db.users.update_one({
         "email": user_email
     }, {
         "$set": {
@@ -408,7 +408,7 @@ async def exchange_google_code_for_token(request: Request,response: Response, co
             path="/api/user/refresh-token",
         )
 
-    user_info = user_collection.find_one({"email": user_email})
+    user_info =await mongo_db.users.find_one({"email": user_email})
     user = {
         "email": user_info.get("email"),
         "username": user_info.get("username"),

services/auth_service.py

@@ -3,7 +3,7 @@ from jose import JWTError, jwt
 from datetime import datetime, timedelta, timezone
 from fastapi import HTTPException, Depends, Request, status
 from schemas.user import RegisterRequest, LoginRequest
-from db.mongoDB import user_collection, blacklist_collection
+from db.mongoDB import mongo_db
 from utils.utils import  verify_password
 from config import SECRET_KEY, ALGORITHM, ACCESS_TOKEN_EXPIRE_MINUTES
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
@@ -31,7 +31,7 @@ async def register_user(user: RegisterRequest):
 
     try:
                 # Kiểm tra username đã tồn tại chưa
-        existing_user = user_collection.find_one({"email": user.email})
+        existing_user =await mongo_db.users.find_one({"email": user.email})
         if existing_user:
             raise HTTPException(status_code=400, detail="Tài khoản đã tồn tại.")
 
@@ -40,7 +40,7 @@ async def register_user(user: RegisterRequest):
 
         avatar = await get_random_unsplash_image()
         # Lưu user mới
-        user_collection.insert_one({
+        await mongo_db.users.insert_one({
             "username": user.username,
             "password": hashed_password,
             "email": user.email,
@@ -53,8 +53,8 @@ async def register_user(user: RegisterRequest):
     except HTTPException as e:
         raise HTTPException(status_code=e.status_code, detail=e.detail)
 # Hàm xác thực đăng nhập
-def authenticate_user(request: LoginRequest):
-    user = user_collection.find_one({"email": request.email})
+async def authenticate_user(request: LoginRequest):
+    user =await mongo_db.users.find_one({"email": request.email})
     if not user:
         raise HTTPException(status_code=401, detail="Sai tài khoản hoặc mật khẩu")
 
@@ -64,8 +64,8 @@ def authenticate_user(request: LoginRequest):
     return user
 
 # Hàm tạo và trả về JWT token sau khi đăng nhập thành công
-def login_user(request: LoginRequest):
-    user = authenticate_user(request)
+async def login_user(request: LoginRequest):
+    user = await authenticate_user(request)
 
     accessToken = create_access_token(data={"sub": request.email})
 
@@ -101,7 +101,7 @@ async def logout_user(req: Request, credentials: HTTPAuthorizationCredentials =
             raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Access token không hợp lệ")
 
         # Blacklist access token
-        await blacklist_collection.insert_one({
+        await mongo_db.token_blacklist.insert_one({
             "token": token,
             "expires_at": datetime.fromtimestamp(exp, tz=timezone.utc)
         })
@@ -110,9 +110,9 @@ async def logout_user(req: Request, credentials: HTTPAuthorizationCredentials =
         refresh_token = req.cookies.get("refresh_token")
         if refresh_token:
             # Invalidate refresh token in database
-            user = await user_collection.find_one({"refresh_token": refresh_token})
+            user = await mongo_db.users.find_one({"refresh_token": refresh_token})
             if user:
-                await user_collection.update_one(
+                await mongo_db.users.update_one(
                     {"_id": user["_id"]},
                     {"$set": {"refresh_token": None, "refresh_token_expiry": None, "revoked": True}}
                 )

services/chat_service.py

@@ -7,7 +7,7 @@ import json
 from utils.utils import save_chat_to_redis, search_term_in_dictionary, minimal_preprocess_for_llm, save_chat_to_mongo, get_langchain_chat_history
 import os
 import logging
-from db.mongoDB import conversations_collection
+from db.mongoDB import mongo_db
 from datetime import datetime, timezone
 import asyncio
 
@@ -51,7 +51,7 @@ async def ask_question_service(app_state, request: QueryRequest, user: UserOut =
                 app_state.redis, chat_id, question_content, answer_def, current_utc_time, assistant_response_time
             )
             await save_chat_to_mongo(
-                conversations_collection, chat_id, user.email, question_content, answer_def, current_utc_time, assistant_response_time
+                mongo_db.conversations, chat_id, user.email, question_content, answer_def, current_utc_time, assistant_response_time
             )
             friendly_answer = f"Xin chào! Về câu hỏi '{question_content}' của bạn, tôi đã tìm thấy thông tin sau:\n\n{answer_def}\n\nHy vọng thông tin này hữu ích cho bạn. Bạn có muốn tìm hiểu thêm về chủ đề này hoặc có câu hỏi nào khác không? 😊"
             return AnswerResponse(
@@ -146,7 +146,7 @@ async def ask_question_service(app_state, request: QueryRequest, user: UserOut =
     # Lưu vào MongoDB
     # Chạy ngầm hoặc sau khi trả lời user để không làm chậm response (nếu có thể)
     await save_chat_to_mongo(
-        conversations_collection, chat_id, user.email, question_content, assistant_response_content, current_utc_time, assistant_response_time
+        mongo_db.conversations, chat_id, user.email, question_content, assistant_response_content, current_utc_time, assistant_response_time
     )
 
     end_time = time.time()
@@ -219,7 +219,7 @@ async def stream_chat_generator(
                     app_state.redis, chat_id, question_content, full_answer_for_saving, current_utc_time, assistant_response_time_dict
                 )
                 asyncio.create_task(save_chat_to_mongo( # Chạy nền
-                    conversations_collection, chat_id, user_email, question_content, full_answer_for_saving, current_utc_time, assistant_response_time_dict
+                    mongo_db.conversations, chat_id, user_email, question_content, full_answer_for_saving, current_utc_time, assistant_response_time_dict
                 ))
                 processing_time_dict = round(time.time() - start_time_total, 2)
                 logger.info(f"Stream: Dictionary answer for {chat_id} sent in {processing_time_dict:.2f}s.")
@@ -321,7 +321,7 @@ async def stream_chat_generator(
         )
         # Chạy lưu MongoDB ngầm để không block
         asyncio.create_task(save_chat_to_mongo(
-            conversations_collection, chat_id, user_email, question_content, full_answer_for_saving, current_utc_time, assistant_response_time
+            mongo_db.conversations, chat_id, user_email, question_content, full_answer_for_saving, current_utc_time, assistant_response_time
         ))
 
         # Cập nhật Langchain history (nếu chain memory không tự làm)

services/user_service.py

@@ -1,4 +1,4 @@
-from db.mongoDB import user_collection
+from db.mongoDB import mongo_db
 from fastapi import HTTPException, Request, Response
 import math
 from datetime import datetime, timedelta, timezone
@@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
 # Initialize password hashing context
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
-def get_users(skip: int = 0, limit: int = 100, query: dict = None):
+async def get_users(skip: int = 0, limit: int = 100, query: dict = None):
     """
     Lấy danh sách người dùng với phân trang và tìm kiếm nâng cao
 
@@ -37,7 +37,7 @@ def get_users(skip: int = 0, limit: int = 100, query: dict = None):
         search_query = query or {}
 
         # Thực hiện truy vấn với phân trang
-        users_cursor = user_collection.find(
+        users_cursor =await mongo_db.users.find(
             search_query,
             {"_id": 0, "password": 0}  # Loại bỏ các trường nhạy cảm
         ).skip(skip).limit(limit)
@@ -56,7 +56,7 @@ def get_users(skip: int = 0, limit: int = 100, query: dict = None):
             detail=f"Lỗi khi lấy danh sách người dùng: {str(e)}"
         )
 
-def count_users(query: dict = None):
+async def count_users(query: dict = None):
     """
     Đếm tổng số người dùng thỏa mãn điều kiện tìm kiếm
 
@@ -68,7 +68,7 @@ def count_users(query: dict = None):
     """
     try:
         search_query = query or {}
-        return user_collection.count_documents(search_query)
+        return await mongo_db.users.count_documents(search_query)
     except Exception as e:
         logger.error(f"Lỗi khi đếm người dùng: {str(e)}")
         raise HTTPException(
@@ -112,7 +112,7 @@ async def get_paginated_users(
         sort_criteria = [(sort_by, sort_order)]
 
         # Thực hiện truy vấn
-        users_cursor = user_collection.find(
+        users_cursor =await mongo_db.users.find(
             query,
             {"_id": 0, "password": 0}  # Loại bỏ các trường nhạy cảm
         ).sort(sort_criteria).skip(skip).limit(limit)
@@ -121,7 +121,7 @@ async def get_paginated_users(
         user_list =  users_cursor.to_list(length=limit)
 
         # Đếm tổng số bản ghi
-        total = user_collection.count_documents(query)
+        total =await mongo_db.users.count_documents(query)
 
         # Tính toán thông tin phân trang
         total_pages = math.ceil(total / limit) if limit > 0 else 0
@@ -156,7 +156,7 @@ async def get_current_user_profile(email: str):
         dict: Thông tin profile của người dùng
     """
     try:
-        user =  user_collection.find_one({"email": email}, {"_id": 0, "password": 0})
+        user =await  mongo_db.users.find_one({"email": email}, {"_id": 0, "password": 0})
         logger.info(f"check user user_service: {user}")
         if not user:
             raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Người dùng không tồn tại")
@@ -171,7 +171,7 @@ async def get_current_user_profile(email: str):
         )
 
 
-def delete_user(user_id: str):
+async def delete_user(user_id: str):
     """
     Xóa người dùng theo ID
 
@@ -182,7 +182,7 @@ def delete_user(user_id: str):
         bool: True nếu xóa thành công, False nếu không tìm thấy người dùng
     """
     try:
-        result = user_collection.delete_one({"_id": user_id})
+        result =await mongo_db.users.delete_one({"_id": user_id})
         return result.deleted_count > 0
     except Exception as e:
         logger.error(f"Lỗi khi xóa người dùng: {str(e)}")
@@ -191,7 +191,7 @@ def delete_user(user_id: str):
             detail=f"Lỗi khi xóa người dùng: {str(e)}"
         )
 
-def change_password(email: str, current_password: str, new_password: str):
+async def change_password(email: str, current_password: str, new_password: str):
     """
     Đổi mật khẩu của người dùng
 
@@ -204,7 +204,7 @@ def change_password(email: str, current_password: str, new_password: str):
         bool: True nếu đổi mật khẩu thành công, False nếu không thành công
     """
     try:
-        user = user_collection.find_one({"email": email})
+        user =await mongo_db.users.find_one({"email": email})
         if not user:
             raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Người dùng không tồn tại")
 
@@ -216,7 +216,7 @@ def change_password(email: str, current_password: str, new_password: str):
         # Hash the new password
         hashed_password = pwd_context.hash(new_password)
         # Update the password in the database
-        result = user_collection.update_one({"email": email}, {"$set": {"password": hashed_password}})
+        result =await mongo_db.users.update_one({"email": email}, {"$set": {"password": hashed_password}})
         return result.modified_count > 0
     except HTTPException as he:
         raise he
@@ -242,13 +242,13 @@ async def reset_password_request(email: str) -> bool:
     """
     try:
         # Check if user exists
-        user =  user_collection.find_one({"email": email.lower()})
+        user = await  mongo_db.users.find_one({"email": email.lower()})
         if not user:
             # Return True even if user not found to prevent email enumeration
             return True
 
         # Check for rate limiting (e.g., max 3 requests per hour)
-        reset_requests =  user_collection.count_documents({
+        reset_requests = await mongo_db.users.count_documents({
             "email": email.lower(),
             "reset_password_timestamp": {
                 "$gte": datetime.now(tz=timezone.utc) - timedelta(hours=1)
@@ -265,7 +265,7 @@ async def reset_password_request(email: str) -> bool:
         expiry = datetime.now() + timedelta(minutes=10)
 
         # Store reset token and timestamp in database
-        user_collection.update_one(
+        await mongo_db.users.update_one(
             {"_id": ObjectId(user["_id"])},
             {
                 "$set": {
@@ -312,7 +312,7 @@ async def reset_password(code: str, newPassword: str) -> bool:
     """
     try:
         # Find user by reset token
-        user =  user_collection.find_one({"reset_password_code": code})
+        user =await  mongo_db.users.find_one({"reset_password_code": code})
         if not user:
             logger.warning(f"Code không hợp lệ: {code}")
             return False
@@ -334,7 +334,7 @@ async def reset_password(code: str, newPassword: str) -> bool:
         hashed_password = pwd_context.hash(newPassword)
 
         # Update user's password and clear reset token
-        result =  user_collection.update_one(
+        result =await  mongo_db.users.update_one(
             {"_id": ObjectId(user["_id"])},
             {
                 "$set": {"password": hashed_password},
@@ -380,7 +380,7 @@ async def generate_and_store_verification_code(email: str) -> bool:
         expiry = datetime.now() + timedelta(minutes=10)
 
         # Store code and expiry in database
-        result =  user_collection.update_one(
+        result = await mongo_db.users.update_one(
             {"email": email.lower()},
             {
                 "$set": {
@@ -425,12 +425,12 @@ async def verify_login_code(email: str, code: str, res: Response): # Bỏ kiểu
         HTTPException: Nếu mã không hợp lệ, đã hết hạn hoặc có lỗi hệ thống.
     """
     try:
-        user =  user_collection.find_one({ # Sử dụng await nếu user_collection là async (ví dụ Motor)
+        user = await mongo_db.users.find_one({ # Sử dụng await nếu mongo_db.users là async (ví dụ Motor)
             "email": email.lower(),
             "login_verification_code": code
         })
-        # Nếu user_collection là đồng bộ (ví dụ PyMongo)
-        # user = user_collection.find_one({ ... })
+        # Nếu mongo_db.users là đồng bộ (ví dụ PyMongo)
+        # user = mongo_db.users.find_one({ ... })
 
         if not user:
             logger.warning(f"Mã xác minh không hợp lệ: {code} cho email: {email}")
@@ -443,7 +443,7 @@ async def verify_login_code(email: str, code: str, res: Response): # Bỏ kiểu
         if not expiry or expiry < datetime.now(expiry.tzinfo if expiry.tzinfo else None): # So sánh aware với aware, naive với naive
             logger.warning(f"Mã xác minh đã hết hạn cho email: {email}")
             # Xóa mã đã hết hạn để tránh sử dụng lại
-            user_collection.update_one(
+            await mongo_db.users.update_one(
                 {"_id": ObjectId(user["_id"])},
                 {
                     "$unset": {
@@ -509,7 +509,7 @@ async def verify_login_code(email: str, code: str, res: Response): # Bỏ kiểu
         # --------------------
 
         # Clear verification code và cập nhật last_login
-        update_result =  user_collection.update_one( # Sử dụng await nếu là async
+        await mongo_db.users.update_one( # Sử dụng await nếu là async
             {"_id": ObjectId(user["_id"])},
             {
                 "$unset": {
@@ -567,7 +567,7 @@ async def authenticate_user(request: LoginRequest) -> dict:
     Raises:
         HTTPException: Nếu thông tin đăng nhập không hợp lệ.
     """
-    user =  user_collection.find_one({"email": request.email.lower()})
+    user = await mongo_db.users.find_one({"email": request.email.lower()})
     if not user or not pwd_context.verify(request.password, user["password"]):
         logger.warning(f"Xác thực thất bại cho email: {request.email}")
         raise HTTPException(
@@ -603,7 +603,7 @@ async def refresh_access_token(req: Request  , res: Response ) -> dict:
             )
 
         # Find user by refresh token in database
-        user =  user_collection.find_one({"refresh_token": refresh_token})
+        user = await mongo_db.users.find_one({"refresh_token": refresh_token})
 
 
         if not user:
@@ -640,7 +640,7 @@ async def refresh_access_token(req: Request  , res: Response ) -> dict:
         new_refresh_token = await create_refresh_token(email)
 
         # Update database with new refresh token
-        user_collection.update_one(
+        await mongo_db.users.update_one(
             {"_id": user["_id"]},
             {
                 "$set": {
@@ -702,7 +702,7 @@ async def verify_forgot_password_code(email: str, code: str) -> dict:
 
     try:
         # Find user by email and code
-        user =  user_collection.find_one({
+        user =await  mongo_db.users.find_one({
             "email": email.lower(),
             "reset_password_code": code
         })

utils/utils.py

@@ -12,7 +12,7 @@ from jose import jwt
 from config import SECRET_KEY, ALGORITHM, ACCESS_TOKEN_EXPIRE_MINUTES
 from typing import List, Dict,  Optional
 from unidecode import unidecode
-from db.mongoDB import user_collection
+from db.mongoDB import mongo_db
 import secrets
 from fastapi import HTTPException, status
 from langchain_community.chat_message_histories import RedisChatMessageHistory
@@ -233,7 +233,7 @@ async def create_refresh_token(email: str) -> str:
 
 
         # Store refresh token in database
-        result =  user_collection.update_one(
+        result = await mongo_db.users.update_one(
             {"email": email.lower()},
             {
                 "$set": {