7|hddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZmZddlmZdZd Zed d d Gdde e Zy)N)AnyDictListOptional) deprecated) Embeddings)get_from_dict_or_env) BaseModel ConfigDictmodel_validator)Selfz'sentence-transformers/all-mpnet-base-v2)feature-extractionz0.2.2z1.0z3langchain_huggingface.HuggingFaceEndpointEmbeddings)sinceremovalalternative_importcteZdZUdZdZeed<dZeed<dZe e ed< dZ e e ed< dZ e e ed< dZ e eed < dZe e ed <ed d ZedededefdZeddefdZdee deeefdZdee deeefdZde deefdZde deefdZy)HuggingFaceHubEmbeddingsawHuggingFaceHub 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_community.embeddings import HuggingFaceHubEmbeddings model = "sentence-transformers/all-mpnet-base-v2" hf = HuggingFaceHubEmbeddings( model=model, task="feature-extraction", huggingfacehub_api_token="my-api-key", ) Nclient async_clientmodelrepo_idrtask model_kwargshuggingfacehub_api_tokenforbid)extraprotected_namespacesbefore)modevaluesreturnc6t|dd} ddlm}m}|j dr |d|d<n,|j dr |d|d<nt |d<t |d<||d|}||d|}||d<||d <|S#t $r t d wxYw) z?Validate that api key and python package exists in environment.rHUGGINGFACEHUB_API_TOKENr)AsyncInferenceClientInferenceClientrr)rtokenrrzfCould not import huggingface_hub python package. Please install it with `pip install huggingface_hub`.)r huggingface_hubr%r&get DEFAULT_MODEL ImportError)clsr!rr%r&rrs m/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/huggingface_hub.pyvalidate_environmentz-HuggingFaceHubEmbeddings.validate_environment5s$8 .0J$    Mzz'"$*7Oy!I&"("3w"/w$1y!$Wo.F 0Wo.L &F8 %1F> "  H  s A2BBaftercj|jtvr td|jdtd|S)z#Post init validation for the class.zGot invalid task z, currently only z are supported)r VALID_TASKS ValueError)selfs r- post_initz"HuggingFaceHubEmbeddings.post_init\s@ 99K '#DII;/""-n>  textsc|Dcgc]}|jdd}}|jxsi}|jjd|i||j}t j |jScc}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.   inputsjsonr)replacerrpostrr<loadsdecoder3r6text _model_kwargs responsess r-embed_documentsz(HuggingFaceHubEmbeddings.embed_documentsfs}6;;TdC(;;))/R KK$$E3]3$))% zz)**,-- c K|Dcgc]}|jdd}}|jxsi}|jj||d|jd{}t j |jScc}w7,w)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. r8r9)r: parametersr;N)r=rrr>rr<r?r@rAs r-aembed_documentsz)HuggingFaceHubEmbeddings.aembed_documentsxs6;;TdC(;;))/R ++00!?dii1  zz)**,-- < sBB?BB  -BrBc0|j|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)rEr3rBresponses r- embed_queryz$HuggingFaceHubEmbeddings.embed_querys ''/2r5cLK|j|gd{d}|S7 w)zAsync Call to HuggingFaceHub's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embeddings for the text. Nr)rHrJs r- aembed_queryz%HuggingFaceHubEmbeddings.aembed_querys,//77;8s $" $)__name__ __module__ __qualname____doc__rr__annotations__rrrstrrrrdictrr model_configr classmethodrr.r r4rfloatrErHrLrNrr5r-rrs8 $FCL#E8C=!GXc]!C.D(3-.&#'L(4.'1.2hsm2H2FL(##$#3#$#J'"4#.T#Y.4U 3D.$.DI.$tE{:K."  U  s tE{ r5r)r<typingrrrrlangchain_core._apirlangchain_core.embeddingsrlangchain_core.utilsr pydanticr r r typing_extensionsr r*r1rrr5r-r_sX ,,*05;;"9 %   L Ly*L  Lr5