logs

auditqa/utils.py

@@ -32,13 +32,18 @@ def save_logs(JSON_DATASET_PATH: str, logs: dict, feedback: str = None) -> None:
         this is to get the usage statistics of app and evaluate model performances.
         Also saves user feedback (when provided).
     """
-    if feedback:
-        logs["feedback"] = feedback
+    try:
+        if feedback:
+            logs["feedback"] = feedback
+            logging.debug(f"Adding feedback to logs: {feedback}")
         
-    with open(JSON_DATASET_PATH, 'a') as f:
-        json.dump(logs, f)
-        f.write("\n")
-    print("logging done")
+        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}")
+    except Exception as e:
+        logging.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':