a bgV@sFddlmZmZmZddlmZddlmZmZGdddeeZ dS))AnyDictList) Embeddings) BaseModelmodel_validatorc@seZdZUdZeed<dZeed<edde e eddd Z ed d d d Z e ee e edddZee edddZd S) AwaEmbeddingszEmbedding documents and queries with Awa DB. Attributes: client: The AwaEmbedding client. model: The name of the model used for embedding. Default is "all-mpnet-base-v2". clientzall-mpnet-base-v2modelbefore)mode)valuesreturnc CsNzddlm}Wn.ty>}ztd|WYd}~n d}~00||d<|S)z)Validate that awadb library is installed.r) AwaEmbeddingzJCould not import awadb library. Please install it with `pip install awadb`Nr )Zawadbr ImportError)clsr rexcrp/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/awa.pyvalidate_environments z"AwaEmbeddings.validate_environmentN) model_namercCs||_||j_dS)zSet the model used for embedding. The default model used is all-mpnet-base-v2 Args: model_name: A string which represents the name of model. N)r r r)selfrrrr set_model"szAwaEmbeddings.set_model)textsrcCs |j|S)zEmbed a list of documents using AwaEmbedding. Args: texts: The list of texts need to be embedded Returns: List of embeddings, one for each text. )r ZEmbeddingBatch)rrrrrembed_documents,s zAwaEmbeddings.embed_documents)textrcCs |j|S)zCompute query embeddings using AwaEmbedding. Args: text: The text to embed. Returns: Embeddings for the text. )r Z Embedding)rrrrr embed_query7s zAwaEmbeddings.embed_query)__name__ __module__ __qualname____doc__r__annotations__r strr classmethodrrrrfloatrrrrrrrs     rN) typingrrrZlangchain_core.embeddingsrZpydanticrrrrrrrs