Connect Your App to AI Gateway

AUTH_TOKEN=$(curl -s --request POST \
    --url '<gateway-base>/oauth/idp/token' \
    --header 'content-type: application/x-www-form-urlencoded' \
    --data grant_type=client_credentials \
    --data client_id=<client-id> \
    --data client_secret=<client-secret> | jq -r .access_token)

from authlib.integrations.requests_client import OAuth2Session

token_endpoint = "<gateway-base>/oauth/idp/token"

client = OAuth2Session(
    client_id="<client-id>",
    client_secret="<client-secret>",
    token_endpoint=token_endpoint,
)

token = client.fetch_token(grant_type="client_credentials")

access_token = token["access_token"]

import * as client from 'openid-client';

const config = await client.discovery(
  new URL('<gateway-base>'),
  '<client-id>',
  '<client-secret>',
);

const tokens = await client.clientCredentialsGrant(config);
const accessToken = tokens.access_token;

import os
from openai import OpenAI

client = OpenAI(
    base_url=os.environ["PROXY_URL"],       # .../llm/v1/providers/my-openai
    api_key=os.environ["AUTH_TOKEN"],        # OIDC access token
)

response = client.chat.completions.create(
    model="gpt-4o",                          # native OpenAI model ID
    messages=[{"role": "user", "content": "Hello from AI Gateway"}],
)
print(response.choices[0].message.content)

import os
from anthropic import Anthropic

client = Anthropic(
    base_url=os.environ["PROXY_URL"],       # .../llm/v1/providers/my-anthropic
    auth_token=os.environ["AUTH_TOKEN"],     # OIDC access token
)

message = client.messages.create(
    model="claude-sonnet-4-6",
    max_tokens=1024,
    messages=[{"role": "user", "content": "Hello from AI Gateway"}],
)
print(message.content[0].text)

import os
from google import genai
from google.genai import types

client = genai.Client(
    api_key=os.environ["AUTH_TOKEN"],        # the SDK requires a key
    http_options=types.HttpOptions(
        base_url=f"{os.environ['PROXY_URL']}/v1beta",  # .../llm/v1/providers/my-google/v1beta
        headers={"X-Redpanda-Cloud-Token": os.environ["AUTH_TOKEN"]},
    ),
)

response = client.models.generate_content(
    model="gemini-2.0-flash",
    contents="Hello from AI Gateway",
)
print(response.text)

import os, httpx

# Bedrock 4.6+ Anthropic models require an inference profile (us./eu./global.).
# Replace with the inference profile your provider exposes.
response = httpx.post(
    f"{os.environ['PROXY_URL']}/model/us.anthropic.claude-sonnet-4-6/invoke",
    headers={"Authorization": f"Bearer {os.environ['AUTH_TOKEN']}"},
    json={
        "anthropic_version": "bedrock-2023-05-31",
        "messages": [{"role": "user", "content": "Hello"}],
        "max_tokens": 1024,
    },
)
print(response.json())

import os
from openai import OpenAI

client = OpenAI(
    base_url=os.environ["PROXY_URL"],       # .../llm/v1/providers/my-vllm
    api_key=os.environ["AUTH_TOKEN"],
)

response = client.chat.completions.create(
    model="meta-llama/Llama-3.3-70B-Instruct",  # as exposed by your upstream
    messages=[{"role": "user", "content": "Hello"}],
)