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;
    }
}