ApplyLobsterPromptInit.java 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. import java.nio.charset.StandardCharsets;
  2. import java.nio.file.Files;
  3. import java.nio.file.Path;
  4. import java.sql.Connection;
  5. import java.sql.DriverManager;
  6. import java.sql.ResultSet;
  7. import java.sql.Statement;
  8. /** Apply lobster_engine_prompt_init.sql (INSERT IGNORE). */
  9. public class ApplyLobsterPromptInit {
  10. private static final String HOST = "cq-cdb-8fjmemkb.sql.tencentcdb.com:27220";
  11. private static final String USER = "root";
  12. private static final String PASS = "Ylrz_1q2w3e4r5t6y";
  13. public static void main(String[] args) throws Exception {
  14. Class.forName("com.mysql.cj.jdbc.Driver");
  15. String db = args.length > 0 ? args[0] : "fs_tenant_cs1";
  16. String sql = Files.readString(Path.of("d:/ylrz_saas_new/java/sql/lobster_engine_prompt_init.sql"), StandardCharsets.UTF_8);
  17. if (sql.startsWith("")) sql = sql.substring(1);
  18. try (Connection c = DriverManager.getConnection(
  19. "jdbc:mysql://" + HOST + "/" + db + "?useSSL=false&serverTimezone=GMT%2B8&characterEncoding=utf8&allowMultiQueries=true",
  20. USER, PASS);
  21. Statement st = c.createStatement()) {
  22. st.execute(sql);
  23. try (ResultSet rs = st.executeQuery(
  24. "SELECT COUNT(*) FROM lobster_system_prompt WHERE company_id IS NULL")) {
  25. rs.next();
  26. System.out.println("platform prompts in " + db + ": " + rs.getInt(1));
  27. }
  28. try (ResultSet rs = st.executeQuery(
  29. "SELECT prompt_key FROM lobster_system_prompt WHERE company_id IS NULL AND prompt_key IN ('proactive_silent_greeting','tenant_product_claim_semantic','universal_deep_dialogue_render')")) {
  30. while (rs.next()) System.out.println(" ok: " + rs.getString(1));
  31. }
  32. }
  33. System.out.println("Done.");
  34. }
  35. }