Update app.py

app.py

@@ -117,7 +117,7 @@ def rag_chain(llm, prompt, db):
 def wandb_log(prompt, completion, rag_option):
     wandb.login(key = wandb_api_key)
     wandb.init(project = "openai-llm-rag", config = config)
-    wandb.log({"prompt": str(completion), "completion": str(completion), "rag_option": rag_option})
+    wandb.log({"prompt": str(prompt), "completion": str(completion), "rag_option": rag_option})
     wandb.finish()
 
 def invoke(openai_api_key, rag_option, prompt):