Update app.py

app.py

@@ -34,13 +34,13 @@ def invoke(openai_api_key, rag_option, prompt):
             #splits = document_loading_splitting()
             #document_storage_chroma(splits)
             
-            completion, chain = rag_chain(openai_api_key, prompt, db)
+            completion, chain = rag_chain(openai_api_key, prompt)
             result = completion["result"]
         elif (rag_option == RAG_MONGODB):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
             
-            completion, chain = rag_chain(openai_api_key, prompt, db)
+            completion, chain = rag_chain(openai_api_key, prompt)
             result = completion["result"]
         else:
             completion, chain = llm_chain(openai_api_key, prompt)
@@ -57,7 +57,7 @@ def invoke(openai_api_key, rag_option, prompt):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        wandb_trace(rag_option, 
+        wandb_trace(rag_option == RAG_OFF, 
                     prompt, 
                     completion, 
                     result,