From e5b10b0fd916c6a244c6c7362be3a1bd2b9aa288 Mon Sep 17 00:00:00 2001
From: Wenaka <121840779+Wenaka2004@users.noreply.github.com>
Date: Fri, 31 Jan 2025 12:28:34 +0800
Subject: [PATCH] Update node.py

---
 node.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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: