Update app.py

app.py

@@ -3,7 +3,7 @@ import os, time
 
 from dotenv import load_dotenv, find_dotenv
 
-from rag import llm_chain, rag_chain
+from rag import llm_chain, rag_chain, rag_batch
 from trace import wandb_trace
 
 _ = load_dotenv(find_dotenv())
@@ -47,8 +47,7 @@ def invoke(openai_api_key, rag_option, prompt):
 
             llm_output = completion.llm_output
         else:
-            #splits = document_loading_splitting()
-            #document_storage_mongodb(splits)
+            #rag_batch(config)
             
             completion, chain = rag_chain(config, openai_api_key, rag_option, prompt)
             result = completion["result"]