Ethanfel/Comfyui-Nodes-Stats
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add independent exception handling to _record_prompt

Browse Source
This commit is contained in:
Ethanfel
2026-04-09 18:08:53 +02:00
parent ed39c5918a
commit 18d098a0da
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+6
View File
@@ -40,10 +40,16 @@ def on_prompt_handler(json_data):
def _record_prompt(class_types, prompt): def _record_prompt(class_types, prompt):
try:
tracker.record_usage(class_types, mapper) tracker.record_usage(class_types, mapper)
except Exception:
logger.warning("nodes-stats: error recording node usage", exc_info=True)
try:
models = model_mapper.extract_models_from_prompt(prompt) models = model_mapper.extract_models_from_prompt(prompt)
if models: if models:
tracker.record_model_usage(models) tracker.record_model_usage(models)
except Exception:
logger.warning("nodes-stats: error recording model usage", exc_info=True)
PromptServer.instance.add_on_prompt_handler(on_prompt_handler) PromptServer.instance.add_on_prompt_handler(on_prompt_handler)