AI21Embeddings

本笔记本介绍如何开始使用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("My query to look up")

embeddings.embed_documents(
    ["This is a content of the document", "This is another document"]
)

相关

• 嵌入模型 概念指南
• 嵌入模型 操作指南