Update app.py

app.py

@@ -3,7 +3,7 @@ import os, time
 
 from dotenv import load_dotenv, find_dotenv
 
-from rag_langchain import llm_chain, rag_chain, rag_ingestion_langchain
+from rag_langchain import LangChainRAG
 from rag_llamaindex import LlamaIndexRAG
 from trace import trace_wandb
 
@@ -33,7 +33,8 @@ def invoke(openai_api_key, prompt, rag_option):
     
     if (RAG_INGESTION):
         if (rag_option == RAG_LANGCHAIN):
-            rag_ingestion_llangchain(config)
+            rag = LangChainRAG()
+            rag.ingestion(config)
         elif (rag_option == RAG_LLAMAINDEX):
             rag = LlamaIndexRAG()
             rag.ingestion(config)
@@ -47,14 +48,16 @@ def invoke(openai_api_key, prompt, rag_option):
         start_time_ms = round(time.time() * 1000)
 
         if (rag_option == RAG_LANGCHAIN):
-            completion, chain, callback = rag_chain(config, prompt)
+            rag = LangChainRAG()
+            completion, chain, callback = rag.rag_chain(config, prompt)
 
             result = completion["result"]
         elif (rag_option == RAG_LLAMAINDEX):
             rag = LlamaIndexRAG()
             result = rag.retrieval(config, prompt)
         else:
-            completion, chain, callback = llm_chain(config, prompt)
+            rag = LangChainRAG()
+            completion, chain, callback = rag.llm_chain(config, prompt)
             
             if (completion.generations[0] != None and 
                 completion.generations[0][0] != None):