import java.sql.*;

public class ProbeAiModels {
    public static void main(String[] a) throws Exception {
        Class.forName("com.mysql.cj.jdbc.Driver");
        String jdbc = "jdbc:mysql://cq-cdb-8fjmemkb.sql.tencentcdb.com:27220/fs_tenant_cs1"
                + "?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8";
        try (Connection c = DriverManager.getConnection(jdbc, "root", "Ylrz_1q2w3e4r5t6y");
             Statement st = c.createStatement()) {
            System.out.println("=== admin_ai_model ===");
            ResultSet rs = st.executeQuery(
                    "SELECT id, model_name, provider_code, model_identifier, status, "
                            + "CASE WHEN api_key IS NULL OR api_key='' THEN 'EMPTY' ELSE CONCAT(LEFT(api_key,8),'...') END ak, "
                            + "LEFT(api_endpoint,60) ep FROM admin_ai_model ORDER BY id");
            while (rs.next()) {
                System.out.println(rs.getLong(1) + " | " + rs.getString(2) + " | " + rs.getString(3)
                        + " | " + rs.getString(4) + " | status=" + rs.getInt(5)
                        + " | key=" + rs.getString(6) + " | ep=" + rs.getString(7));
            }
            System.out.println("\n=== scenes (lobster related) ===");
            rs = st.executeQuery(
                    "SELECT sm.scene_code, sm.pipeline_order, sm.role, m.model_name, m.provider_code, m.status "
                            + "FROM admin_ai_scene_model sm JOIN admin_ai_model m ON sm.model_id=m.id "
                            + "WHERE sm.scene_code IN ('workflow_llm','workflow_evolution','semantic_analysis',"
                            + "'quality_scoring','deep_dialogue','lobster_reply') "
                            + "ORDER BY sm.scene_code, sm.pipeline_order");
            while (rs.next()) {
                System.out.println(rs.getString(1) + " #" + rs.getInt(2) + " " + rs.getString(3)
                        + " -> " + rs.getString(4) + " (" + rs.getString(5) + ") status=" + rs.getInt(6));
            }
            System.out.println("\n=== all scenes ===");
            rs = st.executeQuery(
                    "SELECT DISTINCT scene_code FROM admin_ai_scene_model ORDER BY scene_code");
            while (rs.next()) {
                System.out.println(rs.getString(1));
            }
        }
    }
}