77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
||
"""苍耳技能中心：技能注册表 + 提示词加载 + 工具执行
|
||
用法：python3 skill_center.py [list|load <id>|run <id> [args...]]"""
|
||
import json, os, subprocess, sys
|
||
|
||
SKILL_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
INDEX = os.path.join(SKILL_DIR, "index.json")
|
||
|
||
|
||
def _index():
|
||
with open(INDEX, encoding="utf-8") as f:
|
||
return json.load(f)
|
||
|
||
|
||
def list_skills():
|
||
return _index().get("skills", [])
|
||
|
||
|
||
def load(skill_id):
|
||
"""返回技能定义 + 提示词全文（prompt_file 存在时）。"""
|
||
for s in list_skills():
|
||
if s["id"] == skill_id:
|
||
data = dict(s)
|
||
pf = data.get("prompt_file")
|
||
if pf:
|
||
p = os.path.normpath(os.path.join(SKILL_DIR, pf))
|
||
try:
|
||
with open(p, encoding="utf-8") as f:
|
||
data["prompt"] = f.read()
|
||
except Exception as e:
|
||
data["prompt"] = ""
|
||
data["prompt_error"] = str(e)
|
||
return data
|
||
return None
|
||
|
||
|
||
def run(skill_id, args=None):
|
||
"""执行工具型技能，返回 (ok, output)。"""
|
||
s = load(skill_id)
|
||
if not s:
|
||
return False, "技能不存在: " + skill_id
|
||
if s.get("type") != "tool":
|
||
return False, f"技能「{s.get('name','')}」是提示词型，请注入提示词使用"
|
||
tool = s.get("tool")
|
||
if not tool:
|
||
return False, "技能无执行命令"
|
||
cmd = tool.split() + (args or [])
|
||
try:
|
||
r = subprocess.run(cmd, capture_output=True, text=True, timeout=600)
|
||
return r.returncode == 0, (r.stdout or r.stderr)[-3000:]
|
||
except subprocess.TimeoutExpired:
|
||
return False, "执行超时（10分钟）"
|
||
except Exception as e:
|
||
return False, str(e)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) < 2:
|
||
print(json.dumps(list_skills(), ensure_ascii=False, indent=2)); sys.exit(0)
|
||
cmd = sys.argv[1]
|
||
if cmd == "list":
|
||
print(json.dumps({"skills": list_skills()}, ensure_ascii=False, indent=2)); sys.exit(0)
|
||
elif cmd == "load":
|
||
s = load(sys.argv[2]) if len(sys.argv) > 2 else None
|
||
if s:
|
||
print(json.dumps({"name": s.get("name"), "type": s.get("type"),
|
||
"prompt_len": len(s.get("prompt", "")),
|
||
"tool": s.get("tool")}, ensure_ascii=False))
|
||
else:
|
||
print("技能不存在"); sys.exit(1)
|
||
elif cmd == "run":
|
||
ok, out = run(sys.argv[2], sys.argv[3:])
|
||
print(("OK\n" if ok else "FAIL\n") + out)
|
||
sys.exit(0 if ok else 1)
|
||
else:
|
||
print("未知命令: " + cmd); sys.exit(1)
|