NgddlZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZddlmZdZdZGd d e eZdS) N)AnyListOptional) Embeddings)from_env) BaseModel ConfigDictFieldmodel_validator)Selfz'sentence-transformers/all-mpnet-base-v2)feature-extractionceZdZUdZdZeed<dZeed<dZe e ed< dZ e e ed< dZ e e ed< dZ e eed < eed d  Ze e ed <eddZeddefdZdee deeefdZdee deeefdZde deefdZde deefdZdS)HuggingFaceEndpointEmbeddingsaxHuggingFaceHub 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", ) Nclient async_clientmodelrepo_idr task model_kwargsHUGGINGFACEHUB_API_TOKEN)default)default_factoryhuggingfacehub_api_tokenforbid)extraprotected_namespacesafter)modereturnc|jptjd} ddlm}m}|jr |j|_n,|jr |j|_nt|_t|_||j|}||j|}|j tvr td|j dtd||_ ||_ n#t$rtdwxYw|S) z?Validate that api key and python package exists in environment.HF_TOKENr)AsyncInferenceClientInferenceClient)rtokenzGot invalid task z, currently only z are supportedzfCould not import huggingface_hub python package. Please install it with `pip install huggingface_hub`.)rosgetenvhuggingface_hubr#r$rr DEFAULT_MODELr VALID_TASKS ValueErrorrr ImportError)selfrr#r$rrs q/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_huggingface/embeddings/huggingface_endpoint.pyvalidate_environmentz2HuggingFaceEndpointEmbeddings.validate_environment5se$(#@$ BI E E  $          z -#z  -!\ * , $_j.F 0/j.L y ++ B BB&1BBB!DK ,D     H    s B(CC textscd|D}|jpi}|jd|i||j}t j|S)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. c:g|]}|ddS  replace.0texts r. zAHuggingFaceEndpointEmbeddings.embed_documents..m&;;;TdC((;;;inputsjsonr)rrpostrr@loadsdecoder-r0 _model_kwargs responsess r.embed_documentsz-HuggingFaceEndpointEmbeddings.embed_documentscso<;U;;;)/R K$$E3]3$)%  z)**,,---r=cKd|D}|jpi}|j||d|jd{V}t j|S)zAsync Call 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. c:g|]}|ddSr3r6r8s r.r;zBHuggingFaceEndpointEmbeddings.aembed_documents..r<r=)r> parametersr?N)rrrArr@rBrCrDs r.aembed_documentsz.HuggingFaceEndpointEmbeddings.aembed_documentsus<;U;;;)/R +00!??di1        z)**,,---r=r:c>||gd}|S)zCall out to HuggingFaceHub's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embeddings for the text. r)rGr-r:responses r. embed_queryz)HuggingFaceEndpointEmbeddings.embed_querys#''//2r=cNK||gd{Vd}|S)zAsync Call to HuggingFaceHub's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embeddings for the text. Nr)rKrMs r. aembed_queryz*HuggingFaceEndpointEmbeddings.aembed_querys9//77777777;r=)__name__ __module__ __qualname____doc__rr__annotations__rrrstrrrrdictr rrr model_configr r r/rfloatrGrKrOrQrr=r.rrs$FCL#E8C=!GXc]!!!C.D(3-...&#'L(4.'''1.3e !;TJJJ///hsm:L _'"""+d+++#"+Z.T#Y.4U 3D....$.DI.$tE{:K...."  U     s tE{      r=r)r@r&typingrrrlangchain_core.embeddingsrlangchain_core.utilsrpydanticrr r r typing_extensionsr r)r*rrr=r.r`s &&&&&&&&&&000000))))))BBBBBBBBBBBB""""""9 % NNNNNIzNNNNNr=