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. azure-openai

Portkey SDK Integration with Azure OpenAI

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

First, add your Azure details to Portkey’s Virtual Keys

Here’s a step-by-step guide:

  1. Request access to Azure OpenAI here.
  2. Create a resource in the Azure portal here. (This will be your Resource Name)
  3. Deploy a model in Azure OpenAI Studio here. (This will be your Deployment Name)
  4. Select your Foundation Model from the dropdowon on the modal.
  5. Now, on Azure OpenAI studio, go to any playground (chat or completions), click on a UI element called “View code”. Note down the API version & API key from here. (This will be your Azure API Version & Azure API Key)

When you input these details, the foundation model will be auto populated. More details in this guide.

If you do not want to add your Azure details to Portkey vault, you can also directly pass them while instantiating the Portkey client. More on that here.

Now, let’s make a request using this virtual key!

1. Install the Portkey SDK

Add the Portkey SDK to your application to interact with Azure OpenAI’s API through Portkey’s gateway.

npm install --save portkey-ai

2. Initialize Portkey with the Virtual Key

Set up Portkey with your virtual key as part of the initialization configuration. You can create a virtual key for Azure in the Portkey UI.

import Portkey from 'portkey-ai'

const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY", // defaults to process.env["PORTKEY_API_KEY"]
    virtualKey: "AZURE_VIRTUAL_KEY" // Your Azure Virtual Key
})

3. Invoke Chat Completions with Azure OpenAI

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

const chatCompletion = await portkey.chat.completions.create({
    messages: [{ role: 'user', content: 'Say this is a test' }],
    model: 'gpt4', // This would be your deployment or model name
});

console.log(chatCompletion.choices);

Managing Azure OpenAI Prompts

You can manage all prompts to Azure OpenAI in the Prompt Library. All the current models of OpenAI 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.

Image Generation

Portkey supports multiple modalities for Azure OpenAI and you can make image generation requests through Portkey’s AI Gateway the same way as making completion calls.

import Portkey from 'portkey-ai'

const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY",
    virtualKey: "DALL-E_VIRTUAL_KEY" // Referencing a Dall-E Azure deployment with Virtual Key
})

const image = await portkey.images.generate({
  prompt:"Lucy in the sky with diamonds",
  size:"1024x1024"
})

Portkey’s fast AI gateway captures the information about the request on your Portkey Dashboard. On your logs screen, you’d be able to see this request with the request and response.

Log view for an image generation request on Azure OpenAI

More information on image generation is available in the API Reference.


Making Requests Without Virtual Keys

Here’s how you can pass your Azure OpenAI details & secrets directly without using the Virutal Keys feature.

Key Mapping

In a typical Azure OpenAI request,

curl https://{YOUR_RESOURCE_NAME}.openai.azure.com/openai/deployments/{YOUR_DEPLOYMENT_NAME}/chat/completions?api-version={API_VERSION} \
  -H "Content-Type: application/json" \
  -H "api-key: {YOUR_API_KEY}" \
  -d '{"messages": [{"role": "system", "content": "Hello"}] }'
ParameterNode SDKPython SDKREST Headers
AZURE RESOURCE NAMEazureResourceNameazure_resource_namex-portkey-azure-resource-name
AZURE DEPLOYMENT NAMEazureDeploymentIdazure_deployment_idx-portkey-azure-deployment-id
API VERSIONazureApiVersionazure_api_versionx-portkey-azure-api-version
AZURE API KEYAuthorization: “Bearer + Authorization = “Bearer + Authorization
AZURE MODEL NAMEazureModelNameazure_model_namex-portkey-azure-model-name

Example

import Portkey from 'portkey-ai'

const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY",
    provider: "azure-openai",
    azureResourceName: "AZURE_RESOURCE_NAME",
    azureDeploymentId: "AZURE_DEPLOYMENT_NAME",
    azureApiVersion: "AZURE_API_VERSION",
    azureModelName: "AZURE_MODEL_NAME"
    Authorization: "Bearer API_KEY"
})

How to Pass JWT (JSON Web Tokens)

If you have configured fine-grained access for Azure OpenAI and need to use JSON web token (JWT) in the Authorization header instead of the regular API Key, you can use the forwardHeaders parameter to do this.

import Portkey from 'portkey-ai'

const portkey = new Portkey({
    apiKey: "PORTKEY_API_KEY",
    provider: "azure-openai",
    azureResourceName: "AZURE_RESOURCE_NAME",
    azureDeploymendId: "AZURE_DEPLOYMENT_NAME",
    azureApiVersion: "AZURE_API_VERSION",
    azureModelName: "AZURE_MODEL_NAME",
    Authorization: "Bearer JWT_KEY", // Pass your JWT here
    forwardHeaders: [ "Authorization" ]
})

For further questions on custom Azure deployments or fine-grained access tokens, reach out to us on [email protected]

Next Steps

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 Azure OpenAI requests
  3. Tracing Azure OpenAI requests
  4. Setup a fallback from OpenAI to Azure OpenAI APIs