Update app.py

app.py

@@ -4,7 +4,7 @@ import os, time
 from dotenv import load_dotenv, find_dotenv
 
 from rag_langchain import llm_chain, rag_chain, rag_ingestion_langchain
-from rag_llamaindex import LlamaIndexRAG #rag_ingestion_llamaindex, rag_retrieval
+from rag_llamaindex import LlamaIndexRAG
 from trace import trace_wandb
 
 _ = load_dotenv(find_dotenv())
@@ -35,7 +35,8 @@ def invoke(openai_api_key, prompt, rag_option):
         if (rag_option == RAG_LANGCHAIN):
             rag_ingestion_llangchain(config)
         elif (rag_option == RAG_LLAMAINDEX):
-            rag_ingestion_llamaindex(config)
+            rag = LlamaIndexRAG()
+            rag.ingestion(config)
     
     completion = ""
     result = ""
@@ -50,9 +51,8 @@ def invoke(openai_api_key, prompt, rag_option):
 
             result = completion["result"]
         elif (rag_option == RAG_LLAMAINDEX):
-            #result = rag_retrieval(config, prompt)
             rag = LlamaIndexRAG()
-            result = rag.rag_retrieval(config, prompt)
+            result = rag.retrieval(config, prompt)
         else:
             completion, chain, callback = llm_chain(config, prompt)