From 91d8e0cf3fa2e8094596704b35796db486d64666 Mon Sep 17 00:00:00 2001
From: Dariusz L <dl122@wp.pl>
Date: Thu, 25 Jun 2026 02:41:02 +0200
Subject: [PATCH] Add return_success_on_error to API endpoints and import
 has_known_model_extension

---
 __init__.py                | 7 +++++--
 core/sources/civarchive.py | 1 +
 core/sources/civitai.py    | 1 +
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 052bed0..8b252a7 100644
--- a/__init__.py
+++ b/__init__.py
@@ -158,7 +158,7 @@ class ModelResolverExtension:
                 )
                 download_available = False
 
-            def json_api_endpoint(error_prefix):
+            def json_api_endpoint(error_prefix, return_success_on_error=False):
                 def decorator(func):
                     from functools import wraps
                     @wraps(func)
@@ -169,7 +169,10 @@ class ModelResolverExtension:
                             self.logger.error(
                                 f"Model Resolver {error_prefix} error: {e}", exc_info=True
                             )
-                            return web.json_response({"error": str(e)}, status=500)
+                            response_data = {"error": str(e)}
+                            if return_success_on_error:
+                                response_data["success"] = False
+                            return web.json_response(response_data, status=500)
                     return wrapper
                 return decorator
 
diff --git a/core/sources/civarchive.py b/core/sources/civarchive.py
index b30335a..ba818aa 100644
--- a/core/sources/civarchive.py
+++ b/core/sources/civarchive.py
@@ -17,6 +17,7 @@ import requests
 from ..matcher import (
     calculate_archived_model_confidence,
     calculate_model_title_confidence,
+    has_known_model_extension as _has_known_model_extension,
     normalize_base_model as _normalize_base_model,
     base_model_matches as _base_model_matches,
     base_model_score as _base_model_score,
diff --git a/core/sources/civitai.py b/core/sources/civitai.py
index f079745..e6f6719 100644
--- a/core/sources/civitai.py
+++ b/core/sources/civitai.py
@@ -15,6 +15,7 @@ from urllib.parse import urlparse, parse_qs, quote
 from ..matcher import (
     calculate_filename_confidence,
     calculate_model_title_confidence,
+    has_known_model_extension as _has_known_model_extension,
     normalize_base_model as _normalize_base_model,
     base_model_matches as _base_model_matches,
     base_model_score as _base_model_score,