Files
griptape-ai-ComfyUI-Griptape/nodes/base_tool.py
T
2024-05-22 15:34:57 -07:00

24 lines
468 B
Python

from griptape.tools import BaseTool
class gtUIBaseTool:
"""
Griptape Base Tool
"""
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {"required": {"off_prompt": ("BOOLEAN", {"default": False})}}
RETURN_TYPES = ("TOOL_LIST",)
RETURN_NAMES = ("TOOL",)
FUNCTION = "create"
CATEGORY = "Griptape/Agent Tools"
def create(self, off_prompt):
return ([BaseTool(off_prompt=off_prompt)],)