#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十一 · 步骤 4：验证快照与草稿 max_tokens 均为 2048。"""
from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _agent_db as A
import _config as C
from dify_api import Dify


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        A.verify_max_tokens(d, C.SNAP_ID, C.DRAFT_ID)
        snap = A.read_max_tokens(d, "agent_config_snapshots", C.SNAP_ID)
        draft = A.read_max_tokens(d, "agent_config_drafts", C.DRAFT_ID)
        ok = str(snap) == str(C.MAX_TOKENS_SMALL) and str(draft) == str(C.MAX_TOKENS_SMALL)
        print("\n[OK]" if ok else "\n[FAIL]", "期望均为", C.MAX_TOKENS_SMALL)
        return 0 if ok else 1
    finally:
        d.close()


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