hddlmZddlZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZddlmZdZd ZGd d e eZy) ) annotationsN)AnyOptional) Embeddings)from_env) BaseModel ConfigDictFieldmodel_validator)Selfz'sentence-transformers/all-mpnet-base-v2)feature-extractionceZdZUdZdZded<dZded<dZded< dZded< dZ ded < d Z ded < dZ d ed < e e ddZded<eddZedddZddZddZddZddZy)HuggingFaceEndpointEmbeddingsayHuggingFaceHub embedding models. To use, you should have the ``huggingface_hub`` python package installed, and the environment variable ``HUGGINGFACEHUB_API_TOKEN`` set with your API token, or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_huggingface import HuggingFaceEndpointEmbeddings model = "sentence-transformers/all-mpnet-base-v2" hf = HuggingFaceEndpointEmbeddings( model=model, task="feature-extraction", huggingfacehub_api_token="my-api-key", ) Nrclient async_clientz Optional[str]modelproviderrepo_idr taskzOptional[dict] model_kwargsHUGGINGFACEHUB_API_TOKEN)default)default_factoryhuggingfacehub_api_tokenforbid)extraprotected_namespacesafter)modec@|jxstjd} ddlm}m}|j r|j |_n4|jr|j|_nt|_t|_||j ||j}||j ||j}|jtvr"d|jdtd}t|||_ ||_|S#t$r}d}t||d }~wwxYw) z?Validate that api key and python package exists in environment.HF_TOKENr)AsyncInferenceClientInferenceClient)rtokenrzGot invalid task z, currently only z are supportedzfCould not import huggingface_hub python package. Please install it with `pip install huggingface_hub`.N)rosgetenvhuggingface_hubr#r$rr DEFAULT_MODELrr VALID_TASKS ValueErrorrr ImportError)selfrr#r$rrmsges t/var/www/html/aiagenthome/venv/lib/python3.12/site-packages/langchain_huggingface/embeddings/huggingface_endpoint.pyvalidate_environmentz2HuggingFaceEndpointEmbeddings.validate_environment;s$(#@#@$ BII E  ( *  zz#zz !\\ * , $jj.F 0jj.L yy +' {3&&1].B!o% DK ,D   *H c" )  *sCD D DDc|Dcgc]}|jdd}}|jxsi}|jjdd|i|}|j Scc}w)zCall out to HuggingFaceHub's embedding endpoint for embedding search docs. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text.   textr)replacerrfeature_extractiontolistr-textsr5 _model_kwargs responsess r0embed_documentsz-HuggingFaceEndpointEmbeddings.embed_documentsmse6;;UTdC(U;))/R 2DKK22OOO !! $++>>  '  !! < sA/A(3A/A-A/c,|j|gdS)zCall out to HuggingFaceHub's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embeddings for the text. r)r=r-r5s r0 embed_queryz)HuggingFaceEndpointEmbeddings.embed_querys##TF+A..cHK|j|gd{dS7w)zAsync Call to HuggingFaceHub's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embeddings for the text. Nr)r?rAs r0 aembed_queryz*HuggingFaceEndpointEmbeddings.aembed_querys&++TF33Q773s " ")returnr )r:z list[str]rFzlist[list[float]])r5strrFz list[float])__name__ __module__ __qualname____doc__r__annotations__rrrrrrr rrr model_configr r1r=r?rBrErrCr0rrs&FCL#E="Hm"p"G]!C.D-.&#'L.'1.3 !;TJ/mL '"/#/b"""$ / 8rCr) __future__rr&typingrrlangchain_core.embeddingsrlangchain_core.utilsrpydanticrr r r typing_extensionsr r)r*rrrCr0rTs7" 0)BB"9 % W8IzW8rC