Skip to content

Conversation

ea-rus
Copy link

@ea-rus ea-rus commented Oct 9, 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

Copy link

vercel bot commented Oct 9, 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.

@krrishdholakia
Copy link
Contributor

let me know when this is ready for review! @ea-rus

@ea-rus ea-rus marked this pull request as ready for review October 10, 2025 06:04
@ea-rus
Copy link
Author

ea-rus commented Oct 10, 2025

@krrishdholakia, thank you. It is ready for review, I've just marked it
It has error in unit tests but the same error I have on current main branch.

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.

2 participants