#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""作业 7 · 步骤 3：创建 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()
        for item in (d.get("/agent?page=1&limit=50").get("json") or {}).get("data") or []:
            if item.get("name") == C.HOMEWORK7_APP_NAME:
                print("already exists", item.get("id"), item.get("agent_id"))
                return 0
        r = d.post(
            "/agent",
            body={
                "name": C.HOMEWORK7_APP_NAME,
                "description": "作业7 加分项：vLLM OOM 检测 → docker restart → 钉钉恢复通知",
                "icon_type": "emoji",
                "icon": "🚑",
                "icon_background": "#FEE2E2",
                "role": "ops",
            },
        )
        pp("create agent", r)
        return 0 if r.get("code") in ("200", "201") else 1
    finally:
        d.close()


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