#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""作业 4 · 步骤 1：写入熔断阈值，并把 Dify 单次生成 max_tokens 设为会话轮次上限。"""
from __future__ import annotations

import json
import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify
from 作业4_成本熔断 import (
    apply_max_tokens,
    backup_snapshots,
    cfg_thresholds,
    save_json,
)


def main() -> int:
    th = cfg_thresholds()
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        bak = backup_snapshots(d, th["snap_id"], th["draft_id"])
        print("backup:", bak)
        apply_max_tokens(d, th["snap_id"], th["draft_id"], th["session_max_tokens"])
        save_json("thresholds.json", th)
        print(json.dumps(th, ensure_ascii=False, indent=2))
        print("\n[PASS] 已配置：Dify max_tokens=%d；会话累计上限=%d；日预算=%d"
              % (th["session_max_tokens"], th["session_token_cap"], th["daily_token_budget"]))
        print("下一步：/usr/bin/python3 /opt/workbuddy/scripts_cursor/作业4_02_验收成本熔断.py")
        return 0
    finally:
        d.close()


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