Skip to main content

AI21 Labs

AI21 Labs is a company specializing in Natural Language Processing (NLP), which develops AI systems that can understand and generate natural language.

This page covers how to use the AI21 ecosystem within LangChain.

Installation and Setup​

  • Get an AI21 api key and set it as an environment variable (AI21_API_KEY)
  • Install the Python package:
pip install langchain-ai21

LLMs​

See a usage example.

from langchain_community.llms import AI21

API Reference:

Chat models​

See a usage example.

from langchain_ai21 import ChatAI21

API Reference:

Embedding models​

See a usage example.

from langchain_ai21 import AI21Embeddings

API Reference:


Help us out by providing feedback on this documentation page: