updated names of classes
This commit is contained in:
Jason Schleifer
+106
-89
@@ -18,13 +18,17 @@ from .nodes.combine_nodes import MergeTexts, RulesList, ToolList
|
||||
from .nodes.config_nodes import (
|
||||
gtUIAmazonBedrockStructureConfig,
|
||||
gtUIAnthropicStructureConfig,
|
||||
gtUIEnv,
|
||||
# gtUIEnv,
|
||||
gtUIGoogleStructureConfig,
|
||||
gtUILMStudioStructureConfig,
|
||||
gtUIOllamaStructureConfig,
|
||||
gtUIOpenAiStructureConfig,
|
||||
)
|
||||
from .nodes.display_nodes import gtUIOutputImageNode, gtUIOutputStringNode
|
||||
from .nodes.display_nodes import (
|
||||
# gtUIOutputArtifactNode,
|
||||
gtUIOutputImageNode,
|
||||
gtUIOutputStringNode,
|
||||
)
|
||||
from .nodes.image_drivers import (
|
||||
gtUIAmazonBedrockStableDiffusionImageGenerationDriver,
|
||||
gtUIAmazonBedrockTitanImageGenerationDriver,
|
||||
@@ -88,94 +92,107 @@ NODE_DISPLAY_NAME_MAPPINGS = {}
|
||||
WEB_DIRECTORY = "./js"
|
||||
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
"gtUIInputNode": gtUIInputStringNode,
|
||||
"gtUIFetchImage": gtUIFetchImage,
|
||||
"gtUITextToClipEncode": gtUITextToClipEncode,
|
||||
"gtUICLIPTextEncode": gtUICLIPTextEncode,
|
||||
"gtUIOpenAiImageGenerationDriver": gtUIOpenAiImageGenerationDriver,
|
||||
"gtUIAmazonBedrockStableDiffusionImageGenerationDriver": gtUIAmazonBedrockStableDiffusionImageGenerationDriver,
|
||||
"gtUIAmazonBedrockTitanImageGenerationDriver": gtUIAmazonBedrockTitanImageGenerationDriver,
|
||||
"gtUILeonardoImageGenerationDriver": gtUILeonardoImageGenerationDriver,
|
||||
"gtUIOutputStringNode": gtUIOutputStringNode,
|
||||
"gtUIOutputImageNode": gtUIOutputImageNode,
|
||||
"RunAgent": RunAgent,
|
||||
"CreateAgent": CreateAgent,
|
||||
"ExpandAgent": ExpandAgent,
|
||||
"gtUIOpenAiStructureConfig": gtUIOpenAiStructureConfig,
|
||||
"gtUIAmazonBedrockStructureConfig": gtUIAmazonBedrockStructureConfig,
|
||||
"gtUIGoogleStructureConfig": gtUIGoogleStructureConfig,
|
||||
"gtUIAnthropicStructureConfig": gtUIAnthropicStructureConfig,
|
||||
"gtUIOllamaStructureConfig": gtUIOllamaStructureConfig,
|
||||
"gtUILMStudioStructureConfig": gtUILMStudioStructureConfig,
|
||||
"Calculator": gtUICalculator,
|
||||
"DateTime": gtUIDateTime,
|
||||
"WebScraper": gtUIWebScraper,
|
||||
"gtUIFileManager": gtUIFileManager,
|
||||
"gtAudioTranscriptionClient": gtUIAudioTranscriptionClient,
|
||||
"ToolList": ToolList,
|
||||
"MergeTexts": MergeTexts,
|
||||
"PromptTask": gtUIPromptTask,
|
||||
"ToolTask": gtUIToolTask,
|
||||
"ToolkitTask": gtUIToolkitTask,
|
||||
"AudioTranscriptionTask": gtUIAudioTranscriptionTask,
|
||||
"gtUIOpenAiAudioTranscriptionDriver": gtUIOpenAiAudioTranscriptionDriver,
|
||||
"gtUIKnowledgeBaseTool": gtUIKnowledgeBaseTool,
|
||||
"gtUIWebSearch": gtUIWebSearch,
|
||||
"ImageQueryTask": gtUIImageQueryTask,
|
||||
"ParallelImageQueryTask": gtUIParallelImageQueryTask,
|
||||
"PromptImageGenerationTask": gtUIPromptImageGenerationTask,
|
||||
"PromptImageVariationTask": gtUIPromptImageVariationTask,
|
||||
"TextSummaryTask": gtUITextSummaryTask,
|
||||
"Rule": gtUIRule,
|
||||
"RulesList": RulesList,
|
||||
"EnvironmentConfig": gtUIEnv,
|
||||
"gtUILoadAudio": gtUILoadAudio,
|
||||
# AGENT
|
||||
"Griptape Create: Agent": CreateAgent,
|
||||
"Griptape Run: Agent": RunAgent,
|
||||
"Griptape Expand: Agent Nodes": ExpandAgent,
|
||||
# AGENT CONFIG
|
||||
"Griptape Agent Config: OpenAI": gtUIOpenAiStructureConfig,
|
||||
"Griptape Agent Config: Amazon Bedrock": gtUIAmazonBedrockStructureConfig,
|
||||
"Griptape Agent Config: Google": gtUIGoogleStructureConfig,
|
||||
"Griptape Agent Config: Anthropic": gtUIAnthropicStructureConfig,
|
||||
"Griptape Agent Config: Ollama": gtUIOllamaStructureConfig,
|
||||
"Griptape Agent Config: LM Studio": gtUILMStudioStructureConfig,
|
||||
# AGENT RULES
|
||||
"Griptape Create: Rules": gtUIRule,
|
||||
"Griptape Combine: Rules List": RulesList,
|
||||
# TASKS
|
||||
"Griptape Run: Prompt Task": gtUIPromptTask,
|
||||
"Griptape Run: Text Summary": gtUITextSummaryTask,
|
||||
"Griptape Run: Tool Task": gtUIToolTask,
|
||||
"Griptape Run: Toolkit Task": gtUIToolkitTask,
|
||||
# AGENT TOOLS
|
||||
"Griptape Tool: Audio Transcription": gtUIAudioTranscriptionClient,
|
||||
"Griptape Tool: Calculator": gtUICalculator,
|
||||
"Griptape Tool: DateTime": gtUIDateTime,
|
||||
"Griptape Tool: FileManager": gtUIFileManager,
|
||||
"Griptape Tool: Griptape Cloud KnowledgeBase": gtUIKnowledgeBaseTool,
|
||||
"Griptape Tool: WebScraper": gtUIWebScraper,
|
||||
"Griptape Tool: WebSearch": gtUIWebSearch,
|
||||
"Griptape Combine: Tool List": ToolList,
|
||||
# TEXT
|
||||
"Griptape Create: Text": gtUIInputStringNode,
|
||||
"Griptape Create: CLIP Text Encode": gtUICLIPTextEncode,
|
||||
"Griptape Convert: Text to CLIP Encode": gtUITextToClipEncode,
|
||||
"Griptape Combine: Merge Texts": MergeTexts,
|
||||
# IMAGES
|
||||
"Griptape Create: Image from Text": gtUIPromptImageGenerationTask,
|
||||
"Griptape Create: Image Variation": gtUIPromptImageVariationTask,
|
||||
"Griptape Load: Image From URL": gtUIFetchImage,
|
||||
"Griptape Run: Image Description": gtUIImageQueryTask,
|
||||
"Griptape Run: Parallel Image Description": gtUIParallelImageQueryTask,
|
||||
# IMAGE DRIVERS
|
||||
"Griptape Driver: Amazon Bedrock Stable Diffusion": gtUIAmazonBedrockStableDiffusionImageGenerationDriver,
|
||||
"Griptape Driver: Amazon Bedrock Titan": gtUIAmazonBedrockTitanImageGenerationDriver,
|
||||
"Griptape Driver: Leonardo.AI": gtUILeonardoImageGenerationDriver,
|
||||
"Griptape Driver: OpenAI Image Generation": gtUIOpenAiImageGenerationDriver,
|
||||
# DISPLAY
|
||||
"Griptape Display: Image": gtUIOutputImageNode,
|
||||
"Griptape Display: Text": gtUIOutputStringNode,
|
||||
# AUDIO
|
||||
"Griptape Load: Audio": gtUILoadAudio,
|
||||
"Griptape Run: Audio Transcription": gtUIAudioTranscriptionTask,
|
||||
# AUDIO DRIVER
|
||||
"Griptape Audio Driver: OpenAI": gtUIOpenAiAudioTranscriptionDriver,
|
||||
# "Griptape Display: Artifact": gtUIOutputArtifactNode,
|
||||
# "Griptape Config: Environment Variables": gtUIEnv,
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"gtUIInputNode": "Griptape Create: Text",
|
||||
"gtUICLIPTextEncode": "Griptape Create: CLIP Text Encode",
|
||||
"gtUITextToClipEncode": "Griptape Convert: Text to CLIP Encode",
|
||||
"gtUIFetchImage": "Griptape Load: Image From URL",
|
||||
"CreateAgent": "Griptape Create: Agent",
|
||||
"RunAgent": "Griptape Run: Agent",
|
||||
"PromptImageGenerationTask": "Griptape Create: Image from Text",
|
||||
"PromptImageVariationTask": "Griptape Create: Image Variation",
|
||||
"Rule": "Griptape Create: Rules",
|
||||
"RulesList": "Griptape Create: Rules List",
|
||||
"gtUIOpenAiAudioTranscriptionDriver": "Griptape Driver: OpenAI",
|
||||
"gtUIAmazonBedrockStableDiffusionImageGenerationDriver": "Griptape Driver: Amazon Bedrock Stable Diffusion",
|
||||
"gtUIAmazonBedrockTitanImageGenerationDriver": "Griptape Driver: Amazon Bedrock Titan",
|
||||
"gtUILeonardoImageGenerationDriver": "Griptape Driver: Leonardo.AI",
|
||||
"gtUIOutputStringNode": "Griptape Display: Text",
|
||||
"gtUIOutputImageNode": "Griptape Display: Image",
|
||||
"ImageQueryTask": "Griptape Run: Image Description",
|
||||
"ParallelImageQueryTask": "Griptape Run: Parallel Image Description",
|
||||
"PromptTask": "Griptape Run: Prompt Task",
|
||||
"ToolTask": "Griptape Run: Tool Task",
|
||||
"ToolkitTask": "Griptape Run: Toolkit Task",
|
||||
"TextSummaryTask": "Griptape Run: Text Summary",
|
||||
"AudioTranscriptionTask": "Griptape Run: Audio Transcription",
|
||||
"ExpandAgent": "Griptape Expand: Agent Nodes",
|
||||
"gtUIOpenAiStructureConfig": "Griptape Agent Config: OpenAI",
|
||||
"gtUIAmazonBedrockStructureConfig": "Griptape Agent Config: Amazon Bedrock",
|
||||
"gtUIGoogleStructureConfig": "Griptape Agent Config: Google",
|
||||
"gtUIAnthropicStructureConfig": "Griptape Agent Config: Anthropic",
|
||||
"gtUIOllamaStructureConfig": "Griptape Agent Config: Ollama",
|
||||
"gtUILMStudioStructureConfig": "Griptape Agent Config: LM Studio",
|
||||
"Calculator": "Griptape Tool: Calculator",
|
||||
"DateTime": "Griptape Tool: DateTime",
|
||||
"WebScraper": "Griptape Tool: WebScraper",
|
||||
"gtUIFileManager": "Griptape Tool: FileManager",
|
||||
"gtAudioTranscriptionClient": "Griptape Tool: Audio Transcription",
|
||||
"gtUIKnowledgeBaseTool": "Griptape Tool: Griptape Cloud KnowledgeBase",
|
||||
"gtUIWebSearch": "Griptape Tool: WebSearch",
|
||||
"ToolList": "Griptape Combine: Tool List",
|
||||
"MergeTexts": "Griptape Combine: Merge Texts",
|
||||
"EnvironmentConfig": "Griptape Config: Environment Variables",
|
||||
"gtUIOpenAiImageGenerationDriver": "Griptape Audio Driver: OpenAI",
|
||||
"gtUILoadAudio": "Griptape Load: Audio",
|
||||
}
|
||||
# NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
# "gtUIInputNode": "Griptape Create: Text",
|
||||
# "gtUICLIPTextEncode": "Griptape Create: CLIP Text Encode",
|
||||
# "gtUITextToClipEncode": "Griptape Convert: Text to CLIP Encode",
|
||||
# "gtUIFetchImage": "Griptape Load: Image From URL",
|
||||
# "CreateAgent": "Griptape Create: Agent",
|
||||
# "RunAgent": "Griptape Run: Agent",
|
||||
# "PromptImageGenerationTask": "Griptape Create: Image from Text",
|
||||
# "PromptImageVariationTask": "Griptape Create: Image Variation",
|
||||
# "Rule": "Griptape Create: Rules",
|
||||
# "RulesList": "Griptape Create: Rules List",
|
||||
# "gtUIOpenAiAudioTranscriptionDriver": "Griptape Driver: OpenAI",
|
||||
# "gtUIAmazonBedrockStableDiffusionImageGenerationDriver": "Griptape Driver: Amazon Bedrock Stable Diffusion",
|
||||
# "gtUIAmazonBedrockTitanImageGenerationDriver": "Griptape Driver: Amazon Bedrock Titan",
|
||||
# "gtUILeonardoImageGenerationDriver": "Griptape Driver: Leonardo.AI",
|
||||
# "gtUIOutputStringNode": "Griptape Display: Text",
|
||||
# "gtUIOutputImageNode": "Griptape Display: Image",
|
||||
# "gtUIOutputArtifactNode": "Griptape Display: Artifact",
|
||||
# "ImageQueryTask": "Griptape Run: Image Description",
|
||||
# "ParallelImageQueryTask": "Griptape Run: Parallel Image Description",
|
||||
# "PromptTask": "Griptape Run: Prompt Task",
|
||||
# "ToolTask": "Griptape Run: Tool Task",
|
||||
# "ToolkitTask": "Griptape Run: Toolkit Task",
|
||||
# "TextSummaryTask": "Griptape Run: Text Summary",
|
||||
# "AudioTranscriptionTask": "Griptape Run: Audio Transcription",
|
||||
# "ExpandAgent": "Griptape Expand: Agent Nodes",
|
||||
# "gtUIOpenAiStructureConfig": "Griptape Agent Config: OpenAI",
|
||||
# "gtUIAmazonBedrockStructureConfig": "Griptape Agent Config: Amazon Bedrock",
|
||||
# "gtUIGoogleStructureConfig": "Griptape Agent Config: Google",
|
||||
# "gtUIAnthropicStructureConfig": "Griptape Agent Config: Anthropic",
|
||||
# "gtUIOllamaStructureConfig": "Griptape Agent Config: Ollama",
|
||||
# "gtUILMStudioStructureConfig": "Griptape Agent Config: LM Studio",
|
||||
# "Calculator": "Griptape Tool: Calculator",
|
||||
# "DateTime": "Griptape Tool: DateTime",
|
||||
# "WebScraper": "Griptape Tool: WebScraper",
|
||||
# "gtUIFileManager": "Griptape Tool: FileManager",
|
||||
# "gtAudioTranscriptionClient": "Griptape Tool: Audio Transcription",
|
||||
# "gtUIKnowledgeBaseTool": "Griptape Tool: Griptape Cloud KnowledgeBase",
|
||||
# "gtUIWebSearch": "Griptape Tool: WebSearch",
|
||||
# "ToolList": "Griptape Combine: Tool List",
|
||||
# "MergeTexts": "Griptape Combine: Merge Texts",
|
||||
# "EnvironmentConfig": "Griptape Config: Environment Variables",
|
||||
# "gtUIOpenAiImageGenerationDriver": "Griptape Audio Driver: OpenAI",
|
||||
# "gtUILoadAudio": "Griptape Load: Audio",
|
||||
# }
|
||||
|
||||
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
||||
__all__ = ["NODE_CLASS_MAPPINGS", "WEB_DIRECTORY"]
|
||||
print(" \033[34m- \033[92mDone!\033[0m\n")
|
||||
|
||||
Reference in New Issue
Block a user