Update app.py

app.py

@@ -143,8 +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,
-                   "completion": completion},
+        outputs = {"result": result},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )
@@ -180,8 +179,9 @@ def invoke(openai_api_key, rag_option, prompt):
             completion, chain = rag_chain(llm, prompt, db)
             result = completion["result"]
         else:
-            completion, chain = llm_chain(llm, prompt)
-            result = completion
+            result, chain = llm_chain(llm, prompt)
+            print(result)
+            completion = result
     except Exception as e:
         status_msg = e
         raise gr.Error(e)