AI21嵌入

本文介绍如何开始使用AI21嵌入模型。

安装

!pip install -qU langchain-ai21

环境设置

我们需要获取一个AI21 API密钥，并设置AI21_API_KEY环境变量：

import os
from getpass import getpass
os.environ["AI21_API_KEY"] = getpass()

使用

from langchain_ai21 import AI21Embeddings
embeddings = AI21Embeddings()

embeddings.embed_query("我要查询的内容")

embeddings.embed_documents(
    ["这是一个文档的内容", "这是另一个文档"]
)