From a2fed4059e151222e9c1673e39a8fee53b60700e Mon Sep 17 00:00:00 2001 From: Sunny Wan Date: Wed, 5 Mar 2025 20:32:18 -0500 Subject: [PATCH] added Snowflake config to ProviderConfigManager --- litellm/__init__.py | 1 + litellm/utils.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/litellm/__init__.py b/litellm/__init__.py index db386f7ebd..da89bba901 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -843,6 +843,7 @@ from .llms.databricks.embed.transformation import DatabricksEmbeddingConfig from .llms.predibase.chat.transformation import PredibaseConfig from .llms.replicate.chat.transformation import ReplicateConfig from .llms.cohere.completion.transformation import CohereTextConfig as CohereConfig +from .llms.snowflake.completion.transformation import SnowflakeConfig from .llms.cohere.rerank.transformation import CohereRerankConfig from .llms.cohere.rerank_v2.transformation import CohereRerankV2Config from .llms.azure_ai.rerank.transformation import AzureAIRerankConfig diff --git a/litellm/utils.py b/litellm/utils.py index cbd5e2d0d3..5f01f9f1fc 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -6071,6 +6071,8 @@ class ProviderConfigManager: return litellm.CohereChatConfig() elif litellm.LlmProviders.COHERE == provider: return litellm.CohereConfig() + elif litellm.LlmProviders.SNOWFLAKE == provider: + return litellm.SnowflakeConfig() elif litellm.LlmProviders.CLARIFAI == provider: return litellm.ClarifaiConfig() elif litellm.LlmProviders.ANTHROPIC == provider: