Skip to main content

Cloudflare Workers AI

Cloudflare, Inc. (维基百科) 是一家提供内容分发网络服务、云网络安全、DDoS 缓解和 ICANN 认证的域名注册服务的美国公司。

Cloudflare Workers AI 允许您通过 REST API 从代码在 Cloudflare 网络上运行机器学习模型。

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")

示例

<!--IMPORTS:[{"imported": "CloudflareWorkersAIEmbeddings", "source": "langchain_community.embeddings.cloudflare_workersai", "docs": "https://python.langchain.com/api_reference/community/embeddings/langchain_community.embeddings.cloudflare_workersai.CloudflareWorkersAIEmbeddings.html", "title": "Cloudflare Workers AI"}]-->
from langchain_community.embeddings.cloudflare_workersai 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)

相关


Was this page helpful?


You can also leave detailed feedback on GitHub.

扫我,入群扫我,找书