Spaces:
Paused
Paused
| """skill-defaults.py <skills dir> <enabled skill>...: the image's skill defaults. The listed skills are on, | |
| every other one is off (`disable-model-invocation: true`: not offered to the model, still usable as | |
| /skill:<name>, and turned on in Pi Web's Settings > Skills). Pi Web keeps a user's own choices in the agent | |
| dir and puts them back on every new machine (lib/skill-choices.ts).""" | |
| import pathlib, re, sys | |
| root, enabled = pathlib.Path(sys.argv[1]), set(sys.argv[2:]) | |
| skills = sorted(root.glob("*/SKILL.md")) | |
| assert any(p.parent.name in enabled for p in skills), f"none of {sorted(enabled)} in {root}" | |
| for skill in skills: | |
| name, text = skill.parent.name, skill.read_text() | |
| head = re.match(r"\A---[ \t]*\n(.*?\n)---", text, re.S) | |
| assert head, f"{name}: no frontmatter" | |
| body = re.sub(r"(?m)^disable-model-invocation:.*\n", "", head.group(1)) | |
| if name not in enabled: | |
| body += "disable-model-invocation: true\n" | |
| skill.write_text(text[: head.start(1)] + body + text[head.end(1):]) | |
| print("on " if name in enabled else "off", name) | |