92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
from typing import Dict, Any
|
|
|
|
|
|
class safetyInputs:
|
|
"""Runware Safety Inputs Node for configuring safety and content moderation settings"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"useMode": ("BOOLEAN", {
|
|
"tooltip": "Enable to include mode in payload.",
|
|
"default": False,
|
|
}),
|
|
"mode": (["none", "fast", "full"], {
|
|
"tooltip": "Safety moderation mode: none (disable), fast (quick check), or full (comprehensive check).",
|
|
"default": "none",
|
|
}),
|
|
"useTolerance": ("BOOLEAN", {
|
|
"tooltip": "Enable to include tolerance in payload.",
|
|
"default": False,
|
|
}),
|
|
"tolerance": ("BOOLEAN", {
|
|
"tooltip": "Safety tolerance setting for content moderation.",
|
|
"default": False,
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
"useCheckInputs": ("BOOLEAN", {
|
|
"tooltip": "Enable to include checkInputs in payload.",
|
|
"default": False,
|
|
}),
|
|
"checkInputs": ("BOOLEAN", {
|
|
"tooltip": "Enable checking of input content for safety violations.",
|
|
"default": False,
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
"useCheckContent": ("BOOLEAN", {
|
|
"tooltip": "Enable to include checkContent in payload.",
|
|
"default": False,
|
|
}),
|
|
"checkContent": ("BOOLEAN", {
|
|
"tooltip": "Enable checking of generated content for safety violations.",
|
|
"default": False,
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("RUNWARESAFETYINPUTS",)
|
|
RETURN_NAMES = ("Safety Inputs",)
|
|
FUNCTION = "createSafetyInputs"
|
|
CATEGORY = "Runware"
|
|
DESCRIPTION = "Configure safety and content moderation settings for Runware tasks."
|
|
|
|
def createSafetyInputs(self, **kwargs) -> tuple[Dict[str, Any]]:
|
|
"""Create safety inputs dictionary"""
|
|
useMode = kwargs.get("useMode", False)
|
|
mode = kwargs.get("mode", None)
|
|
useTolerance = kwargs.get("useTolerance", False)
|
|
tolerance = kwargs.get("tolerance", None)
|
|
useCheckInputs = kwargs.get("useCheckInputs", False)
|
|
checkInputs = kwargs.get("checkInputs", None)
|
|
useCheckContent = kwargs.get("useCheckContent", False)
|
|
checkContent = kwargs.get("checkContent", None)
|
|
|
|
safetyInputs = {}
|
|
|
|
if useMode and mode is not None:
|
|
safetyInputs["mode"] = mode
|
|
if useTolerance and tolerance is not None:
|
|
safetyInputs["tolerance"] = tolerance
|
|
if useCheckInputs and checkInputs is not None:
|
|
safetyInputs["checkInputs"] = checkInputs
|
|
if useCheckContent and checkContent is not None:
|
|
safetyInputs["checkContent"] = checkContent
|
|
|
|
return (safetyInputs,)
|
|
|
|
|
|
# Node class mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwareSafetyInputs": safetyInputs,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwareSafetyInputs": "Runware Safety Inputs",
|
|
}
|