o `+ i @s\ddlmZddlmZmZddlmZmZddlm Z m Z ddl m Z Gddde Z dS) )Any)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)BaseDocumentCompressorDocument) BaseRetriever RetrieverLike) ConfigDictc @sleZdZUdZeed< eed< eddZde de de d e e fd d Zde dede d e e fd d ZdS)ContextualCompressionRetrieverzARetriever that wraps a base retriever and compresses the results.base_compressorbase_retrieverT)Zarbitrary_types_allowedquery run_managerkwargsreturncKsF|jj|fdd|ii|}|r!|jj|||d}t|SgS)zGet documents relevant for a query. Args: query: string to find relevant documents for Returns: Sequence of relevant documents config callbacksr)r Zinvoke get_childr Zcompress_documentslistselfr rrdocsZcompressed_docsrw/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/retrievers/contextual_compression.py_get_relevant_documentss z6ContextualCompressionRetriever._get_relevant_documentscsT|jj|fdd|ii|IdH}|r(|jj|||dIdH}t|SgS)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_documents6s    z7ContextualCompressionRetriever._aget_relevant_documentsN)__name__ __module__ __qualname____doc__r__annotations__rr Z model_configstrrrrrrrrrrrrr s6  r N)typingrZlangchain_core.callbacksrrZlangchain_core.documentsrrZlangchain_core.retrieversrrZpydanticr r rrrrs