#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十五 · 步骤 4.1：确认知识库 available 且 embedding 为 bge-m3。"""
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()
        r = d.get("/datasets?page=1&limit=20")
        ok = False
        for ds in (r.get("json") or {}).get("data") or []:
            print(
                ds.get("id"), ds.get("name"), ds.get("indexing_status"),
                ds.get("embedding_model"), ds.get("embedding_model_provider"),
            )
            if ds.get("id") == C.DS_ID and ds.get("indexing_status") == "available":
                ok = True
        if not ok:
            print("[WARN] 目标库 %s 不可用或不在列表" % C.DS_ID)
            return 1
        return 0
    finally:
        d.close()


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