diff --git a/__init__.py b/__init__.py
index 48e2079..f6c18a0 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,27 +1,27 @@
 from .nodes import OpenAIApiLoaderNode, OpenAIImageEncoderNode, OpenAIFileUploaderNode, OpenAITextBlockNode, OpenAIChatNode
 from .gemini_nodes import GeminiApiLoaderNode, GeminiImageEncoderNode, GeminiFileUploaderNode, GeminiTextBlockNode, GeminiChatNode
 NODE_CLASS_MAPPINGS = {
-    "OpenAIApiLoader_Zho": OpenAIApiLoaderNode,
-    "OpenAIImageEncoder_Zho": OpenAIImageEncoderNode,
-    "OpenAIFileUploader_Zho": OpenAIFileUploaderNode,
-    "OpenAITextBlock_Zho": OpenAITextBlockNode,
-    "OpenAIChat_Zho": OpenAIChatNode,
-    "GeminiApiLoader_Zho": GeminiApiLoaderNode,
-    "GeminiImageEncoder_Zho": GeminiImageEncoderNode,
-    "GeminiFileUploader_Zho": GeminiFileUploaderNode,
-    "GeminiTextBlock_Zho": GeminiTextBlockNode,
-    "GeminiChat_Zho": GeminiChatNode,
+    "OpenAIApiLoader": OpenAIApiLoaderNode,
+    "OpenAIImageEncoder": OpenAIImageEncoderNode,
+    "OpenAIFileUploader": OpenAIFileUploaderNode,
+    "OpenAITextBlock": OpenAITextBlockNode,
+    "OpenAIChat": OpenAIChatNode,
+    "GeminiApiLoader": GeminiApiLoaderNode,
+    "GeminiImageEncoder": GeminiImageEncoderNode,
+    "GeminiFileUploader": GeminiFileUploaderNode,
+    "GeminiTextBlock": GeminiTextBlockNode,
+    "GeminiChat": GeminiChatNode,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "OpenAIApiLoader_Zho": "OpenAI API 加载器",
-    "OpenAIImageEncoder_Zho": "OpenAI 图像编码器",
-     "OpenAIFileUploader_Zho": "OpenAI 文件上传器",
-     "OpenAITextBlock_Zho": "OpenAI 文本块",
-    "OpenAIChat_Zho": "OpenAI 聊天节点",
-    "GeminiApiLoader_Zho": "Gemini API 加载器",
-    "GeminiImageEncoder_Zho": "Gemini 图像编码器 (Inline)",
-     "GeminiFileUploader_Zho": "Gemini 文件上传器 (File API)",
-     "GeminiTextBlock_Zho": "Gemini 文本块",
-    "GeminiChat_Zho": "Gemini 聊天节点",
+    "OpenAIApiLoader": "OpenAI API 加载器",
+    "OpenAIImageEncoder": "OpenAI 图像编码器",
+    "OpenAIFileUploader": "OpenAI 文件上传器",
+    "OpenAITextBlock": "OpenAI 文本块",
+    "OpenAIChat": "OpenAI 聊天节点",
+    "GeminiApiLoader": "Gemini API 加载器",
+    "GeminiImageEncoder": "Gemini 图像编码器 (Inline)",
+    "GeminiFileUploader": "Gemini 文件上传器 (File API)",
+    "GeminiTextBlock": "Gemini 文本块",
+    "GeminiChat": "Gemini 聊天节点",
 }