From cd71722b11b53111bba1ce667d28a1667f61857f Mon Sep 17 00:00:00 2001
From: "Alex \"mcmonkey\" Goodwin" <git_commits@alexgoodwin.dev>
Date: Thu, 19 Sep 2024 16:39:18 +0900
Subject: [PATCH] only cache object_info for now

---
 folder_paths.py | 6 +++++-
 server.py       | 2 ++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/folder_paths.py b/folder_paths.py
index c81db8eb2..40d3f1875 100644
--- a/folder_paths.py
+++ b/folder_paths.py
@@ -52,12 +52,16 @@ class CacheHelper:
     """
     def __init__(self):
         self.cache: dict[str, tuple[list[str], dict[str, float], float]] = {}
+        self.active = False
 
     def get(self, key: str, default=None) -> tuple[list[str], dict[str, float], float]:
+        if not self.active:
+            return default
         return self.cache.get(key, default)
     
     def set(self, key: str, value: tuple[list[str], dict[str, float], float]) -> None:
-        self.cache[key] = value
+        if self.active:
+            self.cache[key] = value
 
     def clear(self):
         self.cache.clear()
diff --git a/server.py b/server.py
index 7bf13b4a6..86919b846 100644
--- a/server.py
+++ b/server.py
@@ -553,6 +553,7 @@ class PromptServer():
         @routes.get("/object_info")
         async def get_object_info(request):
             folder_paths.cache_helper.clear()
+            folder_paths.cache_helper.active = True
             out = {}
             for x in nodes.NODE_CLASS_MAPPINGS:
                 try:
@@ -560,6 +561,7 @@ class PromptServer():
                 except Exception as e:
                     logging.error(f"[ERROR] An error occurred while retrieving information for the '{x}' node.")
                     logging.error(traceback.format_exc())
+            folder_paths.cache_helper.active = False
             return web.json_response(out)
 
         @routes.get("/object_info/{node_class}")