diff --git a/node.py b/node.py index dd2a0c2..fe7903d 100644 --- a/node.py +++ b/node.py @@ -64,8 +64,11 @@ class LLMProcessingNode: CATEGORY = "LLM Tag Classifier" def process(self, text, api_key): - # 从文件中加载历史记录 - prompt_history = load_prompt_history("custom_nodes\ComfyUI-TagClassifier\prompt.json") + # 尝试从"custom_nodes\ComfyUI-TagClassifier\prompt.json"或者直接prompt.json中加载历史记录 + try: + prompt_history = load_prompt_history("prompt.json") + except FileNotFoundError: + prompt_history = load_prompt_history("custom_nodes\ComfyUI-TagClassifier\prompt.json") # 调用DeepSeek API try: