ChatAI21
This notebook covers how to get started with AI21 chat models.
Installation
!pip install -qU langchain-ai21
Environment Setup
We'll need to get a AI21 API key and set the AI21_API_KEY
environment variable:
import os
from getpass import getpass
os.environ["AI21_API_KEY"] = getpass()
Usage
from langchain_ai21 import ChatAI21
from langchain_core.prompts import ChatPromptTemplate
chat = ChatAI21(model="j2-ultra")
prompt = ChatPromptTemplate.from_messages(
[
("system", "You are a helpful assistant that translates English to French."),
("human", "Translate this sentence from English to French. {english_text}."),
]
)
chain = prompt | chat
chain.invoke({"english_text": "Hello, how are you?"})
API Reference:
AIMessage(content='Bonjour, comment vas-tu?')