feat: snapshot product suite reference assets

This commit is contained in:
chengma
2026-07-17 17:03:40 +08:00
parent d11dc752b5
commit 1fabd8afcf
12 changed files with 252 additions and 5 deletions
+8
View File
@@ -347,6 +347,13 @@ def build_job_specs(source_assets, base_prompt, settings, item_id, *, template_t
if not assets:
return []
normalized = normalize_suite_settings(settings)
primary_asset_id = int(getattr(assets[0], "id", assets[0]))
reference_asset_ids = []
if not normalized["per_image_primary"]:
for asset in assets[1:8]:
asset_id = int(getattr(asset, "id", asset))
if asset_id != primary_asset_id and asset_id not in reference_asset_ids:
reference_asset_ids.append(asset_id)
specs = []
for category in category_order(normalized):
count = _count(normalized["categories"].get(category, 0))
@@ -358,6 +365,7 @@ def build_job_specs(source_assets, base_prompt, settings, item_id, *, template_t
specs.append(
{
"source_asset_id": int(getattr(asset, "id", asset)),
"reference_asset_ids": list(reference_asset_ids),
"job_type": str(category),
"category": str(category),
"category_index": category_index,