a bg@sRddlmZmZmZmZddlmZmZmZddl m Z dZ Gdddee Z dS))AnyDictListTuple) BaseModel ConfigDictField)BaseCrossEncoderzBAAI/bge-reranker-basecseZdZUdZdZeed<eZe ed<e e dZ e e efed<edfdd Zed d d Zeee e feed ddZZS)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_kwargs)kwargsc sjtjfi|z ddl}Wn.tyL}ztd|WYd}~n d}~00|j|jfi|j|_dS)z$Initialize the sentence_transformer.rNzrCould not import sentence_transformers python package. Please install it with `pip install sentence-transformers`.)super__init__sentence_transformers ImportErrorZ CrossEncoderr rr )selfrrexc __class__|/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/cross_encoders/huggingface.pyr s z HuggingFaceCrossEncoder.__init__Zforbidr)extraZprotected_namespaces) text_pairsreturncCs,|j|}t|jdkr(tdd|}|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. cSs|dS)Nrr)xrrr?z/HuggingFaceCrossEncoder.score..)r Zpredictlenshapemap)rrZscoresrrrscore2s zHuggingFaceCrossEncoder.score)__name__ __module__ __qualname____doc__r r__annotations__DEFAULT_MODEL_NAMEr strrdictrrrrZ model_configrrfloatr$ __classcell__rrrrr s    r N) typingrrrrZpydanticrrrZ'langchain_community.cross_encoders.baser r*r rrrrs