Cerebras provider

You can proxy requests to Cerebras AI models through AI Gateway by creating AI Model Provider and AI Model entities. This reference documents all supported AI capabilities, configuration requirements, and provider-specific details needed for proper integration.

AI Gateway automatically routes requests to the appropriate Cerebras API endpoints. The following table shows the upstream paths used for each capability.

The following tables show the AI capabilities supported by the Cerebras provider when configuring AI Models.

By default, AI Gateway uses the path templates shown in the tables below (e.g., /chat/completions, /embeddings, etc.). To customize these paths, configure the config.paths field in your AI Model entity. Custom paths take the form {configured_path}/{template_path} — for example, if you set a custom path of /v2, requests to /embeddings would be routed to /v2/embeddings.

Text generation

Support for Cerebras text generation capabilities:

The base URL is https://api.cerebras.ai/{capability_path}. The {capability_path} is determined by the AI capability.

AI Gateway uses this URL automatically. You only need to configure a URL if you’re using a self-hosted or Cerebras-compatible endpoint, in which case set the upstream_url option in your AI Model configuration.

To use Cerebras with AI Gateway, configure a new AI Model Provider. You can then access supported AI Models from Cerebras.

Here’s a minimal configuration for chat completions:

curl -X POST https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/model-providers \
    --header "accept: application/json" \
    --header "Content-Type: application/json" \
    --header "Authorization: Bearer $KONNECT_TOKEN" \
    --data '
    {
      "display_name": "Cerebras Production",
      "name": "my-cerebras-account",
      "type": "cerebras",
      "config": {
        "auth": {
          "type": "basic",
          "headers": [
            {
              "name": "Authorization",
              "value": "'$CEREBRAS_API_KEY'"
            }
          ]
        }
      }
    }
    '

ai_gateway_model_providers:
  - ref: my-cerebras-account
    ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
    display_name: Cerebras Production
    name: my-cerebras-account
    type: cerebras
    config:
      auth:
        type: basic
        headers:
        - name: Authorization
          value: !secret {source: !env CEREBRAS_API_KEY}