tj+ TddlmZmZmZddlmZddlmZmZGddeeZ dS))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 tjdi| ddlm}ddlm}n"#t $r}t d|d}~wwxYw||j|j|j |_ dS)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 k/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/embeddings/modelscope_hub.pyrzModelScopeEmbeddings.__init__s""6"""  5 5 5 5 5 5 7 7 7 7 7 7 7   C   X  $-.    s ) AAAforbidr)extraprotected_namespacestextsreturncttd|}d|i}||d}|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. c.|ddS)N  )replace)xs rz6ModelScopeEmbeddings.embed_documents..5s199T3#7#7source_sentenceinputtext_embedding)listmapr tolist)rr!inputs embeddingss rembed_documentsz$ModelScopeEmbeddings.embed_documents,sTS77??@@#U+ZZfZ--.>?   """r*textc|dd}d|gi}||dd}|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 r1)rr5r2 embeddings r embed_queryz ModelScopeEmbeddings.embed_query:sS||D#&&#dV,JJVJ,,-=>qA !!!r*)__name__ __module__ __qualname____doc__r r__annotations__r strr rrr model_configrfloatr4r8 __classcell__)rs@rr r s  E3DHcDDD$(NHSM(((       ":H2FFFL #T#Y #4U 3D # # # # " "U " " " " " " " "r*r N) typingrrrlangchain_core.embeddingsrpydanticrrr rr*rrEs&&&&&&&&&&000000********?"?"?"?"?"9j?"?"?"?"?"r*