From 18d098a0dace56a3dcbdb4f21129e4a6b5785b10 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Thu, 9 Apr 2026 18:08:53 +0200 Subject: [PATCH] fix: add independent exception handling to _record_prompt --- __init__.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/__init__.py b/__init__.py index b9d3c41..3f1c494 100644 --- a/__init__.py +++ b/__init__.py @@ -40,10 +40,16 @@ def on_prompt_handler(json_data): def _record_prompt(class_types, prompt): - tracker.record_usage(class_types, mapper) - models = model_mapper.extract_models_from_prompt(prompt) - if models: - tracker.record_model_usage(models) + try: + 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) + if 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)