LogicGoInfotechSpaces commited on
Commit
3220e2a
·
1 Parent(s): 1ff09f8

Fix MongoDB logging: Add null checks and ensure consistent log format

Browse files

- Add null check for mongo_logs before insert_one in all endpoints
- Ensure log format matches expected structure:
- timestamp: datetime.utcnow() (stored as datetime in MongoDB)
- ts: int(time.time()) (integer timestamp)
- response_time_ms: float (response time in milliseconds)
- All fields match expected api_logs collection format

Files changed (1) hide show
  1. api/main.py +21 -12
api/main.py CHANGED
@@ -701,10 +701,13 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
701
  }
702
  if error_msg:
703
  log_doc["error"] = error_msg
704
- try:
705
- mongo_logs.insert_one(log_doc)
706
- except Exception as mongo_err:
707
- log.error("Mongo log insert failed: %s", mongo_err)
 
 
 
708
 
709
 
710
  @app.post("/inpaint-multipart")
@@ -872,10 +875,13 @@ def inpaint_multipart(
872
  }
873
  if error_msg:
874
  log_doc["error"] = error_msg
875
- try:
876
- mongo_logs.insert_one(log_doc)
877
- except Exception as mongo_err:
878
- log.error("Mongo log insert failed: %s", mongo_err)
 
 
 
879
 
880
 
881
  @app.post("/remove-pink")
@@ -1014,10 +1020,13 @@ def remove_pink_segments(
1014
  }
1015
  if error_msg:
1016
  log_doc["error"] = error_msg
1017
- try:
1018
- mongo_logs.insert_one(log_doc)
1019
- except Exception as mongo_err:
1020
- log.error("Mongo log insert failed: %s", mongo_err)
 
 
 
1021
 
1022
 
1023
  @app.get("/download/{filename}")
 
701
  }
702
  if error_msg:
703
  log_doc["error"] = error_msg
704
+ if mongo_logs is not None:
705
+ try:
706
+ mongo_logs.insert_one(log_doc)
707
+ except Exception as mongo_err:
708
+ log.error("Mongo log insert failed: %s", mongo_err)
709
+ else:
710
+ log.warning("MongoDB not configured, skipping log insert")
711
 
712
 
713
  @app.post("/inpaint-multipart")
 
875
  }
876
  if error_msg:
877
  log_doc["error"] = error_msg
878
+ if mongo_logs is not None:
879
+ try:
880
+ mongo_logs.insert_one(log_doc)
881
+ except Exception as mongo_err:
882
+ log.error("Mongo log insert failed: %s", mongo_err)
883
+ else:
884
+ log.warning("MongoDB not configured, skipping log insert")
885
 
886
 
887
  @app.post("/remove-pink")
 
1020
  }
1021
  if error_msg:
1022
  log_doc["error"] = error_msg
1023
+ if mongo_logs is not None:
1024
+ try:
1025
+ mongo_logs.insert_one(log_doc)
1026
+ except Exception as mongo_err:
1027
+ log.error("Mongo log insert failed: %s", mongo_err)
1028
+ else:
1029
+ log.warning("MongoDB not configured, skipping log insert")
1030
 
1031
 
1032
  @app.get("/download/{filename}")