#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题五 · 步骤 3：查询 tenant_default_models 当前默认 LLM。"""
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()
        sql = (
            "docker exec %s psql -U postgres -d dify -c "
            "\"SELECT model_type, provider_name, model_name FROM tenant_default_models;\"" % C.PG_CONTAINER
        )
        out, err = d.sh(sql)
        print(out or err)
        return 0
    finally:
        d.close()


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