@aws/ml-container-creator 0.2.4 → 0.2.6

package/src/lib/mcp-validator-config.js

@@ -0,0 +1,89 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * MCP Validator Configuration Reader
+ *
+ * Reads smart-mode validator configuration from config/mcp.json.
+ * When a smart-mode validator is configured, it is queried only when --smart is passed.
+ * The actual MCP spawning is a FUTURE integration point — this module implements
+ * configuration reading and provides a stub for the spawning.
+ *
+ * Expected format in config/mcp.json:
+ * {
+ *   "mcpServers": { ... existing servers ... },
+ *   "smartValidators": [
+ *     {
+ *       "name": "bedrock-validator",
+ *       "command": "node",
+ *       "args": ["path/to/validator.js"],
+ *       "timeout": 15000,
+ *       "enabled": true
+ *     }
+ *   ]
+ * }
+ *
+ * Requirements: 15.6
+ */
+
+import { existsSync, readFileSync } from 'node:fs';
+import path from 'node:path';
+
+/**
+ * Load smart-mode validator configuration from config/mcp.json.
+ *
+ * @param {string} [configPath] - Override path to mcp.json (defaults to config/mcp.json relative to project root)
+ * @returns {{ validators: Array, loaded: boolean }}
+ */
+export function loadSmartValidatorConfig(configPath) {
+    const resolvedPath = configPath || path.resolve('config', 'mcp.json');
+
+    if (!existsSync(resolvedPath)) {
+        return { validators: [], loaded: false };
+    }
+
+    try {
+        const content = readFileSync(resolvedPath, 'utf8');
+        const config = JSON.parse(content);
+
+        const smartValidators = config.smartValidators || [];
+
+        // Filter to only enabled validators
+        const enabledValidators = smartValidators.filter(v => v.enabled !== false);
+
+        return {
+            validators: enabledValidators.map(v => ({
+                name: v.name || 'unnamed-smart-validator',
+                command: v.command || 'node',
+                args: v.args || [],
+                timeout: v.timeout || 15000,
+                enabled: true
+            })),
+            loaded: true
+        };
+    } catch {
+        return { validators: [], loaded: false };
+    }
+}
+
+/**
+ * Spawn a smart-mode validator child process and pass context via stdio.
+ * FUTURE INTEGRATION POINT — currently returns an empty findings array.
+ *
+ * @param {Object} validatorConfig - Configuration for the validator
+ * @param {string} validatorConfig.name - Validator name
+ * @param {string} validatorConfig.command - Command to spawn
+ * @param {Array} validatorConfig.args - Arguments for the command
+ * @param {number} validatorConfig.timeout - Timeout in milliseconds
+ * @param {Object} context - The full ValidationContext (JSON-serializable)
+ * @param {Object} [options]
+ * @param {Array} [options.priorFindings] - Findings from static validators
+ * @returns {Promise<Array>} Array of findings from the MCP validator (currently empty stub)
+ */
+export async function spawnSmartValidator(validatorConfig, context, _options = {}) {
+    // FUTURE: Spawn child process, pass context via stdio, collect findings
+    // For now, return empty findings array as a stub
+    return [];
+}
+
+export default { loadSmartValidatorConfig, spawnSmartValidator };

package/src/lib/payload-builder.js

@@ -0,0 +1,153 @@
+/* eslint-disable eqeqeq */
+/**
+ * Constructs AWS API payloads from configuration values.
+ * Produces a JSON-serializable ValidationContext for the validation pipeline.
+ *
+ * Requirements: 3.1, 3.2, 3.3, 3.4, 3.5, 3.6, 3.7, 3.8
+ */
+export default class PayloadBuilder {
+    /**
+     * Build all payloads for a deployment configuration.
+     * @param {Object} config - Configuration from do/config or generator answers
+     * @param {string} deploymentTarget - 'realtime-inference' | 'async-inference' | 'batch-transform'
+     * @returns {Object} ValidationContext (JSON-serializable)
+     */
+    build(config, deploymentTarget) {
+        const payloads = {};
+
+        const endpointConfig = this.buildCreateEndpointConfig(config);
+        if (endpointConfig && Object.keys(endpointConfig).length > 0) {
+            payloads['sagemaker:CreateEndpointConfig'] = endpointConfig;
+        }
+
+        const icPayload = this.buildCreateInferenceComponent(config);
+        if (icPayload && Object.keys(icPayload).length > 0) {
+            payloads['sagemaker:CreateInferenceComponent'] = icPayload;
+        }
+
+        if (deploymentTarget === 'async-inference') {
+            const modelPayload = this.buildCreateModel(config);
+            if (modelPayload && Object.keys(modelPayload).length > 0) {
+                payloads['sagemaker:CreateModel'] = modelPayload;
+            }
+        }
+
+        if (deploymentTarget === 'batch-transform') {
+            const transformPayload = this.buildCreateTransformJob(config);
+            if (transformPayload && Object.keys(transformPayload).length > 0) {
+                payloads['sagemaker:CreateTransformJob'] = transformPayload;
+            }
+        }
+
+        // Filter out undefined values from config to ensure JSON-serializability
+        const cleanConfig = {};
+        for (const [key, value] of Object.entries(config)) {
+            if (value !== undefined) {
+                cleanConfig[key] = value;
+            }
+        }
+
+        return {
+            payloads,
+            config: cleanConfig,
+            deploymentTarget,
+            metadata: {
+                generatedAt: new Date().toISOString(),
+                generatorVersion: '0.2.5',
+                services: [...new Set(Object.keys(payloads).map(k => k.split(':')[0]))]
+            }
+        };
+    }
+
+    /**
+     * Build CreateEndpointConfig payload.
+     * @param {Object} config
+     * @returns {Object} AWS API payload
+     */
+    buildCreateEndpointConfig(config) {
+        const variant = {};
+
+        if (config.INSTANCE_TYPE != null) variant.InstanceType = config.INSTANCE_TYPE;
+        if (config.INFERENCE_AMI_VERSION != null) variant.InferenceAmiVersion = config.INFERENCE_AMI_VERSION;
+        if (config.ENDPOINT_VARIANT_NAME != null) variant.VariantName = config.ENDPOINT_VARIANT_NAME;
+        if (config.ENDPOINT_INITIAL_INSTANCE_COUNT != null) variant.InitialInstanceCount = config.ENDPOINT_INITIAL_INSTANCE_COUNT;
+        if (config.ENDPOINT_VOLUME_SIZE != null) variant.VolumeSizeInGB = config.ENDPOINT_VOLUME_SIZE;
+
+        if (Object.keys(variant).length === 0) return {};
+
+        return {
+            ProductionVariants: [variant]
+        };
+    }
+
+    /**
+     * Build CreateInferenceComponent payload.
+     * @param {Object} config
+     * @returns {Object} AWS API payload
+     */
+    buildCreateInferenceComponent(config) {
+        const payload = {};
+        const computeResources = {};
+
+        if (config.IC_CPU_COUNT != null) computeResources.NumberOfCpuCoresRequired = config.IC_CPU_COUNT;
+        if (config.IC_MEMORY_SIZE != null) computeResources.MinMemoryRequiredInMb = config.IC_MEMORY_SIZE;
+        if (config.IC_GPU_COUNT != null) computeResources.NumberOfAcceleratorDevicesRequired = config.IC_GPU_COUNT;
+
+        if (Object.keys(computeResources).length > 0) {
+            payload.Specification = { ComputeResourceRequirements: computeResources };
+        }
+
+        if (config.IC_COPY_COUNT != null) {
+            payload.RuntimeConfig = { CopyCount: config.IC_COPY_COUNT };
+        }
+
+        return payload;
+    }
+
+    /**
+     * Build CreateModel payload (async inference).
+     * @param {Object} config
+     * @returns {Object} AWS API payload
+     */
+    buildCreateModel(config) {
+        const payload = {};
+
+        if (config.CONTAINER_IMAGE != null) {
+            payload.PrimaryContainer = { Image: config.CONTAINER_IMAGE };
+            if (config.MODEL_DATA_URL != null) {
+                payload.PrimaryContainer.ModelDataUrl = config.MODEL_DATA_URL;
+            }
+        }
+
+        if (config.ROLE_ARN != null) payload.ExecutionRoleArn = config.ROLE_ARN;
+
+        return payload;
+    }
+
+    /**
+     * Build CreateTransformJob payload (batch transform).
+     * @param {Object} config
+     * @returns {Object} AWS API payload
+     */
+    buildCreateTransformJob(config) {
+        const payload = {};
+        const transformResources = {};
+
+        if (config.INSTANCE_TYPE != null) transformResources.InstanceType = config.INSTANCE_TYPE;
+        if (config.BATCH_INSTANCE_COUNT != null) transformResources.InstanceCount = config.BATCH_INSTANCE_COUNT;
+
+        if (Object.keys(transformResources).length > 0) {
+            payload.TransformResources = transformResources;
+        }
+
+        if (config.BATCH_SPLIT_TYPE != null) {
+            payload.TransformInput = { SplitType: config.BATCH_SPLIT_TYPE };
+        }
+
+        if (config.BATCH_STRATEGY != null) {
+            payload.BatchStrategy = config.BATCH_STRATEGY;
+        }
+
+        return payload;
+    }
+}