Using elephant-1 on Vercel AI Gateway

Implementation guide · Elephant · Baseten

Serverless
Open Source

Vercel AI Gateway exposes elephant-1 through model ID baseten/elephant-1. Use the setup steps, capabilities, and official provider links below to validate this route before deployment.

Last refreshed 2026-10-21. Next refresh: weekly.

Quick Start

  1. 1
    Create an account at Vercel AI Gateway and generate an API key.
  2. 2
    Use the Vercel AI Gateway SDK or REST API to call baseten/elephant-1 — see the documentation for request format.

Code Examples

Install
pip install openai
API key
AI_GATEWAY_API_KEY
Model ID
baseten/elephant-1

creator/model-name e.g. kwaipilot/kat-coder-pro-v2

import os
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["AI_GATEWAY_API_KEY"],
    base_url="https://ai-gateway.vercel.sh/v1"
)
response = client.chat.completions.create(
    model="baseten/elephant-1",
    messages=[{"role": "user", "content": "Hello"}]
)
print(response.choices[0].message.content)

Capabilities

Structured Outputs

About elephant-1

elephant-1 is a 9B open-weight decision model from Base Labs, Baseten's research lab, released October 21, 2026. It is a full fine-tune of Qwen3.5-9B that turns a state and typed questions into calibrated probabilities for choice, yes/no and score answers in a single forward pass, without generating free-form text, for routing, classification and rubric scoring. It shares mammoth-1's block-bidirectional attention and 6.21M-example training mix and takes text-only input. Weights are on Hugging Face at baselabs/elephant-1 under MIT. Baseten Model APIs serve it through the System One API (/v1/systemone) as decision-models/baseten-elephant-9b-latest, free through October 27, 2026.

Model Specs

Released2026-10-21
Parameters9B
Context16K

Provider

Vercel AI Gateway

Vercel