From ab115ee5930462d54ac1c04424a05b142d624067 Mon Sep 17 00:00:00 2001
From: Gabe <gabe.develop@gmail.com>
Date: Sun, 5 Apr 2026 10:45:46 -0500
Subject: [PATCH] fixes

---
 node.py | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/node.py b/node.py
index 1c92866..cc5233f 100644
--- a/node.py
+++ b/node.py
@@ -56,8 +56,7 @@ class OpenRouterNode:
                 "web_search": ("BOOLEAN", {"default": False}),
                 "cheapest": ("BOOLEAN", {"default": True}),
                 "fastest": ("BOOLEAN", {"default": False}),
-                "image_generation": ("BOOLEAN", {"default": False}),
-                "image_size": (["1K", "2K", "4K"], {"default": "1K"}),
+                "image_resolution": (["1K", "2K", "4K"], {"default": "1K"}),
                 "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff, "control_after_generate": "fixed"}),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -158,7 +157,7 @@ class OpenRouterNode:
 
     def generate_response(self, api_key, system_prompt, user_message_box, model,
                          web_search, cheapest, fastest, temperature, pdf_engine, chat_mode,
-                         image_generation=False, image_size="1K", seed=0,
+                         image_resolution="1K", seed=0,
                          pdf_data=None, user_message_input=None, **kwargs):
         """
         Sends a completion request to the OpenRouter chat completion endpoint.
@@ -313,12 +312,10 @@ class OpenRouterNode:
             "seed": seed
         }
 
-        # Only add modalities parameter if explicitly requested by user
-        # This prevents "Multi-modal output is not supported" errors on text-only models
-        if image_generation:
-            data["modalities"] = ["image", "text"]
-            data["image_config"] = {"image_size": image_size}
-            print(f"Image generation enabled by user setting")
+        # Always enable image+text modalities and send image_config
+        data["modalities"] = ["image", "text"]
+        data["image_config"] = {"image_size": image_resolution}
+        print(f"Payload: modalities={data['modalities']}, image_config={data['image_config']}, model={modified_model}")
 
         # Add plugins if a specific PDF engine is selected
         if pdf_engine != "auto":
@@ -578,7 +575,7 @@ class OpenRouterNode:
     @classmethod
     def IS_CHANGED(cls, api_key, system_prompt, user_message_box, model,
                    web_search, cheapest, fastest, temperature, pdf_engine, chat_mode,
-                   image_generation=False, image_size="1K", seed=0,
+                   image_resolution="1K", seed=0,
                    pdf_data=None, user_message_input=None, **kwargs):
         """
         Check if any input that affects the output has changed.
@@ -633,7 +630,7 @@ class OpenRouterNode:
         # Use primitive types where possible for reliable hashing/comparison
         return (api_key, system_prompt, user_message_box, model,
                 web_search, cheapest, fastest, temp_float, pdf_engine, chat_mode,
-                image_generation, image_size, seed, tuple(image_hashes), pdf_hash, user_message_input)
+                image_resolution, seed, tuple(image_hashes), pdf_hash, user_message_input)
 
 # Node class mappings
 NODE_CLASS_MAPPINGS = {