Update app.py

app.py

@@ -13,11 +13,11 @@ RAG_CHROMA  = "Chroma"
 RAG_MONGODB = "MongoDB"
 
 config = {
-    "chunk_overlap": 150,
-    "chunk_size": 1500,
-    "k": 3,
-    "model_name": "gpt-4-0613",
-    "temperature": 0,
+    "chunk_overlap": 150,       # document splitting
+    "chunk_size": 1500,         # document splitting
+    "k": 3,                     # document retrieval
+    "model_name": "gpt-4-0613", # llm
+    "temperature": 0,           # llm
 }
 
 def invoke(openai_api_key, rag_option, prompt):
@@ -38,26 +38,20 @@ def invoke(openai_api_key, rag_option, prompt):
     try:
         start_time_ms = round(time.time() * 1000)
 
-        if (rag_option == RAG_CHROMA):
-            #splits = document_loading_splitting()
-            #document_storage_chroma(splits)
-            
-            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)
-            result = completion["result"]
-        else:
-            completion, chain = llm_chain(openai_api_key, prompt)
+        if (rag_option == RAG_OFF):
+            completion, chain = llm_chain(config, openai_api_key, prompt)
             
             if (completion.generations[0] != None and completion.generations[0][0] != None):
                 result = completion.generations[0][0].text
                 generation_info = completion.generations[0][0].generation_info
 
             llm_output = completion.llm_output
+        else:
+            #splits = document_loading_splitting()
+            #document_storage_mongodb(splits)
+            
+            completion, chain = rag_chain(config, openai_api_key, rag_option, prompt)
+            result = completion["result"]
     except Exception as e:
         err_msg = e