OpenAI相容-Batch（檔案輸入）

/**
 * 阿里雲百鍊 Batch API - 上傳檔案
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：apiKey: 'sk-xxx'
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 安裝依賴：npm install openai
 */
const OpenAI = require('openai');
const fs = require('fs');

// 新加坡地區配置（預設）
const BASE_URL = 'https://dashscope-intl.aliyuncs.com/compatible-mode/v1';
// 如果使用北京地區，請將上面的 BASE_URL 替換為：
// const BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1';
// 注意：切換地區時，API Key也需要對應更換

const apiKey = process.env.DASHSCOPE_API_KEY;
if (!apiKey) {
    console.error('錯誤: 請設定環境變數 DASHSCOPE_API_KEY');
    console.error('或在代碼中設定: const apiKey = "sk-xxx";');
    process.exit(1);
}

const client = new OpenAI({
    apiKey: apiKey,
    baseURL: BASE_URL
});

const fileStream = fs.createReadStream('test.jsonl');
const fileObject = await client.files.create({
    file: fileStream,
    purpose: 'batch'
});
console.log(fileObject.id);

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Scanner;
import java.util.regex.Pattern;
import java.util.regex.Matcher;

/**
 * 阿里雲百鍊 Batch API - 上傳檔案
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：API_KEY = "sk-xxx"
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 地區配置：
 * - 北京地區：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - 新加坡地區：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：切換地區時，API Key也需要對應更換
 */
public class BatchAPIUploadFile {

    // 新加坡地區配置
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 如果使用北京地區，請將上面的 BASE_URL 替換為：
    // private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1";
    // 注意：切換地區時，API Key也需要對應更換

    private static String API_KEY;

    public static void main(String[] args) throws Exception {
        API_KEY = System.getenv("DASHSCOPE_API_KEY");
        if (API_KEY == null || API_KEY.isEmpty()) {
            System.err.println("錯誤: 請設定環境變數 DASHSCOPE_API_KEY");
            System.err.println("或在代碼中設定: API_KEY = \"sk-xxx\";");
            System.exit(1);
        }

String fileId = uploadFile("test.jsonl");
        System.out.println("檔案ID: " + fileId);
    }

    // === 工具方法 ===

    private static String uploadFile(String filePath) throws Exception {
        String boundary = "----WebKitFormBoundary" + System.currentTimeMillis();
        URL url = new URL(BASE_URL + "/files");
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setDoOutput(true);
        conn.setRequestMethod("POST");
        conn.setRequestProperty("Authorization", "Bearer " + API_KEY);
        conn.setRequestProperty("Content-Type", "multipart/form-data; boundary=" + boundary);

        try (DataOutputStream out = new DataOutputStream(conn.getOutputStream())) {
            // 添加 purpose 欄位
            out.writeBytes("--" + boundary + "\r\n");
            out.writeBytes("Content-Disposition: form-data; name=\"purpose\"\r\n\r\n");
            out.writeBytes("batch\r\n");

            // 添加檔案
            out.writeBytes("--" + boundary + "\r\n");
            out.writeBytes("Content-Disposition: form-data; name=\"file\"; filename=\"" + filePath + "\"\r\n");
            out.writeBytes("Content-Type: application/octet-stream\r\n\r\n");

            byte[] fileBytes = Files.readAllBytes(Paths.get(filePath));
            out.write(fileBytes);
            out.writeBytes("\r\n");
            out.writeBytes("--" + boundary + "--\r\n");
        }

        String response = readResponse(conn);
        return parseField(response, "\"id\":\\s*\"([^\"]+)\"");
    }

    private static String readResponse(HttpURLConnection conn) throws Exception {
        int responseCode = conn.getResponseCode();
        InputStream is = (responseCode < 400) ? conn.getInputStream() : conn.getErrorStream();
        try (Scanner scanner = new Scanner(is, "UTF-8").useDelimiter("\\A")) {
            return scanner.hasNext() ? scanner.next() : "";
        }
    }

    private static String parseField(String json, String regex) {
        Pattern pattern = Pattern.compile(regex);
        Matcher matcher = pattern.matcher(json);
        return matcher.find() ? matcher.group(1) : null;
    }
}

/**
 * 阿里雲百鍊 Batch API - 建立Batch任務
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：apiKey: 'sk-xxx'
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 安裝依賴：npm install openai
 */
const OpenAI = require('openai');

// 新加坡地區配置（預設）
const BASE_URL = 'https://dashscope-intl.aliyuncs.com/compatible-mode/v1';
// 如果使用北京地區，請將上面的 BASE_URL 替換為：
// const BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1';
// 注意：切換地區時，API Key也需要對應更換

const apiKey = process.env.DASHSCOPE_API_KEY;
if (!apiKey) {
    console.error('錯誤: 請設定環境變數 DASHSCOPE_API_KEY');
    console.error('或在代碼中設定: const apiKey = "sk-xxx";');
    process.exit(1);
}

const client = new OpenAI({
    apiKey: apiKey,
    baseURL: BASE_URL
});

const batch = await client.batches.create({
    input_file_id: 'file-batch-xxx',
    endpoint: '/v1/chat/completions',
    completion_window: '24h',
    metadata: {'ds_name': '任務名稱', 'ds_description': '任務描述'}
});
console.log(batch.id);

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Scanner;
import java.util.regex.Pattern;
import java.util.regex.Matcher;

/**
 * 阿里雲百鍊 Batch API - 建立Batch任務
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：API_KEY = "sk-xxx"
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 地區配置：
 * - 北京地區：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - 新加坡地區：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：切換地區時，API Key也需要對應更換
 */
public class BatchAPICreateBatch {

    // 新加坡地區配置（預設）
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 如果使用北京地區，請將上面的 BASE_URL 替換為：
    // private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1";
    // 注意：切換地區時，API Key也需要對應更換

    private static String API_KEY;

    public static void main(String[] args) throws Exception {
        API_KEY = System.getenv("DASHSCOPE_API_KEY");
        if (API_KEY == null || API_KEY.isEmpty()) {
            System.err.println("錯誤: 請設定環境變數 DASHSCOPE_API_KEY");
            System.err.println("或在代碼中設定: API_KEY = \"sk-xxx\";");
            System.exit(1);
        }

        String jsonBody = "{\"input_file_id\":\"file-batch-xxx\",\"endpoint\":\"/v1/chat/completions\",\"completion_window\":\"24h\",\"metadata\":{\"ds_name\":\"任務名稱\",\"ds_description\":\"任務描述\"}}";
String response = sendRequest("POST", "/batches", jsonBody);
        String batchId = parseField(response, "\"id\":\\s*\"([^\"]+)\"");
        System.out.println("Batch任務ID: " + batchId);
    }

    // === 工具方法 ===

    private static String sendRequest(String method, String path, String jsonBody) throws Exception {
        URL url = new URL(BASE_URL + path);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setRequestMethod(method);
        conn.setRequestProperty("Authorization", "Bearer " + API_KEY);

        if (jsonBody != null) {
            conn.setDoOutput(true);
            conn.setRequestProperty("Content-Type", "application/json");
            try (OutputStream os = conn.getOutputStream()) {
                os.write(jsonBody.getBytes("UTF-8"));
            }
        }

        return readResponse(conn);
    }

    private static String readResponse(HttpURLConnection conn) throws Exception {
        int responseCode = conn.getResponseCode();
        InputStream is = (responseCode < 400) ? conn.getInputStream() : conn.getErrorStream();
        try (Scanner scanner = new Scanner(is, "UTF-8").useDelimiter("\\A")) {
            return scanner.hasNext() ? scanner.next() : "";
        }
    }

    private static String parseField(String json, String regex) {
        Pattern pattern = Pattern.compile(regex);
        Matcher matcher = pattern.matcher(json);
        return matcher.find() ? matcher.group(1) : null;
    }
}

/**
 * 阿里雲百鍊 Batch API - 下載結果檔案
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：apiKey: 'sk-xxx'
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 安裝依賴：npm install openai
 */
const OpenAI = require('openai');
const fs = require('fs');

// 新加坡地區配置（預設）
const BASE_URL = 'https://dashscope-intl.aliyuncs.com/compatible-mode/v1';
// 如果使用北京地區，請將上面的 BASE_URL 替換為：
// const BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1';
// 注意：切換地區時，API Key也需要對應更換

const apiKey = process.env.DASHSCOPE_API_KEY;
if (!apiKey) {
    console.error('錯誤: 請設定環境變數 DASHSCOPE_API_KEY');
    console.error('或在代碼中設定: const apiKey = "sk-xxx";');
    process.exit(1);
}

const client = new OpenAI({
    apiKey: apiKey,
    baseURL: BASE_URL
});

// 下載結果檔案
const content = await client.files.content('file-batch_output-xxx');
const text = await content.text();
console.log(text);

// 儲存到本地檔案
fs.writeFileSync('result.jsonl', text);
console.log('結果已儲存到 result.jsonl');

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Scanner;
import java.util.regex.Pattern;
import java.util.regex.Matcher;

/**
 * 阿里雲百鍊 Batch API - 下載結果檔案
 *
 * 若沒有配置環境變數，可在代碼中寫入程式碼API Key：API_KEY = "sk-xxx"
 * 但不建議在生產環境中直接將API Key寫入程式碼到代碼中，以減少API Key泄露風險。
 * 新加坡和北京地區的API Key不同。
 *
 * 地區配置：
 * - 北京地區：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - 新加坡地區：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：切換地區時，API Key也需要對應更換
 */
public class BatchAPIDownloadFile {

    // 新加坡地區配置
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 如果使用北京地區，請將上面的 BASE_URL 替換為：
    // private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1";
    // 注意：切換地區時，API Key也需要對應更換

    private static String API_KEY;

    public static void main(String[] args) throws Exception {
        API_KEY = System.getenv("DASHSCOPE_API_KEY");
        if (API_KEY == null || API_KEY.isEmpty()) {
            System.err.println("錯誤: 請設定環境變數 DASHSCOPE_API_KEY");
            System.err.println("或在代碼中設定: API_KEY = \"sk-xxx\";");
            System.exit(1);
        }

// 下載結果檔案
String content = sendRequest("GET", "/files/file-batch_output-xxx/content", null);
System.out.println(content);

// 儲存到本地檔案
        Files.write(Paths.get("result.jsonl"), content.getBytes());
        System.out.println("結果已儲存到 result.jsonl");
    }

    // === 工具方法 ===

    private static String sendRequest(String method, String path, String jsonBody) throws Exception {
        URL url = new URL(BASE_URL + path);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setRequestMethod(method);
        conn.setRequestProperty("Authorization", "Bearer " + API_KEY);

        if (jsonBody != null) {
            conn.setDoOutput(true);
            conn.setRequestProperty("Content-Type", "application/json");
            try (OutputStream os = conn.getOutputStream()) {
                os.write(jsonBody.getBytes("UTF-8"));
            }
        }

        return readResponse(conn);
    }

    private static String readResponse(HttpURLConnection conn) throws Exception {
        int responseCode = conn.getResponseCode();
        InputStream is = (responseCode < 400) ? conn.getInputStream() : conn.getErrorStream();
        try (Scanner scanner = new Scanner(is, "UTF-8").useDelimiter("\\A")) {
            return scanner.hasNext() ? scanner.next() : "";
        }
    }

    private static String parseField(String json, String regex) {
        Pattern pattern = Pattern.compile(regex);
        Matcher matcher = pattern.matcher(json);
        return matcher.find() ? matcher.group(1) : null;
    }
}