o Zh+ @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__Zmodelscope.pipelinesr Zmodelscope.utils.constantr ImportErrorZsentence_embeddingr r r )selfr r re __class__rd/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/modelscope_hub.pyrs"  zModelScopeEmbeddings.__init__Zforbidr)extraZprotected_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)rrinputsZ 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. rrr!r"r$r)rr r')rr*r(Z embeddingrrr embed_query:s z ModelScopeEmbeddings.embed_query)__name__ __module__ __qualname____doc__r r__annotations__r strr rrrZ model_configrfloatr)r+ __classcell__rrrrrs   rN) typingrrrZlangchain_core.embeddingsrZpydanticrrrrrrrs