百川文本嵌入

截至今日（2024年1月25日），百川文本嵌入在 C-MTEB（中文多任务嵌入基准）排行榜中名列第一。

排行榜（总体 -> 中文部分）：https://huggingface.co/spaces/mteb/leaderboard

官方网站：https://platform.baichuan-ai.com/docs/text-Embedding

使用该嵌入模型需要 API 密钥。您可以通过在 https://platform.baichuan-ai.com/docs/text-Embedding 注册来获取 API 密钥。

百川文本嵌入支持512个标记窗口，并生成1024维向量。

请注意，百川文本嵌入仅支持中文文本嵌入。多语言支持即将推出。

from langchain_community.embeddings import BaichuanTextEmbeddings
embeddings = BaichuanTextEmbeddings(baichuan_api_key="sk-*")

或者，您可以通过以下方式设置 API 密钥：

import os
os.environ["BAICHUAN_API_KEY"] = "YOUR_API_KEY"

text_1 = "今天天气不错"
text_2 = "今天阳光很好"
query_result = embeddings.embed_query(text_1)
query_result

doc_result = embeddings.embed_documents([text_1, text_2])
doc_result