SuHoU commited on
Commit
21cfc32
·
verified ·
1 Parent(s): 6c395e3

Update src/rag.py

Browse files
Files changed (1) hide show
  1. src/rag.py +2 -1
src/rag.py CHANGED
@@ -6,7 +6,8 @@ from langchain_core.output_parsers import StrOutputParser
6
  from langchain_core.prompts import ChatPromptTemplate
7
  from langchain_qdrant import QdrantVectorStore
8
  from langchain_community.cross_encoders import HuggingFaceCrossEncoder
9
- from langchain_community.retrievers.document_compressors import CrossEncoderReranker
 
10
  from sentence_transformers import CrossEncoder
11
 
12
 
 
6
  from langchain_core.prompts import ChatPromptTemplate
7
  from langchain_qdrant import QdrantVectorStore
8
  from langchain_community.cross_encoders import HuggingFaceCrossEncoder
9
+ from langchain.retrievers.document_compressors import CrossEncoderReranker
10
+ from langchain.retrievers import ContextualCompressionRetriever
11
  from sentence_transformers import CrossEncoder
12
 
13