# 提示词水管工式 "Agent"

if user_input == "写代码":
    result = llm("请帮我写代码")
elif user_input == "调试":
    result = llm("请帮我调试")
else:
    result = llm("我不知道你在说什么")

# 这是一个 if-else 外壳包着 LLM
# 不是 Agent，是有妄想的 shell 脚本

def agent_loop(messages: list):
    while True:
        response = client.messages.create(
            model=MODEL, system=SYSTEM,
            messages=messages,
            tools=TOOLS, max_tokens=8000,
        )
        messages.append(
          {"role":"assistant",
           "content": response.content})
        if response.stop_reason != "tool_use":
            return
        results = []
        for block in response.content:
            if block.type == "tool_use":
                output = run_bash(
                  block.input["command"])
                results.append({
                  "type":"tool_result",
                  "tool_use_id": block.id,
                  "content": output})
        messages.append(
          {"role":"user",
           "content": results})

TOOLS = [{
    "name": "bash",
    "description": "Run a shell command.",
    "input_schema": {
      "type": "object",
      "properties": {
        "command": {"type":"string"}
      },
      "required": ["command"],
    },
}]

TOOL_HANDLERS = {
    "bash":
      lambda **kw: run_bash(kw["command"]),
    "read_file":
      lambda **kw: run_read(kw["path"]),
    "write_file":
      lambda **kw: run_write(
        kw["path"], kw["content"]),
    "edit_file":
      lambda **kw: run_edit(
        kw["path"],kw["old_text"],
        kw["new_text"]),
}
# 循环中的调用方式：
handler = TOOL_HANDLERS.get(block.name)
output = handler(**block.input)

def run_subagent(prompt: str) -> str:
    sub_messages = [
      {"role":"user",
       "content": prompt}]
    for _ in range(30):
        response = client.messages.create(
          model=MODEL, system=SUBAGENT_SYSTEM,
          messages=sub_messages,
          tools=CHILD_TOOLS, max_tokens=8000)
        sub_messages.append(...)
        if response.stop_reason != "tool_use":
            break
        # ... execute tools, append results
    return "".join(
      b.text for b in response.content
      if hasattr(b, "text"))

class TaskManager:
    def create(self, subject, description=""):
        task = {
          "id": self._next_id,
          "subject": subject,
          "status": "pending",
          "blockedBy": [],
        }
        self._save(task)

    def _clear_dependency(self, completed_id):
        """完成后自动解除下游依赖"""
        for f in self.dir.glob("task_*.json"):
            task = json.loads(f.read_text())
            if completed_id in task["blockedBy"]:
                task["blockedBy"].remove(
                  completed_id)
                self._save(task)

class MessageBus:
    def send(self, sender, to, content,
             msg_type="message"):
        msg = {"type": msg_type,
               "from": sender,
               "content": content,
               "timestamp": time.time()}
        path = self.dir / f"{to}.jsonl"
        with open(path, "a") as f:
            f.write(json.dumps(msg) + "\n")

    def read_inbox(self, name):
        path = self.dir / f"{name}.jsonl"
        msgs = [json.loads(l)
          for l in path.read_text()
          .strip().splitlines()
          if l]
        path.write_text("") # drain
        return msgs

def create(self, name, task_id=None,
         base_ref="HEAD"):
    path = self.dir / name
    branch = f"wt/{name}"
    self._run_git(
      ["worktree", "add",
       "-b", branch,
       str(path), base_ref])

    entry = {"name": name,
             "path": str(path),
             "branch": branch,
             "task_id": task_id,
             "status": "active"}
    # 保存到 index.json
    idx = self._load_index()
    idx["worktrees"].append(entry)
    # 绑定任务
    if task_id:
      self.tasks.bind_worktree(
        task_id, name)