diff --git a/litellm/llms/snowflake/common_utils.py b/litellm/llms/snowflake/common_utils.py index 5ceeee7403..ab8d37bcf1 100644 --- a/litellm/llms/snowflake/common_utils.py +++ b/litellm/llms/snowflake/common_utils.py @@ -1,8 +1,6 @@ import httpx from typing import List, Optional - - class SnowflakeBase: def validate_environment( self, diff --git a/litellm/llms/snowflake/completion/handler.py b/litellm/llms/snowflake/completion/handler.py index a330ca83c9..039ae78351 100644 --- a/litellm/llms/snowflake/completion/handler.py +++ b/litellm/llms/snowflake/completion/handler.py @@ -19,7 +19,7 @@ class SnowflakeChatCompletion(OpenAILikeChatHandler,SnowflakeBase): model: str, messages: List[Dict[str, Any]], api_base: str, - acompletion: str, + acompletion: bool, custom_prompt_dict: Dict[str, Any], model_response: ModelResponse, print_verbose: Callable, diff --git a/snowflake_testing.py b/snowflake_testing.py index 0ed59cc43c..7e779b5803 100644 --- a/snowflake_testing.py +++ b/snowflake_testing.py @@ -1,8 +1,8 @@ import os from litellm import completion -os.environ["SNOWFLAKE_ACCOUNT_ID"] = "EBSRFJH-BI29448" -os.environ["SNOWFLAKE_JWT"] = "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJFQlNSRkpILkJJMjk0NDguU0hBMjU2OjZXdVlwazZPSTBUNHhMb0VGaVVWRWN0R3V2cWsrOC9oVmJibzcwcUIrOFk9Iiwic3ViIjoiRUJTUkZKSC5CSTI5NDQ4IiwiaWF0IjoxNzQwOTc5NzEwLCJleHAiOjE3NDEwNjYxMTB9.XpI50hT1O6SbnNCeAfz2TFke_V4y3fBoaNaS230lg2eTTzhfVKoda0azCQDeYf8BTLSJjAjtjPuXbEgnoB1J0keQW9H8hJUItvRhfYnqN3ci_Ln4IoiLvwYM2BneoQ7pZdYrC3nxz0PBRxuMpkNTSp4FFoFwtbPhvzgHH5TMBJA3Kyt7Usr1RpNxJIIcR43M9wjCpovj_9wJlG2ry1dpqrB_aZTssnynLFnE9533V8WgLbtiX-balobjpZcPNUMZB_fv-aHGUT6wq5SOP2G0opbVBGq_NpW5R1ZF-oYVIXiaKxzfN_PK9RhbjHVVxZU-As4llKKlAYmC8ArFMOVsrA" +os.environ["SNOWFLAKE_ACCOUNT_ID"] = "YOUR ACCOUNT" +os.environ["SNOWFLAKE_JWT"] = "YOUR JWT" messages = [{"role": "user", "content": "Write me a poem about the blue sky"}]