Update app.py

app.py

@@ -138,9 +138,9 @@ def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms
     else:
         result = completion["result"]
         documents = completion["source_documents"]
-        document_0 = completion["source_documents"][0].metadata.source
-        document_1 = completion["source_documents"][1].metadata.source
-        document_2 = completion["source_documents"][2].metadata.source
+        document_0 = completion["source_documents"][0].metadata["source"]
+        document_1 = completion["source_documents"][1].metadata["source"]
+        document_2 = completion["source_documents"][2].metadata["source"]
     trace = Trace(
         kind = "chain",
         name = type(chain).__name__ if (chain != None) else "",