> ## Documentation Index
> Fetch the complete documentation index at: https://docs.obiguard.ai/llms.txt
> Use this file to discover all available pages before exploring further.

# AI21

Obiguard provides a robust and secure gateway to facilitate the integration of various Large Language Models (LLMs) into your applications, including [AI21](https://ai21.com).

With Obiguard, 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](/virtual-keys) system.

<Note>
  Provider Slug. **ai21**
</Note>

## Obiguard SDK Integration with AI21 Models

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

### 1. Install the Obiguard SDK

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

<Tabs>
  <Tab title="Python SDK">
    ```sh theme={null}
    pip install obiguard
    ```
  </Tab>
</Tabs>

### 2. Initialize Obiguard with the Virtual Key

To use AI21 with Obiguard, [get your API key from here](https://studio.ai21.com/account/api-key), then add it to Obiguard to create the virtual key.

<Tabs>
  <Tab title="Python SDK">
    ```python theme={null}
    from obiguard import Obiguard

    client = Obiguard(
      obiguard_api_key="vk-obg***",  # Your Obiguard virtual key
      virtual_key="VIRTUAL_KEY"   # Replace with your virtual key for AI21
    )
    ```
  </Tab>
</Tabs>

### 3. Invoke Chat Completions with AI21

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

<Tabs>
  <Tab title="Python SDK">
    ```python theme={null}
    completion = client.chat.completions.create(
        messages= [{ "role": 'user', "content": 'Say this is a test' }],
        model= 'jamba-1-5-large'
    )

    print(completion)
    ```
  </Tab>
</Tabs>

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

<Card title="SDK" href="/api-reference/sdk/python" />
