Skip to content

Conversation

ea-rus
Copy link
Contributor

@ea-rus ea-rus commented Oct 20, 2025

Title

Improved snowflake provider support: embeddings, PAT, account_id

Relevant issues

Pre-Submission checklist

Please complete all items before asking a LiteLLM maintainer to review your PR

  • I have Added testing in the tests/litellm/ directory, Adding at least 1 test is a hard requirement - see details
  • I have added a screenshot of my new test passing locally
  • My PR passes all unit tests on make test-unit
  • My PR's scope is as isolated as possible, it only solves 1 specific problem

Type

🆕 New Feature

Changes

Improved snowflake provider support:

  1. Added cortex embeddings support

Example:

response = litellm.embedding(
    "snowflake/snowflake-arctic-embed-l-v2.0",
    input=['test',],
)
  1. Added support of programmatic access tokens (PAT)
    The lifetime of PAT is much longer than JWT key

Example:

response = litellm.completion(
    "snowflake/mistral-large",
    messages=[...],
    pat_key="xxxxx"  # <--
)
  1. Added option to pass snowflake account id as parameter. (At the moment it is possible to do only in via ENV variable)

Example:

response = litellm.completion(
    "snowflake/mistral-large",
    messages=[...],
    account_id="AAAA-BBBB"  # <--
)

pat_key and account_id parameters work both for embeding and completion.

Tests:
image

@vercel
Copy link

vercel bot commented Oct 20, 2025

@ea-rus is attempting to deploy a commit to the CLERKIEAI Team on Vercel.

A member of the Team first needs to authorize it.

def get_config(cls):
return super().get_config()

class SnowflakeBaseConfig:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

refactor this base config into a separate file - e.g. the utils.py file you wrote inside snowflake/

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

done, moved to utils.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants