langchain_community.embeddings.cloudflare_workersai.CloudflareWorkersAIEmbeddings

langchain_community.embeddings.cloudflare_workersai.CloudflareWorkersAIEmbeddings[source]

基类:BaseModelEmbeddings

Cloudflare Workers AI 嵌入模型。

要使用,您需要提供API令牌和账户ID以访问Cloudflare Workers AI。

示例

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
)

初始化Cloudflare Workers AI客户端。

参数 account_id: str [必需]
参数 api_base_url: str = 'https://api.cloudflare.com/client/v4/accounts'
参数 api_token: str [必需]
参数 batch_size: int = 50
参数 headers: Dict[str, str] = {'Authorization': 'Bearer '}
参数 model_name: str = '@cf/baai/bge-base-en-v1.5'
参数 strip_new_lines: bool = True
async aembed_documents(texts: List[str]) List[List[float]]

异步嵌入搜索文档。

参数

texts (列表str) – 嵌入的文本列表。

返回值

嵌入列表。

返回类型

列表[列表[浮点数]]

async aembed_query(Text: str) List[float]

异步嵌入查询文本。

参数

text (str) – 需要嵌入的文本。

返回值

嵌入。

返回类型

列表[浮点数]

embed_documents(texts: List[str]) List[List[float]][源代码]

使用Cloudflare Workers AI计算文档嵌入。

参数

texts (列表str) – 需要嵌入的文本列表。

返回值

每个文本对应的嵌入列表。

返回类型

列表[列表[浮点数]]

embed_query(text: str) List[float][源代码]

使用Cloudflare Workers AI计算查询嵌入。

参数

text (str) – 需要嵌入的文本。

返回值

文本的嵌入。

返回类型

列表[浮点数]

使用CloudflareWorkersAIEmbeddings的示例