Update app.py

app.py

@@ -124,10 +124,10 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
     if (str(err_msg) != ""):
         result = ""
     elif (rag_option == RAG_OFF):
-        result = completion.generations[0][0].text if (completion.generations[0] != None and completion.generations[0][0] != None) else ""
+        result = completion.generations[0][0].text if (completion.generations[0] != None and 
+                                                       completion.generations[0][0] != None) else ""
     else:
         result = completion["result"]
-        docs_meta = str([doc.metadata for doc in completion["source_documents"]])
     wandb.init(project = "openai-llm-rag")
     trace = Trace(
         kind = "chain",
@@ -143,8 +143,7 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
         },
         inputs = {"rag_option": rag_option if (str(err_msg) == "") else "",
                   "prompt": str(prompt if (str(err_msg) == "") else ""),
-                  "prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(err_msg) == "") else ""),
-                  "docs_meta": "" if (str(err_msg) != "" or rag_option == RAG_OFF) else docs_meta},
+                  "prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(err_msg) == "") else "")}
         outputs = {"result": result,
                    "completion": str(completion)},
         start_time_ms = start_time_ms,
@@ -183,7 +182,8 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             completion, chain = llm_chain(llm, prompt)
-            result = completion.generations[0][0].text if (completion.generations[0] != None and completion.generations[0][0] != None) else ""
+            result = completion.generations[0][0].text if (completion.generations[0] != None and 
+                                                           completion.generations[0][0] != None) else ""
     except Exception as e:
         err_msg = e
         raise gr.Error(e)