Cohere provider

You can proxy requests to Cohere 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 Cohere API endpoints. The following table shows the upstream paths used for each capability.

The following tables show the AI capabilities supported by the Cohere 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 Cohere text generation capabilities:

Embeddings

Support for Cohere embeddings generation:

Rerank

Support for Cohere reranking capabilities:

The base URL is https://api.cohere.com:443/{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 Cohere-compatible endpoint, in which case set the upstream_url option in your AI Model configuration.

By default, AI Gateway uses OpenAI-compatible request formats. Configure a native format in your AI Model to use Cohere-specific APIs and features.

The following native Cohere formats are supported:

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

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": "Cohere Production",
      "name": "my-cohere-account",
      "type": "cohere",
      "config": {
        "auth": {
          "type": "basic",
          "headers": [
            {
              "name": "Authorization",
              "value": "'$COHERE_API_KEY'"
            }
          ]
        }
      }
    }
    '

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