o Ag+ @sFddlmZmZmZddlmZddlmZmZGdddeeZ dS))AnyListOptional) Embeddings) BaseModel ConfigDictcseZdZUdZdZeed<dZeed< dZ e eed<deffdd Z e d d d Z d eedeeefddZdedeefddZZS)ModelScopeEmbeddingsaModelScopeHub embedding models. To use, you should have the ``modelscope`` python package installed. Example: .. code-block:: python from langchain_community.embeddings import ModelScopeEmbeddings model_id = "damo/nlp_corom_sentence-embedding_english-base" embed = ModelScopeEmbeddings(model_id=model_id, model_revision="v1.0.0") Nembedz.damo/nlp_corom_sentence-embedding_english-basemodel_idmodel_revisionkwargsc sltjdi|zddlm}ddlm}Wnty(}ztd|d}~ww||j|j|j d|_ dS)zInitialize the modelscoper)pipeline)TaskszVCould not import some python packages.Please install it with `pip install modelscope`.N)modelr ) super__init__modelscope.pipelinesr modelscope.utils.constantr ImportErrorsentence_embeddingr r r )selfr r re __class__rt/var/www/html/development/chatbot/venv/lib/python3.10/site-packages/langchain_community/embeddings/modelscope_hub.pyrs"  zModelScopeEmbeddings.__init__forbidr)extraprotected_namespacestextsreturncCs2ttdd|}d|i}|j|dd}|S)zCompute doc embeddings using a modelscope embedding model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. cSs |ddS)N  )replace)xrrr5s z6ModelScopeEmbeddings.embed_documents..source_sentenceinputtext_embedding)listmapr tolist)rrinputs embeddingsrrrembed_documents,s z$ModelScopeEmbeddings.embed_documentstextcCs2|dd}d|gi}|j|ddd}|S)zCompute query embeddings using a modelscope embedding model. Args: text: The text to embed. Returns: Embeddings for the text. r!r"r&r'r)r)r#r r,)rr0r- embeddingrrr embed_query:s z ModelScopeEmbeddings.embed_query)__name__ __module__ __qualname____doc__r r__annotations__r strr rrr model_configrfloatr/r2 __classcell__rrrrrs   rN) typingrrrlangchain_core.embeddingsrpydanticrrrrrrrs