Update app.py

app.py

@@ -8,9 +8,7 @@ from trace import wandb_trace
 
 _ = load_dotenv(find_dotenv())
 
-RAG_OFF     = "Off"
-RAG_CHROMA  = "Chroma"
-RAG_MONGODB = "MongoDB"
+RAG_BATCH = False # document loading, splitting, persisting
 
 config = {
     "chunk_overlap": 150,       # document splitting
@@ -20,6 +18,10 @@ config = {
     "temperature": 0,           # llm
 }
 
+RAG_OFF     = "Off"
+RAG_CHROMA  = "Chroma"
+RAG_MONGODB = "MongoDB"
+
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -47,7 +49,8 @@ def invoke(openai_api_key, rag_option, prompt):
 
             llm_output = completion.llm_output
         else:
-            #rag_batch(config)
+            if (RAG_BATCH):
+                rag_batch(config)
             
             completion, chain = rag_chain(config, openai_api_key, rag_option, prompt)
             result = completion["result"]