#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十四 · 步骤 6：新开 Agent 对话验收（conversation_id 留空）。"""
from __future__ import annotations

import sys

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


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        payload = {
            "inputs": {},
            "query": C.PROBE_QUERY_KB,
            "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)
        return 0 if r.get("code") == "200" else 1
    finally:
        d.close()


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