From 69b47cf738c47b74027f4da31c771233107c19ff Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 13 Mar 2025 20:10:41 -0700 Subject: [PATCH] fix code quality check --- litellm/llms/snowflake/common_utils.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/litellm/llms/snowflake/common_utils.py b/litellm/llms/snowflake/common_utils.py index ab8d37bcf1..40c8270f95 100644 --- a/litellm/llms/snowflake/common_utils.py +++ b/litellm/llms/snowflake/common_utils.py @@ -1,5 +1,5 @@ -import httpx -from typing import List, Optional +from typing import Optional + class SnowflakeBase: def validate_environment( @@ -17,22 +17,18 @@ class SnowflakeBase: "Accept": "application/json", "Authorization": "Bearer " + , "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT" - } + } """ if JWT is None: - raise ValueError( - "Missing Snowflake JWT key" - ) + raise ValueError("Missing Snowflake JWT key") headers.update( { "Content-Type": "application/json", "Accept": "application/json", "Authorization": "Bearer " + JWT, - "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT" + "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT", } ) return headers - - \ No newline at end of file