78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
"""
|
|
提示词内容提取预设节点 - V3 版本
|
|
"""
|
|
|
|
import hashlib
|
|
from typing import Dict, Any
|
|
|
|
from comfy_api.latest import io
|
|
from .base.base_node import BaseNode
|
|
|
|
|
|
class KontextPresetNode(BaseNode, io.ComfyNode):
|
|
"""
|
|
提示词内容提取预设节点（V3）
|
|
允许用户选择一个 KonText（上下文），读取其配置并格式化输出预设的系统提示词，
|
|
将选定的参数组（包含模型和温度）也输出，作为后续节点的强制输入参数。
|
|
"""
|
|
|
|
# 静态缓存
|
|
_cached_config: Dict[str, Any] = {}
|
|
|
|
@classmethod
|
|
def _load_config(cls) -> Dict[str, Any]:
|
|
"""按需获取配置，V3由于有类方法缓存机制，依然可以使用"""
|
|
from ..config_manager import config_manager
|
|
|
|
config = config_manager.get_system_prompts()
|
|
if not config:
|
|
return {}
|
|
|
|
return config
|
|
|
|
@classmethod
|
|
def define_schema(cls):
|
|
config = cls._load_config()
|
|
kontext_options = []
|
|
|
|
# 解析可用的 kontext 选项
|
|
if config and "kontexts" in config:
|
|
for kontext in config["kontexts"]:
|
|
kontext_options.append(kontext["name"])
|
|
|
|
if not kontext_options:
|
|
kontext_options = ["默认提取预设"]
|
|
|
|
return io.Schema(
|
|
node_id="KontextPresetNode",
|
|
display_name="✨KonText Extractor",
|
|
category="✨Prompt Assistant",
|
|
description="Extract and format prompt templates from a selected KonText",
|
|
inputs=[
|
|
io.Combo.Input(
|
|
"kontext",
|
|
options=kontext_options,
|
|
default=kontext_options[0] if kontext_options else None
|
|
),
|
|
],
|
|
outputs=[
|
|
io.String.Output("system_prompt"),
|
|
],
|
|
)
|
|
|
|
@classmethod
|
|
def fingerprint_inputs(cls, kontext=None):
|
|
return hash((kontext,))
|
|
|
|
@classmethod
|
|
def execute(cls, kontext):
|
|
config = cls._load_config()
|
|
system_prompt = ""
|
|
|
|
if config and "kontexts" in config:
|
|
for k in config["kontexts"]:
|
|
if k["name"] == kontext:
|
|
system_prompt = k.get("prompt", "")
|
|
break
|
|
|
|
return io.NodeOutput(system_prompt) |