#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题五 · 步骤 4 · 修改后：把 tenant 默认 LLM 改为 qwen3.6-27b 并重启 api/worker/plugin。"""
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()
        tenant_sql = (
            "docker exec %s psql -U postgres -d dify -tc "
            "\"SELECT id FROM tenants WHERE email='%s'\"" % (C.PG_CONTAINER, C.DIFY_EMAIL)
        )
        out, err = d.sh(tenant_sql)
        tenant_id = out.strip()
        if not tenant_id:
            print("未找到 tenant:", err)
            return 1
        print("tenant_id:", tenant_id)
        upd = (
            "docker exec %s psql -U postgres -d dify -c "
            "\"UPDATE tenant_default_models SET provider_name='%s', model_name='%s' "
            "WHERE tenant_id='%s' AND model_type='llm';\"" % (
                C.PG_CONTAINER, C.OPENAI_PROVIDER, C.LLM_MODEL, tenant_id
            )
        )
        out2, err2 = d.sh(upd)
        print(out2 or err2)
        out3, err3 = d.sh(
            "cd %s && docker compose restart api worker plugin_daemon" % C.DIFY_DOCKER_DIR,
            t=180,
        )
        print(out3 or err3)
        return 0
    finally:
        d.close()


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