diff --git a/comfy_extras/nodes_primitive.py b/comfy_extras/nodes_primitive.py index 5a1aeba80077..487e08fc25ad 100644 --- a/comfy_extras/nodes_primitive.py +++ b/comfy_extras/nodes_primitive.py @@ -40,6 +40,24 @@ def execute(cls, value: str) -> io.NodeOutput: return io.NodeOutput(value) +class RandInt(io.ComfyNode): + @classmethod + def define_schema(cls): + return io.Schema( + node_id="PrimitiveRandomInt", + display_name="RandomInt", + category="utils/primitive", + inputs=[ + io.Int.Input("value", min=-sys.maxsize, max=sys.maxsize, control_after_generate=True), + ], + outputs=[io.Int.Output()], + ) + + @classmethod + def execute(cls, value: int) -> io.NodeOutput: + return io.NodeOutput(value) + + class Int(io.ComfyNode): @classmethod def define_schema(cls): @@ -48,7 +66,7 @@ def define_schema(cls): display_name="Int", category="utils/primitive", inputs=[ - io.Int.Input("value", min=-sys.maxsize, max=sys.maxsize, control_after_generate=True), + io.Int.Input("value", min=-sys.maxsize, max=sys.maxsize), ], outputs=[io.Int.Output()], ) @@ -100,6 +118,7 @@ async def get_node_list(self) -> list[type[io.ComfyNode]]: return [ String, StringMultiline, + RandInt, Int, Float, Boolean,