From 61ee71745a8fe02d69b36df7b5b49bc34effccd4 Mon Sep 17 00:00:00 2001 From: Sunny Wan Date: Mon, 3 Mar 2025 01:42:48 -0500 Subject: [PATCH] [CHORE] Added proper typing --- litellm/llms/snowflake/completion/handler.py | 22 ++++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/litellm/llms/snowflake/completion/handler.py b/litellm/llms/snowflake/completion/handler.py index 039ae783514..85ec676606e 100644 --- a/litellm/llms/snowflake/completion/handler.py +++ b/litellm/llms/snowflake/completion/handler.py @@ -1,11 +1,11 @@ from litellm.llms.base import BaseLLM from typing import Any, List, Optional -from typing import List, Dict, Callable, Optional, Any,cast +from typing import List, Dict, Callable, Optional, Any, cast, Union import litellm from litellm.utils import ModelResponse from litellm.types.llms.openai import AllMessageValues - +from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler from litellm.llms.openai_like.chat.handler import OpenAILikeChatHandler from ..common_utils import SnowflakeBase @@ -19,18 +19,18 @@ class SnowflakeChatCompletion(OpenAILikeChatHandler,SnowflakeBase): model: str, messages: List[Dict[str, Any]], api_base: str, - acompletion: bool, - custom_prompt_dict: Dict[str, Any], + custom_prompt_dict: dict, model_response: ModelResponse, print_verbose: Callable, - encoding: Any, + encoding, JWT: str, - logging_obj: Any, - optional_params: Optional[Dict[str, Any]] = None, - litellm_params: Optional[Dict[str, Any]] = None, - logger_fn: Optional[Callable] = None, - headers: Optional[Dict[str, str]] = None, - client: Optional[Any] = None, + logging_obj, + optional_params: dict, + acompletion=None, + litellm_params=None, + logger_fn=None, + headers: Optional[dict] = None, + client: Optional[Union[HTTPHandler, AsyncHTTPHandler]] = None, ) -> None: messages = litellm.SnowflakeConfig()._transform_messages(