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: