#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十五 · 步骤 0：创建前检查 LLM / Embedding / Rerank / 知识库 / 外部探针。"""
from __future__ import annotations

import sys

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


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        pp("llm", d.get("/workspaces/current/models/model-types/llm"))
        pp("emb", d.get("/workspaces/current/models/model-types/text-embedding"))
        pp("rerank", d.get("/workspaces/current/models/model-types/rerank"))
        pp("datasets", d.get("/datasets?page=1&limit=20"))

        print("\n=== vLLM ===")
        out, err = d.sh(
            "curl -s -m 10 -H 'Authorization: Bearer %s' %s/models"
            % (C.VLLM_KEY, C.VLLM_BASE)
        )
        print(out or err)

        print("\n=== Xinference ===")
        out2, err2 = d.sh(
            "curl -s -m 15 -H 'Authorization: Bearer %s' %s/v1/models"
            % (C.XF_KEY, C.XF_URL)
        )
        print(out2 or err2)
        return 0
    finally:
        d.close()


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