Update app.py

app.py

@@ -115,8 +115,9 @@ def rag_chain(llm, prompt, db):
                                             retriever = db.as_retriever(search_kwargs = {"k": 3}), 
                                             return_source_documents = True)
     completion = rag_chain({"query": prompt})
-    wandb.log({"prompt": prompt, "completion": completion})
-    return completion["result"]
+    completion_result = completion["result"]
+    wandb.log({"prompt": prompt, "completion": completion_result})
+    return completion_result
 
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):