7|h+ LddlmZmZmZddlmZddlmZmZGddeeZ y))AnyListOptional) Embeddings) BaseModel ConfigDictceZdZUdZdZeed<dZeed< dZ e eed<deffd Z e d d Z d eed eeefdZded eefdZxZS)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 t|di| ddlm}ddlm}||j|j|j|_ y#t $r}t d|d}~wwxYw)zInitialize the modelscoper)pipeline)TaskszVCould not import some python packages.Please install it with `pip install modelscope`.N)modelr ) super__init__modelscope.pipelinesrmodelscope.utils.constantr ImportErrorsentence_embeddingr r r )selfrrre __class__s l/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/modelscope_hub.pyrzModelScopeEmbeddings.__init__so "6"  5 7   $ $--..   C  s A A' A""A'forbidr)extraprotected_namespacestextsreturncttd|}d|i}|j|d}|jS)zCompute doc embeddings using a modelscope embedding model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. c&|jddS)N  )replace)xs rz6ModelScopeEmbeddings.embed_documents..5s199T3#7source_sentenceinputtext_embedding)listmapr tolist)rr!inputs embeddingss rembed_documentsz$ModelScopeEmbeddings.embed_documents,sES7?@#U+ZZfZ-.>?   ""r*textc|jdd}d|gi}|j|dd}|jS)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 r1)rr5r2 embeddings r embed_queryz ModelScopeEmbeddings.embed_query:sJ||D#&#dV,JJVJ,-=>qA !!r*)__name__ __module__ __qualname____doc__r r__annotations__r strr rrr model_configrfloatr4r8 __classcell__)rs@rr r s E3DHcD$(NHSM(  "H2FL #T#Y #4U 3D # " "U "r*r N) typingrrrlangchain_core.embeddingsrpydanticrrr rr*rrEs&&0*?"9j?"r*