a bg+ @sFddlmZmZmZddlmZddlmZmZGdddeeZ dS))AnyListOptional) Embeddings) BaseModel ConfigDictcseZdZUdZdZeed<dZeed<dZ e eed<edfdd Z e d d d Z eeeeed ddZeeedddZZS)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_revision)kwargsc sxtjfi|zddlm}ddlm}Wn.ty\}ztd|WYd}~n d}~00||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__{/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/modelscope_hub.pyrs zModelScopeEmbeddings.__init__Zforbidr)extraZprotected_namespaces)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)xrrr5z6ModelScopeEmbeddings.embed_documents..source_sentenceinputtext_embedding)listmapr tolist)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. 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