From f3d533c1fd30045b28ca2da18f64d97d41e8a90a Mon Sep 17 00:00:00 2001
From: Claude <noreply@anthropic.com>
Date: Wed, 11 Feb 2026 23:50:32 +0000
Subject: [PATCH] Fix safety_tolerance to render as INT spinner instead of
 broken dropdown

The string-list combo was rendering as an INT widget defaulting to 300
(the timeout value). Changed to a proper INT type with min=1, max=6,
default=4 so it renders correctly in the ComfyUI UI.

https://claude.ai/code/session_01VtRzQMdLschx9awGyTGGH7
---
 nodes.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 91899fa..a15a212 100644
--- a/nodes.py
+++ b/nodes.py
@@ -27,7 +27,6 @@ OUTPUT_FORMATS = ["png", "jpeg", "webp"]
 
 RESOLUTIONS = ["1K", "2K", "4K"]
 
-SAFETY_TOLERANCE_OPTIONS = ["1", "2", "3", "4", "5", "6"]
 
 
 def _resolve_api_key(fal_api_key=""):
@@ -213,7 +212,7 @@ class NanoBananaProImageEdit:
                 "seed": ("INT", {"default": -1, "min": -1, "max": 2**31 - 1}),
                 "limit_generations": ("BOOLEAN", {"default": False}),
                 "enable_web_search": ("BOOLEAN", {"default": False}),
-                "safety_tolerance": (SAFETY_TOLERANCE_OPTIONS, {"default": "4"}),
+                "safety_tolerance": ("INT", {"default": 4, "min": 1, "max": 6}),
                 "timeout": ("INT", {"default": 300, "min": 30, "max": 600}),
             },
         }
@@ -238,7 +237,7 @@ class NanoBananaProImageEdit:
         seed=-1,
         limit_generations=False,
         enable_web_search=False,
-        safety_tolerance="4",
+        safety_tolerance=4,
         timeout=300,
     ):
         api_key = _resolve_api_key(fal_api_key)
@@ -255,7 +254,7 @@ class NanoBananaProImageEdit:
             "output_format": output_format,
             "resolution": resolution,
             "enable_web_search": enable_web_search,
-            "safety_tolerance": int(safety_tolerance),
+            "safety_tolerance": safety_tolerance,
         }
         if seed >= 0:
             payload["seed"] = seed