AI21 Labs website screenshot

AI21 Labs

AI21 Labs is an enterprise foundation-model company best known for the Jamba family of open-weight hybrid Mamba/Transformer models and AI21 Maestro, a dynamic planning system that orchestrates tools, retrieval, and validated output during inference. The platform exposes a Bearer-token REST API at api.ai21.com covering chat completions, conversational RAG over a managed file library, batch processing, fine-tuning, and function calling. Official Python and TypeScript SDKs wrap the API with sync and async clients and integrate natively into LangChain and LlamaIndex. Jamba weights are also published on Hugging Face for self-hosted vLLM deployment.

8 APIs 0 Features
AIFoundation ModelsLLMJambaMambaRAGAgentsMaestroInferenceEnterprise AIFine-Tuning

APIs

AI21 Jamba Chat Completions API

Conversational completions endpoint backed by the Jamba family of hybrid Mamba/Transformer models (Jamba 1.5 Mini, Jamba 1.5 Large, Jamba 1.6). Accepts a message array, system p...

AI21 Maestro API

Dynamic planning system that, at inference time, determines the optimal sequence of actions to solve a task. Exposes run creation and retrieval endpoints, supports validated out...

AI21 File Library API

Managed file storage for RAG workflows. Upload, list, retrieve, update, and delete documents that Maestro and Jamba endpoints can ground on at inference time.

AI21 Batch API

Asynchronous batch processing for large volumes of Jamba completions. Submit a batch job, poll for status, and download results when the run completes.

AI21 Fine-Tuning API

Fine-tuning service for Jamba models supporting full, LoRA, and QLoRA strategies. Create training jobs from uploaded datasets and deploy the resulting model variants behind the ...

AI21 Python SDK

Official Python SDK (ai21 on PyPI) wrapping the AI21 Studio REST API with sync and async clients, streaming support, LangChain / LlamaIndex integrations, and helpers for chat, M...

AI21 TypeScript / JavaScript SDK

Official TypeScript and JavaScript client for the AI21 Studio API, with typed request and response models and streaming helpers for browser and Node.js runtimes.

AI21 Jamba on Hugging Face

Open-weight releases of the Jamba family on Hugging Face for self-hosted inference via vLLM, Transformers, and other runtimes.

Collections

AI21 Studio API

OPEN

GraphQL

AI21 Labs GraphQL Schema

This conceptual GraphQL schema maps the AI21 Labs REST API surface — covering the Jamba chat completion, Maestro agentic planning, conversational RAG file library, batch process...

GRAPHQL

Pricing Plans

Ai21 Labs Plans Pricing

1 plans

PLANS

Rate Limits

Ai21 Labs Rate Limits

2 limits

RATE LIMITS

FinOps

Ai21 Labs Finops

FINOPS

Event Specifications

AI21 Studio Streaming API

AsyncAPI description of AI21 Labs' documented streaming surface. The AI21 Studio Jamba chat completions endpoint streams partial responses over HTTP using Server-Sent Events (SS...

ASYNCAPI

Resources

🔗
Website
Website
🔗
Documentation
Documentation
🔗
APIReference
APIReference
👥
GitHubOrganization
GitHubOrganization
🔗
HuggingFace
HuggingFace
📰
Blog
Blog
💰
Pricing
Pricing
📄
ChangeLog
ChangeLog
🔗
LinkedIn
LinkedIn
🔗
X
X
🔗
LlmsText
LlmsText

Sources

Raw ↑ 
opencollection: 1.0.0
info:
  name: AI21 Studio API
  version: 1.0.0
request:
  auth:
    type: bearer
    token: '{{bearerToken}}'
items:
- info:
    name: Chat
    type: folder
  items:
  - info:
      name: Create a chat completion with a Jamba model
      type: http
    http:
      method: POST
      url: https://api.ai21.com/studio/v1/chat/completions
      body:
        type: json
        data: '{}'
    docs: Create a chat completion with a Jamba model
- info:
    name: Maestro
    type: folder
  items:
  - info:
      name: Create a Maestro run
      type: http
    http:
      method: POST
      url: https://api.ai21.com/studio/v1/maestro/runs
      body:
        type: json
        data: '{}'
    docs: Create a Maestro run
  - info:
      name: Retrieve a Maestro run by ID
      type: http
    http:
      method: GET
      url: https://api.ai21.com/studio/v1/maestro/runs/:run_id
      params:
      - name: run_id
        value: ''
        type: path
    docs: Retrieve a Maestro run by ID
- info:
    name: Library
    type: folder
  items:
  - info:
      name: List documents in the file library
      type: http
    http:
      method: GET
      url: https://api.ai21.com/studio/v1/library/files
      params:
      - name: name
        value: ''
        type: query
      - name: status
        value: ''
        type: query
      - name: labels
        value: ''
        type: query
    docs: List documents in the file library
  - info:
      name: Upload a file to the library
      type: http
    http:
      method: POST
      url: https://api.ai21.com/studio/v1/library/files
      body:
        type: multipart-form
        data:
        - name: file
          type: text
          value: ''
        - name: labels
          type: text
          value: ''
        - name: publicUrl
          type: text
          value: ''
    docs: Upload a file to the library
bundled: true