This commit is contained in:
Gabe
2026-04-05 10:45:46 -05:00
parent d481e92c49
commit ab115ee593
+8 -11
View File
@@ -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 = {