import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;


import com.google.gson.Gson;


public class Main {


    static class Message {
        String role;
        String content;


        public Message(String role, String content) {
            this.role = role;
            this.content = content;
        }
    }


    static class RequestBody {
        String model;
        Message[] messages;


        public RequestBody(String model, Message[] messages) {
            this.model = model;
            this.messages = messages;
        }
    }


    public static void main(String[] args) {
        try {
            // 創建請求體
            RequestBody requestBody = new RequestBody(
                    "qwen-plus",
                    new Message[] {
                            new Message("system", "You are a helpful assistant."),
                            new Message("user", "你是誰？")
                    }
            );


            // 將請求體轉換為 JSON
            Gson gson = new Gson();
            String jsonInputString = gson.toJson(requestBody);


            // 創建 URL 對象
            URL url = new URL("https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions");
            HttpURLConnection httpURLConnection = (HttpURLConnection) url.openConnection();


            // 設置請求方法為 POST
            httpURLConnection.setRequestMethod("POST");
            httpURLConnection.setRequestProperty("Content-Type", "application/json; utf-8");
            httpURLConnection.setRequestProperty("Accept", "application/json");


            // 若沒有配置環境變量，請用百煉API Key將下行替換為：String apiKey = "sk-xxx";
            String apiKey = System.getenv("DASHSCOPE_API_KEY");
            String auth = "Bearer " + apiKey;
            httpURLConnection.setRequestProperty("Authorization", auth);


            // 啟用輸入輸出流
            httpURLConnection.setDoOutput(true);


            // 寫入請求體
            try (OutputStream os = httpURLConnection.getOutputStream()) {
                byte[] input = jsonInputString.getBytes(StandardCharsets.UTF_8);
                os.write(input, 0, input.length);
            }


            // 獲取響應碼
            int responseCode = httpURLConnection.getResponseCode();
            System.out.println("Response Code: " + responseCode);


            // 讀取響應體
            try (BufferedReader br = new BufferedReader(new InputStreamReader(httpURLConnection.getInputStream(), StandardCharsets.UTF_8))) {
                StringBuilder response = new StringBuilder();
                String responseLine;
                while ((responseLine = br.readLine()) != null) {
                    response.append(responseLine.trim());
                }
                System.out.println("Response Body: " + response);
            }


        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            System.exit(0);
        }
    }
}