AI21 Labs

AI21 Labs es el laboratorio israelí detrás de Jamba, un híbrido Mamba-Transformer que se mantiene barato en documentos largos porque no paga el coste completo de Transformer en cada token. 256K de contexto. Las cuentas nuevas reciben $10 de crédito durante tres meses.

Obtener clave de API

Dos modelos en nuestro radar, compatible con OpenAI, sin tarjeta para empezar. El antiguo listado /studio/v1/models devuelve ahora 410 — coge el id Jamba actual de la consola. Úsalo cuando el documento es el protagonista; menos cuando quieres un chat generalista.

Modelos gratuitos: 2
Contexto máximo: 256K
Nivel gratuito: $10 de crédito durante 3 meses
Requisito: Cuenta de correo, sin tarjeta

Punto de conexión

URL basehttps://api.ai21.com/studio/v1

Variable de entornoAI21_API_KEY

Límite de tasa$10 of credits for 3 months

Compatible con OpenAI
SDK nativo: AI21 SDK
texto

Ejemplos de código

npm install openai

import OpenAI from 'openai'

const client = new OpenAI({
  baseURL: 'https://api.ai21.com/studio/v1',
  apiKey: process.env.AI21_API_KEY
})

const response = await client.chat.completions.create({
  model: 'jamba-mini-2',
  messages: [{ role: 'user', content: 'Explain closures in one paragraph.' }]
})

console.log(response.choices[0].message.content)

pip install openai

import os
from openai import OpenAI

client = OpenAI(
    base_url="https://api.ai21.com/studio/v1",
    api_key=os.environ["AI21_API_KEY"],
)

response = client.chat.completions.create(
    model="jamba-mini-2",
    messages=[{"role": "user", "content": "Explain closures in one paragraph."}],
)

print(response.choices[0].message.content)

curl https://api.ai21.com/studio/v1/chat/completions \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $AI21_API_KEY" \
  -d '{
    "model": "jamba-mini-2",
    "messages": [{"role": "user", "content": "Explain closures in one paragraph."}]
  }'

pip install ai21

import os
from ai21 import AI21Client
from ai21.models.chat import ChatMessage

client = AI21Client(api_key=os.environ["AI21_API_KEY"])

response = client.chat.completions.create(
    model="jamba-mini-2",
    messages=[ChatMessage(role="user", content="Explain closures in one paragraph.")],
)

print(response.choices[0].message.content)

Vale la pena saber

El listado /studio/v1/models ahora devuelve 410. Coge los ids Jamba actuales de la consola.

Modelos que obtienes gratis2

Jamba Mini 2jamba-mini-2
Jamba Large 1.6jamba-large-1.6