ProbeAiModels.java 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. import java.sql.*;
  2. public class ProbeAiModels {
  3. public static void main(String[] a) throws Exception {
  4. Class.forName("com.mysql.cj.jdbc.Driver");
  5. String jdbc = "jdbc:mysql://cq-cdb-8fjmemkb.sql.tencentcdb.com:27220/fs_tenant_cs1"
  6. + "?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8";
  7. try (Connection c = DriverManager.getConnection(jdbc, "root", "Ylrz_1q2w3e4r5t6y");
  8. Statement st = c.createStatement()) {
  9. System.out.println("=== admin_ai_model ===");
  10. ResultSet rs = st.executeQuery(
  11. "SELECT id, model_name, provider_code, model_identifier, status, "
  12. + "CASE WHEN api_key IS NULL OR api_key='' THEN 'EMPTY' ELSE CONCAT(LEFT(api_key,8),'...') END ak, "
  13. + "LEFT(api_endpoint,60) ep FROM admin_ai_model ORDER BY id");
  14. while (rs.next()) {
  15. System.out.println(rs.getLong(1) + " | " + rs.getString(2) + " | " + rs.getString(3)
  16. + " | " + rs.getString(4) + " | status=" + rs.getInt(5)
  17. + " | key=" + rs.getString(6) + " | ep=" + rs.getString(7));
  18. }
  19. System.out.println("\n=== scenes (lobster related) ===");
  20. rs = st.executeQuery(
  21. "SELECT sm.scene_code, sm.pipeline_order, sm.role, m.model_name, m.provider_code, m.status "
  22. + "FROM admin_ai_scene_model sm JOIN admin_ai_model m ON sm.model_id=m.id "
  23. + "WHERE sm.scene_code IN ('workflow_llm','workflow_evolution','semantic_analysis',"
  24. + "'quality_scoring','deep_dialogue','lobster_reply') "
  25. + "ORDER BY sm.scene_code, sm.pipeline_order");
  26. while (rs.next()) {
  27. System.out.println(rs.getString(1) + " #" + rs.getInt(2) + " " + rs.getString(3)
  28. + " -> " + rs.getString(4) + " (" + rs.getString(5) + ") status=" + rs.getInt(6));
  29. }
  30. System.out.println("\n=== all scenes ===");
  31. rs = st.executeQuery(
  32. "SELECT DISTINCT scene_code FROM admin_ai_scene_model ORDER BY scene_code");
  33. while (rs.next()) {
  34. System.out.println(rs.getString(1));
  35. }
  36. }
  37. }
  38. }