DemonGatanjieuandClaude Opus 5.5
5667fa2d9e
Types go by their usual names, base model chips narrow the list, the version stands apart from the name, and a model's versions are one card whose version opens the list to show another; a pick goes with it. The grouping, the families and the type names are the Models page's own. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
"""The quick search's model list in a few words (api/quick_index.py)."""
|
|
|
|
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
|
|
PLUGIN = Path(__file__).resolve().parents[1]
|
|
sys.path[:0] = [str(PLUGIN), str(PLUGIN.parents[1])]
|
|
from api import quick_index
|
|
|
|
|
|
def model(**fields):
|
|
base = {"type": "loras", "path_idx": 0, "subfolder": "chars", "filename": "a.safetensors",
|
|
"size_bytes": 10, "preview_url": "/anomalous/image?x", "metadata": {}}
|
|
base.update(fields)
|
|
return base
|
|
|
|
|
|
class CompactModelsTest(unittest.TestCase):
|
|
def test_names_base_and_words(self):
|
|
found = quick_index.compact_models([model(metadata={
|
|
"custom_name": " Ahri ", "name": "Ahri LoRA", "baseModel": "Illustrious",
|
|
"trainedWords": ["ahri", "fox ears", "", "a", "b", "c", "d", "e"], "description": "long",
|
|
})])
|
|
self.assertEqual(found, [{
|
|
"type": "loras", "path_idx": 0, "subfolder": "chars", "filename": "a.safetensors",
|
|
"name": "Ahri", "civitai_name": "Ahri LoRA", "base": "Illustrious", "base_guess": "", "page": 0, "version": 0, "version_name": "",
|
|
"words": ["ahri", "fox ears", "a", "b", "c", "d"], "preview_url": "/anomalous/image?x",
|
|
}])
|
|
|
|
def test_one_line_per_file_and_no_node_pack_files(self):
|
|
found = quick_index.compact_models([
|
|
model(type="unet", subfolder="", filename="flux.safetensors"),
|
|
model(type="diffusion_models", subfolder="", filename="flux.safetensors"),
|
|
model(type="custom_nodes", filename="pack.pt"),
|
|
model(metadata={"trigger_words": "one word"}),
|
|
])
|
|
self.assertEqual([item["type"] for item in found], ["unet", "loras"])
|
|
self.assertEqual(found[1]["words"], ["one word"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|