#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""作业 5 · 步骤 3：创建 vLLM 健康检查 Agent 空壳。"""
from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify, pp


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        r_list = d.get("/agent?page=1&limit=50")
        data = (r_list.get("json") or {}).get("data") or []
        for item in data:
            if item.get("name") == C.HOMEWORK5_APP_NAME:
                print(
                    "already exists app_id=",
                    item.get("id"),
                    "agent_id=",
                    item.get("agent_id") or item.get("id"),
                )
                print("下一步：运行 作业5_04_查ID并挂工具.py")
                return 0

        body = {
            "name": C.HOMEWORK5_APP_NAME,
            "description": "本周实战作业：AIOps 最小闭环 · 检查 vLLM 容器 / GPU / 排队",
            "icon_type": "emoji",
            "icon": "🩺",
            "icon_background": "#E0F2FE",
            "role": "ops",
        }
        r = d.post("/agent", body)
        pp("create agent", r)
        j = r.get("json") or {}
        print("app_id", j.get("id"))
        print("agent_id", j.get("agent_id") or j.get("id"))
        if r.get("code") not in ("200", "201"):
            return 1
        print("下一步：运行 作业5_04_查ID并挂工具.py")
        return 0
    finally:
        d.close()


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