ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
Show Stats 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