Stability AI

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

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.

Portkey SDK Integration with Stability AI

Portkey provides a consistent API to interact with image generation models from various providers. To integrate Stability AI with Portkey:

1. Install the Portkey SDK

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

NodeJS

npm install --save portkey-ai

Python

pip install 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 Stability AI in your Portkey account.

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 Stability AI 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 Stability AI
)

3. Invoke Image Generation with Stability AI

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

NodeJS SDK

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

Python SDK

image = portkey.images.generate(
  model="stable-diffusion-v1-6",
  prompt="Lucy in the sky with diamonds",
  size="1024x1024"
)

Notice how we're using the OpenAI's image generation signature to prompt Stability allowing greater flexibility to change models and providers later if necessary.

Next Steps

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

pagePortkey SDK Client

You'll find more information in the relevant sections:

1. Add metadata to your requests

2. Add gateway configs to your Stability AI requests

3. Tracing Stability AI's requests

4. Setup a fallback from OpenAI to Stability

5. Image generation API Reference