
    +#h                          G d  d      Z y)c                   @    e Zd ZdZ	 d	defdZdedee   dee   fdZy)
 SentenceTransformersCrossEncoderz;Wrapper for sentence-transformers cross-encoder model.
    model_name_or_pathc                 \    	 ddl m}  ||      | _        y # t        $ r t        d      w xY w)N    )CrossEncoderzZYou need to install sentence-transformers library to use SentenceTransformersCrossEncoder.)#sentence_transformers.cross_encoderr   ImportErrorModuleNotFoundErrormodel)selfr   r   s      N/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dsp/modules/sbert.py__init__z)SentenceTransformersCrossEncoder.__init__   s:    	H
 ""45
	  	%l 	s    +querypassagereturnc                 ~    | j                   j                  |D cg c]  }||g c}      j                         S c c}w )N)r   predicttolist)r   r   r   ps       r   __call__z)SentenceTransformersCrossEncoder.__call__   s2    zz!!w"?!E1:"?@GGII"?s   :N)z%cross-encoder/ms-marco-MiniLM-L-12-v2)	__name__
__module____qualname____doc__strr   listfloatr        r   r   r      s@     )P	6"%	6Jc JDI J$u+ Jr   r   N)r   r   r   r   <module>r       s   J Jr   