diff --git a/litellm/__init__.py b/litellm/__init__.py
index ac2b420d7..83e30d775 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -724,6 +724,8 @@ from .utils import (
     get_supported_openai_params,
     get_api_base,
     get_first_chars_messages,
+    ModelResponse,
+    ImageResponse,
 )
 from .llms.huggingface_restapi import HuggingfaceConfig
 from .llms.anthropic import AnthropicConfig