From c4cf8ae6cac8d8564467081b264ea2f8df28eaee Mon Sep 17 00:00:00 2001 From: yunxiang wei <121092793+yiwangsimple@users.noreply.github.com> Date: Wed, 21 Aug 2024 10:53:49 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Egemini=5Fflash=E8=8A=82?= =?UTF-8?q?=E7=82=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- __init__.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/__init__.py b/__init__.py index 6e4a635..e7d3040 100644 --- a/__init__.py +++ b/__init__.py @@ -23,25 +23,22 @@ from .nodes.deepseek_chat import NODE_CLASS_MAPPINGS as DEEPSEEK_CHAT_CLASS_MAPP from .nodes.deepseek_chat import NODE_DISPLAY_NAME_MAPPINGS as DEEPSEEK_CHAT_DISPLAY_MAPPINGS from .nodes.error_log import NODE_CLASS_MAPPINGS as ERROR_LOG_CLASS_MAPPINGS from .nodes.error_log import NODE_DISPLAY_NAME_MAPPINGS as ERROR_LOG_DISPLAY_MAPPINGS - -# 导入执行时间插件 from .nodes.execution_time import NODE_CLASS_MAPPINGS as EXECUTION_TIME_CLASS_MAPPINGS from .nodes.execution_time import NODE_DISPLAY_NAME_MAPPINGS as EXECUTION_TIME_DISPLAY_MAPPINGS - -# 导入gemma2相关节点 from .nodes.gemma_node import NODE_CLASS_MAPPINGS as GEMMA_NODE_CLASS_MAPPINGS from .nodes.gemma_node import NODE_DISPLAY_NAME_MAPPINGS as GEMMA_NODE_DISPLAY_MAPPINGS from .nodes.gemma2prompt import NODE_CLASS_MAPPINGS as GEMMA2_PROMPT_CLASS_MAPPINGS from .nodes.gemma2prompt import NODE_DISPLAY_NAME_MAPPINGS as GEMMA2_PROMPT_DISPLAY_MAPPINGS - -# 修改 github_link_node 的导入方式 from .nodes.github_link_node import NODE_CLASS_MAPPINGS as GITHUB_LINK_NODE_CLASS_MAPPINGS from .nodes.github_link_node import NODE_DISPLAY_NAME_MAPPINGS as GITHUB_LINK_NODE_DISPLAY_MAPPINGS from .nodes.github_link_node import WEB_DIRECTORY, initialize_github_links - from .nodes.ollama_nodes import NODE_CLASS_MAPPINGS as OLLAMA_CLASS_MAPPINGS from .nodes.ollama_nodes import NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_DISPLAY_MAPPINGS +# 导入新的 Gemini 1.5 Flash 节点 +from .nodes.gemini_flash import NODE_CLASS_MAPPINGS as GEMINI_FLASH_CLASS_MAPPINGS +from .nodes.gemini_flash import NODE_DISPLAY_NAME_MAPPINGS as GEMINI_FLASH_DISPLAY_MAPPINGS + # 调用初始化函数 initialize_github_links() @@ -73,7 +70,8 @@ NODE_CLASS_MAPPINGS = { **GEMMA2_PROMPT_CLASS_MAPPINGS, **GEMMA_NODE_CLASS_MAPPINGS, **GITHUB_LINK_NODE_CLASS_MAPPINGS, - **OLLAMA_CLASS_MAPPINGS # 添加 Ollama 节点 + **OLLAMA_CLASS_MAPPINGS, + **GEMINI_FLASH_CLASS_MAPPINGS # 添加 Gemini 1.5 Flash 节点 } NODE_DISPLAY_NAME_MAPPINGS = { @@ -92,7 +90,8 @@ NODE_DISPLAY_NAME_MAPPINGS = { **GEMMA2_PROMPT_DISPLAY_MAPPINGS, **GEMMA_NODE_DISPLAY_MAPPINGS, **GITHUB_LINK_NODE_DISPLAY_MAPPINGS, - **OLLAMA_DISPLAY_MAPPINGS # 添加 Ollama 节点 + **OLLAMA_DISPLAY_MAPPINGS, + **GEMINI_FLASH_DISPLAY_MAPPINGS # 添加 Gemini 1.5 Flash 节点 } __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "load_api_key"] \ No newline at end of file