@aws/ml-container-creator 0.10.0 → 0.12.1

package/servers/workload-picker/manifest.json

@@ -0,0 +1,16 @@
+{
+    "name": "@amzn/ml-container-creator-workload-picker",
+    "version": "1.0.0",
+    "description": "MCP server that provides named benchmark workload profiles for do/benchmark.",
+    "modes": {
+        "static": true,
+        "smart": false,
+        "discover": false
+    },
+    "catalogs": {
+        "workload-profiles": "./catalogs/workload-profiles.json"
+    },
+    "tool": {
+        "name": "get_workload_profile"
+    }
+}

package/servers/workload-picker/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "@amzn/ml-container-creator-workload-picker",
+  "private": true,
+  "version": "1.0.0",
+  "description": "MCP server that provides named benchmark workload profiles for ML Container Creator. Defines token distributions, concurrency levels, and streaming modes for standardized benchmarking.",
+  "type": "module",
+  "main": "index.js",
+  "license": "Apache-2.0",
+  "scripts": {
+    "test": "node test.js"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.0",
+    "zod": "^3.22.0"
+  }
+}

package/src/app.js

@@ -16,7 +16,7 @@ import CommentGenerator from './lib/comment-generator.js';
 import ConfigurationManager from './lib/configuration-manager.js';
 import RegistryLoader from './lib/registry-loader.js';
 import { resolvePrefixedEnvVars } from './lib/engine-prefix-resolver.js';
-import { _ensureTemplateVariables, _validateEnvironmentVariables, _mergeEnvVarsWithPrecedence } from './lib/template-variable-resolver.js';
+import { _ensureTemplateVariables, _validateEnvironmentVariables } from './lib/template-variable-resolver.js';
 import ejs from 'ejs';
 
 const __filename = fileURLToPath(import.meta.url);
@@ -344,13 +344,15 @@ export async function writeProject(templateDir, destDir, answers, registryConfig
     // Exclude do/benchmark when benchmarking is not selected
     if (!answers.includeBenchmark) {
         ignorePatterns.push('**/do/benchmark');
+        ignorePatterns.push('**/do/.benchmark_writer.py');
         ignorePatterns.push('**/do/optimize');
     }
 
-    // Exclude do/adapter and do/adapters/ when LoRA is not enabled
+    // Exclude do/adapter, do/adapters/, and adapter sidecar when LoRA is not enabled
     if (!answers.enableLora) {
         ignorePatterns.push('**/do/adapter');
         ignorePatterns.push('**/do/adapters/**');
+        ignorePatterns.push('**/code/adapter_sidecar.py');
     }
 
     // Exclude tune files when framework is NOT transformers OR deploymentTarget is batch-transform
@@ -375,6 +377,12 @@ export async function writeProject(templateDir, destDir, answers, registryConfig
         ignorePatterns.push('**/do/lib/feedback.sh');
     }
 
+    // Exclude do/stage when model is already S3-sourced (nothing to stage)
+    const modelName = answers.modelName || answers.customModelName || '';
+    if (answers.modelSource === 's3' || modelName.startsWith('s3://')) {
+        ignorePatterns.push('**/do/stage');
+    }
+
     // Exclude do/test when hosted-model-endpoint is not selected
     const testTypes = answers.testTypes || [];
     if (!testTypes.includes('hosted-model-endpoint')) {
@@ -816,7 +824,8 @@ function _setExecutablePermissions(destDir, answers = {}) {
         'do/add-ic',
         'do/adapter',
         'do/tune',
-        'do/train'
+        'do/train',
+        'do/stage'
     ];
 
     const shellScripts = architecture === 'marketplace' ? marketplaceScripts : defaultScripts;