U h @sBddlmZmZmZddlmZddlmZGdddeeZdS))AnyListOptional) Embeddings) BaseModelcseZdZUdZeed<dZeed<dZe eed<edfdd Z Gd d d Z e ee e e d d dZee e dddZZS)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") embedz.damo/nlp_corom_sentence-embedding_english-basemodel_idNmodel_revision)kwargsc srtjf|zddlm}ddlm}Wn,tk rV}ztd|W5d}~XYnX||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 r e __class__Q/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/modelscope_hub.pyrs zModelScopeEmbeddings.__init__c@seZdZdZdS)zModelScopeEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrrrConfig*sr)textsreturncCs2ttdd|}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)xrrr6z6ModelScopeEmbeddings.embed_documents..source_sentenceinputtext_embedding)listmaprtolist)rrinputsZ embeddingsrrrembed_documents-s z$ModelScopeEmbeddings.embed_documents)textrcCs2|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. rr r%r&r(r)r!rr+)rr.r,Z embeddingrrr embed_query;s  z ModelScopeEmbeddings.embed_query)rrr__doc__r__annotations__r strr rrrrfloatr-r/ __classcell__rrrrrs   rN) typingrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rrrrrrs