7|hXddlmZmZmZmZddlmZmZmZddl m Z dZ Gddee Z y))AnyDictListTuple) BaseModel ConfigDictField)BaseCrossEncoderzBAAI/bge-reranker-baseceZdZUdZdZeed<eZe ed< e e Z e e efed< deffd Zed d Zd eee e fd eefdZxZS)HuggingFaceCrossEncoderaHuggingFace cross encoder models. Example: .. code-block:: python from langchain_community.cross_encoders import HuggingFaceCrossEncoder model_name = "BAAI/bge-reranker-base" model_kwargs = {'device': 'cpu'} hf = HuggingFaceCrossEncoder( model_name=model_name, model_kwargs=model_kwargs ) Nclient model_name)default_factory model_kwargskwargsc t|di| ddl}|j|j fi|j |_y#t$r}td|d}~wwxYw)z$Initialize the sentence_transformer.rNzrCould not import sentence_transformers python package. Please install it with `pip install sentence-transformers`.)super__init__sentence_transformers ImportError CrossEncoderrrr )selfrrexc __class__s m/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/cross_encoders/huggingface.pyrz HuggingFaceCrossEncoder.__init__ sq "6"  (9+88 OO #00   N  sA A AAforbidr)extraprotected_namespaces text_pairsreturnc|jj|}t|jdkDr t d|}|S)zCompute similarity scores using a HuggingFace transformer model. Args: text_pairs: The list of text text_pairs to score the similarity. Returns: List of scores, one for each pair. c |dS)Nr#r)xs rz/HuggingFaceCrossEncoder.score..?s 1Q4)r predictlenshapemap)rr scoress rscorezHuggingFaceCrossEncoder.score2s<$$Z0 v|| q 0F r')__name__ __module__ __qualname____doc__r r__annotations__DEFAULT_MODEL_NAMErstrr dictrrrr model_configrrfloatr- __classcell__)rs@rr r sy FC(J(#(#>L$sCx.>1  H2FLU38_ 5$u+r'r N) typingrrrrpydanticrrr 'langchain_community.cross_encoders.baser r3r rr'rr<s)))11D-6i)96r'