Agent Teams：多 Agent 协作

import json, threading, time, uuid
from pathlib import Path
from datetime import datetime

TEAM_DIR = Path(".team")
INBOX_DIR = TEAM_DIR / "inbox"
CONFIG_PATH = TEAM_DIR / "config.json"
TEAM_DIR.mkdir(exist_ok=True)
INBOX_DIR.mkdir(exist_ok=True)

class TeammateManager:
    def __init__(self):
        self._active_team: str | None = None   # 同 Claude Code：一个 leader 只管一个 team

    def create_team(self, team_name: str, description: str = "") -> str:
        """对应 TeamCreateTool.call()"""
        if self._active_team:
            return f"Already leading team '{self._active_team}'. Delete it first."

        config = self._load_config()
        if config.get("members"):
            # 名字冲突时自动加后缀（对应 generateUniqueTeamName）
            team_name = f"{team_name}-{uuid.uuid4().hex[:6]}"

        # 确定性 lead ID
        lead_id = f"team-lead@{team_name}"

        config = {
            "team_name": team_name,
            "description": description,
            "created_at": datetime.now().isoformat(),
            "lead_agent_id": lead_id,
            "members": {
                "team-lead": {
                    "agent_id": lead_id,
                    "status": "IDLE",
                    "system_prompt": "你是团队 leader，负责分解任务并分配给成员。",
                    "started_at": datetime.now().isoformat(),
                }
            }
        }
        self._save_config(config)
        self._active_team = team_name

        # 确保 leader 收件箱存在
        (INBOX_DIR / "team-lead.jsonl").touch()
        return f"Team '{team_name}' created. Lead ID: {lead_id}"

    def spawn(self, name: str, system_prompt: str) -> str:
        """启动一个新 Agent 成员"""
        config = self._load_config()
        if name in config.get("members", {}):
            return f"Agent '{name}' already exists"

        team_name = config.get("team_name", "default")
        agent_id = f"{name}@{team_name}"  # 确定性 ID

        config.setdefault("members", {})[name] = {
            "agent_id": agent_id,
            "status": "WORKING",
            "system_prompt": system_prompt,
            "started_at": datetime.now().isoformat(),
        }
        self._save_config(config)

        inbox_path = INBOX_DIR / f"{name}.jsonl"
        if not inbox_path.exists():
            inbox_path.touch()

        thread = threading.Thread(
            target=self._run_agent,
            args=(name, system_prompt),
            daemon=True
        )
        thread.start()
        return f"Spawned agent: {name} (id: {agent_id})"

    def send(self, to: str, message: str, sender: str = "user",
             summary: str | None = None) -> str:
        """发送消息到指定 Agent 的收件箱"""
        inbox = INBOX_DIR / f"{to}.jsonl"
        if not inbox.exists():
            return f"Agent '{to}' not found"
        entry = json.dumps({
            "from": sender,
            "message": message,
            "summary": summary,          # 同源码：summary 用于 UI 预览
            "timestamp": datetime.now().isoformat(),
            "read": False,
        })
        with open(inbox, "a") as f:
            f.write(entry + "\n")
        return f"Message sent to {to}"

    def read_inbox(self, agent_name: str) -> str:
        """读取并标记收件箱中的未读消息"""
        inbox = INBOX_DIR / f"{agent_name}.jsonl"
        if not inbox.exists():
            return "No messages."
        lines = inbox.read_text().splitlines()
        unread = []
        updated = []
        for line in lines:
            if not line.strip():
                continue
            msg = json.loads(line)
            if not msg.get("read"):
                unread.append(f"[{msg['from']}]: {msg['message']}")
                msg["read"] = True
            updated.append(json.dumps(msg))
        inbox.write_text("\n".join(updated) + "\n" if updated else "")
        return "\n".join(unread) if unread else "No new messages."

    def status(self) -> str:
        config = self._load_config()
        members = config.get("members", {})
        if not members:
            return "No team members."
        lines = []
        for name, info in members.items():
            lines.append(f"- {name}: {info['status']}")
        return "\n".join(lines)

    def _run_agent(self, name: str, system_prompt: str):
        while True:
            new_msgs = self.read_inbox(name)
            if new_msgs != "No new messages.":
                self._process_as_agent(name, system_prompt, new_msgs)
                self._set_status(name, "IDLE")
            time.sleep(5)

    def _process_as_agent(self, name: str, system_prompt: str, inbox_content: str):
        self._set_status(name, "WORKING")
        messages = [{
            "role": "user",
            "content": f"你是 {name}。你的收件箱有新消息：\n\n{inbox_content}"
        }]
        # ... 标准 agent_loop 逻辑 ...

    def _load_config(self) -> dict:
        if CONFIG_PATH.exists():
            return json.loads(CONFIG_PATH.read_text())
        return {"members": {}}

    def _save_config(self, config: dict):
        CONFIG_PATH.write_text(json.dumps(config, indent=2))

    def _set_status(self, name: str, status: str):
        config = self._load_config()
        if name in config.get("members", {}):
            config["members"][name]["status"] = status
            self._save_config(config)

team = TeammateManager()

TOOL_HANDLERS["create_team"] = lambda **kw: team.create_team(kw["team_name"], kw.get("description", ""))
TOOL_HANDLERS["spawn"]       = lambda **kw: team.spawn(kw["name"], kw["system_prompt"])
TOOL_HANDLERS["send"]        = lambda **kw: team.send(kw["to"], kw["message"],
                                                       kw.get("from", "leader"),
                                                       kw.get("summary"))
TOOL_HANDLERS["read_inbox"]  = lambda **kw: team.read_inbox(kw["agent_name"])