-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy path__init__.py
138 lines (121 loc) · 7.65 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import os
from .api_utils import load_api_key
from .nodes.groqchat import NODE_CLASS_MAPPINGS as GROQ_CLASS_MAPPINGS
from .nodes.groqchat import NODE_DISPLAY_NAME_MAPPINGS as GROQ_DISPLAY_MAPPINGS
from .nodes.moonshot_chat_nodes import NODE_CLASS_MAPPINGS as MOONSHOT_CLASS_MAPPINGS
from .nodes.moonshot_chat_nodes import NODE_DISPLAY_NAME_MAPPINGS as MOONSHOT_DISPLAY_MAPPINGS
from .nodes.SD3LongCaptioner_v2 import NODE_CLASS_MAPPINGS as SD3_V2_CLASS_MAPPINGS
from .nodes.SD3LongCaptioner_v2 import NODE_DISPLAY_NAME_MAPPINGS as SD3_V2_DISPLAY_MAPPINGS
from .nodes.file_based_chat import NODE_CLASS_MAPPINGS as FILE_CHAT_CLASS_MAPPINGS
from .nodes.file_based_chat import NODE_DISPLAY_NAME_MAPPINGS as FILE_CHAT_DISPLAY_MAPPINGS
from .nodes.dwimage2 import NODE_CLASS_MAPPINGS as MOONDREAM2_V2_NODE_CLASS_MAPPINGS
from .nodes.dwimage2 import NODE_DISPLAY_NAME_MAPPINGS as MOONDREAM2_V2_NODE_DISPLAY_NAME_MAPPINGS
from .nodes.prompt_extractor import NODE_CLASS_MAPPINGS as PROMPT_EXTRACTOR_CLASS_MAPPINGS
from .nodes.prompt_extractor import NODE_DISPLAY_NAME_MAPPINGS as PROMPT_EXTRACTOR_DISPLAY_MAPPINGS
from .nodes.sdprompt_agent import NODE_CLASS_MAPPINGS as SD_PROMPT_AGENT_CLASS_MAPPINGS
from .nodes.sdprompt_agent import NODE_DISPLAY_NAME_MAPPINGS as SD_PROMPT_AGENT_DISPLAY_NAME_MAPPINGS
from .nodes.ollama_prompt_extractor import NODE_CLASS_MAPPINGS as OLLAMA_PROMPT_EXTRACTOR_CLASS_MAPPINGS
from .nodes.ollama_prompt_extractor import NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_PROMPT_EXTRACTOR_DISPLAY_MAPPINGS
from .nodes.deepseek_translater import NODE_CLASS_MAPPINGS as DEEPSEEK_TRANSLATOR_CLASS_MAPPINGS
from .nodes.deepseek_translater import NODE_DISPLAY_NAME_MAPPINGS as DEEPSEEK_TRANSLATOR_DISPLAY_MAPPINGS
from .nodes.deepseek_chat import NODE_CLASS_MAPPINGS as DEEPSEEK_CHAT_CLASS_MAPPINGS
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
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
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
# 导入新的 GeminiFluxPrompt 节点
from .nodes.gemini_flux_prompt import NODE_CLASS_MAPPINGS as GEMINI_FLUX_PROMPT_CLASS_MAPPINGS
from .nodes.gemini_flux_prompt import NODE_DISPLAY_NAME_MAPPINGS as GEMINI_FLUX_PROMPT_DISPLAY_MAPPINGS
# 导入新的 PaliGemma3bCaptioner 节点
from .nodes.PaliGemma3bCaptioner import NODE_CLASS_MAPPINGS as PALI_GEMMA_3B_CAPTIONER_CLASS_MAPPINGS
from .nodes.PaliGemma3bCaptioner import NODE_DISPLAY_NAME_MAPPINGS as PALI_GEMMA_3B_CAPTIONER_DISPLAY_MAPPINGS
# 导入新的 Qwen2VLCaption 节点
from .nodes.Qwen2VLCaption import NODE_CLASS_MAPPINGS as QWEN2VL_CAPTION_CLASS_MAPPINGS
from .nodes.Qwen2VLCaption import NODE_DISPLAY_NAME_MAPPINGS as QWEN2VL_CAPTION_DISPLAY_MAPPINGS
# 导入新的 Qwen2VLLocalCaption 节点
from .nodes.Qwen2VLLocalCaption import NODE_CLASS_MAPPINGS as QWEN2VL_LOCAL_CAPTION_CLASS_MAPPINGS
from .nodes.Qwen2VLLocalCaption import NODE_DISPLAY_NAME_MAPPINGS as QWEN2VL_LOCAL_CAPTION_DISPLAY_MAPPINGS
# 导入新的 PromptEngineeringNode 节点
from .nodes.PromptEngineeringNode import NODE_CLASS_MAPPINGS as PROMPT_ENGINEERING_NODE_CLASS_MAPPINGS
from .nodes.PromptEngineeringNode import NODE_DISPLAY_NAME_MAPPINGS as PROMPT_ENGINEERING_NODE_DISPLAY_MAPPINGS
# 调用初始化函数
initialize_github_links()
# 定义执行时间统计函数
def load_javascript(web_directory):
js_file_path = os.path.join(web_directory, "executionTime.js")
if os.path.exists(js_file_path):
with open(js_file_path, "r") as js_file:
javascript = js_file.read()
return javascript
return None
# 确保在 NODE_CLASS_MAPPINGS 定义之前添加这行
WEB_DIRECTORY = os.path.join(os.path.dirname(os.path.realpath(__file__)), "web")
NODE_CLASS_MAPPINGS = {
**GROQ_CLASS_MAPPINGS,
**MOONSHOT_CLASS_MAPPINGS,
**SD3_V2_CLASS_MAPPINGS,
**FILE_CHAT_CLASS_MAPPINGS,
**MOONDREAM2_V2_NODE_CLASS_MAPPINGS,
**PROMPT_EXTRACTOR_CLASS_MAPPINGS,
**SD_PROMPT_AGENT_CLASS_MAPPINGS,
**OLLAMA_PROMPT_EXTRACTOR_CLASS_MAPPINGS,
**DEEPSEEK_TRANSLATOR_CLASS_MAPPINGS,
**DEEPSEEK_CHAT_CLASS_MAPPINGS,
**ERROR_LOG_CLASS_MAPPINGS,
**EXECUTION_TIME_CLASS_MAPPINGS,
**GEMMA2_PROMPT_CLASS_MAPPINGS,
**GEMMA_NODE_CLASS_MAPPINGS,
**GITHUB_LINK_NODE_CLASS_MAPPINGS,
**OLLAMA_CLASS_MAPPINGS,
**GEMINI_FLASH_CLASS_MAPPINGS, # 添加 Gemini 1.5 Flash 节点
**GEMINI_FLUX_PROMPT_CLASS_MAPPINGS, # 添加 GeminiFluxPrompt 节点
**PALI_GEMMA_3B_CAPTIONER_CLASS_MAPPINGS, # 添加 PaliGemma3bCaptioner 节点
**QWEN2VL_CAPTION_CLASS_MAPPINGS, # 添加 Qwen2VLCaption 节点
**QWEN2VL_LOCAL_CAPTION_CLASS_MAPPINGS, # 添加 Qwen2VLLocalCaption 节点
**PROMPT_ENGINEERING_NODE_CLASS_MAPPINGS, # 添加 PromptEngineeringNode 节点
}
from .nodes.PromptEngineeringNode import NODE_CLASS_MAPPINGS as PROMPT_ENGINEERING_NODE
NODE_CLASS_MAPPINGS.update(PROMPT_ENGINEERING_NODE)
NODE_DISPLAY_NAME_MAPPINGS = {
**GROQ_DISPLAY_MAPPINGS,
**MOONSHOT_DISPLAY_MAPPINGS,
**SD3_V2_DISPLAY_MAPPINGS,
**FILE_CHAT_DISPLAY_MAPPINGS,
**MOONDREAM2_V2_NODE_DISPLAY_NAME_MAPPINGS,
**PROMPT_EXTRACTOR_DISPLAY_MAPPINGS,
**SD_PROMPT_AGENT_DISPLAY_NAME_MAPPINGS,
**OLLAMA_PROMPT_EXTRACTOR_DISPLAY_MAPPINGS,
**DEEPSEEK_TRANSLATOR_DISPLAY_MAPPINGS,
**DEEPSEEK_CHAT_DISPLAY_MAPPINGS,
**ERROR_LOG_DISPLAY_MAPPINGS,
**EXECUTION_TIME_DISPLAY_MAPPINGS,
**GEMMA2_PROMPT_DISPLAY_MAPPINGS,
**GEMMA_NODE_DISPLAY_MAPPINGS,
**GITHUB_LINK_NODE_DISPLAY_MAPPINGS,
**OLLAMA_DISPLAY_MAPPINGS,
**GEMINI_FLASH_DISPLAY_MAPPINGS, # 添加 Gemini 1.5 Flash 节点
**GEMINI_FLUX_PROMPT_DISPLAY_MAPPINGS, # 添加 GeminiFluxPrompt 节点
**PALI_GEMMA_3B_CAPTIONER_DISPLAY_MAPPINGS, # 添加 PaliGemma3bCaptioner 节点
**QWEN2VL_CAPTION_DISPLAY_MAPPINGS, # 添加 Qwen2VLCaption 节点
**QWEN2VL_LOCAL_CAPTION_DISPLAY_MAPPINGS, # 添加 Qwen2VLLocalCaption 节点
**PROMPT_ENGINEERING_NODE_DISPLAY_MAPPINGS, # 添加 PromptEngineeringNode 节点
}
from .nodes.FluxPromptEngineeringNode import NODE_CLASS_MAPPINGS as FLUX_PROMPT_ENGINEERING_NODE_CLASS_MAPPINGS
from .nodes.FluxPromptEngineeringNode import NODE_DISPLAY_NAME_MAPPINGS as FLUX_PROMPT_ENGINEERING_NODE_DISPLAY_MAPPINGS
# 在 NODE_CLASS_MAPPINGS 和 NODE_DISPLAY_NAME_MAPPINGS 中添加新的映射
NODE_CLASS_MAPPINGS.update(FLUX_PROMPT_ENGINEERING_NODE_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(FLUX_PROMPT_ENGINEERING_NODE_DISPLAY_MAPPINGS)
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "load_api_key"]