Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
| #!/usr/bin/env python3 | |
| """Fail CI when parser, sandbox guide, or public-skill inventory drifts.""" | |
| from __future__ import annotations | |
| import sys | |
| from pathlib import Path | |
| REPOSITORY = Path(__file__).resolve().parents[2] | |
| sys.path.insert(0, str(REPOSITORY / "standalone_cli" / "src")) | |
| from pwc_cli.cli import build_parser # noqa: E402 | |
| EDIT_COMMANDS = ("auth login", "auth status", "auth logout", "paper edit export", "paper edit preview", "paper edit submit") | |
| COMMANDS = ( | |
| "search", | |
| "paper info", | |
| "paper evaluations", | |
| "paper read", | |
| "paper list", | |
| "paper recent", | |
| "paper trending", | |
| "paper related", | |
| "paper lineage list", | |
| "task", | |
| "task list", | |
| "method", | |
| "method list", | |
| "conference", | |
| "conference list", | |
| "organization", | |
| "organization list", | |
| "framework", | |
| "framework list", | |
| "benchmark", | |
| "benchmark list", | |
| "skills add", | |
| "version", | |
| ) | |
| RESEARCH_COMMANDS = tuple(command for command in COMMANDS if command != "skills add") | |
| def parser_commands() -> set[str]: | |
| parser = build_parser() | |
| discovered: set[str] = set() | |
| def visit(current, prefix: str = ""): | |
| for action in current._actions: | |
| choices = getattr(action, "choices", None) | |
| if not isinstance(choices, dict): | |
| continue | |
| for name, child in choices.items(): | |
| command = f"{prefix} {name}".strip() | |
| nested = any( | |
| isinstance(getattr(item, "choices", None), dict) | |
| for item in child._actions | |
| ) | |
| if nested: | |
| if "handler" in child._defaults: | |
| discovered.add(command) | |
| visit(child, command) | |
| else: | |
| discovered.add(command) | |
| visit(parser) | |
| return discovered | |
| def main() -> int: | |
| expected = set(COMMANDS) | set(EDIT_COMMANDS) | |
| if parser_commands() != expected: | |
| raise SystemExit("standalone parser command inventory drifted") | |
| skill = (REPOSITORY / "standalone_cli" / "SKILL.md").read_text() | |
| for command in EDIT_COMMANDS: | |
| if f"pwc {command}" not in skill: | |
| raise SystemExit(f"missing edit command: {command}") | |
| for command in RESEARCH_COMMANDS: | |
| if command == "version": | |
| continue | |
| invocation = f"pwc {command}" | |
| if invocation not in skill: | |
| raise SystemExit(f"missing read-only inventory entry: {invocation}") | |
| return 0 | |
| if __name__ == "__main__": | |
| raise SystemExit(main()) | |