from .nodes import OpenAIApiLoaderNode, OpenAIImageEncoderNode, OpenAIFileUploaderNode, OpenAITextBlockNode, OpenAIChatNode
from .gemini_nodes import GeminiApiLoaderNode, GeminiImageEncoderNode, GeminiFileUploaderNode, GeminiTextBlockNode, GeminiChatNode
NODE_CLASS_MAPPINGS = {
    "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": "OpenAI API 加载器",
    "OpenAIImageEncoder": "OpenAI 图像编码器",
    "OpenAIFileUploader": "OpenAI 文件上传器",
    "OpenAITextBlock": "OpenAI 文本块",
    "OpenAIChat": "OpenAI 聊天节点",
    "GeminiApiLoader": "Gemini API 加载器",
    "GeminiImageEncoder": "Gemini 图像编码器 (Inline)",
    "GeminiFileUploader": "Gemini 文件上传器 (File API)",
    "GeminiTextBlock": "Gemini 文本块",
    "GeminiChat": "Gemini 聊天节点",
}