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

fix code quality check

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-13 20:10:41 -07:00
parent 7dd55ce70c
commit 69b47cf738
1 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/llms/snowflake/common_utils.py
View file

@ -1,5 +1,5 @@
import httpx from typing import Optional
from typing import List, Optional
class SnowflakeBase: class SnowflakeBase:
def validate_environment( def validate_environment(
@ -17,22 +17,18 @@ class SnowflakeBase:
"Accept": "application/json", "Accept": "application/json",
"Authorization": "Bearer " + <JWT>, "Authorization": "Bearer " + <JWT>,
"X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT" "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT"
} }
""" """
if JWT is None: if JWT is None:
raise ValueError( raise ValueError("Missing Snowflake JWT key")
"Missing Snowflake JWT key"
)
headers.update( headers.update(
{ {
"Content-Type": "application/json", "Content-Type": "application/json",
"Accept": "application/json", "Accept": "application/json",
"Authorization": "Bearer " + JWT, "Authorization": "Bearer " + JWT,
"X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT" "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT",
} }
) )
return headers return headers