U h@s\ddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z Gddde Z dS) )AnyList)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)Document) BaseRetriever RetrieverLike)BaseDocumentCompressorc@s`eZdZUdZeed<eed<GdddZee e e e dddZ eee e e dd d Zd S) ContextualCompressionRetrieverzARetriever that wraps a base retriever and compresses the results.base_compressorbase_retrieverc@seZdZdZdS)z%ContextualCompressionRetriever.ConfigTN)__name__ __module__ __qualname__Zarbitrary_types_allowedrrO/tmp/pip-unpacked-wheel-bo69hh5q/langchain/retrievers/contextual_compression.pyConfigsr)query run_managerkwargsreturncKsJ|jj|fdd|ii|}|rB|jj|||d}t|SgSdS)zGet documents relevant for a query. Args: query: string to find relevant documents for Returns: Sequence of relevant documents config callbacksrN)r Zinvoke get_childr Zcompress_documentslistselfrrrZdocsZcompressed_docsrrr_get_relevant_documentss z6ContextualCompressionRetriever._get_relevant_documentscsV|jj|fdd|ii|IdH}|rN|jj|||dIdH}t|SgSdS)zGet documents relevant for a query. Args: query: string to find relevant documents for Returns: List of relevant documents rrNr)r Zainvokerr Zacompress_documentsrrrrr_aget_relevant_documents5s   z7ContextualCompressionRetriever._aget_relevant_documentsN)r rr__doc__r __annotations__rrstrrrrrrrrrrrrr s  r N)typingrrZlangchain_core.callbacksrrZlangchain_core.documentsrZlangchain_core.retrieversrrZ.langchain.retrievers.document_compressors.baser r rrrrs