#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题二 · 步骤 4B：REST 启动 bge-m3（model_uid 固定 bge-m3）。"""
from __future__ import annotations

import json
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:
        body = json.dumps({
            "model_uid": C.EMBED_MODEL,
            "model_name": C.EMBED_MODEL,
            "model_type": "embedding",
        })
        out, err = d.sh(
            "curl -s -X POST '%s/v1/models' -H 'Authorization: Bearer %s' "
            "-H 'Content-Type: application/json' -d '%s'"
            % (C.XF_URL, C.XF_KEY, body)
        )
        print("launch response:", out or err)

        print("\n=== 等待列表出现 bge-m3 ===")
        out2, err2 = d.sh(
            "curl -s -H 'Authorization: Bearer %s' %s/v1/models" % (C.XF_KEY, C.XF_URL)
        )
        print(out2 or err2)
        return 0 if C.EMBED_MODEL in (out2 or "") else 1
    finally:
        d.close()


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