AI21嵌入模型
这将帮助您使用LangChain开始使用AI21嵌入模型。有关AI21Embeddings
功能和配置选项的详细文档,请参阅API参考。
概述
集成细节
Provider | Package |
---|---|
AI21 | langchain-ai21 |
设置
要访问AI21 嵌入模型,您需要创建一个AI21账户,获取API密钥,并安装langchain-ai21
集成包。
凭证
前往https://docs.ai21.com/注册AI21并生成API密钥。完成后设置AI21_API_KEY
环境变量:
import getpass
import os
if not os.getenv("AI21_API_KEY"):
os.environ["AI21_API_KEY"] = getpass.getpass("Enter your AI21 API key: ")
如果您想要自动跟踪模型调用,可以通过取消注释下面的内容来设置您的 LangSmith API 密钥:
# os.environ["LANGCHAIN_TRACING_V2"] = "true"
# os.environ["LANGCHAIN_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
安装
LangChain AI21 集成位于 langchain-ai21
包中:
%pip install -qU langchain-ai21
实例化
现在我们可以实例化我们的模型对象并生成聊天完成:
from langchain_ai21 import AI21Embeddings
embeddings = AI21Embeddings(
# Can optionally increase or decrease the batch_size
# to improve latency.
# Use larger batch sizes with smaller documents, and
# smaller batch sizes with larger documents.
# batch_size=256,
)