Merge pull request #4258 from nejch/fix/typing

fix: add more type hints to init methods

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

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