From 2ecd614a73cd1d1d998a3ff97ea796306858570e Mon Sep 17 00:00:00 2001
From: Nejc Habjan <nejc.habjan@siemens.com>
Date: Tue, 18 Jun 2024 12:09:39 +0200
Subject: [PATCH] fix: add more type hints to init methods

---
 litellm/integrations/custom_logger.py | 2 +-
 litellm/types/utils.py                | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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):