a bg_ @sFddlmZmZddlmZddlmZmZdZGdddeeZ dS))AnyList) Embeddings) BaseModel ConfigDictzBhttps://tfhub.dev/google/universal-sentence-encoder-multilingual/3cs|eZdZUdZdZeed<eZe ed<edfdd Z e dd d Z e e e e ed d d Ze e edddZZS)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_url)kwargsc sttjfi|z ddl}Wnty8tdYn0z ddl}Wnty`tdYn0||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 r r __class__{/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/tensorflow_hub.pyr s      z TensorflowHubEmbeddings.__init__Zforbidr)extraZprotected_namespaces)textsreturncCs(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)xrrr<z9TensorflowHubEmbeddings.embed_documents..)listmaprnumpytolist)rrZ embeddingsrrrembed_documents3s z'TensorflowHubEmbeddings.embed_documents)textrcCs(|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 embeddingrrr embed_query@s z#TensorflowHubEmbeddings.embed_query)__name__ __module__ __qualname____doc__rr__annotations__DEFAULT_MODEL_URLr strr rZ model_configrfloatr#r% __classcell__rrrrr s    rN) typingrrZlangchain_core.embeddingsrZpydanticrrr+rrrrrs