mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-09 14:38:09 +08:00
b18833f912
* feat(agent): scripted generation of agent skill docs with operation/field notes Generate skills/moviepilot-api API docs from scripts/generate_agent_skill_docs.py instead of hand-editing: - surface canonical MediaType agent keys in the API contract - add subscribeactivity table note and global operation notes - preserve upstream search.torrents execution semantics via field notes Co-Authored-By: pavel2 <pavelyuan468@gmail.com> * fix(agent): preserve generated skill contracts and English guidance --------- Co-authored-by: jxxghp <jxxghp@gmail.com>
85 lines
3.8 KiB
Python
85 lines
3.8 KiB
Python
"""验证技能文档生成器保留操作契约并输出可重复的英文文档。"""
|
|
|
|
import re
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from scripts import generate_agent_skill_docs as generator
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def api_categories() -> dict[str, str]:
|
|
"""从真实 MCP 元数据生成分类文档，供契约断言复用。"""
|
|
categories, _, _ = generator._split_api_docs(generator._render_api_docs())
|
|
return categories
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("category", "operation", "field"),
|
|
[
|
|
("search", "search.torrents", "mtype"),
|
|
("site", "site.searchable", "media_type*"),
|
|
("site", "site.resource", "mtype"),
|
|
("filter", "filter.group.add", "media_type"),
|
|
("subtitle", "subtitle.search.media", "mtype"),
|
|
],
|
|
)
|
|
def test_agent_media_keys_are_scoped_to_supported_operations(
|
|
api_categories: dict[str, str], category: str, operation: str, field: str
|
|
) -> None:
|
|
"""仅在接口接受 Agent 媒体键时显示英文取值。"""
|
|
section = api_categories[category].split(f"### `{operation}`\n", 1)[1].split("\n### `", 1)[0]
|
|
assert f"`{field}` (string(movie,tv,music)" in section
|
|
|
|
|
|
def test_distinct_media_type_contracts_are_preserved(api_categories: dict[str, str]) -> None:
|
|
"""推荐的 all、订阅枚举和数据库过滤字段不得被同名字段覆盖。"""
|
|
assert "`media_type` (string(all,movie,tv,music); default `all`)" in api_categories["recommendation"]
|
|
assert "`mtype` (MediaType|null)" in api_categories["subscription"]
|
|
assert "`mtype*` (string):" in api_categories["subscription"]
|
|
assert "`mtype` (string|null):" in api_categories["library"]
|
|
assert "Agent-facing values" not in api_categories["subscription"]
|
|
assert "Agent-facing values" not in api_categories["library"]
|
|
|
|
|
|
def test_regeneration_retains_operation_guidance(api_categories: dict[str, str]) -> None:
|
|
"""生成器必须保留无法从原始 schema 得出的业务和超时说明。"""
|
|
assert "Music uses exact identity only." in api_categories["subscription"]
|
|
assert "paused subscription without enabling it" in api_categories["subscription"]
|
|
assert "Omitted fields keep their stored values." in api_categories["subscription"]
|
|
assert "MUSIC_METADATA_TO_SIMPLIFIED" in api_categories["music"]
|
|
assert "290 seconds" in api_categories["search"]
|
|
assert "300 seconds" in api_categories["search"]
|
|
assert "600 seconds by default" in api_categories["search"]
|
|
assert "context containing `media_info`, not a candidate list" in api_categories["media"]
|
|
assert "unavailable source does not prove there are no resources" in api_categories["search"]
|
|
|
|
|
|
def test_full_regeneration_is_english_current_and_idempotent(
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
"""在临时目录重建全部技能，验证已提交产物同步且重复执行无变化。"""
|
|
source_root = generator.PROJECT_ROOT
|
|
shutil.copytree(source_root / "skills", tmp_path / "skills")
|
|
schema_path = Path("app/agent/policy/resources/api_mcp_schema.json")
|
|
(tmp_path / schema_path).parent.mkdir(parents=True)
|
|
shutil.copyfile(source_root / schema_path, tmp_path / schema_path)
|
|
monkeypatch.setattr(generator, "PROJECT_ROOT", tmp_path)
|
|
expected = {
|
|
path.relative_to(tmp_path): path.read_bytes()
|
|
for path in (tmp_path / "skills").rglob("*.md")
|
|
}
|
|
|
|
for _ in range(2):
|
|
assert generator.main() == 0
|
|
actual = {
|
|
path.relative_to(tmp_path): path.read_bytes()
|
|
for path in (tmp_path / "skills").rglob("*.md")
|
|
}
|
|
assert actual == expected
|
|
for skill in ("moviepilot-api", "downloader-operation", "mediaserver-operation", "database-operation"):
|
|
for path in (tmp_path / "skills" / skill).rglob("*.md"):
|
|
assert not re.search(r"[\u3400-\u9fff]", path.read_text(encoding="utf-8")), path
|