Update app.py

app.py

@@ -121,8 +121,8 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
     trace = Trace(
         kind = "chain",
         name = "LLMChain" if (rag_option == "Off") else "RetrievalQA",
-        status_code = "SUCCESS" if (status_msg == "") else "ERROR",
-        status_message = status_msg,
+        status_code = "SUCCESS" if (str(status_msg) == "") else "ERROR",
+        status_message = str(status_msg),
         metadata={
             "chunk_overlap": "" if (rag_option == "Off") else config["chunk_overlap"],
             "chunk_size": "" if (rag_option == "Off") else config["chunk_size"],
@@ -130,9 +130,9 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
             "model": config["model"],
             "temperature": config["temperature"],
         },
-        inputs = {"rag_option": rag_option if (status_msg == "") else "",
-                  "prompt": str(prompt if (status_msg == "") else ""), 
-                  "prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (status_msg == "") else "")},
+        inputs = {"rag_option": rag_option if (str(status_msg) == "") else "",
+                  "prompt": str(prompt if (str(status_msg) == "") else ""), 
+                  "prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
         outputs = {"completion": str(completion)},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms