phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added Snowflake config to ProviderConfigManager

Browse source
This commit is contained in:
Sunny Wan 2025-03-05 20:32:18 -05:00
parent ce55a2d06e
commit a2fed4059e
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

@ -843,6 +843,7 @@ from .llms.databricks.embed.transformation import DatabricksEmbeddingConfig
from .llms.predibase.chat.transformation import PredibaseConfig from .llms.predibase.chat.transformation import PredibaseConfig
from .llms.replicate.chat.transformation import ReplicateConfig from .llms.replicate.chat.transformation import ReplicateConfig
from .llms.cohere.completion.transformation import CohereTextConfig as CohereConfig 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.transformation import CohereRerankConfig
from .llms.cohere.rerank_v2.transformation import CohereRerankV2Config from .llms.cohere.rerank_v2.transformation import CohereRerankV2Config
from .llms.azure_ai.rerank.transformation import AzureAIRerankConfig from .llms.azure_ai.rerank.transformation import AzureAIRerankConfig

2
litellm/utils.py
View file

@ -6071,6 +6071,8 @@ class ProviderConfigManager:
return litellm.CohereChatConfig() return litellm.CohereChatConfig()
elif litellm.LlmProviders.COHERE == provider: elif litellm.LlmProviders.COHERE == provider:
return litellm.CohereConfig() return litellm.CohereConfig()
elif litellm.LlmProviders.SNOWFLAKE == provider:
return litellm.SnowflakeConfig()
elif litellm.LlmProviders.CLARIFAI == provider: elif litellm.LlmProviders.CLARIFAI == provider:
return litellm.ClarifaiConfig() return litellm.ClarifaiConfig()
elif litellm.LlmProviders.ANTHROPIC == provider: elif litellm.LlmProviders.ANTHROPIC == provider: