Update src/rag.py

src/rag.py

@@ -5,8 +5,8 @@ from langchain_core.runnables import RunnablePassthrough
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_qdrant import QdrantVectorStore
-from langchain.retrievers import ContextualCompressionRetriever
-from langchain.retrievers.document_compressors import CrossEncoderReranker
+from langchain_community.cross_encoders import HuggingFaceCrossEncoder
+from langchain_community.retrievers.document_compressors import CrossEncoderReranker
 from sentence_transformers import CrossEncoder
 
 
@@ -50,7 +50,7 @@ base_retriever = vector_store.as_retriever(
     search_kwargs={"k": 20}
 )
 
-model_r = CrossEncoder("BAAI/bge-reranker-v2-m3")
+model_r = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-v2-m3")
 compressor = CrossEncoderReranker(model=model_r, top_n=5)
 
 retriever = ContextualCompressionRetriever(