fix

db/mongoDB.py

@@ -51,7 +51,7 @@ async def connect_to_mongo():
 
         # 3. Gán các đối tượng database và collection
         mongo_db.db = mongo_db.client[config.DB_NAME]
-        mongo_db.user = mongo_db.db["users"]
+        mongo_db.users = mongo_db.db["users"]
         mongo_db.token_blacklist = mongo_db.db["token_blacklist"]
         mongo_db.conversations = mongo_db.db["conversations"]
 
@@ -63,7 +63,9 @@ async def connect_to_mongo():
             await mongo_db.token_blacklist.create_index("expires_at", expireAfterSeconds=0)
             logger.info("🔸 Successfully created TTL index for 'expires_at' in 'token_blacklist'.")
 
+
         logger.info("✅ MongoDB connection successful and collections are ready.")
+        return mongo_db
 
     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)

dependencies.py

@@ -70,8 +70,8 @@ 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 await 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.")
+    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.", mongo_db.users)
         raise HTTPException(status_code=500, detail="Lỗi kết nối tới database.")
 
     app_state.google_api_key = os.environ.get("GOOGLE_API_KEY")

main.py

@@ -24,13 +24,11 @@ logger = logging.getLogger(__name__)
 async def lifespan(app: FastAPI):
     logger.info("✅ [Lifespan] STARTING UP...")
 
-    await connect_to_mongo()
-
-
     current_app_state_instance = AppState()
     initialization_successful = False
     try:
         logger.info("✅ [Lifespan] Calling initialize_api_components...")
+        await connect_to_mongo()
         await initialize_api_components(current_app_state_instance)
         app.state.app_state = current_app_state_instance
         initialization_successful = True
@@ -99,6 +97,6 @@ if __name__ == "__main__":
         host=config.API_HOST if hasattr(config, 'API_HOST') else "0.0.0.0",
         port=int(config.API_PORT) if hasattr(config, 'API_PORT') else 5000,
         reload=True, # reload=True chỉ nên dùng cho development
-        timeout_keep_alive=180, # Tăng thời gian giữ kết nối
+        timeout_keep_alive=120, # Tăng thời gian giữ kết nối
         log_level="info" # Hoặc "debug" nếu bạn muốn nhiều thông tin hơn
     )

schemas/chat.py

@@ -14,6 +14,7 @@ class AppState(BaseModel):
     redis: Optional[Any] = None
     retriever: Optional[Any] = None
     weaviateDB: Optional[Any] = None
+    # mongoDB: Optional[Any] = None
     reranker: Optional[Any] = None
 
 class QueryRequest(BaseModel):