a bg @sfdZddlZddlmZmZddlZddlmZddlm Z m Z m Z dZ dZ dZGd d d e eZdS) z+Wrapper around Bookend AI embedding models.N)AnyList) Embeddings) BaseModel ConfigDictFieldzhttps://api.bookend.ai/Z embeddingsz/models/predictcseZdZUdZeed<eed<eed<eedZeed<e ddZ e d fd d Z e ee e ed d dZee edddZZS)BookendEmbeddingsaBookend AI sentence_transformers embedding models. Example: .. code-block:: python from langchain_community.embeddings import BookendEmbeddings bookend = BookendEmbeddings( domain={domain} api_token={api_token} model_id={model_id} ) bookend.embed_documents([ "Please put on these earmuffs because I can't you hear.", "Baby wipes are made of chocolate stardust.", ]) bookend.embed_query( "She only paints with bold colors; she does not like pastels." ) domain api_tokenmodel_id)default_factory auth_header)Zprotected_namespaces)kwargsc s(tjfi|dd|ji|_dS)N AuthorizationzBasic {})super__init__formatr r )selfr __class__rt/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/bookend.pyr/szBookendEmbeddings.__init__)textsreturncCstg}|j}d|d<|jtd}|D]L}t|dddd}tjdt|jt |||d}| |dd q"|S) zEmbed documents using a Bookend deployed embeddings model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. zapplication/json; charset=utf-8z Content-Type)r taskN)textquestioncontextZ instructionPOST)headersparamsdatarr!) r r DEFAULT_TASKjsondumpsrequestsrequestAPI_URLr PATHappend)rrresultrr rr!rrrrembed_documents3s&   z!BookendEmbeddings.embed_documents)rrcCs||gdS)zEmbed a query using a Bookend deployed embeddings model. Args: text: The text to embed. Returns: Embeddings for the text. r)r,)rrrrr embed_querySs zBookendEmbeddings.embed_query)__name__ __module__ __qualname____doc__str__annotations__rdictr rZ model_configrrrfloatr,r- __classcell__rrrrrs   r)r1r#typingrrr%Zlangchain_core.embeddingsrZpydanticrrrr'r"r(rrrrrs