AI21 Labs

AI21 Labs 是 Jamba 背后的以色列实验室：混合 Mamba-Transformer 架构，不必在每个 token 上支付完整 Transformer 成本，所以长文档依然便宜。256K 上下文。新账户三个月内 $10 赠款。

获取 API 密钥

我们追踪两个模型，OpenAI 兼容，起步不要卡。旧的 /studio/v1/models 现在返回 410 —— 从控制台取当前的 Jamba id。文档是主角时用它；要通用对话模型就别。

免费模型: 2
最大上下文: 256K
免费额度: 3 个月 $10 赠款
注册要求: 只要邮箱，不要卡

端点

接口地址https://api.ai21.com/studio/v1

环境变量AI21_API_KEY

速率限制$10 of credits for 3 months

OpenAI 兼容
官方 SDK：AI21 SDK
文本

代码示例

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)

值得注意

/studio/v1/models 现在返回 410。请从控制台取当前的 Jamba id。

你能免费使用的模型2

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