Snowflake Cortex

Portkey provides a robust and secure gateway to facilitate the integration of various Large Language Models (LLMs) into your applications, including Snowlfake Cortex 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 Snowflake Cortex Models

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

​

1. Install the Portkey SDK

Add the Portkey SDK to your application to interact with Snowflake Cortex AI’s API through Portkey’s gateway.

npm install --save portkey-ai

npm install --save portkey-ai

pip install portkey-ai

​

2. Initialize Portkey with the Virtual Key

To use Snowflake Cortex with Portkey, get your API key/JWT Token from the Snowflake Platform, then add it to Portkey to create the virtual key.

import Portkey from 'portkey-ai'

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

import Portkey from 'portkey-ai'

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

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 Snowflake Cortex
)

​

3. Invoke Chat Completions with Snowflake Cortex

Use the Portkey instance to send requests to Snowflake Cortex. 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: 'claude-3-5-sonnet',
});

console.log(chatCompletion.choices);

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

console.log(chatCompletion.choices);

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

print(completion)

​

Managing Snowflake Cortex Prompts

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

You’ll find more information in the relevant sections:

1. Add metadata to your requests
2. Add gateway configs to your Snowflake Cortex requests
3. Tracing Snowflake Cortex requests
4. Setup a fallback from OpenAI to Snowflake Cortex APIs