Vercel AI SDK

Log LLM requests with Keywords AI gateway

1. Get Keywords AI API key

After you create an account on Keywords AI, you can get your API key from the API keys page.

2. Add provider credentials

You have to add your own credentials to activate AI gateway otherwise your LLM calls will cause errors. We will use your credentials to call LLMs on your behalf.

We won’t use your credentials for any other purposes and no extra charges will be applied.

Go to the Providers page to add your credentials.

Learn how to add credentials to a specific provider here.

---

3. Compatibility at a glance

SDK helper	Works via Keywords AI?	Switch models?
@ai-sdk/openai	✅ Yes	✅ Yes
@ai-sdk/anthropic	✅ Yes (Anthropic models only)	❌ No
@ai-sdk/google	✅ Yes	✅ Yes

---

4. Integrate Keywords AI gateway

OpenAI

Here’s an example of how to use the Vercel AI SDK with OpenAI:

OpenAI.tsx

import { createOpenAI, OpenAIProvider } from '@ai-sdk/openai'
import { streamText, streamObject } from 'ai'

async function main() {
  // Initialize OpenAI with Keywords gateways
  const client: OpenAIProvider = createOpenAI({
    baseURL: 'https://api.keywordsai.co',
    apiKey: 'YOUR_KEYWORDS_AI_API_KEY',
    compatibility: 'strict',
  })


  const requestParamsDefault: Parameters<typeof streamText>[0] = {
    model: client.chat('gpt-3.5-turbo'),
    messages: [
      {
        role: 'user',
        content: 'Hello! How are you doing today?',
      },
    ],
    temperature: 0.5,
  }

  try {
    console.log('Calling OpenAI with Keywords proxy...')

    const { textStream: proxyTextStream } = await streamText(requestParamsDefault)
    for await (const textPart of proxyTextStream) {
      console.log('Keywords Proxy Response:', textPart)
    }
  } catch (error) {
    console.error('Error:', error)
  }
}

main()

Here is an example of using the Vercel AI SDK with response API.

TypeScript

import { streamText, streamObject, generateText, tool } from "ai";
import { createOpenAI, openai } from "@ai-sdk/openai";
import { z } from "zod";

// Keywords AI parameters to be sent in the header
const keywordsAIHeaderContent = {
    "customer_identifier": "test_customer_identifier_from_header"
}
const encoded = Buffer.from(JSON.stringify(keywordsAIHeaderContent)).toString('base64');

const client = createOpenAI({
  headers: {
    "X-Data-Keywordsai-Params": encoded
  },
  baseURL: "https://api.keywordsai.co/api",
  apiKey: process.env.KEYWORDSAI_API_KEY,
});

const result = await generateText({
  model: client.responses('gpt-4o-mini'),
  prompt: 'What happened in San Francisco last week?',
  tools: {
    web_search_preview: client.tools.webSearchPreview({
      searchContextSize: 'high',
      userLocation: {
        type: 'approximate',
        city: 'San Francisco',
        region: 'California',
      },
    }),
  },
});

console.log(result.text);
console.log(result.sources);

You can find the full source code for this example on GitHub: TypeScript Example

Anthropic

Here’s an example of how to use the Vercel AI SDK with Anthropic:

Anthropic.tsx

import { createAnthropic } from '@ai-sdk/anthropic';
import { streamText, convertToCoreMessages } from 'ai';

// Allow streaming responses up to 30 seconds
export const maxDuration = 30;

export const anthropic: AnthropicProvider = createAnthropic({
  baseURL: "https://api.keywordsai.co/api/anthropic/v1",
  apiKey: process.env.KEYWORDS_AI_API_KEY,
});
   

export async function POST(req: Request) {
  const { messages } = await req.json();

  const result = await streamText({
    model: anthropic('claude-3-5-sonnet-20240620'), // choose any model
    messages: convertToCoreMessages(messages),
  });

  return result.toDataStreamResponse();
}

Google Gemini

Here’s an example of how to use the Vercel AI SDK with Google Gemini:

Google Gemini.tsx

import {
  createGoogleGenerativeAI,
  GoogleGenerativeAIProviderOptions,
} from "@ai-sdk/google";
import { generateText, streamText } from "ai";


const url = `https://api.keywordsai.co/api/google/`;

const google = createGoogleGenerativeAI({
  // custom settings
  baseURL: url,
  apiKey: process.env.KEYWORDSAI_API_KEY,
  
});

const { text } = await generateText({
  model: google("gemini-2.0-flash"),
  providerOptions: {
    google: {
      responseModalities: ["TEXT", "IMAGE"],
    } satisfies GoogleGenerativeAIProviderOptions,
  },
  prompt: "Hello, how are you?",
  temperature: 0.5,
});

5. Add Keywords AI parameters

Adding Keywords AI parameters to the Vercel AI SDK is different than other frameworks. Here is an example of how to do it:

1. Specify Keywords AI params in an object

   You should create an object with the Keywords AI parameters you want to use. Add parameters you want to use as keys in the object.

   const keywordsAIHeaderContent = {
       "customer_params": {
           "customer_identifier": "customer_123",
           "name": "Hendrix Liu", //optional
           "email": "[e-mail]" //optional
       }
       // "cache_enabled": true or other parameters
   }
   
2. Encode the object as a string

   You should encode the object as a string and then you can send it as a header in your request.

   const encoded = Buffer.from(JSON.stringify(keywordsAIHeaderContent)).toString('base64');
   
3. Add the header to your request

   You should send it in the X-Data-Keywordsai-Params header.

   OpenAI

   const client = createOpenAI({
     baseURL: process.env.KEYWORDSAI_ENDPOINT_LOCAL,
     apiKey: process.env.KEYWORDSAI_API_KEY_TEST,
     compatibility: "strict",
     headers: {
       "X-Data-Keywordsai-Params": encoded
     }
   });
   
4. Full example

   OpenAI

   import { streamText, streamObject } from "ai";
   import { createOpenAI } from "@ai-sdk/openai";
   
   const keywordsAIHeaderContent = {
       "customer_identifier": "test_customer_identifier_from_header"
   }
   const encoded = Buffer.from(JSON.stringify(keywordsAIHeaderContent)).toString('base64');
   
   const client = createOpenAI({
     baseURL: process.env.KEYWORDSAI_ENDPOINT_LOCAL,
     apiKey: process.env.KEYWORDSAI_API_KEY_TEST,
     compatibility: "strict",
     headers: {
       "X-Data-Keywordsai-Params": encoded
     }
   });
   
   const requestParamsDefault: Parameters<typeof streamText>[0] = {
     model: client.chat("gpt-4o"),
     messages: [
       {
         role: "user",
         content: "Hello! How are you doing today?",
       },
     ],
     temperature: 0.5,
   };
   
   try {
     console.log("Calling OpenAI with Keywords proxy...");
   
     const { textStream: proxyTextStream } = await streamText(
       requestParamsDefault
     );
     for await (const textPart of proxyTextStream) {
       console.log("Keywords Proxy Response:", textPart);
     }
   } catch (error) {
     console.error("Error:", error);
   }