昺hP TddlmZmZmZddlmZddlmZmZGddeeZ dS))AnyListOptional) Embeddings) BaseModelExtraceZdZUdZeed<dZeed< dZe eed<deffd Z Gd d Z d e ed e e e fd Zded e e fdZxZS)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_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 u/var/www/html/services/rag-service/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 ) AAAc eZdZdZejZdS)ModelScopeEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__rforbidextrarrConfigr*s55 r&r'textsreturncttd|}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..8s199T3#7#7r&source_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-r1r2r4r)r.r r7)rr;r8 embeddings r embed_queryz ModelScopeEmbeddings.embed_query=sS||D#&&#dV,JJVJ,,-=>qA !!!r&)r r!r"r#r__annotations__r strr rrr'rfloatr:r> __classcell__)rs@rr r s   JJJDHcDDD$(NHSM(((       " #T#Y #4U 3D # # # # " "U " " " " " " " "r&r N) typingrrrlangchain_core.embeddingsrlangchain_core.pydantic_v1rrr rr&rrFs&&&&&&&&&&00000077777777B"B"B"B"B"9jB"B"B"B"B"r&