Update app.py

app.py

@@ -120,10 +120,7 @@ def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms
         result = completion
     else:
         result = completion["result"]
-        docs_meta = str([doc.metadata for doc in completion["source_documents"]])
-        #docs[] = completion["source_documents"][0].metadata
-        #docs_meta_1 = completion["source_documents"][1].metadata
-        #docs_meta_2 = completion["source_documents"][2].metadata
+        docs_meta = json.loads([doc.metadata for doc in completion["source_documents"]])
     wandb.init(project = "openai-llm-rag")
     trace = Trace(
         kind = "chain",
@@ -141,9 +138,6 @@ def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms
                   "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 ""),
                   "docs_meta": "" if (rag_option == "Off" or str(status_msg) != "") else docs_meta},
-                  #"docs_meta_0": "" if (rag_option == "Off" or str(status_msg) != "") else str(docs_meta_0),
-                  #"docs_meta_1": "" if (rag_option == "Off" or str(status_msg) != "") else str(docs_meta_1),
-                  #"docs_meta_2": "" if (rag_option == "Off" or str(status_msg) != "") else str(docs_meta_2)},
         outputs = {"result": result},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms