mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-28 01:47:32 +08:00
2d21956ac7
Path-derived tags for nested model layouts (e.g. models/checkpoints/flux/foo.safetensors) emitted only the slash-joined shape `["models", "checkpoints/flux"]`, which broke the frontend combo-widget set-membership filter `include_tags=models,checkpoints` — the literal `checkpoints` token was no longer present in the asset's tag set. Add `expand_bucket_prefixes` at the tag-write layer. When a tag's first slash segment is a registered model category (or input/output/temp root), the bucket is inserted as a standalone token immediately after the slash-joined form. This preserves tag[1] as the slash-joined positional contract cloud emits while restoring the set-membership token the frontend filter requires. The expansion is bounded to known buckets so free-form user labels with slashes (`my-org/team-a`) pass through unchanged. The helper is applied uniformly in `set_reference_tags`, `add_tags_to_reference`, and `batch_insert_seed_assets` so HTTP uploads, user-tag mutations, and path-scanning ingest all converge on the same canonical shape. Also align the upload-route category validator with `resolve_destination_from_tags` by extracting the first slash segment of tag[1], so HTTP uploads matching cloud's slash-joined emission shape are no longer rejected as `unknown models category`.
70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
"""Unit tests for app.assets.helpers."""
|
|
|
|
from app.assets.helpers import expand_bucket_prefixes
|
|
|
|
|
|
class TestExpandBucketPrefixes:
|
|
def test_flat_category_unchanged(self):
|
|
# `checkpoints` is already a standalone token, no expansion needed.
|
|
assert expand_bucket_prefixes(["models", "checkpoints"]) == [
|
|
"models",
|
|
"checkpoints",
|
|
]
|
|
|
|
def test_nested_category_inserts_bucket(self):
|
|
# Path-derived shape for `models/checkpoints/flux/foo.safetensors` —
|
|
# the standalone bucket has to be present so the FE set-membership
|
|
# filter (`include_tags=models,checkpoints`) matches the asset.
|
|
assert expand_bucket_prefixes(["models", "checkpoints/flux"]) == [
|
|
"models",
|
|
"checkpoints/flux",
|
|
"checkpoints",
|
|
]
|
|
|
|
def test_deeply_nested_only_first_segment_expands(self):
|
|
# Only the FIRST slash segment ever gets emitted as a standalone —
|
|
# intermediate path segments don't have routing significance.
|
|
assert expand_bucket_prefixes(
|
|
["models", "diffusers/kolors/text_encoder"]
|
|
) == ["models", "diffusers/kolors/text_encoder", "diffusers"]
|
|
|
|
def test_unknown_prefix_does_not_expand(self):
|
|
# Free-form user labels with slashes whose first segment is not a
|
|
# registered bucket pass through opaquely.
|
|
assert expand_bucket_prefixes(["models", "my-org/team-a"]) == [
|
|
"models",
|
|
"my-org/team-a",
|
|
]
|
|
|
|
def test_idempotent(self):
|
|
# Re-applying the helper is a no-op once the bucket is in the set.
|
|
expanded = expand_bucket_prefixes(["models", "checkpoints/flux"])
|
|
assert expand_bucket_prefixes(expanded) == expanded
|
|
|
|
def test_does_not_duplicate_existing_bucket(self):
|
|
# If the caller already supplied the standalone bucket, don't add a
|
|
# second copy.
|
|
assert expand_bucket_prefixes(
|
|
["models", "checkpoints/flux", "checkpoints"]
|
|
) == ["models", "checkpoints/flux", "checkpoints"]
|
|
|
|
def test_preserves_caller_order(self):
|
|
# User tags after path tags must stay after; the inserted bucket
|
|
# token slots in immediately after its slash-joined parent so the
|
|
# microsecond stagger lands it at path-tier before user-tier.
|
|
assert expand_bucket_prefixes(
|
|
["models", "loras/style", "favorite", "v2"]
|
|
) == ["models", "loras/style", "loras", "favorite", "v2"]
|
|
|
|
def test_empty_input(self):
|
|
assert expand_bucket_prefixes([]) == []
|
|
|
|
def test_input_root_with_subpath_no_expansion(self):
|
|
# `portraits` isn't a registered model category, so the input
|
|
# subpath stays opaque (FE filter doesn't have a checkpoint-loader
|
|
# analogue for input subfolders).
|
|
assert expand_bucket_prefixes(["input", "portraits/2026"]) == [
|
|
"input",
|
|
"portraits/2026",
|
|
]
|