From 1cd97e41e9a5fe17e6aac08206c9b80b6b9d49cb Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 24 May 2025 12:26:31 -0700
Subject: [PATCH] test: update tests

---
 litellm/images/__init__.py                         | 0
 litellm/proxy/image_endpoints/__init__.py          | 3 +++
 tests/proxy_unit_tests/test_key_generate_prisma.py | 1 +
 3 files changed, 4 insertions(+)
 create mode 100644 litellm/images/__init__.py
 create mode 100644 litellm/proxy/image_endpoints/__init__.py

diff --git a/litellm/images/__init__.py b/litellm/images/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/litellm/proxy/image_endpoints/__init__.py b/litellm/proxy/image_endpoints/__init__.py
new file mode 100644
index 00000000000..2e24ea01e5a
--- /dev/null
+++ b/litellm/proxy/image_endpoints/__init__.py
@@ -0,0 +1,3 @@
+from .endpoints import image_generation
+
+__all__ = ["image_generation"]
diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py
index cff4747bc0e..e1890732ec0 100644
--- a/tests/proxy_unit_tests/test_key_generate_prisma.py
+++ b/tests/proxy_unit_tests/test_key_generate_prisma.py
@@ -77,6 +77,7 @@ from litellm.proxy.proxy_server import (
     moderations,
     user_api_key_auth,
 )
+from litellm.proxy.image_endpoints import image_generation
 from litellm.proxy.management_endpoints.customer_endpoints import (
     new_end_user,
 )