Update src/Database/operations.py

src/Database/operations.py

@@ -6,6 +6,8 @@ from typing import Any, Dict, List, Optional
 from pymongo import DESCENDING
 from Database.connections import get_img_gen_collection
 
+from core.logger import logger
+
 
 def insert_img_gen_run_result(
     *,
@@ -40,7 +42,7 @@ def insert_img_gen_run_result(
         "created_by": created_by,
         "created_at": datetime.utcnow(),
     }
-    log.info(doc)
+    logger.info(doc)
     res = col.insert_one(doc)
     return str(res.inserted_id)