feat(misc): mark model versions in library for Civitai user models

This commit is contained in:
Will Miao
2025-10-09 15:23:42 +08:00
parent b1dd733493
commit 2b6910bd55
2 changed files with 29 additions and 4 deletions

View File

@@ -687,9 +687,6 @@ class ModelLibraryHandler:
except (TypeError, ValueError):
continue
if await scanner.check_model_version_exists(version_id_int):
continue
images = version.get("images") or []
thumbnail_url = None
if images and isinstance(images, list):
@@ -697,6 +694,8 @@ class ModelLibraryHandler:
if isinstance(first_image, dict):
thumbnail_url = first_image.get("url")
in_library = await scanner.check_model_version_exists(version_id_int)
versions.append(
{
"modelId": model_id_int,
@@ -707,6 +706,7 @@ class ModelLibraryHandler:
"tags": tags,
"baseModel": version.get("baseModel"),
"thumbnailUrl": thumbnail_url,
"inLibrary": in_library,
}
)