Update trace.py

trace.py

@@ -18,16 +18,16 @@ def trace_wandb(config,
     trace = Trace(
         kind = "LLM",
         name = "Context-Aware Reasoning Application",
-        status_code = "success" if (str(err_msg) == "") else "error",
-        status_message = str(err_msg),
-        inputs = {"prompt": str(prompt),
-                  "rag_option": str(rag_option),
-                  "config": str(config)
-                 } if (str(err_msg) == "") else {},
-        outputs = {"result": str(result),
-                   "callback": str(callback),
-                   "completion": str(completion)
-                  } if (str(err_msg) == "") else {},
+        status_code = "success" if (err_msg == "") else "error",
+        status_message = err_msg,
+        inputs = {"prompt": prompt,
+                  "rag_option": rag_option,
+                  "config": config
+                 } if (err_msg == "") else {},
+        outputs = {"result": result,
+                   "callback": callback,
+                   "completion": completion
+                  } if (err_msg == "") else {},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )