+#hddlZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZddlmZdZdZGd d e eZy) N)AnyListOptional) Embeddings)from_env) BaseModel ConfigDictFieldmodel_validator)Selfz'sentence-transformers/all-mpnet-base-v2)feature-extractionc`eZdZUdZdZeed<dZeed<dZe e ed< dZ e e ed< dZ e e ed< dZ e eed < eed d  Ze e ed <eddZeddefdZdee deeefdZdee deeefdZde deefdZde deefdZy)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|jxstjd} ddlm}m}|j r|j |_n4|jr|j|_nt|_t|_||j |}||j |}|jtvr td|jdtd||_ ||_ |S#t$r tdwxYw) 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 r/var/www/html/sandstorm/venv/lib/python3.12/site-packages/langchain_huggingface/embeddings/huggingface_endpoint.pyvalidate_environmentz2HuggingFaceEndpointEmbeddings.validate_environment5s$(#@#@$ BII E  $   zz#zz !\\ * , $jj.F 0jj.L yy + ' {3&&1].B!DK ,D   H  s CC))C>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)replacerrpostrr6loadsdecoder-r0text _model_kwargs responsess r.embed_documentsz-HuggingFaceEndpointEmbeddings.embed_documentscs}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. r2r3)r4 parametersr5N)r7rrr8rr6r9r:r;s r.aembed_documentsz.HuggingFaceEndpointEmbeddings.aembed_documentsus6;;TdC(;;))/R ++00!?dii1  zz)**,-- < sBB?BB  -Br<c0|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)r?r-r<responses r. embed_queryz)HuggingFaceEndpointEmbeddings.embed_querys ''/2cLK|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)rBrDs r. aembed_queryz*HuggingFaceEndpointEmbeddings.aembed_querys,//77;8s $" $)__name__ __module__ __qualname____doc__rr__annotations__rrrstrrrrdictr rrr model_configr r r/rfloatr?rBrFrIrrGr.rrs#$FCL#E8C=!GXc]!C.D(3-.&#'L(4.'1.3 !;TJ/hsmL '"+d+#+Z.T#Y.4U 3D.$.DI.$tE{:K."  U  s tE{ rGr)r6r&typingrrrlangchain_core.embeddingsrlangchain_core.utilsrpydanticrr r r typing_extensionsr r)r*rrrGr.rXs: &&0)BB"9 % NIzNrG