Update rag.py

rag.py

@@ -1,8 +1,8 @@
-from langchain.document_loaders import DirectoryLoader
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
-from langchain.llms import Ollama  
+from langchain_community.document_loaders import DirectoryLoader
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
+from langchain_community.retrievers import BM25Retriever, EnsembleRetriever
+from langchain_community.llms import Ollama
 
 loader = DirectoryLoader('.', glob="all_dialogues.txt") 
 docs = loader.load()
@@ -20,7 +20,7 @@ db = FAISS.load_local(
     allow_dangerous_deserialization=True  
 )
 
-from langchain.retrievers import BM25Retriever, EnsembleRetriever
+
 
 # Vector Store Retriever
 vector_retriever = db.as_retriever(search_kwargs={"k": 3})
@@ -38,11 +38,7 @@ ensemble_retriever = EnsembleRetriever(
 # Use in ask_question()
 
 
-from langchain_community.llms import Ollama
-from langchain_core.prompts import ChatPromptTemplate
 
-from langchain_community.llms import Ollama
-from langchain_core.prompts import ChatPromptTemplate
 from langchain_community.llms import HuggingFaceHub
 from langchain_core.prompts import ChatPromptTemplate
 
@@ -84,7 +80,7 @@ def respond_rag_huggingface(
 
     return response.content
 
-
+__all__ = ["respond_rag_huggingface"]
 # def respond_rag_ollama(
 #     message: str,
 #     system_message: str = "you are game of thrones measter answer the given question strictly based on the context provived.if u donot know the answer reply i dont know donot give gibberish answers ",