feat: support download multiple actual files (#196)

This commit is contained in:
Hayden
2025-08-11 09:10:20 +08:00
committed by GitHub
parent 8b9f3a0e65
commit ac4a168f13
5 changed files with 89 additions and 16 deletions

View File

@@ -69,8 +69,8 @@ class CivitaiModelSearcher(ModelSearcher):
models: list[dict] = []
for version in model_versions:
model_files: list[dict] = version.get("files", [])
model_files = utils.filter_with(model_files, {"type": "Model"})
version_files: list[dict] = version.get("files", [])
model_files = utils.filter_with(version_files, {"type": "Model"})
shortname = version.get("name", None) if len(model_files) > 0 else None
@@ -108,7 +108,7 @@ class CivitaiModelSearcher(ModelSearcher):
description_parts.append("")
model = {
"id": file.get("id"),
"id": version.get("id"),
"shortname": shortname or basename,
"basename": basename,
"extension": extension,
@@ -122,6 +122,7 @@ class CivitaiModelSearcher(ModelSearcher):
"downloadPlatform": "civitai",
"downloadUrl": file.get("downloadUrl"),
"hashes": file.get("hashes"),
"files": version_files if len(version_files) > 1 else None,
}
models.append(model)