Add return_success_on_error to API endpoints and import has_known_model_extension

__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
 

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,

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,