Update app.py

app.py

@@ -4,7 +4,7 @@ import logging, os, sys, threading, time
 from dotenv import load_dotenv, find_dotenv
 
 from rag_langchain import LangChainRAG
-from rag_llamaindex import LlamaIndexRAG
+#from rag_llamaindex import LlamaIndexRAG
 from trace import trace_wandb
 
 lock = threading.Lock()
@@ -43,9 +43,9 @@ def invoke(openai_api_key, prompt, rag_option):
             if (rag_option == RAG_LANGCHAIN):
                 rag = LangChainRAG()
                 rag.ingestion(config)
-            elif (rag_option == RAG_LLAMAINDEX):
-                rag = LlamaIndexRAG()
-                rag.ingestion(config)
+            #elif (rag_option == RAG_LLAMAINDEX):
+            #    rag = LlamaIndexRAG()
+            #    rag.ingestion(config)
     
         completion = ""
         result = ""
@@ -59,9 +59,9 @@ def invoke(openai_api_key, prompt, rag_option):
                 rag = LangChainRAG()
                 completion, callback = rag.rag_chain(config, prompt)
                 result = completion["result"]
-            elif (rag_option == RAG_LLAMAINDEX):
-                rag = LlamaIndexRAG()
-                result, callback = rag.retrieval(config, prompt)
+            #elif (rag_option == RAG_LLAMAINDEX):
+            #    rag = LlamaIndexRAG()
+            #    result, callback = rag.retrieval(config, prompt)
             else:
                 rag = LangChainRAG()
                 completion, callback = rag.llm_chain(config, prompt)