#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十五 · 步骤 9.2：新开 Agent 对话 + messages 表验收。"""
from __future__ import annotations

import subprocess
import sys
from pathlib import Path

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


def main() -> int:
    chat_script = Path(__file__).resolve().parent / "专题十三_09_Agent验收对话.py"
    rc = subprocess.call([sys.executable, str(chat_script)])

    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        out, err = d.sh(
            """docker exec %s psql -U postgres -d dify -c "
SELECT id, status, message_tokens, answer_tokens,
       left(query, 40), left(coalesce(error,''), 120)
FROM messages ORDER BY created_at DESC LIMIT 5;
"
"""
            % C.PG_CONTAINER
        )
        print("\n=== messages ===")
        print(out or err)
    finally:
        d.close()
    return rc


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