feat: rename BASE_URL to IMAGE_BASE_URL

Author: danielcopper

.env.example

@@ -26,8 +26,13 @@ GEMINI_API_KEY=
 # Path to store generated images
 STORAGE_PATH=./generated_images
 
-# Base URL for serving images (e.g., http://localhost:8000 or https://api.example.com)
-BASE_URL=http://localhost:8000
+# IMPORTANT: URL where THIS API serves images (not Open WebUI or other services!)
+# This should be the URL where this image-generation API is accessible.
+# Examples:
+#   - Local: http://localhost:8000
+#   - Docker: http://openapi-image-gen:8000 (use container name)
+#   - External: https://images.example.com
+IMAGE_BASE_URL=http://localhost:8000
 
 # -----------------------------------------------------------------------------
 # Security (Optional)
@@ -36,8 +41,15 @@ BASE_URL=http://localhost:8000
 API_BEARER_TOKEN=
 
 # -----------------------------------------------------------------------------
-# Model Registry
+# Model Configuration
 # -----------------------------------------------------------------------------
+# Default model for image generation (optional, uses first available if not set)
+# Examples: gemini/gemini-2.0-flash-exp-image-generation, dall-e-3, gpt-image-1
+DEFAULT_MODEL=
+
+# Filter LiteLLM models to only show image generation models (default: true)
+FILTER_IMAGE_MODELS=true
+
 # Cache TTL for model list in seconds (default: 3600 = 1 hour)
 MODEL_CACHE_TTL=3600
 

app/services/storage_service.py

@@ -34,9 +34,7 @@ async def save_image(self, image_data: bytes, extension: str = "png") -> str:
 
         relative_url = f"/images/{filename}"
 
-        if settings.BASE_URL:
-            return f"{settings.BASE_URL.rstrip('/')}{relative_url}"
-        return relative_url
+        return f"{settings.IMAGE_BASE_URL.rstrip('/')}{relative_url}"
 
 
 storage_service = StorageService()

tests/test_storage_service.py

@@ -13,7 +13,7 @@ async def test_save_image(temp_storage):
     # Create service with temp storage
     with patch("app.services.storage_service.settings") as mock_settings:
         mock_settings.STORAGE_PATH = str(temp_storage)
-        mock_settings.BASE_URL = "http://localhost:8000"
+        mock_settings.IMAGE_BASE_URL = "http://localhost:8000"
 
         service = StorageService()
 
@@ -42,7 +42,7 @@ async def test_save_image_different_extensions(temp_storage):
     """
     with patch("app.services.storage_service.settings") as mock_settings:
         mock_settings.STORAGE_PATH = str(temp_storage)
-        mock_settings.BASE_URL = "http://localhost:8000"
+        mock_settings.IMAGE_BASE_URL = "http://localhost:8000"
 
         service = StorageService()
 
@@ -53,3 +53,18 @@ async def test_save_image_different_extensions(temp_storage):
 
             filename = url.split("/")[-1]
             assert (temp_storage / filename).exists()
+
+
+@pytest.mark.asyncio
+async def test_save_image_with_custom_base_url(temp_storage):
+    """
+    Test that IMAGE_BASE_URL is used correctly.
+    """
+    with patch("app.services.storage_service.settings") as mock_settings:
+        mock_settings.STORAGE_PATH = str(temp_storage)
+        mock_settings.IMAGE_BASE_URL = "http://image-api:8000"
+
+        service = StorageService()
+        url = await service.save_image(b"test", "png")
+
+        assert url.startswith("http://image-api:8000/images/")