diff --git a/nodes/agent/BaseAgent.py b/nodes/agent/BaseAgent.py index 740a628..ab4b3fc 100644 --- a/nodes/agent/BaseAgent.py +++ b/nodes/agent/BaseAgent.py @@ -33,26 +33,46 @@ class BaseAgent: "AGENT", { "forceInput": True, + "tooltip": "An existing agent to use.\nIf not provided, a new agent will be created.", }, ), "config": ( "CONFIG", { "forceInput": True, + "tooltip": "The configuration for the agent. If not provided, the default configuration will be used.", + }, + ), + "tools": ( + "TOOL_LIST", + { + "forceInput": True, + "INPUT_IS_LIST": True, + "tooltip": "One or more tools to use with the agent.", + }, + ), + "rulesets": ( + "RULESET", + { + "forceInput": True, + "tooltip": "One or more rules to use with the agent.\nUse these to control the agent behavior.", }, ), - "tools": ("TOOL_LIST", {"forceInput": True, "INPUT_IS_LIST": True}), - "rulesets": ("RULESET", {"forceInput": True}), "input_string": ( "STRING", { "forceInput": True, "dynamicPrompts": True, + "tooltip": "Additional text be appended to the STRING with a newline character.", }, ), "STRING": ( "STRING", - {"multiline": True, "dynamicPrompts": True}, + { + "multiline": True, + "dynamicPrompts": True, + "tooltip": "The prompt text.", + }, ), }, } @@ -65,6 +85,10 @@ class BaseAgent: "OUTPUT", "AGENT", ) + OUTPUT_TOOLTIPS = ( + "Text response from the Agent", + "The Agent. Can be connected to other nodes.", + ) FUNCTION = "run" OUTPUT_NODE = True