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

[CHORE] Fixed some style issues and leaks

Browse source
This commit is contained in:
Sunny Wan 2025-03-03 01:34:44 -05:00
parent fd090c8043
commit b87704cc34
3 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,6 @@
import httpx import httpx
from typing import List, Optional from typing import List, Optional
class SnowflakeBase: class SnowflakeBase:
def validate_environment( def validate_environment(
self, self,

2
litellm/llms/snowflake/completion/handler.py
View file

@ -19,7 +19,7 @@ class SnowflakeChatCompletion(OpenAILikeChatHandler,SnowflakeBase):
model: str, model: str,
messages: List[Dict[str, Any]], messages: List[Dict[str, Any]],
api_base: str, api_base: str,
acompletion: str, acompletion: bool,
custom_prompt_dict: Dict[str, Any], custom_prompt_dict: Dict[str, Any],
model_response: ModelResponse, model_response: ModelResponse,
print_verbose: Callable, print_verbose: Callable,

4
snowflake_testing.py
View file

@ -1,8 +1,8 @@
import os import os
from litellm import completion from litellm import completion
os.environ["SNOWFLAKE_ACCOUNT_ID"] = "EBSRFJH-BI29448" os.environ["SNOWFLAKE_ACCOUNT_ID"] = "YOUR ACCOUNT"
os.environ["SNOWFLAKE_JWT"] = "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJFQlNSRkpILkJJMjk0NDguU0hBMjU2OjZXdVlwazZPSTBUNHhMb0VGaVVWRWN0R3V2cWsrOC9oVmJibzcwcUIrOFk9Iiwic3ViIjoiRUJTUkZKSC5CSTI5NDQ4IiwiaWF0IjoxNzQwOTc5NzEwLCJleHAiOjE3NDEwNjYxMTB9.XpI50hT1O6SbnNCeAfz2TFke_V4y3fBoaNaS230lg2eTTzhfVKoda0azCQDeYf8BTLSJjAjtjPuXbEgnoB1J0keQW9H8hJUItvRhfYnqN3ci_Ln4IoiLvwYM2BneoQ7pZdYrC3nxz0PBRxuMpkNTSp4FFoFwtbPhvzgHH5TMBJA3Kyt7Usr1RpNxJIIcR43M9wjCpovj_9wJlG2ry1dpqrB_aZTssnynLFnE9533V8WgLbtiX-balobjpZcPNUMZB_fv-aHGUT6wq5SOP2G0opbVBGq_NpW5R1ZF-oYVIXiaKxzfN_PK9RhbjHVVxZU-As4llKKlAYmC8ArFMOVsrA" os.environ["SNOWFLAKE_JWT"] = "YOUR JWT"
messages = [{"role": "user", "content": "Write me a poem about the blue sky"}] messages = [{"role": "user", "content": "Write me a poem about the blue sky"}]