Update app.py

app.py

@@ -120,8 +120,8 @@ def wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_n
     trace = Trace(
         name = chain_name,
         kind = "chain",
-        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": config["chunk_overlap"],
             "chunk_size": config["chunk_size"],
@@ -157,7 +157,7 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
             prompt_template = rag_template
-            chain_name = RetrievalQA.__class__
+            chain_name = "RetrievalQA"
         elif (rag_option == "MongoDB"):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
@@ -165,12 +165,12 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
             prompt_template = rag_template
-            chain_name = RetrievalQA.__class__
+            chain_name = "RetrievalQA"
         else:
             result = llm_chain(llm, prompt)
             completion = result
             prompt_template = llm_template
-            chain_name = LLMChain.__class__
+            chain_name = "LLMChain"
     except Exception as e:
         status_msg = e
         raise gr.Error(e)