#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题三 · 步骤 1：列出 workspace 所有知识库及索引/embedding 状态。"""
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:
        ok, _, _ = d.login()
        print("login ok:", ok)
        r = d.get("/datasets")
        kb_list = (r.get("json") or {}).get("data", [])
        if not kb_list:
            print("当前 workspace 没有任何知识库")
            return 0
        for kb in kb_list:
            print("---")
            print("name           :", kb.get("name"))
            print("id             :", kb.get("id"))
            print("indexing_status:", kb.get("indexing_status"))
            print("embedding_model:", kb.get("embedding_model"))
            print("provider       :", kb.get("embedding_model_provider"))
        return 0
    finally:
        d.close()


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