Release Candidate v4.11.0 (#278)

Author: milosmns

.version

@@ -1 +1 @@
-4.10.0
+4.11.0

src/features/chat/chat_imaging_service.py

@@ -150,7 +150,7 @@ def __restore_image(self) -> tuple[Result, URLList, ErrorList]:
         return result, urls, errors
 
     def __edit_image(self) -> tuple[Result, URLList, ErrorList]:
-        log.t(f"Editing {len(self.__attachments)} images")
+        log.t(f"Editing {len(self.__attachments)} images in aspect ratio {self.__aspect_ratio}")
         result = ChatImagingService.Result.success
         urls: URLList = []
         errors: ErrorList = []

src/features/chat/llm_tools/llm_tool_library.py

@@ -42,7 +42,8 @@ def process_attachments(
         attachment_ids: [mandatory] A comma-separated list of verbatim, unique 📎 attachment IDs that need to be processed (located in each message); include any dashes, underscores or other symbols; these IDs are not to be cleaned or truncated
         operation: [mandatory] The action to perform on the attachments
         context: [optional] Additional task context or guidance, e.g. the user's message/question/caption, if available
-        aspect_ratio: [optional] The desired aspect ratio for image editing operations. Valid options: 1:1, 2:3, 3:2, 3:4, 4:3, 16:9, 9:16. Defaults to the aspect ratio of the input image
+        aspect_ratio: [optional] The desired image's aspect ratio for image editing. Valid options: 1:1, 2:3, 3:2, 3:4, 4:3, 16:9, 9:16. If not explicitly requested, don't send; default is the aspect ratio of the input image
+        aspect_ratio: [optional] The desired image's aspect ratio. Valid options: 1:1, 2:3, 3:2, 3:4, 4:3, 16:9, 9:16. If not explicitly requested, don't send; default is 2:3
     """
     try:
         operation = operation.lower().strip()
@@ -57,6 +58,7 @@ def process_attachments(
                 raise ValueError("Failed to resolve attachments")
             return json.dumps({"result": result.value, "attachments": describer.result})
         elif operation in editing_operations:
+            log.d(f"LLM requested to process {len(attachment_ids_list)} images in aspect ratio {aspect_ratio}")
             # Generate images based on the provided context
             result, details = di.chat_imaging_service(attachment_ids_list, operation, context, aspect_ratio).execute()
             if result == ChatImagingService.Result.failed:
@@ -85,9 +87,10 @@ def generate_image(
 
     Args:
         prompt: [mandatory] The user's description or prompt for the generated image
-        aspect_ratio: [optional] The desired aspect ratio for generation. Valid options: 1:1, 2:3 (default if not specified), 3:2, 3:4, 4:3, 16:9, 9:16.
+        aspect_ratio: [optional] The desired image's aspect ratio. Valid options: 1:1, 2:3, 3:2, 3:4, 4:3, 16:9, 9:16. If not explicitly requested, don't send; default is 2:3
     """
     try:
+        log.d(f"LLM requested to generate an image in aspect ratio {aspect_ratio}")
         copywriter_tool = di.tool_choice_resolver.require_tool(
             SmartStableDiffusionGenerator.COPYWRITER_TOOL_TYPE,
             SmartStableDiffusionGenerator.DEFAULT_COPYWRITER_TOOL,

src/features/external_tools/external_tool_library.py

@@ -83,6 +83,13 @@
     types = [ToolType.chat, ToolType.reasoning, ToolType.copywriting, ToolType.vision],
 )
 
+GPT_5_2 = ExternalTool(
+    id = "gpt-5.2",
+    name = "GPT 5.2",
+    provider = OPEN_AI,
+    types = [ToolType.chat, ToolType.reasoning, ToolType.copywriting, ToolType.vision],
+)
+
 GPT_4O = ExternalTool(
     id = "gpt-4o",
     name = "GPT 4o",
@@ -213,6 +220,13 @@
     types = [ToolType.images_gen],
 )
 
+GEMINI_3_FLASH = ExternalTool(
+    id = "gemini-3-flash-preview",
+    name = "Gemini 3 Flash (Preview)",
+    provider = GOOGLE_AI,
+    types = [ToolType.chat, ToolType.reasoning, ToolType.copywriting, ToolType.vision],
+)
+
 GEMINI_3_PRO = ExternalTool(
     id = "gemini-3-pro-preview",
     name = "Gemini 3 Pro (Preview)",
@@ -305,13 +319,34 @@
     types = [ToolType.images_inpainting],
 )
 
-IMAGE_GENERATION_FLUX = ExternalTool(
+IMAGE_GENERATION_FLUX_1_1 = ExternalTool(
     id = "black-forest-labs/flux-1.1-pro",
-    name = "Black Forest's Flux Pro 1.1",
+    name = "Black Forest's Flux 1.1 Pro",
     provider = REPLICATE,
     types = [ToolType.images_gen],
 )
 
+IMAGE_GENERATION_EDITING_FLUX_2_PRO = ExternalTool(
+    id = "black-forest-labs/flux-2-pro",
+    name = "Black Forest's Flux 2 Pro",
+    provider = REPLICATE,
+    types = [ToolType.images_gen, ToolType.images_edit],
+)
+
+IMAGE_GENERATION_EDITING_FLUX_2_MAX = ExternalTool(
+    id = "black-forest-labs/flux-2-max",
+    name = "Black Forest's Flux 2 Max",
+    provider = REPLICATE,
+    types = [ToolType.images_gen, ToolType.images_edit],
+)
+
+IMAGE_GENERATION_EDITING_GPT_IMAGE_1_5 = ExternalTool(
+    id = "openai/gpt-image-1.5",
+    name = "OpenAI's GPT Image 1.5",
+    provider = REPLICATE,
+    types = [ToolType.images_gen, ToolType.images_edit],
+)
+
 IMAGE_GENERATION_GEMINI_2_5_FLASH_IMAGE = ExternalTool(
     id = "google/gemini-2.5-flash-image",
     name = "Google's Gemini 2.5 Flash Image",
@@ -368,6 +403,7 @@
     GPT_5_MINI,
     GPT_5_NANO,
     GPT_5_1,
+    GPT_5_2,
     GPT_4O,
     GPT_4O_MINI,
     GPT_O3_MINI,
@@ -388,6 +424,7 @@
     GEMINI_2_5_FLASH,
     GEMINI_2_5_PRO,
     GEMINI_2_5_FLASH_IMAGE,
+    GEMINI_3_FLASH,
     GEMINI_3_PRO,
     # Perplexity
     SONAR,
@@ -404,7 +441,10 @@
     BACKGROUND_REMOVAL,
     IMAGE_RESTORATION,
     IMAGE_INPAINTING,
-    IMAGE_GENERATION_FLUX,
+    IMAGE_GENERATION_FLUX_1_1,
+    IMAGE_GENERATION_EDITING_FLUX_2_PRO,
+    IMAGE_GENERATION_EDITING_FLUX_2_MAX,
+    IMAGE_GENERATION_EDITING_GPT_IMAGE_1_5,
     IMAGE_GENERATION_GEMINI_2_5_FLASH_IMAGE,
     IMAGE_EDITING_FLUX_KONTEXT_PRO,
     IMAGE_EDITING_SEED_EDIT,

src/features/images/aspect_ratio_utils.py

@@ -1,5 +1,7 @@
 import re
 
+from util import log
+
 VALID_ASPECT_RATIOS = [
     "9:16",  # Ultra-tall portrait
     "2:3",   # Tall portrait
@@ -26,6 +28,7 @@ def validate_aspect_ratio(aspect_ratio: str | None, default: str) -> str:
     Returns:
         A valid aspect ratio string or the default value
     """
+    log.d(f"Validating aspect ratio… requested '{aspect_ratio}', default: '{default}'")
     if not aspect_ratio:
         return default
 

src/features/images/image_editor.py

@@ -21,6 +21,7 @@
 class ImageEditor:
 
     DEFAULT_ASPECT_RATIO: str = "match_input_image"
+    DEFAULT_IMAGE_SIZE: str = "4K"
     DEFAULT_TOOL: ExternalTool = IMAGE_EDITING_FLUX_KONTEXT_PRO
     TOOL_TYPE: ToolType = ToolType.images_edit
 
@@ -63,16 +64,28 @@ def execute(self) -> str | None:
                 with open(temp_file.name, "rb") as file:
                     input_data = {
                         "prompt": self.__context or "",
+                        "prompt_upsampling": False,
                         "image": file,
                         "input_image": file,
                         "image_input": [file],
+                        "input_images": [file],
                         "aspect_ratio": self.__aspect_ratio,
                         "output_format": "png",
-                        "safety_tolerance": 0,
+                        "output_mime_type": "image/png",
+                        "output_quality": 100,
+                        "num_inference_steps": 30,
+                        "safety_tolerance": 1,
                         "guidance_scale": 5.5,
-                        "size": "4K",
+                        "num_outputs": 1,
                         "max_images": 1,
                         "sequential_image_generation": "disabled",
+                        "size": ImageEditor.DEFAULT_IMAGE_SIZE,
+                        "quality": "high",
+                        "background": "auto",
+                        "moderation": "low",
+                        "input_fidelity": "high",
+                        "number_of_images": 1,
+                        "output_compression": 90,
                     }
                     tool, _, _ = self.__configured_tool
                     result = self.__replicate.run(tool.id, input = input_data)

src/features/images/simple_stable_diffusion_generator.py

@@ -6,7 +6,7 @@
 
 from di.di import DI
 from features.external_tools.external_tool import ExternalTool, ToolType
-from features.external_tools.external_tool_library import IMAGE_GENERATION_FLUX
+from features.external_tools.external_tool_library import IMAGE_GENERATION_FLUX_1_1
 from features.external_tools.external_tool_provider_library import GOOGLE_AI, REPLICATE
 from features.external_tools.tool_choice_resolver import ConfiguredTool
 from features.images.aspect_ratio_utils import validate_aspect_ratio
@@ -20,8 +20,9 @@ class SimpleStableDiffusionGenerator:
 
     DEFAULT_ASPECT_RATIO: str = "2:3"
     DEFAULT_IMAGE_SIZE: str = "4K"
+    DEFAULT_IMAGE_RESOLUTION: str = "4 MP"
     DEFAULT_OUTPUT_MIME_TYPE: str = "image/png"
-    DEFAULT_TOOL: ExternalTool = IMAGE_GENERATION_FLUX
+    DEFAULT_TOOL: ExternalTool = IMAGE_GENERATION_FLUX_1_1
     TOOL_TYPE: ToolType = ToolType.images_gen
 
     error: str | None
@@ -86,18 +87,25 @@ def __generate_with_replicate(self) -> str | None:
             tool.id,
             input = {
                 "prompt": self.__prompt,
-                "prompt_upsampling": True,
+                "prompt_upsampling": False,
                 "aspect_ratio": self.__aspect_ratio,
                 "output_format": "png",
                 "output_mime_type": SimpleStableDiffusionGenerator.DEFAULT_OUTPUT_MIME_TYPE,
                 "output_quality": 100,
                 "num_inference_steps": 30,
-                "safety_tolerance": 0,
+                "safety_tolerance": 1,
                 "guidance_scale": 5.5,
                 "num_outputs": 1,
-                "size": SimpleStableDiffusionGenerator.DEFAULT_IMAGE_SIZE,
                 "max_images": 1,
                 "sequential_image_generation": "disabled",
+                "size": SimpleStableDiffusionGenerator.DEFAULT_IMAGE_SIZE,
+                "resolution": SimpleStableDiffusionGenerator.DEFAULT_IMAGE_RESOLUTION,
+                "quality": "high",
+                "background": "auto",
+                "moderation": "low",
+                "input_fidelity": "high",
+                "number_of_images": 1,
+                "output_compression": 90,
             },
         )
         log.d("Result", result)

src/features/images/smart_stable_diffusion_generator.py

@@ -49,7 +49,7 @@ def __init__(
 
     def execute(self) -> Result:
         invoker_chat = self.__di.require_invoker_chat()
-        log.d(f"Generating image for chat '{invoker_chat.chat_id}'")
+        log.d(f"Generating image for chat '{invoker_chat.chat_id}' in aspect ratio {self.__aspect_ratio}")
         self.error = None
 
         # let's correct/prettify and translate the prompt first

src/util/log.py

@@ -49,8 +49,13 @@ def _format_args(*args: Any) -> tuple[str, list[Exception]]:
 
 
 def _log_message(level: str, message: str, exceptions: list[Exception]) -> str:
+    # build enriched return value with exception messages (always, regardless of logging)
+    return_parts = [message]
+    for exception in exceptions:
+        return_parts.append(f"{type(exception).__name__}: {str(exception)}")
+
     if not _should_log(level) and not exceptions:
-        return message
+        return "\n ├─ ".join(return_parts)
 
     # for uvicorn, use the uvicorn logger
     if config.log_level != "local":
@@ -66,7 +71,7 @@ def _log_message(level: str, message: str, exceptions: list[Exception]) -> str:
                         logger.warning(message)
                     case "ERROR":
                         logger.error(message)
-            # log the exceptions
+            # log the exceptions (message + traceback)
             for exception in exceptions:
                 logger.error(f"Message: {str(exception)}")
                 if trace := exception.__traceback__:
@@ -83,7 +88,7 @@ def _log_message(level: str, message: str, exceptions: list[Exception]) -> str:
                     trace_lines = traceback.format_tb(trace)
                     indented_trace = "".join(trace_lines).strip()
                     print(indented_trace, file = sys.stderr)
-        return message
+        return "\n ├─ ".join(return_parts)
 
     # for local execution, print to stdout/stderr
     print(f"[{level[0]}] {message}")
@@ -93,7 +98,7 @@ def _log_message(level: str, message: str, exceptions: list[Exception]) -> str:
             trace_lines = traceback.format_tb(trace)
             indented_trace = "".join(("    " + line.strip()) for line in trace_lines)
             print(indented_trace, file = sys.stderr)
-    return message
+    return "\n ├─ ".join(return_parts)
 
 
 def t(*args: Any) -> str:

test/features/chat/test_chat_imaging_service.py

@@ -173,7 +173,14 @@ def test_execute_remove_background_exception(self):
 
         self.assertEqual(result, ChatImagingService.Result.failed)
         expected_details = [
-            {"url": None, "error": "Failed to remove background from attachment 'attachment1'\n ├─ ! Exception (see below)"},
+            {
+                "url": None,
+                "error": (
+                    "Failed to remove background from attachment 'attachment1'\n"
+                    " ├─ ! Exception (see below)\n"
+                    " ├─ Exception: Test exception"
+                ),
+            },
         ]
         self.assertEqual(details, expected_details)
         mock_remover_instance.execute.assert_called_once()