AI21
LiteLLM supports j2-light, j2-mid and j2-ultra from AI21.
They're available to use without a waitlist.
API KEYS
import os
os.environ["AI21_API_KEY"] = ""
AI21 Models
| Model Name | Function Call | Required OS Variables |
|---|---|---|
| j2-light | completion('j2-light', messages) | os.environ['AI21_API_KEY'] |
| j2-mid | completion('j2-mid', messages) | os.environ['AI21_API_KEY'] |
| j2-ultra | completion('j2-ultra', messages) | os.environ['AI21_API_KEY'] |