3fiddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZddlmZmZmZGdd eZy ) ) annotations)deepcopy)AnyDictListOptionalSequenceUnion) Callbacks)BaseDocumentCompressorDocument)get_from_dict_or_env) ConfigDictFieldmodel_validatorceZdZUdZdZded< dZded< dZded < edd Z ded < e d d dZ e de ddZdd ddZ d ddZy)DashScopeRerankz5Document compressor that uses `DashScope Rerank API`.Nrclientz Optional[str]model Optional[int]top_napi_key)aliasdashscope_api_keyTforbid)populate_by_namearbitrary_types_allowedextrabefore)modec|jdsG ddl}|j|d<t |dd|d<|jj j |d<|S#t$r tdwxYw)z?Validate that api key and python package exists in environment.rrNzZCould not import dashscope python package. Please install it with `pip install dashscope`.rDASHSCOPE_API_KEYr)get dashscope ImportError TextReRankrModels gte_rerank)clsvaluesr%s w/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/document_compressors/dashscope_rerank.pyvalidate_environmentz$DashScopeRerank.validate_environment"s zz(#   )33F8 *>+-@+F& '(2299DDF7O  !F s AA0)rct|dk(rgS|Dcgc] }t|tr |jn|"}}||dkDr|n |j}|j j |j|||d|j}g}|jjD]*}|j|j|jd,|Scc}w)aUReturns an ordered list of documents ordered by their relevance to the provided query. Args: query: The query to use for reranking. documents: A sequence of documents to rerank. top_n : The number of results to return. If None returns all results. Defaults to self.top_n. rF)rquery documentsrreturn_documentsr)indexrelevance_score)len isinstancer page_contentrrcallrroutputresultsappendr3r4) selfr1r0rdocdocsr: result_dictsress r,rerankzDashScopeRerank.rerank8s y>Q INW GJ 3 9C  s B   -5194::++""**"** #  >>)) C    YY'*':':  / s%C cg}|j||D]W}||d}t|jt|j}|d|jd<|j |Y|S)aZ Compress documents using DashScope's rerank API. Args: documents: A sequence of documents to compress. query: The query to use for compressing the documents. callbacks: Callbacks to run during the compression process. Returns: A sequence of compressed documents. r3)metadatar4)rAr r7rrCr;)r<r1r0 callbacks compressedr@r=doc_copys r,compress_documentsz"DashScopeRerank.compress_documentscsy" ;;y%0 (CCL)C 0 08CLL;QRH367H3IH  / 0   h '  ( )r+rreturnr)r1z$Sequence[Union[str, Document, dict]]r0strrrrIzList[Dict[str, Any]])N)r1Sequence[Document]r0rJrDzOptional[Callbacks]rIrK)__name__ __module__ __qualname____doc__r__annotations__rrrrr model_configr classmethodr-rArGrHr,rr s?FC<E=%E=(',T'C}C $L (#$2 " )7))  )  )^*. %'   rHrN) __future__rcopyrtypingrrrrr r langchain_core.callbacks.baser langchain_core.documentsr r langchain_core.utilsrpydanticrrrrrSrHr,r[s/"==3E577n,nrH