o Zh_ @sFddlmZmZddlmZddlmZmZdZGdddeeZ dS))AnyList) Embeddings) BaseModel ConfigDictzBhttps://tfhub.dev/google/universal-sentence-encoder-multilingual/3cseZdZUdZdZeed<eZe ed< deffdd Z e dd d Z d e e d e e efd dZde d e efddZZS)TensorflowHubEmbeddingsaTensorflowHub embedding models. To use, you should have the ``tensorflow_text`` python package installed. Example: .. code-block:: python from langchain_community.embeddings import TensorflowHubEmbeddings url = "https://tfhub.dev/google/universal-sentence-encoder-multilingual/3" tf = TensorflowHubEmbeddings(model_url=url) Nembed model_urlkwargsc sltjdi|zddl}Wn tytdwzddl}Wn ty,tdw||j|_dS)z2Initialize the tensorflow_hub and tensorflow_text.rNzeCould not import tensorflow-hub python package. Please install it with `pip install tensorflow-hub``.zgCould not import tensorflow_text python package. Please install it with `pip install tensorflow_text``.)super__init__tensorflow_hub ImportErrortensorflow_textloadr r)selfr rr __class__r d/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/tensorflow_hub.pyr s     z TensorflowHubEmbeddings.__init__Zforbidr )extraZprotected_namespacestextsreturncCs(ttdd|}||}|S)zCompute doc embeddings using a TensorflowHub embedding model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. cSs |ddS)N  )replace)xr r r<s z9TensorflowHubEmbeddings.embed_documents..)listmaprnumpytolist)rrZ embeddingsr r rembed_documents3s z'TensorflowHubEmbeddings.embed_documentstextcCs(|dd}||gd}|S)zCompute query embeddings using a TensorflowHub embedding model. Args: text: The text to embed. Returns: Embeddings for the text. rrr)rrr r!)rr#Z embeddingr r r embed_query@s z#TensorflowHubEmbeddings.embed_query)__name__ __module__ __qualname____doc__rr__annotations__DEFAULT_MODEL_URLr strr rZ model_configrfloatr"r$ __classcell__r r rrr s   rN) typingrrZlangchain_core.embeddingsrZpydanticrrr*rr r r rs