diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py index 1d447da1f23..5a6282994c3 100644 --- a/litellm/integrations/custom_logger.py +++ b/litellm/integrations/custom_logger.py @@ -10,7 +10,7 @@ import traceback class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class # Class variables or attributes - def __init__(self): + def __init__(self) -> None: pass def log_pre_api_call(self, model, messages, kwargs): diff --git a/litellm/types/utils.py b/litellm/types/utils.py index f021fcd3452..a541dbb9023 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -528,7 +528,7 @@ class ModelResponse(OpenAIObject): response_ms=None, hidden_params=None, **params, - ): + ) -> None: if stream is not None and stream is True: object = "chat.completion.chunk" if choices is not None and isinstance(choices, list):