#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题五 · 步骤 2：从 Dify 服务器验证到 vLLM 的网络可达性。"""
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()
        cmd = (
            "curl -s -m 10 %s/v1/models -H 'Authorization: Bearer %s' | head -c 800"
            % (C.VLLM_URL, C.VLLM_KEY)
        )
        out, err = d.sh(cmd, t=30)
        text = out or err
        print(text)
        if C.LLM_MODEL in text:
            print("\n[OK] 从 Dify 服务器可达 vLLM，且列出 %s" % C.LLM_MODEL)
            return 0
        print("\n[FAIL] 未在响应中看到 %s" % C.LLM_MODEL)
        return 1
    finally:
        d.close()


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