h ddlmZmZmZddlZddlmZddlmZddl m Z m Z dZ eddd Gd d e eZ y) )AnyDictListN) deprecated) Embeddings) BaseModel ConfigDictz@cf/baai/bge-base-en-v1.5z0.3.23z1.0z2langchain_cloudflare.CloudflareWorkersAIEmbeddings)sinceremovalalternative_importceZdZUdZdZeed<eed<eed<eZeed<dZ e ed<d Z e ed <d d iZ eeefed <deffd ZeddZdeedeeefdZdedeefdZxZS)CloudflareWorkersAIEmbeddingsaPCloudflare Workers AI embedding model. To use, you need to provide an API token and account ID to access Cloudflare Workers AI. Example: .. code-block:: python from langchain_community.embeddings import CloudflareWorkersAIEmbeddings account_id = "my_account_id" api_token = "my_secret_api_token" model_name = "@cf/baai/bge-small-en-v1.5" cf = CloudflareWorkersAIEmbeddings( account_id=account_id, api_token=api_token, model_name=model_name ) z-https://api.cloudflare.com/client/v4/accounts api_base_url account_id api_token model_name2 batch_sizeTstrip_new_lines AuthorizationBearer headerskwargsc Pt|di|dd|ji|_y)z,Initialize the Cloudflare Workers AI client.rrN)super__init__rr)selfr __class__s q/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain_community/embeddings/cloudflare_workersai.pyrz&CloudflareWorkersAIEmbeddings.__init__.s+ "6"'74>>2B)CD forbidr)extraprotected_namespacestextsreturnc|jr|Dcgc]}|jdd}}tdt||jDcgc]}||||jz}}g}|D]q}t j |jd|jd|j|jd|i}|j|jdd s|Scc}wcc}w) zCompute doc embeddings using Cloudflare Workers AI. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text.   r//ai/run/textrjsonresultdata) rreplacerangelenrrequestspostrrrrextendr.)rr%r,ibatches embeddingsbatchresponses r embed_documentsz-CloudflareWorkersAIEmbeddings.embed_documents6s   9>?T\\$,?E?1c%j$//:  !a$//) *    AE}}$$%Qt&7x?PQ e_H   hmmoh7? @  A!@ s C#C(r,c |jr|jddn|}tj|jd|j d|j |jd|gi}|jddd S) zCompute query embeddings using Cloudflare Workers AI. Args: text: The text to embed. Returns: Embeddings for the text. r(r)r*r+r,r-r/r0r) rr1r4r5rrrrr.)rr,r;s r embed_queryz)CloudflareWorkersAIEmbeddings.embed_queryRs+/*>*>t||D#&D==  !4??"38DOO;L MLL4&!  }}x(033r!)__name__ __module__ __qualname____doc__rstr__annotations__DEFAULT_MODEL_NAMErrintrboolrrrrr model_configrfloatr<r> __classcell__)rs@r rr s *HL#GON(J(J OT . :GT#s(^:EE H2FLT#Y4U 3D844U 4r!r)typingrrrr4langchain_core._api.deprecationrlangchain_core.embeddingsrpydanticrr rErrr!r rOsK""60*0  K Q4IzQ4  Q4r!