o Zh: @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 d e ffd d Z d e ed e e efddZded e efddZZS)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_namespaceskwargsc s(tjdi|dd|ji|_dS)N AuthorizationzBasic {}r)super__init__formatr r )selfr __class__r]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/bookend.pyr/szBookendEmbeddings.__init__textsreturncCstg}|j}d|d<|jtd}|D]&}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_documentsrcCs||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_queryXs 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