logs

auditqa/utils.py

@@ -9,7 +9,12 @@ import requests
 from datetime import datetime
 from uuid import uuid4
 
-
+# TESTING DEBUG LOG
+from auditqa.logging_config import setup_logging
+setup_logging()
+import logging
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)  # Ensure debug logging is enabled
 
 
 # def save_logs(scheduler, JSON_DATASET_PATH, logs, feedback=None) -> None:
@@ -40,11 +45,12 @@ def save_logs(JSON_DATASET_PATH: str, logs: dict, feedback: str = None) -> None:
         with open(JSON_DATASET_PATH, 'a') as f:
             json.dump(logs, f)
             f.write("\n")
-        logging.info(f"Successfully saved logs to {JSON_DATASET_PATH}")
+        logger.info(f"Successfully saved logs to {JSON_DATASET_PATH}")
     except Exception as e:
-        logging.error(f"Failed to save logs to {JSON_DATASET_PATH}: {str(e)}")
+        logger.error(f"Failed to save logs to {JSON_DATASET_PATH}: {str(e)}")
         raise
 
+
 def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
     if type == 'NVIDIA':
         messages =  [{"role": "system", "content": SYSTEM_PROMPT},