Agent/.claude/skills/agent/invoke.py

#!/usr/bin/env python
"""
薄 CLI 脚本：透传命令行参数到 cyber-agent 的 invoke_agent() SDK。
远端 / 本地由 agent_type 前缀决定，同步返回 JSON 到 stdout。
"""

import argparse
import asyncio
import json
import sys

from agent import invoke_agent


def main() -> int:
    parser = argparse.ArgumentParser(description="调用 Agent（远端或本地）")
    parser.add_argument("--agent_type", required=True,
                        help="Agent 类型。'remote_' 前缀走 KnowHub 服务器；否则本地执行")
    parser.add_argument("--task", required=True, help="任务描述")
    parser.add_argument("--skills", default=None,
                        help="逗号分隔的 skill 列表（如 ask_strategy,upload_strategy）")
    parser.add_argument("--continue_from", default=None,
                        help="已有 sub_trace_id，传入则续跑")
    parser.add_argument("--messages", default=None,
                        help="预置消息（JSON 数组字符串，OpenAI 格式）")
    parser.add_argument("--project_root", default=None,
                        help="本地 agent 必填——项目目录（含 config.py）")
    args = parser.parse_args()

    skills = [s.strip() for s in args.skills.split(",") if s.strip()] if args.skills else None
    messages = json.loads(args.messages) if args.messages else None

    result = asyncio.run(invoke_agent(
        agent_type=args.agent_type,
        task=args.task,
        skills=skills,
        continue_from=args.continue_from,
        messages=messages,
        project_root=args.project_root,
    ))

    print(json.dumps(result, ensure_ascii=False, indent=2))
    return 0 if result.get("status") == "completed" else 1


if __name__ == "__main__":
    sys.exit(main())