Update app.py

app.py

@@ -3,7 +3,7 @@ import os, time
 
 from dotenv import load_dotenv, find_dotenv
 
-from rag import get_llm, llm_chain, rag_chain
+from rag import llm_chain, rag_chain
 from trace import wandb_trace
 
 _ = load_dotenv(find_dotenv())
@@ -34,14 +34,12 @@ def invoke(openai_api_key, rag_option, prompt):
             #splits = document_loading_splitting()
             #document_storage_chroma(splits)
             
-            db = document_retrieval_chroma(openai_api_key, prompt)
             completion, chain = rag_chain(openai_api_key, prompt, db)
             result = completion["result"]
         elif (rag_option == RAG_MONGODB):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
             
-            db = document_retrieval_mongodb(openai_api_key, prompt)
             completion, chain = rag_chain(openai_api_key, prompt, db)
             result = completion["result"]
         else: