OpenAI 互換 - バッチ (ファイル入力)

/**
 * Model Studio バッチ API - ファイルのアップロード
 *
 * 環境変数が設定されていない場合は、apiKey: 'sk-xxx' を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * 依存関係のインストール：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 キーも合わせて更新してください。

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;

/**
 * Model Studio バッチ API - ファイルのアップロード
 *
 * 環境変数が設定されていない場合は、API_KEY = "sk-xxx" を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * リージョン設定：
 * - 北京リージョン：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - シンガポールリージョン：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：リージョンを切り替える際は、API キーも合わせて更新してください。
 */
public class BatchAPIUploadFile {

    // シンガポールリージョンの設定
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 北京リージョンを使用する場合は、private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"; を使用し、API キーを更新してください。

    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("File 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;
    }
}

/**
 * Model Studio バッチ API - バッチタスクの作成
 *
 * 環境変数が設定されていない場合は、apiKey: 'sk-xxx' を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * 依存関係のインストール：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 キーも合わせて更新してください。

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;

/**
 * Model Studio バッチ API - バッチタスクの作成
 *
 * 環境変数が設定されていない場合は、API_KEY = "sk-xxx" を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * リージョン設定：
 * - 北京リージョン：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - シンガポールリージョン：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：リージョンを切り替える際は、API キーも合わせて更新してください。
 */
public class BatchAPICreateBatch {

    // シンガポールリージョンの設定 (デフォルト)
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 北京リージョンを使用する場合は、private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"; を使用し、API キーを更新してください。

    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("バッチタスク 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;
    }
}

/**
 * Model Studio バッチ API - 結果ファイルのダウンロード
 *
 * 環境変数が設定されていない場合は、apiKey: 'sk-xxx' を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * 依存関係のインストール：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 キーも合わせて更新してください。

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;

/**
 * Model Studio バッチ API - 結果ファイルのダウンロード
 *
 * 環境変数が設定されていない場合は、API_KEY = "sk-xxx" を使用します (本番環境では漏洩のリスクがあるため非推奨)。
 * API キーはリージョンごとに異なります。
 *
 * リージョン設定：
 * - 北京リージョン：https://dashscope.aliyuncs.com/compatible-mode/v1
 * - シンガポールリージョン：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
 * 注意：リージョンを切り替える際は、API キーも合わせて更新してください。
 */
public class BatchAPIDownloadFile {

    // シンガポールリージョンの設定
    private static final String BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
    // 北京リージョンを使用する場合は、private static final String BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"; を使用し、API キーを更新してください。

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