diff --git a/litellm/__init__.py b/litellm/__init__.py index 0086e20899..db386f7ebd 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -600,7 +600,6 @@ ollama_models = ["llama2"] maritalk_models = ["maritalk"] -# Probably shouldn't hard code this, change later snowflake_models = [ "snowflake/deepseek-r1", "snowflake/claude-3-5-sonnet", diff --git a/litellm/litellm_core_utils/get_llm_provider_logic.py b/litellm/litellm_core_utils/get_llm_provider_logic.py index 5e9489d427..0bf74c5dca 100644 --- a/litellm/litellm_core_utils/get_llm_provider_logic.py +++ b/litellm/litellm_core_utils/get_llm_provider_logic.py @@ -577,7 +577,7 @@ def _get_openai_compatible_provider_info( # noqa: PLR0915 or get_secret("SNOWFLAKE_API_BASE") or f"https://{get_secret('SNOWFLAKE_ACCOUNT_ID')}.snowflakecomputing.com/api/v2/cortex/inference:complete" ) # type: ignore - dynamic_api_key = api_key or get_secret("SNOWFLAKE_JWT") # Snowflake doesn't use API keys so this will have to change. Support of OAuth and JWT + dynamic_api_key = api_key or get_secret("SNOWFLAKE_JWT") if api_base is not None and not isinstance(api_base, str): raise Exception("api base needs to be a string. api_base={}".format(api_base))