langchain_community.embeddings.cloudflare_workersai
.CloudflareWorkersAIEmbeddings¶
- 类 langchain_community.embeddings.cloudflare_workersai.CloudflareWorkersAIEmbeddings[source]¶
基类:
BaseModel
,Embeddings
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) – 需要嵌入的文本。
- 返回值
文本的嵌入。
- 返回类型
列表[浮点数]