Update app.py

app.py

@@ -124,7 +124,7 @@ 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.content
+        result = completion
     else:
         result = completion["result"]
         docs_meta = str([doc.metadata for doc in completion["source_documents"]])
@@ -184,7 +184,8 @@ def invoke(openai_api_key, rag_option, prompt):
         else:
             completion, chain = llm_chain(llm, prompt)
             print(completion)
-            result = completion.content
+            print(completion.message)
+            result = completion
     except Exception as e:
         err_msg = e
         raise gr.Error(e)