Files
jinchan ce1d493769 feat(module): 添加 AIGC BYOK 密钥解密节点及批量套版 AI hook 功能
- 新增 byok_crypto.py,提供兼容 Java AES-256-GCM encryptedAk 解密工具
- 新增 nodes_byok.py,提供 AigcByokDecryptNode 节点支持工作流中 encryptedAk 解密
- 在 node.py 注册 AigcByokDecryptNode 节点及显示名称
- 扩展 nodes_template_compose.py 支持工具任务服务调用,新增 tool_base_url 等参数
- 支持按新版 templateWorkflow 协议(version 2)执行工作流 AI hook,处理依赖链及缓存
- 增加工作流协议解析与校验实现 template_workflow.py,处理图层索引、JSON Pointer 读写及依赖排序
- 优化模板校验,捕获 WorkflowProtocolError 并返回统一错误结构,完善日志打印
- 支持工具任务上下文解析,剔除保留字段保护 requestId 和 toolCall 不被覆盖
- 修改套版服务调用逻辑,Headers 支持合并 Bearer Token 和自定义头,优先使用 auth_token
- 增加详细错误定位信息,方便调试复杂多节点 AI 工作流
- 更新.gitignore,添加 tests/ 目录忽略规则
2026-09-14 20:42:43 +08:00

63 lines
2.5 KiB
Python

from .nodes_ai_generation import ADICOpenAIGPTImage1, IdeaLabImageGenerate
from .nodes_byok import AigcByokDecryptNode
from .nodes_api_basic import (
ADIC_COMMON_API,
ImageTranslateAPI,
ImageTranslateParamsBuilder,
ImageTranslateResultExtractor,
LoadImagesFromUrls,
MarketImageGenerateWithPolling,
PythonCodeExecutor,
StringToJsonArray,
)
from .nodes_image_utils import (
FluxKontextImageScale,
ImageConcatFromBatch,
ImageStitch,
ReferenceLatent,
)
from .nodes_storage import AliCloudOSSUpload
from .nodes_template_compose import RemoteTemplateBatchCompose
NODE_CLASS_MAPPINGS = {
"ImageTranslateAPI": ImageTranslateAPI,
"ImageTranslateParamsBuilder": ImageTranslateParamsBuilder,
"ImageTranslateResultExtractor": ImageTranslateResultExtractor,
"ADIC_COMMON_API": ADIC_COMMON_API,
"LoadImagesFromUrls": LoadImagesFromUrls,
"PythonCodeExecutor": PythonCodeExecutor,
"StringToJsonArray": StringToJsonArray,
"MarketImageGenerateWithPolling": MarketImageGenerateWithPolling,
"MaletteImageStitch": ImageStitch,
"MaletteReferenceLatent": ReferenceLatent,
"MaletteFluxKontextImageScale": FluxKontextImageScale,
"MaletteImageConcatFromBatch": ImageConcatFromBatch,
"AliCloudOSSUpload": AliCloudOSSUpload,
"ADICOpenAIGPTImage1": ADICOpenAIGPTImage1,
"IdeaLabImageGenerate": IdeaLabImageGenerate,
"RemoteTemplateBatchCompose": RemoteTemplateBatchCompose,
"AigcByokDecrypt": AigcByokDecryptNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"ImageTranslateAPI": "图片翻译 API",
"ImageTranslateParamsBuilder": "图片翻译参数构建器",
"ImageTranslateResultExtractor": "图片翻译结果提取器",
"ADIC_COMMON_API": "ADIC Common API",
"LoadImagesFromUrls": "从URL列表加载图片",
"PythonCodeExecutor": "Python代码执行器",
"StringToJsonArray": "字符串转JSON数组",
"MarketImageGenerateWithPolling": "营销图生图任务(带轮询)",
"MaletteImageStitch": "图片拼接",
"MaletteReferenceLatent": "参考潜变量",
"MaletteFluxKontextImageScale": "Flux Kontext 图片缩放",
"MaletteImageConcatFromBatch": "图片拼接",
"AliCloudOSSUpload": "阿里云OSS文件上传",
"OpenAIGPTImage1": "ADIC OpenAIGPTImage1",
"ADICOpenAIGPTImage1": "ADIC OpenAIGPTImage1",
"IdeaLabImageGenerate": "IdeaLab Image Generate",
"RemoteTemplateBatchCompose": "批量套版(远端合成)",
"AigcByokDecrypt": "AIGC BYOK 密钥解密",
}