From 8abbdaf218176c0103f41d5bc5f5b96c931de28e Mon Sep 17 00:00:00 2001
From: Ninthkitten <165195091+Ninthkitten@users.noreply.github.com>
Date: Thu, 16 Jul 2026 23:47:33 +0200
Subject: [PATCH] feat(tools): add native function schema for generate_image

generate_image existed only in the text-block dialect, so API models could never call it and instead hallucinated broken text invocations. Add a native schema plus argument conversion. Fixes #5520.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
---
 src/tool_schemas.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/src/tool_schemas.py b/src/tool_schemas.py
index 6adb087c52..0f3ed8a9d0 100644
--- a/src/tool_schemas.py
+++ b/src/tool_schemas.py
@@ -25,6 +25,7 @@
     "read_file": ("path",),
     "write_file": ("path",),
     "edit_file": ("path",),
+    "generate_image": ("prompt",),
 }
 
 # ---------------------------------------------------------------------------
@@ -1003,6 +1004,25 @@
             }
         }
     },
+    {
+        # generate_image existed only as a fenced prompt-block tool — API models
+        # (native function calling) could never invoke it and hallucinated
+        # malformed text calls instead. Give it a schema like its sibling edit_image.
+        "type": "function",
+        "function": {
+            "name": "generate_image",
+            "description": "Generate a new image from a text prompt using the configured image model. Use for any 'create/draw/generate an image of X' request.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "prompt": {"type": "string", "description": "Detailed description of the image to generate"},
+                    "size": {"type": "string", "description": "WxH, e.g. 1024x1024 (optional)"},
+                    "quality": {"type": "string", "enum": ["low", "medium", "high"], "description": "Optional quality hint"},
+                },
+                "required": ["prompt"]
+            }
+        }
+    },
     {
         "type": "function",
         "function": {
@@ -1507,6 +1527,15 @@ def function_call_to_tool_block(name: str, arguments: str) -> Optional[ToolBlock
         # them once; pre-escaping here caused literal ``\u00f1`` sequences to
         # remain visible in the debug panel.
         content = json.dumps(args, ensure_ascii=False)
+    elif tool_type == "generate_image":
+        # The executor parses line-format
+        # (prompt \n model \n size \n quality) \u2014 see ai_interaction.py.
+        content = "\n".join([
+            str(args.get("prompt", "")).strip(),
+            str(args.get("model", "") or ""),
+            str(args.get("size", "") or ""),
+            str(args.get("quality", "") or ""),
+        ]).rstrip("\n")
     else:
         content = json.dumps(args)