Skip to main content
Open In Colab在 GitHub 上打开

Cloudflare 工人 AI

Cloudflare， Inc.（维基百科）是一家美国公司，提供内容交付网络服务、云网络安全、DDoS 缓解和 ICANN 认证的域名注册服务。

Cloudflare Workers AI 允许您在Cloudflare网络，通过 REST API 从您的代码中获取。

Cloudflare AI 文档列出了所有可用的文本嵌入模型。

建立

Cloudflare 帐户 ID 和 API 令牌都是必需的。从此文档中了解如何获取它们。

import getpass

my_account_id = getpass.getpass("Enter your Cloudflare account ID:\n\n")
my_api_token = getpass.getpass("Enter your Cloudflare API token:\n\n")

例

from langchain_cloudflare.embeddings import (
    CloudflareWorkersAIEmbeddings,
)
embeddings = CloudflareWorkersAIEmbeddings(
    account_id=my_account_id,
    api_token=my_api_token,
    model_name="@cf/baai/bge-small-en-v1.5",
)
# single string embeddings
query_result = embeddings.embed_query("test")
len(query_result), query_result[:3]
(384, [-0.033627357333898544, 0.03982774540781975, 0.03559349477291107])
# string embeddings in batches
batch_query_result = embeddings.embed_documents(["test1", "test2", "test3"])
len(batch_query_result), len(batch_query_result[0])
(3, 384)