#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题一 · 步骤 2：控制台 API 添加自定义 Qwen3.6-27B 凭据。"""
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()
        body = {
            "model": C.QWEN_MODEL,
            "model_type": "llm",
            "credentials": {
                "openai_api_key": C.VLLM_KEY,
                "openai_api_base": C.VLLM_BASE,
                "openai_organization": "",
                "api_protocol": "chat",
            },
            "name": C.QWEN_MODEL,
        }
        path = "/workspaces/current/model-providers/%s/models/credentials" % C.OPENAI_PROVIDER
        r = d.get(
            "/workspaces/current/model-providers/%s/models/credentials?model_type=llm&model=%s"
            % (C.OPENAI_PROVIDER, C.QWEN_MODEL)
        )
        cid = (r.get("json") or {}).get("current_credential_id")
        if cid:
            body["credential_id"] = cid
            r = d.req("PUT", path, body=body)
            pp("update qwen cred", r)
        else:
            r = d.post(path, body=body)
            pp("add qwen cred", r)
        r2 = d.get("/workspaces/current/models/model-types/llm")
        pp("llm list", r2)
        return 0 if r.get("code") in ("200", "201") else 1
    finally:
        d.close()


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