3fibddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZddlmZmZGdd eZy ) ) annotations)deepcopy)AnyDictListOptionalSequenceUnion) Callbacks)BaseDocumentCompressorDocument)get_from_dict_or_env) ConfigDictmodel_validatorceZdZUdZdZded< dZded< dZded< dZd ed < d Z d ed < d Z ded< e dddZ e deddZdd ddZ d ddZy)VolcengineRerankz6Document compressor that uses `Volcengine Rerank API`.Nrclientz Optional[str]akskapi-vikingdb.volces.comstrregion cn-beijinghost Optional[int]top_nTforbid)populate_by_namearbitrary_types_allowedextrabefore)modec |jds? ddlm}t |dd|d<t |dd|d<|d d d d d |d|d |d<|S#t$r tdwxYw)z?Validate that api key and python package exists in environment.rrVikingDBServicezCould not import volcengine python package. Please install it with `pip install volcengine` or `pip install --user volcengine`.r VOLC_API_AKr VOLC_API_SKrrhttps)rrschemeconnection_timeoutsocket_timeoutrr)getvolcengine.viking_dbr& ImportErrorr)clsvaluesr&s x/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/document_compressors/volcengine_rerank.pyvalidate_environmentz%VolcengineRerank.validate_environment)s zz(# @0mLF4L/mLF4L..##%!$<$< F8  ) !: s AA()rc|t|dk(rgS|Dcgc]#}|t|tr |jn|d%}}ddlm}|j }|j|}g} t|D]\} } | j| | d| jdd||dkDr|n |j}| d|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. r)querycontentr%)indexrelevance_scorec |dS)Nr:)xs r3z)VolcengineRerank.rerank..js *;(<T)keyreverseN) len isinstancer page_contentr/r&r batch_rerank enumerateappendsortr) self documentsr7rdocdocsr&rresults result_dictsr9scores r3rerankzVolcengineRerank.rerankGs y>Q I !  /9#x/H3++c    9"&++%%d+ %g. LLE5   %E J K L rhs,"==3E50z-zr?