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 List, Optional
from typing import Optional
class SnowflakeBase:
def validate_environment(
@ -17,22 +17,18 @@ class SnowflakeBase:
"Accept": "application/json",
"Authorization": "Bearer " + <JWT>,
"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