Update app.py

app.py

@@ -146,7 +146,7 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
                   "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},
         outputs = {"result": result,
-                   "completion": completion},
+                   "completion": str(completion)},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )
@@ -183,6 +183,8 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             completion, chain = llm_chain(llm, prompt)
+            print(completion)
+            print(completion.llm_output)
             result = completion.text
     except Exception as e:
         err_msg = e