mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-09-21 11:11:26 -03:00
Will Miao
942717f0b6
A repository whose uploader wrote no README still gets a card. ModelScope
answers with a placeholder notice ("the contributor provided no further
description"), a block of SDK/git download instructions, and a closing
invitation to complete the card. None of it describes the model, yet it was
being sent to the LLM and, worse, stored as `modelDescription` — so a Krea 2
LoRA whose only real text was the author's summary showed 841 characters of
`pip install modelscope` scaffolding on its description tab.
Add `_strip_generated_card_boilerplate()` and run it on both paths:
`clean_readme_for_llm()` (the prompt) and `convert_readme_to_html()` (the
stored description). Markers are matched as substrings because the notices
are prose and because non-Latin scripts are not space-delimited — the notice
continues with a full-width period, so the `title == keyword` matching used
for the English boilerplate headings never fired.
A marker heading takes its whole section with it, which is what removes the
download block hanging off the notice; a stand-alone notice line is dropped
alone. Content the author added later, under a heading of equal or higher
level, is kept, so a card that was improved after the placeholder is not
thrown away.
Verified on the live repositories: the placeholder card's description went
from 841 characters to the 86-character author summary, while the repo with
a genuinely author-written card is byte-for-byte unchanged.
608 lines
20 KiB
Python
608 lines
20 KiB
Python
"""Tests for ``readme_processor.py`` — HF README processing for enrich_hf_metadata.
|
|
|
|
Import via ``importlib`` to avoid the ``folder_paths`` dependency in
|
|
``py.services.agent.__init__``.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
_MODULE_PATH = Path(__file__).parents[2] / "py" / "services" / "agent" / "skills" / "enrich_hf_metadata" / "readme_processor.py"
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def R():
|
|
"""Load the ``readme_processor`` module once per session."""
|
|
spec = importlib.util.spec_from_file_location("readme_processor", str(_MODULE_PATH))
|
|
assert spec is not None
|
|
assert spec.loader is not None
|
|
mod = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(mod)
|
|
return mod
|
|
|
|
|
|
# ======================================================================
|
|
# extract_gallery_images
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractGalleryImages:
|
|
def test_empty(self, R):
|
|
assert R.extract_gallery_images("", "repo") == []
|
|
assert R.extract_gallery_images("no frontmatter", "repo") == []
|
|
|
|
def test_no_widget(self, R):
|
|
readme = "---\ntags: [test]\n---\nbody"
|
|
assert R.extract_gallery_images(readme, "repo") == []
|
|
|
|
def test_widget_simple_text(self, R):
|
|
"""YAML ``text: 'plain'`` → extracted as-is."""
|
|
readme = """---
|
|
widget:
|
|
- text: 'a cute cat'
|
|
output:
|
|
url: images/cat.png
|
|
---"""
|
|
imgs = R.extract_gallery_images(readme, "user/repo")
|
|
assert len(imgs) == 1
|
|
assert imgs[0]["meta"]["prompt"] == "a cute cat"
|
|
assert "images/cat.png" in imgs[0]["url"]
|
|
|
|
def test_widget_unquoted_text(self, R):
|
|
"""YAML ``text: plain value`` without quotes."""
|
|
readme = """---
|
|
widget:
|
|
- text: simple text
|
|
output:
|
|
url: img.png
|
|
---"""
|
|
imgs = R.extract_gallery_images(readme, "user/repo")
|
|
assert len(imgs) == 1
|
|
assert imgs[0]["meta"]["prompt"] == "simple text"
|
|
|
|
def test_widget_block_scalar(self, R):
|
|
"""YAML ``text: >-`` folded block scalar — extract actual content."""
|
|
readme = """---
|
|
widget:
|
|
- text: >-
|
|
Long toons, a close-up of a cartoon characters face is featured in a
|
|
vibrant red backdrop.
|
|
output:
|
|
url: images/LT4.png
|
|
---"""
|
|
imgs = R.extract_gallery_images(readme, "user/repo")
|
|
assert len(imgs) == 1
|
|
prompt = imgs[0]["meta"]["prompt"]
|
|
assert "Long toons" in prompt
|
|
assert "vibrant red backdrop" in prompt
|
|
assert prompt != ">-"
|
|
|
|
def test_widget_dash_prefix_output(self, R):
|
|
"""YAML ``- output:`` (dash prefix) — regression for widget parsing."""
|
|
readme = """---
|
|
widget:
|
|
- output:
|
|
url: images/test.png
|
|
text: dash test
|
|
---"""
|
|
imgs = R.extract_gallery_images(readme, "user/repo")
|
|
assert len(imgs) == 1
|
|
assert imgs[0]["meta"]["prompt"] == "dash test"
|
|
assert "images/test.png" in imgs[0]["url"]
|
|
|
|
def test_widget_mixed_entries(self, R):
|
|
"""Multiple widget entries with different text styles."""
|
|
readme = """---
|
|
widget:
|
|
- text: >-
|
|
First entry description.
|
|
output:
|
|
url: img1.png
|
|
- text: second entry
|
|
output:
|
|
url: img2.png
|
|
- text: 'third entry'
|
|
output:
|
|
url: img3.png
|
|
---"""
|
|
imgs = R.extract_gallery_images(readme, "user/repo")
|
|
assert len(imgs) == 3
|
|
assert imgs[0]["meta"]["prompt"] == "First entry description."
|
|
assert imgs[1]["meta"]["prompt"] == "second entry"
|
|
assert imgs[2]["meta"]["prompt"] == "third entry"
|
|
|
|
|
|
# ======================================================================
|
|
# extract_simple_markdown_images
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractSimpleMarkdownImages:
|
|
def test_empty(self, R):
|
|
assert R.extract_simple_markdown_images("", "repo") == []
|
|
|
|
def test_basic_markdown_image(self, R):
|
|
"""```` → absolute URL."""
|
|
imgs = R.extract_simple_markdown_images("", "u/r")
|
|
assert len(imgs) == 1
|
|
assert "image_0.png" in imgs[0]["url"]
|
|
assert imgs[0]["meta"]["prompt"] == "test"
|
|
|
|
def test_absolute_url(self, R):
|
|
"""```` → keep as-is."""
|
|
imgs = R.extract_simple_markdown_images(
|
|
"", "u/r"
|
|
)
|
|
assert len(imgs) == 1
|
|
assert imgs[0]["url"] == "https://example.com/img.png"
|
|
|
|
def test_skips_code_fences(self, R):
|
|
"""Inside ``` blocks should be ignored."""
|
|
text = """outside
|
|
```
|
|
|
|
```
|
|
outside again
|
|
"""
|
|
imgs = R.extract_simple_markdown_images(text, "u/r")
|
|
assert len(imgs) == 1
|
|
assert "valid.png" in imgs[0]["url"]
|
|
|
|
def test_deduplicates(self, R):
|
|
text = "\n"
|
|
imgs = R.extract_simple_markdown_images(text, "u/r")
|
|
assert len(imgs) == 1 # deduplicated
|
|
|
|
|
|
# ======================================================================
|
|
# extract_html_img_tags
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractHtmlImgTags:
|
|
def test_double_quoted_src(self, R):
|
|
imgs = R.extract_html_img_tags('<img src="./img.png">', "u/r")
|
|
assert len(imgs) == 1
|
|
assert "img.png" in imgs[0]["url"]
|
|
|
|
def test_single_quoted_src(self, R):
|
|
imgs = R.extract_html_img_tags("<img src='./img.png'>", "u/r")
|
|
assert len(imgs) == 1
|
|
assert "img.png" in imgs[0]["url"]
|
|
|
|
def test_absolute_url(self, R):
|
|
imgs = R.extract_html_img_tags(
|
|
'<img src="https://cdn.example.com/img.png">', "u/r"
|
|
)
|
|
assert len(imgs) == 1
|
|
assert imgs[0]["url"] == "https://cdn.example.com/img.png"
|
|
|
|
def test_deduplicates_across_formats(self, R):
|
|
text = '<img src="./img.png">\n<img src=\'./img.png\'>'
|
|
imgs = R.extract_html_img_tags(text, "u/r")
|
|
assert len(imgs) == 1
|
|
|
|
|
|
# ======================================================================
|
|
# extract_gallery_table_images
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractGalleryTableImages:
|
|
def test_gallery_table(self, R):
|
|
text = """| Preview | Prompt |
|
|
|--------|--------|
|
|
|  | a cat |
|
|
|  | a dog |"""
|
|
imgs = R.extract_gallery_table_images(text, "u/r")
|
|
assert len(imgs) == 2
|
|
assert imgs[0]["meta"]["prompt"] == "a cat"
|
|
assert "a.png" in imgs[0]["url"]
|
|
assert imgs[1]["meta"]["prompt"] == "a dog"
|
|
|
|
def test_skips_non_gallery_table(self, R):
|
|
text = """| Parameter | Value |
|
|
|----------|-------|
|
|
| Steps | 4 |"""
|
|
imgs = R.extract_gallery_table_images(text, "u/r")
|
|
assert len(imgs) == 0
|
|
|
|
|
|
# ======================================================================
|
|
# clean_readme_for_llm + strip helpers
|
|
# ======================================================================
|
|
|
|
|
|
class TestCleanReadmeForLlm:
|
|
def test_preserves_plain_code_block(self, R):
|
|
"""`` ``` `` without language tag → preserved (trigger words)."""
|
|
text = """Before
|
|
```
|
|
pixel art sprite, game asset
|
|
```
|
|
After"""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "pixel art sprite" in cleaned
|
|
assert "game asset" in cleaned
|
|
|
|
def test_strips_fenced_code_with_lang(self, R):
|
|
"""`` ```python `` → stripped."""
|
|
text = "before\n```python\nimport torch\n```\nafter"
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "import torch" not in cleaned
|
|
assert "before" in cleaned
|
|
assert "after" in cleaned
|
|
|
|
def test_preserves_markdown_image_url(self, R):
|
|
"""```` → URL kept for LLM preview extraction."""
|
|
text = ""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "./preview.png" in cleaned
|
|
|
|
def test_converts_html_img_tag_to_markdown_image(self, R):
|
|
"""``<img src="...">`` → ```` preserving URL for LLM."""
|
|
text = 'before\n<img src="logo.png">\nafter'
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "" in cleaned
|
|
assert "logo.png" in cleaned # URL preserved for LLM extraction
|
|
|
|
def test_widget_stripped_frontmatter_preserved(self, R):
|
|
"""Widget YAML stripped but ``base_model:`` kept."""
|
|
text = """---
|
|
tags: [test]
|
|
widget:
|
|
- text: >-
|
|
long description here
|
|
output:
|
|
url: img.png
|
|
base_model: black-forest-labs/FLUX.1-dev
|
|
instance_prompt: test
|
|
---"""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "widget:" not in cleaned
|
|
assert "black-forest-labs/FLUX.1-dev" in cleaned
|
|
assert "instance_prompt: test" in cleaned
|
|
|
|
def test_training_table_stripped(self, R):
|
|
"""Training-parameter table → stripped."""
|
|
text = """before
|
|
| LR Scheduler | constant |
|
|
|--------------|---------|
|
|
| Optimizer | AdamW |
|
|
after"""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "LR Scheduler" not in cleaned
|
|
assert "Optimizer" not in cleaned
|
|
assert "before" in cleaned
|
|
assert "after" in cleaned
|
|
|
|
def test_best_dimensions_table_kept(self, R):
|
|
"""Non-training table (Best Dimensions) → kept."""
|
|
text = """## Best Dimensions
|
|
- 768 x 1024 (Best)
|
|
- 1024 x 1024 (Default)"""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "768 x 1024" in cleaned
|
|
|
|
def test_boilerplate_section_stripped(self, R):
|
|
text = """stuff
|
|
## Download model
|
|
[link](url)
|
|
## Next section
|
|
content"""
|
|
cleaned = R.clean_readme_for_llm(text)
|
|
assert "Download model" not in cleaned
|
|
assert "Next section" in cleaned
|
|
assert "content" in cleaned
|
|
|
|
def test_returns_empty_for_none(self, R):
|
|
assert R.clean_readme_for_llm(None) == ""
|
|
|
|
def test_returns_empty_for_empty(self, R):
|
|
assert R.clean_readme_for_llm("") == ""
|
|
|
|
|
|
# ======================================================================
|
|
# _is_heading / _heading_level
|
|
# ======================================================================
|
|
|
|
|
|
class TestHeadingDetection:
|
|
@pytest.mark.parametrize(
|
|
"line,expected",
|
|
[
|
|
("# Title", 1),
|
|
("## Sub", 2),
|
|
("### Subsub", 3),
|
|
("#### Subsubsub", 4),
|
|
("<h1>Title</h1>", 1),
|
|
("<h2>Sub</h2>", 2),
|
|
("<h3 class='x'>Sub</h3>", 3),
|
|
("<h4 id='y'>Sub</h4>", 4),
|
|
("not a heading", 0),
|
|
("###", 0), # no text after ###
|
|
("</h2>", 0), # closing tag, not a heading
|
|
("", 0),
|
|
],
|
|
)
|
|
def test_heading_level(self, R, line, expected):
|
|
assert R._heading_level(line) == expected
|
|
|
|
@pytest.mark.parametrize(
|
|
"line,expected",
|
|
[
|
|
("# Title", True),
|
|
("<h2>Sub</h2>", True),
|
|
("</h2>", False), # closing tag
|
|
("not heading", False),
|
|
],
|
|
)
|
|
def test_is_heading(self, R, line, expected):
|
|
assert R._is_heading(line) == expected
|
|
|
|
|
|
# ======================================================================
|
|
# extract_relevant_section
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractRelevantSection:
|
|
def test_fallback_full_readme(self, R):
|
|
"""No match → full README returned."""
|
|
readme = "# Title\n\nsome content"
|
|
assert R.extract_relevant_section(readme, "nonexistent") == readme
|
|
|
|
def test_empty_basename_returns_full(self, R):
|
|
readme = "# Title"
|
|
assert R.extract_relevant_section(readme, "") == readme
|
|
|
|
def test_match_heading_includes_yaml(self, R):
|
|
"""Matching heading should still include YAML frontmatter."""
|
|
readme = """---
|
|
base_model: foo
|
|
---
|
|
# My-Model-Title
|
|
|
|
content
|
|
## Subsection
|
|
more"""
|
|
section = R.extract_relevant_section(readme, "My-Model")
|
|
assert "base_model: foo" in section
|
|
assert "content" in section
|
|
assert "Subsection" in section
|
|
|
|
def test_match_heading_includes_subheadings(self, R):
|
|
"""``# Title`` match includes all ``##`` children."""
|
|
readme = """# Main Title
|
|
|
|
## Child A
|
|
content A
|
|
## Child B
|
|
content B
|
|
## Child C
|
|
content C"""
|
|
section = R.extract_relevant_section(readme, "Main Title")
|
|
assert "Child A" in section
|
|
assert "Child B" in section
|
|
assert "Child C" in section
|
|
|
|
def test_match_download_link(self, R):
|
|
"""Download link containing basename → section extracted."""
|
|
readme = """# Collection
|
|
## Model A
|
|
[Download](./model_a.safetensors)
|
|
## MyModel
|
|
[Download](./mymodel.safetensors)
|
|
content here
|
|
## Model B
|
|
other"""
|
|
section = R.extract_relevant_section(readme, "mymodel")
|
|
assert "content here" in section
|
|
assert "Model A" not in section # should not include sibling
|
|
|
|
def test_heading_closing_tag_not_boundary(self, R):
|
|
"""``</h2>`` should NOT be treated as a section boundary."""
|
|
readme = """# Title
|
|
<p>some text</p>
|
|
</h2>
|
|
## Real Section
|
|
content"""
|
|
section = R.extract_relevant_section(readme, "Title")
|
|
assert "Real Section" in section # forward walk should not stop at </h2>
|
|
assert "content" in section
|
|
|
|
|
|
# ======================================================================
|
|
# _extract_frontmatter
|
|
# ======================================================================
|
|
|
|
|
|
class TestExtractFrontmatter:
|
|
def test_basic(self, R):
|
|
assert R._extract_frontmatter("---\ntags: [a]\n---\nbody") == "\ntags: [a]\n"
|
|
|
|
def test_no_frontmatter(self, R):
|
|
assert R._extract_frontmatter("no dashes") == ""
|
|
|
|
def test_empty_string(self, R):
|
|
assert R._extract_frontmatter("") == ""
|
|
|
|
|
|
# ======================================================================
|
|
# _strip_widget_section
|
|
# ======================================================================
|
|
|
|
|
|
class TestStripWidgetSection:
|
|
def test_strip_widget_keep_base_model(self, R):
|
|
"""Widget stripped but ``base_model:`` preserved."""
|
|
text = """---
|
|
tags: [test]
|
|
widget:
|
|
- text: >-
|
|
long text
|
|
output:
|
|
url: img.png
|
|
base_model: black-forest-labs/FLUX.1-dev
|
|
---"""
|
|
result = R._strip_widget_section(text)
|
|
assert "widget:" not in result
|
|
assert "black-forest-labs/FLUX.1-dev" in result
|
|
|
|
def test_no_widget_no_change(self, R):
|
|
text = "---\ntags: [a]\n---"
|
|
assert R._strip_widget_section(text) == text
|
|
|
|
def test_widget_at_end_of_frontmatter(self, R):
|
|
"""Widget is the last YAML key before closing ---."""
|
|
text = """---
|
|
base_model: a
|
|
widget:
|
|
- text: x
|
|
output:
|
|
url: y.png
|
|
---"""
|
|
result = R._strip_widget_section(text)
|
|
assert "widget:" not in result
|
|
assert "base_model: a" in result
|
|
|
|
|
|
# ======================================================================
|
|
# _strip_fenced_code_blocks
|
|
# ======================================================================
|
|
|
|
|
|
class TestStripFencedCodeBlocks:
|
|
def test_strips_with_language(self, R):
|
|
text = "a\n```python\ncode\n```\nb"
|
|
assert R._strip_fenced_code_blocks(text) == "a\nb"
|
|
|
|
def test_keeps_plain_fence(self, R):
|
|
"""`` ``` `` without language → preserved."""
|
|
text = "a\n```\ntrigger words\n```\nb"
|
|
assert "trigger words" in R._strip_fenced_code_blocks(text)
|
|
|
|
def test_pattern(self, R):
|
|
text = "x\n```yaml\nkey: val\n```\ny"
|
|
assert "key: val" not in R._strip_fenced_code_blocks(text)
|
|
|
|
|
|
# ======================================================================
|
|
# Site-generated placeholder cards
|
|
# ======================================================================
|
|
|
|
#: The card ModelScope renders when the uploader wrote no README. Copied from
|
|
#: a live repository so the marker strings stay honest.
|
|
PLACEHOLDER_CARD = """---
|
|
base_model: krea/Krea-2-Turbo
|
|
license: Apache License 2.0
|
|
tags:
|
|
- LoRA
|
|
- text-to-image
|
|
- \u68a6\u5e7b\u5149\u5f71
|
|
---
|
|
### \u5f53\u524d\u6a21\u578b\u7684\u8d21\u732e\u8005\u672a\u63d0\u4f9b\u66f4\u52a0\u8be6\u7ec6\u7684\u6a21\u578b\u4ecb\u7ecd\u3002\u6a21\u578b\u6587\u4ef6\u548c\u6743\u91cd\uff0c\u53ef\u6d4f\u89c8\u201c\u6a21\u578b\u6587\u4ef6\u201d\u9875\u9762\u83b7\u53d6\u3002
|
|
#### \u60a8\u53ef\u4ee5\u901a\u8fc7\u5982\u4e0bgit clone\u547d\u4ee4\uff0c\u6216\u8005ModelScope SDK\u6765\u4e0b\u8f7d\u6a21\u578b
|
|
|
|
SDK\u4e0b\u8f7d
|
|
```bash
|
|
#\u5b89\u88c5ModelScope
|
|
pip install modelscope
|
|
```
|
|
Git\u4e0b\u8f7d
|
|
```
|
|
#Git\u6a21\u578b\u4e0b\u8f7d
|
|
git clone https://www.modelscope.cn/yan303145427/krea2-CcFQWZ-Portrait.git
|
|
```
|
|
|
|
<p style="color: lightgrey;">\u5982\u679c\u60a8\u662f\u672c\u6a21\u578b\u7684\u8d21\u732e\u8005\uff0c\u6211\u4eec\u9080\u8bf7\u60a8\u6839\u636e<a href="x">\u6a21\u578b\u8d21\u732e\u6587\u6863</a>\uff0c\u53ca\u65f6\u5b8c\u5584\u6a21\u578b\u5361\u7247\u5185\u5bb9\u3002</p>
|
|
"""
|
|
|
|
#: A real, author-written card (ModelScope AIGC training output).
|
|
REAL_CARD = """---
|
|
base_model: krea/Krea-2-Turbo
|
|
---
|
|
# krea\u8138\u6a21
|
|
|
|
## \u6a21\u578b\u4ecb\u7ecd
|
|
|
|
\u672c\u6a21\u578b\u4f9d\u6258\u9b54\u642d\u793e\u533a\u5b8c\u6210\u8bad\u7ec3\u3002
|
|
|
|
## \u63a8\u7406\u4ee3\u7801
|
|
|
|
\u5b89\u88c5 DiffSynth-Studio\uff1a
|
|
"""
|
|
|
|
|
|
class TestStripGeneratedCardBoilerplate:
|
|
def test_removes_the_whole_placeholder_body(self, R):
|
|
stripped = R._strip_generated_card_boilerplate(PLACEHOLDER_CARD)
|
|
assert "\u5f53\u524d\u6a21\u578b\u7684\u8d21\u732e\u8005" not in stripped
|
|
assert "SDK\u4e0b\u8f7d" not in stripped
|
|
assert "git clone" not in stripped
|
|
assert "\u9080\u8bf7\u60a8" not in stripped
|
|
# The frontmatter is the only thing that survives.
|
|
assert "base_model: krea/Krea-2-Turbo" in stripped
|
|
|
|
def test_leaves_a_real_card_untouched(self, R):
|
|
assert R._strip_generated_card_boilerplate(REAL_CARD) == REAL_CARD
|
|
|
|
def test_drops_a_standalone_invitation_line(self, R):
|
|
text = "real body\n<p>\u5982\u679c\u60a8\u662f\u672c\u6a21\u578b\u7684\u8d21\u732e\u8005\uff0c\u8bf7\u5b8c\u5584</p>\nmore body"
|
|
stripped = R._strip_generated_card_boilerplate(text)
|
|
assert "real body" in stripped
|
|
assert "more body" in stripped
|
|
assert "\u8d21\u732e\u8005" not in stripped
|
|
|
|
def test_keeps_content_added_after_the_placeholder(self, R):
|
|
"""An author who later wrote a real section must not lose it."""
|
|
text = (
|
|
"### \u5f53\u524d\u6a21\u578b\u7684\u8d21\u732e\u8005\u672a\u63d0\u4f9b\u66f4\u52a0\u8be6\u7ec6\u7684\u6a21\u578b\u4ecb\u7ecd\u3002\n"
|
|
"#### \u60a8\u53ef\u4ee5\u901a\u8fc7\u5982\u4e0bgit clone\u547d\u4ee4\u4e0b\u8f7d\u6a21\u578b\n"
|
|
"```\ngit clone x\n```\n"
|
|
"## \u6211\u7684\u771f\u5b9e\u4ecb\u7ecd\n"
|
|
"\u8fd9\u662f\u4f5c\u8005\u540e\u6765\u8865\u5199\u7684\u5185\u5bb9\u3002\n"
|
|
)
|
|
stripped = R._strip_generated_card_boilerplate(text)
|
|
assert "\u6211\u7684\u771f\u5b9e\u4ecb\u7ecd" in stripped
|
|
assert "\u4f5c\u8005\u540e\u6765\u8865\u5199\u7684\u5185\u5bb9" in stripped
|
|
assert "git clone" not in stripped
|
|
|
|
def test_handles_html_headings(self, R):
|
|
text = (
|
|
"<h3>\u5f53\u524d\u6a21\u578b\u7684\u8d21\u732e\u8005\u672a\u63d0\u4f9b\u66f4\u52a0\u8be6\u7ec6\u7684\u6a21\u578b\u4ecb\u7ecd\u3002</h3>\n"
|
|
"<p>pip install modelscope</p>\n"
|
|
)
|
|
assert R._strip_generated_card_boilerplate(text).strip() == ""
|
|
|
|
|
|
class TestCleanReadmeForLlmPlaceholder:
|
|
def test_boilerplate_is_gone_but_frontmatter_survives(self, R):
|
|
cleaned = R.clean_readme_for_llm(PLACEHOLDER_CARD)
|
|
assert "pip install modelscope" not in cleaned
|
|
assert "git clone" not in cleaned
|
|
assert "\u5f53\u524d\u6a21\u578b\u7684\u8d21\u732e\u8005" not in cleaned
|
|
# Metadata the LLM still needs.
|
|
assert "base_model: krea/Krea-2-Turbo" in cleaned
|
|
assert "\u68a6\u5e7b\u5149\u5f71" in cleaned
|
|
|
|
def test_a_real_card_keeps_its_body(self, R):
|
|
cleaned = R.clean_readme_for_llm(REAL_CARD)
|
|
assert "krea\u8138\u6a21" in cleaned
|
|
assert "\u6a21\u578b\u4ecb\u7ecd" in cleaned
|
|
|
|
|
|
class TestConvertReadmeToHtmlPlaceholder:
|
|
def test_a_placeholder_card_renders_to_nothing(self, R):
|
|
assert R.convert_readme_to_html(PLACEHOLDER_CARD) == ""
|
|
|
|
def test_a_real_card_still_renders(self, R):
|
|
html = R.convert_readme_to_html(REAL_CARD)
|
|
assert "<h1>krea\u8138\u6a21</h1>" in html
|
|
assert "DiffSynth-Studio" in html
|