#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十三 · 步骤 9：新开 Agent 对话验收（修改后）。"""
from __future__ import annotations

import sys

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


QUERY = "法定假日加班的制度是什么，加班费怎么算？"


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        payload = {
            "inputs": {},
            "query": QUERY,
            "response_mode": "streaming",
            "files": [],
            "conversation_id": "",
        }
        r = d.post(
            "/agent/%s/chat-messages" % C.AGENT_ID,
            body=payload,
            extra="-m 420 --max-time 420",
            t=450,
        )
        print("HTTP", r.get("code"))
        text = r.get("text") or ""
        print(text[-1500:] if len(text) > 1500 else text)
        if r.get("stderr"):
            print("[stderr]", r["stderr"][:400])
        return 0 if r.get("code") == "200" else 1
    finally:
        d.close()


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