From 5eac01bbcf94e3b9d03bc0887c6604420fcdf429 Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Wed, 26 Jun 2024 13:41:56 +1200 Subject: [PATCH] updated names of classes --- __init__.py | 195 ++++++++++++++++++++++++++++------------------------ 1 file changed, 106 insertions(+), 89 deletions(-) diff --git a/__init__.py b/__init__.py index 56424b1..07efd36 100644 --- a/__init__.py +++ b/__init__.py @@ -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")