import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.sql.*;
import java.time.Duration;

public class ProbeExactModelConfig {
    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();
             ResultSet rs = st.executeQuery(
                     "SELECT model_name, model_identifier, api_key, api_endpoint, max_tokens, temperature "
                             + "FROM admin_ai_model WHERE provider_code='doubao' AND status=1 LIMIT 1")) {
            rs.next();
            String ep = rs.getString("api_endpoint");
            String key = rs.getString("api_key");
            String modelId = rs.getString("model_identifier");
            int maxTokens = rs.getInt("max_tokens");
            double temp = rs.getDouble("temperature");
            System.out.println("ep=" + ep);
            System.out.println("modelId=" + modelId);
            System.out.println("maxTokens=" + maxTokens + " temp=" + temp);
            String body = "{\"model\":\"" + modelId + "\",\"max_tokens\":" + maxTokens
                    + ",\"temperature\":" + temp
                    + ",\"messages\":[{\"role\":\"system\",\"content\":\"\u4f60\u662f\u767d\u9152\u9500\u552e\u987e\u95ee\"},"
                    + "{\"role\":\"user\",\"content\":\"\u4f60\u597d\uff0c\u60f3\u4e86\u89e3\u4e00\u4e0b\u5497\u4eec\u767d\u9152\u600e\u4e48\u5356\uff1f\"}]}";
            HttpClient client = HttpClient.newBuilder().connectTimeout(Duration.ofSeconds(15)).build();
            HttpRequest req = HttpRequest.newBuilder().uri(URI.create(ep)).timeout(Duration.ofSeconds(60))
                    .header("Authorization", "Bearer " + key)
                    .header("Content-Type", "application/json")
                    .POST(HttpRequest.BodyPublishers.ofString(body, StandardCharsets.UTF_8)).build();
            HttpResponse<String> resp = client.send(req, HttpResponse.BodyHandlers.ofString(StandardCharsets.UTF_8));
            System.out.println("HTTP " + resp.statusCode());
            System.out.println(resp.body());
        }
    }
}