alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-08-25 23:11:26 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat(versions): add file-variant badge and hide download button for in-library versions (#1058)

Browse Source
This commit is contained in:
Will Miao
2026-08-24 23:17:10 +08:00
parent a0bb6df2b8
commit da071e8452
17 changed files with 557 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/routes/test_model_update_handler.py
+18
View File
@@ -693,3 +693,21 @@ async def test_enrich_early_access_details_skips_permanent_paid(monkeypatch):
enriched_map = {v.version_id: v for v in enriched.versions}
assert enriched_map[200].early_access_ends_at == "2099-01-01T00:00:00.000Z"
assert enriched_map[100].early_access_ends_at is None
def test_serialize_version_includes_file_count():
version = ModelVersionRecord(
version_id=11, name="v11", base_model=None, released_at=None, size_bytes=None,
preview_url=None, is_in_library=True, should_ignore=False, file_count=2,
)
serialized = ModelUpdateHandler._serialize_version(version, None)
assert serialized["fileCount"] == 2
def test_serialize_version_file_count_defaults_to_none():
version = ModelVersionRecord(
version_id=12, name="v12", base_model=None, released_at=None, size_bytes=None,
preview_url=None, is_in_library=False, should_ignore=False,
)
serialized = ModelUpdateHandler._serialize_version(version, None)
assert serialized["fileCount"] is None