a bg @sRddlmZmZmZddlZddlmZddlmZm Z dZ GdddeeZ dS))AnyDictListN) Embeddings) BaseModel ConfigDictz@cf/baai/bge-base-en-v1.5cseZdZUdZdZeed<eed<eed<eZeed<dZ e ed<d Z e ed <d d iZ eeefed <edfdd ZedddZeeeeedddZeeedddZZS)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 headers)kwargsc s(tjfi|dd|ji|_dS)z,Initialize the Cloudflare Workers AI client.rrN)super__init__r r)selfr __class__/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/cloudflare_workersai.pyr(sz&CloudflareWorkersAIEmbeddings.__init__Zforbidr)extraZprotected_namespaces)textsreturncsjrddDfddtdtjD}g}|D]F}tjjdjdjj d|id}| | d d q>|S) zCompute doc embeddings using Cloudflare Workers AI. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. cSsg|]}|ddqS)  )replace).0textrrr :zACloudflareWorkersAIEmbeddings.embed_documents..csg|]}||jqSr)r)r!irrrrr#<sr//ai/run/r"rjsonresultdata) rrangelenrrequestspostr r r rextendr*)rrZbatchesZ embeddingsbatchresponserr&rembed_documents0s  z-CloudflareWorkersAIEmbeddings.embed_documents)r"rcCsX|jr|ddn|}tj|jd|jd|j|jd|gid}|ddd S) zCompute query embeddings using Cloudflare Workers AI. Args: text: The text to embed. Returns: Embeddings for the text. rrr'r(r"r)r+r,r) rr r/r0r r r rr*)rr"r3rrr embed_queryLs z)CloudflareWorkersAIEmbeddings.embed_query)__name__ __module__ __qualname____doc__r str__annotations__DEFAULT_MODEL_NAMEr rintrboolrrrrrZ model_configrfloatr4r5 __classcell__rrrrr s      r) typingrrrr/Zlangchain_core.embeddingsrZpydanticrrr<rrrrrs