Update app.py

app.py

@@ -15,7 +15,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import os
 import transformers
 import torch
-from langchain_retrieval import BaseRetrieverChain
+# from langchain_retrieval import BaseRetrieverChain
 # from dotenv import load_dotenv
 
 # load_dotenv()
@@ -78,11 +78,7 @@ def get_context_retriever_chain(vector_store,llm):
 #     stuff_documents_chain = create_stuff_documents_chain(llm,prompt)
     
 #     return create_retrieval_chain(retriever_chain, stuff_documents_chain)
-def get_conversational_rag_chain(
-    retriever_chain: Optional[langchain_retrieval.BaseRetrieverChain],
-    llm: Callable[[str], str],
-    chat_history: Optional[langchain_core.prompts.chat.ChatPromptValue] = None,
-) -> langchain_retrieval.BaseRetrieverChain:
+def get_conversational_rag_chain(retriever_chain,llm):
 
     if not retriever_chain:
         raise ValueError("`retriever_chain` cannot be None or an empty object.")