AI21

Portkey provides a robust and secure gateway to facilitate the integration of various Large Language Models (LLMs) into your applications, including AI21.

With Portkey, you can take advantage of features like fast AI gateway access, observability, prompt management, and more, all while ensuring the secure management of your LLM API keys through a virtual key system.

Provider Slug**: **ai21

Portkey SDK Integration with AI21 Models

Portkey provides a consistent API to interact with models from various providers. To integrate AI21 with Portkey:

1. Install the Portkey SDK

Add the Portkey SDK to your application to interact with AI21 AI's API through Portkey's gateway.

NodeJS

npm install --save portkey-ai

Python

pip install portkey-ai

2. Initialize Portkey with the Virtual Key

To use AI21 with Portkey, get your API key from here, then add it to Portkey to create the virtual key.

NodeJS SDK

import Portkey from 'portkey-ai'
 
const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY", // defaults to process.env["PORTKEY_API_KEY"]
    virtualKey: "VIRTUAL_KEY" // Your AI21 Virtual Key
})

Python SDK

from portkey_ai import Portkey

portkey = Portkey(
    api_key="PORTKEY_API_KEY",  # Replace with your Portkey API key
    virtual_key="VIRTUAL_KEY"   # Replace with your virtual key for AI21
)

3. Invoke Chat Completions with AI21

Use the Portkey instance to send requests to AI21. You can also override the virtual key directly in the API call if needed.

NodeJS SDK

const chatCompletion = await portkey.chat.completions.create({
    messages: [{ role: 'user', content: 'Say this is a test' }],
    model: 'jamba-1-5-large',
});

console.log(chatCompletion.choices);

Python SDK

completion = portkey.chat.completions.create(
    messages= [{ "role": 'user', "content": 'Say this is a test' }],
    model= 'jamba-1-5-large'
)

print(completion)

Managing AI21 Prompts

You can manage all prompts to AI21 in the Prompt Library. All the current models of AI21 are supported and you can easily start testing different prompts.

Once you're ready with your prompt, you can use the portkey.prompts.completions.create interface to use the prompt in your application.

The complete list of features supported in the SDK are available on the link below.

SDK

You'll find more information in the relevant sections

1. Add metadata to your requests

2. Add gateway configs to your AI21 requests

3. Tracing AI21 requests

4. Setup a fallback from OpenAI to AI21 APIs