Update app.py

app.py

@@ -122,7 +122,7 @@ def rag_chain(llm, prompt, db):
 
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
     if (rag_option == RAG_OFF or str(status_msg) != ""):
-        result = completion["text"]
+        result = completion
     else:
         result = completion["result"]
         docs_meta = str([doc.metadata for doc in completion["source_documents"]])
@@ -143,7 +143,7 @@ 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 == RAG_OFF) else rag_template) if (str(status_msg) == "") else ""),
                   "docs_meta": "" if (rag_option == RAG_OFF or str(status_msg) != "") else docs_meta},
-        outputs = {"result": result,
+        outputs = {#"result": result,
                    "completion": completion},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
@@ -181,7 +181,7 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             completion, chain = llm_chain(llm, prompt)
-            result = completion["text"]
+            result = completion
     except Exception as e:
         status_msg = e
         raise gr.Error(e)