| 123456789101112131415161718192021222324252627282930313233343536373839 |
- import java.nio.charset.StandardCharsets;
- import java.nio.file.Files;
- import java.nio.file.Path;
- import java.sql.Connection;
- import java.sql.DriverManager;
- import java.sql.ResultSet;
- import java.sql.Statement;
- /** Apply lobster_engine_prompt_init.sql (INSERT IGNORE). */
- public class ApplyLobsterPromptInit {
- private static final String HOST = "cq-cdb-8fjmemkb.sql.tencentcdb.com:27220";
- private static final String USER = "root";
- private static final String PASS = "Ylrz_1q2w3e4r5t6y";
- public static void main(String[] args) throws Exception {
- Class.forName("com.mysql.cj.jdbc.Driver");
- String db = args.length > 0 ? args[0] : "fs_tenant_cs1";
- String sql = Files.readString(Path.of("d:/ylrz_saas_new/java/sql/lobster_engine_prompt_init.sql"), StandardCharsets.UTF_8);
- if (sql.startsWith("")) sql = sql.substring(1);
- try (Connection c = DriverManager.getConnection(
- "jdbc:mysql://" + HOST + "/" + db + "?useSSL=false&serverTimezone=GMT%2B8&characterEncoding=utf8&allowMultiQueries=true",
- USER, PASS);
- Statement st = c.createStatement()) {
- st.execute(sql);
- try (ResultSet rs = st.executeQuery(
- "SELECT COUNT(*) FROM lobster_system_prompt WHERE company_id IS NULL")) {
- rs.next();
- System.out.println("platform prompts in " + db + ": " + rs.getInt(1));
- }
- try (ResultSet rs = st.executeQuery(
- "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')")) {
- while (rs.next()) System.out.println(" ok: " + rs.getString(1));
- }
- }
- System.out.println("Done.");
- }
- }
|