ai-functions 2.1.3 → 2.3.0

package/dist/batch/bedrock.js

@@ -1,17 +1,19 @@
 /**
  * AWS Bedrock Batch Inference Adapter
  *
- * Implements batch processing using AWS Bedrock's batch inference API.
- * Bedrock batch inference provides cost-effective processing for large workloads.
+ * Bedrock has a true batch inference API (S3-driven) and a runtime invoke API.
+ * The "batch" adapter here uses concurrent runtime invocations as a fallback
+ * (no S3 setup required); `createBedrockBatchJob` is exported separately for
+ * callers who want to drive the real S3-based batch flow directly.
  *
  * @see https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference.html
  *
  * @packageDocumentation
  */
-import { registerBatchAdapter, registerFlexAdapter, } from '../batch-queue.js';
-import { schema as convertSchema } from '../schema.js';
+import { getLogger } from '../logger.js';
+import { LocalJobStore, processConcurrently, registerBatchAdapter, registerFlexAdapter, tryParseJson, } from './provider.js';
 // ============================================================================
-// AWS Configuration
+// AWS configuration
 // ============================================================================
 let awsRegion;
 let awsAccessKeyId;
@@ -19,12 +21,9 @@ let awsSecretAccessKey;
 let awsSessionToken;
 let s3Bucket;
 let roleArn;
-// AI Gateway configuration (optional - for routing through Cloudflare AI Gateway)
 let gatewayUrl;
 let gatewayToken;
-/**
- * Configure AWS credentials and settings
- */
+/** Configure AWS credentials and settings. */
 export function configureAWSBedrock(options) {
     if (options.region)
         awsRegion = options.region;
@@ -44,16 +43,14 @@ export function configureAWSBedrock(options) {
         gatewayToken = options.gatewayToken;
 }
 function getConfig() {
-    const region = awsRegion || process.env.AWS_REGION || process.env.AWS_DEFAULT_REGION || 'us-east-1';
-    const accessKeyId = awsAccessKeyId || process.env.AWS_ACCESS_KEY_ID;
-    const secretAccessKey = awsSecretAccessKey || process.env.AWS_SECRET_ACCESS_KEY;
-    const sessionToken = awsSessionToken || process.env.AWS_SESSION_TOKEN;
-    const bucket = s3Bucket || process.env.BEDROCK_BATCH_S3_BUCKET;
-    const role = roleArn || process.env.BEDROCK_BATCH_ROLE_ARN;
-    // Check for AI Gateway configuration
-    const gwUrl = gatewayUrl || process.env.AI_GATEWAY_URL;
-    const gwToken = gatewayToken || process.env.AI_GATEWAY_TOKEN;
-    // If using gateway, we don't need AWS credentials
+    const region = awsRegion || process.env['AWS_REGION'] || process.env['AWS_DEFAULT_REGION'] || 'us-east-1';
+    const accessKeyId = awsAccessKeyId || process.env['AWS_ACCESS_KEY_ID'];
+    const secretAccessKey = awsSecretAccessKey || process.env['AWS_SECRET_ACCESS_KEY'];
+    const sessionToken = awsSessionToken || process.env['AWS_SESSION_TOKEN'];
+    const bucket = s3Bucket || process.env['BEDROCK_BATCH_S3_BUCKET'];
+    const role = roleArn || process.env['BEDROCK_BATCH_ROLE_ARN'];
+    const gwUrl = gatewayUrl || process.env['AI_GATEWAY_URL'];
+    const gwToken = gatewayToken || process.env['AI_GATEWAY_TOKEN'];
     if (gwUrl && gwToken) {
         return {
             region,
@@ -72,14 +69,21 @@ function getConfig() {
     if (!bucket) {
         throw new Error('S3 bucket for Bedrock batch not configured. Set BEDROCK_BATCH_S3_BUCKET');
     }
-    return { region, accessKeyId, secretAccessKey, sessionToken, bucket, role, gatewayUrl: undefined, gatewayToken: undefined };
+    return {
+        region,
+        accessKeyId,
+        secretAccessKey,
+        sessionToken,
+        bucket,
+        role,
+        gatewayUrl: undefined,
+        gatewayToken: undefined,
+    };
 }
 // ============================================================================
-// AWS Signature V4 (Simplified)
+// AWS SigV4 (delegated to optional @smithy/signature-v4 if available)
 // ============================================================================
 async function signRequest(method, url, body, config, service) {
-    // In production, use @aws-sdk/signature-v4 or similar
-    // This is a simplified implementation for demonstration
     const headers = new Headers({
         'Content-Type': 'application/json',
         'X-Amz-Date': new Date().toISOString().replace(/[:-]|\.\d{3}/g, ''),
@@ -87,18 +91,13 @@ async function signRequest(method, url, body, config, service) {
     if (config.sessionToken) {
         headers.set('X-Amz-Security-Token', config.sessionToken);
     }
-    // For actual implementation, compute proper AWS Signature V4
-    // This requires crypto operations that vary by environment
-    // Fallback: Use AWS SDK if available
     try {
-        // Dynamic import to avoid build-time dependency
+        // Optional dependency — present in production, absent in dev/test.
         // @ts-expect-error - Optional dependency
         const signatureV4Module = await import('@smithy/signature-v4');
         // @ts-expect-error - Optional dependency
         const sha256Module = await import('@aws-crypto/sha256-js');
-        const SignatureV4 = signatureV4Module.SignatureV4;
-        const Sha256 = sha256Module.Sha256;
-        const signer = new SignatureV4({
+        const signer = new signatureV4Module.SignatureV4({
             service,
             region: config.region,
             credentials: {
@@ -106,7 +105,7 @@ async function signRequest(method, url, body, config, service) {
                 secretAccessKey: config.secretAccessKey,
                 sessionToken: config.sessionToken,
             },
-            sha256: Sha256,
+            sha256: sha256Module.Sha256,
         });
         const signedRequest = await signer.sign({
             method,
@@ -118,185 +117,140 @@ async function signRequest(method, url, body, config, service) {
         return new Headers(signedRequest.headers);
     }
     catch {
-        // AWS SDK not available - return basic headers
-        // In production, the SDK should always be available
-        console.warn('AWS SDK not available for request signing. Install @smithy/signature-v4 and @aws-crypto/sha256-js');
+        getLogger().warn('AWS SDK not available for request signing. Install @smithy/signature-v4 and @aws-crypto/sha256-js');
         return headers;
     }
 }
 // ============================================================================
-// In-memory job tracking
+// Local job tracking
 // ============================================================================
-const pendingJobs = new Map();
-let jobCounter = 0;
+const jobs = new LocalJobStore('bedrock_batch');
 // ============================================================================
-// Bedrock Batch Adapter
+// Bedrock batch adapter (BatchProvider port)
 // ============================================================================
-/**
- * AWS Bedrock batch adapter
- *
- * Bedrock batch inference:
- * 1. Uploads input JSONL to S3
- * 2. Creates a batch inference job
- * 3. Results are written to S3
- * 4. Download and parse results
- *
- * Note: This requires S3 bucket access and proper IAM roles.
- */
 const bedrockAdapter = {
     async submit(items, options) {
         const config = getConfig();
-        const jobId = `bedrock_batch_${++jobCounter}_${Date.now()}`;
-        // Default to Claude on Bedrock
         const model = options.model || 'anthropic.claude-3-sonnet-20240229-v1:0';
-        // Store job state
-        pendingJobs.set(jobId, {
-            items,
-            options,
-            results: [],
-            status: 'pending',
-            createdAt: new Date(),
-        });
-        // For true Bedrock batch processing:
-        // 1. Create JSONL file with requests
-        // 2. Upload to S3
-        // 3. Create batch inference job via Bedrock API
-        // 4. Poll for completion
-        // 5. Download and parse results from S3
-        // For now, we implement a concurrent processing approach
-        // (similar to Cloudflare) that works without S3 setup
-        const completion = processBedrockRequestsConcurrently(jobId, items, config, model, options);
+        const { id, state } = jobs.create(items, options);
+        // Drive the job state machine in the background.
+        const completion = (async () => {
+            state.status = 'in_progress';
+            const results = await processConcurrently(items, (item) => processBedrockItem(item, config, model), {
+                concurrency: 5, // Bedrock has stricter rate limits.
+                delayBetweenWaves: 1000,
+                onWaveComplete: (partial) => {
+                    state.results = partial;
+                },
+            });
+            state.results = results;
+            state.status = results.every((r) => r.status === 'completed') ? 'completed' : 'failed';
+            state.completedAt = new Date();
+            return results;
+        })();
         const job = {
-            id: jobId,
+            id,
             provider: 'bedrock',
             status: 'pending',
             totalItems: items.length,
             completedItems: 0,
             failedItems: 0,
-            createdAt: new Date(),
-            webhookUrl: options.webhookUrl,
+            createdAt: state.createdAt,
+            ...(options.webhookUrl !== undefined && { webhookUrl: options.webhookUrl }),
         };
         return { job, completion };
     },
     async getStatus(batchId) {
-        const job = pendingJobs.get(batchId);
-        if (!job) {
-            throw new Error(`Batch not found: ${batchId}`);
-        }
-        const completedItems = job.results.filter((r) => r.status === 'completed').length;
-        const failedItems = job.results.filter((r) => r.status === 'failed').length;
-        return {
-            id: batchId,
-            provider: 'bedrock',
-            status: job.status,
-            totalItems: job.items.length,
-            completedItems,
-            failedItems,
-            createdAt: job.createdAt,
-            completedAt: job.completedAt,
-        };
+        return jobs.snapshot(batchId, 'bedrock');
     },
     async cancel(batchId) {
-        const job = pendingJobs.get(batchId);
-        if (job) {
-            job.status = 'cancelled';
-            // If we have a Bedrock job ARN, cancel it
-            if (job.jobArn) {
-                const config = getConfig();
-                const url = `https://bedrock.${config.region}.amazonaws.com/model-invocation-job/${encodeURIComponent(job.jobArn)}/stop`;
-                try {
-                    await fetch(url, {
-                        method: 'POST',
-                        headers: await signRequest('POST', url, '', config, 'bedrock'),
-                    });
-                }
-                catch (error) {
-                    console.warn('Failed to cancel Bedrock job:', error);
-                }
+        if (!jobs.has(batchId))
+            return;
+        const state = jobs.get(batchId);
+        state.status = 'cancelled';
+        const jobArn = state.meta?.['jobArn'];
+        if (jobArn) {
+            const config = getConfig();
+            const url = `https://bedrock.${config.region}.amazonaws.com/model-invocation-job/${encodeURIComponent(jobArn)}/stop`;
+            try {
+                await fetch(url, {
+                    method: 'POST',
+                    headers: await signRequest('POST', url, '', config, 'bedrock'),
+                });
+            }
+            catch (error) {
+                getLogger().warn('Failed to cancel Bedrock job:', error);
             }
         }
     },
     async getResults(batchId) {
-        const job = pendingJobs.get(batchId);
-        if (!job) {
-            throw new Error(`Batch not found: ${batchId}`);
-        }
-        return job.results;
+        return jobs.get(batchId).results;
     },
     async waitForCompletion(batchId, pollInterval = 5000) {
-        const job = pendingJobs.get(batchId);
-        if (!job) {
-            throw new Error(`Batch not found: ${batchId}`);
-        }
-        while (job.status !== 'completed' && job.status !== 'failed' && job.status !== 'cancelled') {
-            await new Promise((resolve) => setTimeout(resolve, pollInterval));
-        }
-        return job.results;
+        return jobs.waitForCompletion(batchId, pollInterval);
     },
 };
 // ============================================================================
-// Processing (Concurrent Mode)
+// Per-item processing
 // ============================================================================
-/**
- * Process Bedrock requests concurrently
- * This is a fallback when true batch inference isn't configured
- */
-async function processBedrockRequestsConcurrently(jobId, items, config, model, options) {
-    const job = pendingJobs.get(jobId);
-    if (!job) {
-        throw new Error(`Job not found: ${jobId}`);
-    }
-    job.status = 'in_progress';
-    // Process with concurrency limit
-    const CONCURRENCY = 5; // Bedrock has stricter rate limits
-    const results = [];
-    for (let i = 0; i < items.length; i += CONCURRENCY) {
-        const batch = items.slice(i, i + CONCURRENCY);
-        const batchResults = await Promise.all(batch.map(async (item) => {
-            try {
-                return await processBedrockItem(item, config, model);
-            }
-            catch (error) {
-                return {
-                    id: item.id,
-                    customId: item.id,
-                    status: 'failed',
-                    error: error instanceof Error ? error.message : 'Unknown error',
-                };
-            }
-        }));
-        results.push(...batchResults);
-        job.results = results;
-        // Respect rate limits
-        if (i + CONCURRENCY < items.length) {
-            await new Promise((resolve) => setTimeout(resolve, 1000));
-        }
-    }
-    job.status = results.every((r) => r.status === 'completed') ? 'completed' : 'failed';
-    job.completedAt = new Date();
-    return results;
-}
 async function processBedrockItem(item, config, model) {
-    // Check if using AI Gateway
     if (config.gatewayUrl && config.gatewayToken) {
         return processBedrockItemViaGateway(item, config, model);
     }
     const url = `https://bedrock-runtime.${config.region}.amazonaws.com/model/${encodeURIComponent(model)}/invoke`;
-    // Build the request body based on the model type
-    let body;
+    const body = buildBedrockRequestBody(item, model);
+    const bodyStr = JSON.stringify(body);
+    const headers = await signRequest('POST', url, bodyStr, config, 'bedrock');
+    const response = await fetch(url, { method: 'POST', headers, body: bodyStr });
+    if (!response.ok) {
+        const error = await response.text();
+        throw new Error(`Bedrock API error: ${response.status} ${error}`);
+    }
+    return parseBedrockResponse(item, await response.json());
+}
+/**
+ * Process a Bedrock item via Cloudflare AI Gateway.
+ *
+ * Note: AI Gateway routes the request but doesn't handle authentication —
+ * Bedrock still requires AWS SigV4 signing.
+ * @see https://developers.cloudflare.com/ai-gateway/usage/providers/bedrock/
+ */
+async function processBedrockItemViaGateway(item, config, model) {
+    const url = `${config.gatewayUrl}/aws-bedrock/bedrock-runtime/${config.region}/model/${encodeURIComponent(model)}/invoke`;
+    const body = {
+        anthropic_version: 'bedrock-2023-05-31',
+        max_tokens: item.options?.maxTokens || 4096,
+        messages: [{ role: 'user', content: item.prompt }],
+        ...(item.options?.system !== undefined && { system: item.options.system }),
+        ...(item.options?.temperature !== undefined && { temperature: item.options.temperature }),
+    };
+    const bodyStr = JSON.stringify(body);
+    if (!config.accessKeyId || !config.secretAccessKey) {
+        throw new Error('Bedrock via AI Gateway still requires AWS credentials for SigV4 signing. ' +
+            'Set AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.');
+    }
+    const headers = await signRequest('POST', url, bodyStr, config, 'bedrock');
+    headers.set('cf-aig-authorization', `Bearer ${config.gatewayToken}`);
+    const response = await fetch(url, { method: 'POST', headers, body: bodyStr });
+    if (!response.ok) {
+        const error = await response.text();
+        throw new Error(`Bedrock via Gateway error: ${response.status} ${error}`);
+    }
+    return parseBedrockResponse(item, await response.json());
+}
+/** Build the Bedrock invoke body for the model family. */
+function buildBedrockRequestBody(item, model) {
     if (model.includes('anthropic')) {
-        // Anthropic models on Bedrock
-        body = {
+        return {
             anthropic_version: 'bedrock-2023-05-31',
             max_tokens: item.options?.maxTokens || 4096,
             messages: [{ role: 'user', content: item.prompt }],
-            system: item.options?.system,
-            temperature: item.options?.temperature,
+            ...(item.options?.system !== undefined && { system: item.options.system }),
+            ...(item.options?.temperature !== undefined && { temperature: item.options.temperature }),
         };
     }
-    else if (model.includes('amazon')) {
-        // Amazon Titan models
-        body = {
+    if (model.includes('amazon')) {
+        return {
             inputText: item.prompt,
             textGenerationConfig: {
                 maxTokenCount: item.options?.maxTokens || 4096,
@@ -304,50 +258,35 @@ async function processBedrockItem(item, config, model) {
             },
         };
     }
-    else if (model.includes('meta')) {
-        // Meta Llama models
-        body = {
+    if (model.includes('meta')) {
+        return {
             prompt: item.prompt,
             max_gen_len: item.options?.maxTokens || 4096,
             temperature: item.options?.temperature || 0.7,
         };
     }
-    else if (model.includes('mistral')) {
-        // Mistral models
-        body = {
+    if (model.includes('mistral')) {
+        return {
             prompt: `<s>[INST] ${item.prompt} [/INST]`,
             max_tokens: item.options?.maxTokens || 4096,
             temperature: item.options?.temperature || 0.7,
         };
     }
-    else {
-        // Generic format (Claude-style)
-        body = {
-            anthropic_version: 'bedrock-2023-05-31',
-            max_tokens: item.options?.maxTokens || 4096,
-            messages: [{ role: 'user', content: item.prompt }],
-            temperature: item.options?.temperature,
-        };
-    }
-    const bodyStr = JSON.stringify(body);
-    const headers = await signRequest('POST', url, bodyStr, config, 'bedrock');
-    const response = await fetch(url, {
-        method: 'POST',
-        headers,
-        body: bodyStr,
-    });
-    if (!response.ok) {
-        const error = await response.text();
-        throw new Error(`Bedrock API error: ${response.status} ${error}`);
-    }
-    const data = await response.json();
-    // Extract content based on model response format
+    // Default: Claude-style.
+    return {
+        anthropic_version: 'bedrock-2023-05-31',
+        max_tokens: item.options?.maxTokens || 4096,
+        messages: [{ role: 'user', content: item.prompt }],
+        ...(item.options?.temperature !== undefined && { temperature: item.options.temperature }),
+    };
+}
+/** Parse a Bedrock invoke response across model families. */
+function parseBedrockResponse(item, raw) {
+    const data = raw;
     let content;
     let usage;
     if (data.content) {
-        // Anthropic format
-        const textContent = data.content.find((c) => c.type === 'text');
-        content = textContent?.text;
+        content = data.content.find((c) => c.type === 'text')?.text;
         if (data.usage) {
             usage = {
                 promptTokens: data.usage.input_tokens,
@@ -357,16 +296,14 @@ async function processBedrockItem(item, config, model) {
         }
     }
     else if (data.results?.[0]) {
-        // Titan format
         content = data.results[0].outputText;
         usage = {
-            promptTokens: 0, // Titan doesn't return this
+            promptTokens: 0,
             completionTokens: data.results[0].tokenCount || 0,
             totalTokens: data.results[0].tokenCount || 0,
         };
     }
     else if (data.generation) {
-        // Llama/Mistral format
         content = data.generation;
         if (data.generation_token_count !== undefined) {
             usage = {
@@ -376,103 +313,23 @@ async function processBedrockItem(item, config, model) {
             };
         }
     }
-    let result = content;
-    // Try to parse JSON if schema was provided
-    if (item.schema && content) {
-        try {
-            result = JSON.parse(content);
-        }
-        catch {
-            // Keep as string
-        }
-    }
     return {
         id: item.id,
         customId: item.id,
         status: 'completed',
-        result,
-        usage,
-    };
-}
-/**
- * Process a Bedrock item via Cloudflare AI Gateway
- *
- * NOTE: Unlike OpenAI and Google, Bedrock via AI Gateway still requires AWS Signature V4 signing.
- * The gateway routes the request but doesn't handle authentication.
- * @see https://developers.cloudflare.com/ai-gateway/usage/providers/bedrock/
- *
- * Gateway URL format: {gateway_url}/aws-bedrock/bedrock-runtime/{region}/model/{model}/invoke
- */
-async function processBedrockItemViaGateway(item, config, model) {
-    // AI Gateway URL for Bedrock - requires full path including region
-    // Format: {gateway_url}/aws-bedrock/bedrock-runtime/{region}/model/{model}/invoke
-    const url = `${config.gatewayUrl}/aws-bedrock/bedrock-runtime/${config.region}/model/${encodeURIComponent(model)}/invoke`;
-    // Build the request body (Anthropic format for Claude models)
-    const body = {
-        anthropic_version: 'bedrock-2023-05-31',
-        max_tokens: item.options?.maxTokens || 4096,
-        messages: [{ role: 'user', content: item.prompt }],
-        system: item.options?.system,
-        temperature: item.options?.temperature,
-    };
-    const bodyStr = JSON.stringify(body);
-    // NOTE: Bedrock via Gateway still requires AWS SigV4 signing
-    // We need both the gateway token AND AWS credentials
-    if (!config.accessKeyId || !config.secretAccessKey) {
-        throw new Error('Bedrock via AI Gateway still requires AWS credentials for SigV4 signing. ' +
-            'Set AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.');
-    }
-    const headers = await signRequest('POST', url, bodyStr, config, 'bedrock');
-    headers.set('cf-aig-authorization', `Bearer ${config.gatewayToken}`);
-    const response = await fetch(url, {
-        method: 'POST',
-        headers,
-        body: bodyStr,
-    });
-    if (!response.ok) {
-        const error = await response.text();
-        throw new Error(`Bedrock via Gateway error: ${response.status} ${error}`);
-    }
-    const data = await response.json();
-    // Extract content (Anthropic format)
-    const textContent = data.content?.find((c) => c.type === 'text');
-    let content = textContent?.text;
-    let usage;
-    if (data.usage) {
-        usage = {
-            promptTokens: data.usage.input_tokens,
-            completionTokens: data.usage.output_tokens,
-            totalTokens: data.usage.input_tokens + data.usage.output_tokens,
-        };
-    }
-    let result = content;
-    // Try to parse JSON if schema was provided
-    if (item.schema && content) {
-        try {
-            result = JSON.parse(content);
-        }
-        catch {
-            // Keep as string
-        }
-    }
-    return {
-        id: item.id,
-        customId: item.id,
-        status: 'completed',
-        result,
-        usage,
+        result: tryParseJson(content, !!item.schema),
+        ...(usage && { usage }),
     };
 }
 // ============================================================================
-// True Batch Inference (S3-based)
+// True S3-based batch inference (separate from the BatchProvider adapter)
 // ============================================================================
 /**
- * Create and submit a true Bedrock batch inference job
- * This requires S3 bucket access and proper IAM setup
+ * Create and submit a true Bedrock batch inference job.
+ * Requires S3 bucket access and proper IAM setup.
  */
 export async function createBedrockBatchJob(items, model, options) {
     const config = getConfig();
-    // Build JSONL content
     const jsonlLines = items.map((item) => {
         const request = {
             recordId: item.id,
@@ -480,16 +337,16 @@ export async function createBedrockBatchJob(items, model, options) {
                 anthropic_version: 'bedrock-2023-05-31',
                 max_tokens: item.options?.maxTokens || 4096,
                 messages: [{ role: 'user', content: item.prompt }],
-                system: item.options?.system,
-                temperature: item.options?.temperature,
+                ...(item.options?.system !== undefined && { system: item.options.system }),
+                ...(item.options?.temperature !== undefined && {
+                    temperature: item.options.temperature,
+                }),
             },
         };
         return JSON.stringify(request);
     });
     const inputKey = `${options.s3InputPrefix || 'bedrock-batch/input'}/${options.jobName}.jsonl`;
     const outputPrefix = `${options.s3OutputPrefix || 'bedrock-batch/output'}/${options.jobName}/`;
-    // Upload to S3
-    // In production, use @aws-sdk/client-s3
     const s3Url = `https://${config.bucket}.s3.${config.region}.amazonaws.com/${inputKey}`;
     const content = jsonlLines.join('\n');
     const s3Response = await fetch(s3Url, {
@@ -500,21 +357,16 @@ export async function createBedrockBatchJob(items, model, options) {
     if (!s3Response.ok) {
         throw new Error(`Failed to upload to S3: ${s3Response.status}`);
     }
-    // Create batch inference job
     const jobUrl = `https://bedrock.${config.region}.amazonaws.com/model-invocation-job`;
     const jobBody = JSON.stringify({
         jobName: options.jobName,
         modelId: model,
         roleArn: options.roleArn,
         inputDataConfig: {
-            s3InputDataConfig: {
-                s3Uri: `s3://${config.bucket}/${inputKey}`,
-            },
+            s3InputDataConfig: { s3Uri: `s3://${config.bucket}/${inputKey}` },
         },
         outputDataConfig: {
-            s3OutputDataConfig: {
-                s3Uri: `s3://${config.bucket}/${outputPrefix}`,
-            },
+            s3OutputDataConfig: { s3Uri: `s3://${config.bucket}/${outputPrefix}` },
         },
     });
     const jobResponse = await fetch(jobUrl, {
@@ -526,59 +378,24 @@ export async function createBedrockBatchJob(items, model, options) {
         const error = await jobResponse.text();
         throw new Error(`Failed to create Bedrock batch job: ${jobResponse.status} ${error}`);
     }
-    const jobData = await jobResponse.json();
+    const jobData = (await jobResponse.json());
     return jobData;
 }
 // ============================================================================
-// Register Adapter
-// ============================================================================
+// Bedrock flex adapter (FlexAdapter port)
 // ============================================================================
-// Bedrock Flex Adapter
-// ============================================================================
-/**
- * AWS Bedrock Flex Adapter
- *
- * Flex processing uses concurrent requests for medium-sized batches (5-500 items).
- * This provides a balance between:
- * - Immediate execution (fast but full price, <5 items)
- * - Full batch inference (50% discount but 24hr turnaround, 500+ items)
- *
- * Flex tier uses concurrent API calls with rate limiting, providing results
- * in minutes rather than hours while still benefiting from efficient processing.
- */
 const bedrockFlexAdapter = {
     async submitFlex(items, options) {
         const config = getConfig();
         const model = options.model || 'anthropic.claude-3-sonnet-20240229-v1:0';
-        const CONCURRENCY = 8; // Bedrock has stricter rate limits than OpenAI
-        const results = [];
-        // Process items concurrently with rate limiting
-        for (let i = 0; i < items.length; i += CONCURRENCY) {
-            const batch = items.slice(i, i + CONCURRENCY);
-            const batchResults = await Promise.all(batch.map(async (item) => {
-                try {
-                    return await processBedrockItem(item, config, model);
-                }
-                catch (error) {
-                    return {
-                        id: item.id,
-                        customId: item.id,
-                        status: 'failed',
-                        error: error instanceof Error ? error.message : 'Unknown error',
-                    };
-                }
-            }));
-            results.push(...batchResults);
-            // Add delay between batches to respect rate limits
-            if (i + CONCURRENCY < items.length) {
-                await new Promise((resolve) => setTimeout(resolve, 500));
-            }
-        }
-        return results;
+        return processConcurrently(items, (item) => processBedrockItem(item, config, model), {
+            concurrency: 8,
+            delayBetweenWaves: 500,
+        });
     },
 };
 // ============================================================================
-// Register Adapters
+// Register adapters
 // ============================================================================
 registerBatchAdapter('bedrock', bedrockAdapter);
 registerFlexAdapter('bedrock', bedrockFlexAdapter);