diff --git a/libs/langchain/langchain/retrievers/contextual_compression.py b/libs/langchain/langchain/retrievers/contextual_compression.py index d41ea489ded..9f5c09e3d1c 100644 --- a/libs/langchain/langchain/retrievers/contextual_compression.py +++ b/libs/langchain/langchain/retrievers/contextual_compression.py @@ -5,7 +5,7 @@ from langchain_core.callbacks import ( CallbackManagerForRetrieverRun, ) from langchain_core.documents import Document -from langchain_core.retrievers import BaseRetriever +from langchain_core.retrievers import BaseRetriever, RetrieverLike from langchain.retrievers.document_compressors.base import ( BaseDocumentCompressor, @@ -18,7 +18,7 @@ class ContextualCompressionRetriever(BaseRetriever): base_compressor: BaseDocumentCompressor """Compressor for compressing retrieved documents.""" - base_retriever: BaseRetriever + base_retriever: RetrieverLike """Base Retriever to use for getting relevant documents.""" class Config: