#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题二 · 步骤 6：设 bge-m3 为默认 Embedding。"""
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()
        r = d.post(
            "/workspaces/current/default-model",
            body={
                "model_settings": [{
                    "model_type": "text-embedding",
                    "provider": C.XINFERENCE_PROVIDER,
                    "model": C.EMBED_MODEL,
                }]
            },
        )
        pp("set default embedding", r)
        r2 = d.get("/workspaces/current/models/model-types/text-embedding")
        pp("embedding list", r2)
        return 0 if r.get("code") in ("200", "201") else 1
    finally:
        d.close()


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