@aws/ml-container-creator 0.2.0

package/servers/lib/custom-validators.js

@@ -0,0 +1,46 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * Custom value validation patterns for MCP servers.
+ * Each server defines a regex pattern and validation function
+ * for user-provided custom values.
+ */
+
+export const CUSTOM_VALIDATORS = {
+    'base-image-picker': {
+        pattern: /^[a-zA-Z0-9][a-zA-Z0-9._\-\/]*(:[a-zA-Z0-9._\-]+)?$/,
+        validate(value) {
+            if (!value || value.trim() === '') return 'Value is required'
+            if (!this.pattern.test(value.trim())) return 'Invalid image format. Expected: [registry/]repository[:tag]'
+            return true
+        },
+        errorMessage: 'Invalid image format. Expected: [registry/]repository[:tag]'
+    },
+    'instance-recommender': {
+        pattern: /^ml\.[a-z0-9]+\.[a-z0-9]+$/,
+        validate(value) {
+            if (!value || value.trim() === '') return 'Value is required'
+            if (!this.pattern.test(value.trim())) return 'Invalid instance type. Expected: ml.<family>.<size>'
+            return true
+        },
+        errorMessage: 'Invalid instance type. Expected: ml.<family>.<size>'
+    },
+    'region-picker': {
+        pattern: /^[a-z]{2,4}-[a-z]+-\d+$/,
+        validate(value) {
+            if (!value || value.trim() === '') return 'Value is required'
+            if (!this.pattern.test(value.trim())) return 'Invalid region code. Expected: <partition>-<geo>-<number>'
+            return true
+        },
+        errorMessage: 'Invalid region code. Expected: <partition>-<geo>-<number>'
+    },
+    'hyperpod-cluster-picker': {
+        pattern: /^.+$/,
+        validate(value) {
+            if (!value || value.trim() === '') return 'Cluster name is required'
+            return true
+        },
+        errorMessage: 'Cluster name is required'
+    }
+}

package/servers/lib/dynamic-resolver.js

@@ -0,0 +1,36 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * DynamicResolver — abstract base class for all dynamic data resolvers.
+ *
+ * Subclasses implement `fetch()` to retrieve data from an external source
+ * (registry API, AWS API, etc.) and `supportedKeys()` to declare which
+ * identifiers they handle.
+ *
+ * Concrete implementations:
+ *   - ImageResolver (base-image-picker) — fetches container images
+ *   - ClusterResolver (hyperpod-cluster-picker) — fetches HyperPod clusters
+ */
+class DynamicResolver {
+    /**
+     * Fetch data for a given key.
+     *
+     * @param {string} key - The identifier to resolve (e.g., framework name, cluster filter)
+     * @param {object} [options] - Resolver-specific options (limit, timeout, etc.)
+     * @returns {Promise<{items: object[], defaultItem: *|null}>}
+     */
+    async fetch(key, options = {}) {
+        throw new Error('fetch() must be implemented by subclass')
+    }
+
+    /**
+     * Returns the list of keys this resolver can handle.
+     * @returns {string[]}
+     */
+    supportedKeys() {
+        throw new Error('supportedKeys() must be implemented by subclass')
+    }
+}
+
+export { DynamicResolver }

package/servers/lib/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "@amzn/ml-container-creator-lib",
+  "private": true,
+  "version": "1.0.0",
+  "description": "Shared library for ML Container Creator bundled MCP servers. Provides a reusable Bedrock client for context-aware recommendations.",
+  "type": "module",
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@aws-sdk/client-bedrock-runtime": "^3.700.0"
+  }
+}

package/servers/lib/schemas/image-catalog.schema.json

@@ -0,0 +1,185 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "$id": "image-catalog.schema.json",
+    "definitions": {
+        "imageEntry": {
+            "type": "object",
+            "required": [
+                "image",
+                "tag",
+                "architecture",
+                "created",
+                "labels",
+                "registry",
+                "repository"
+            ],
+            "properties": {
+                "image": {
+                    "type": "string",
+                    "minLength": 1
+                },
+                "tag": {
+                    "type": "string",
+                    "minLength": 1
+                },
+                "architecture": {
+                    "type": "string",
+                    "enum": [
+                        "amd64",
+                        "arm64"
+                    ]
+                },
+                "created": {
+                    "type": "string",
+                    "format": "date-time"
+                },
+                "labels": {
+                    "type": "object",
+                    "additionalProperties": {
+                        "type": "string"
+                    }
+                },
+                "registry": {
+                    "type": "string",
+                    "enum": [
+                        "dockerhub",
+                        "ngc",
+                        "ecr",
+                        "ecr-public"
+                    ]
+                },
+                "repository": {
+                    "type": "string",
+                    "minLength": 1
+                },
+                "defaults": {
+                    "type": "object",
+                    "properties": {
+                        "envVars": {
+                            "type": "object",
+                            "additionalProperties": {
+                                "type": "string"
+                            }
+                        },
+                        "inferenceAmiVersion": {
+                            "type": "string"
+                        },
+                        "recommendedInstanceTypes": {
+                            "type": "array",
+                            "items": {
+                                "type": "string"
+                            }
+                        }
+                    },
+                    "additionalProperties": false
+                },
+                "accelerator": {
+                    "type": "object",
+                    "required": [
+                        "type",
+                        "version",
+                        "versionRange"
+                    ],
+                    "properties": {
+                        "type": {
+                            "type": "string",
+                            "enum": [
+                                "cuda",
+                                "neuron",
+                                "cpu",
+                                "rocm"
+                            ]
+                        },
+                        "version": {
+                            "type": "string"
+                        },
+                        "versionRange": {
+                            "type": "object",
+                            "required": [
+                                "min",
+                                "max"
+                            ],
+                            "properties": {
+                                "min": {
+                                    "type": "string"
+                                },
+                                "max": {
+                                    "type": "string"
+                                }
+                            },
+                            "additionalProperties": false
+                        }
+                    },
+                    "additionalProperties": false
+                },
+                "validationLevel": {
+                    "type": "string",
+                    "enum": [
+                        "tested",
+                        "community-validated",
+                        "experimental",
+                        "untested"
+                    ]
+                },
+                "profiles": {
+                    "type": "object",
+                    "additionalProperties": {
+                        "type": "object",
+                        "required": [
+                            "displayName",
+                            "description"
+                        ],
+                        "properties": {
+                            "displayName": {
+                                "type": "string"
+                            },
+                            "description": {
+                                "type": "string"
+                            },
+                            "envVars": {
+                                "type": "object",
+                                "additionalProperties": {
+                                    "type": "string"
+                                }
+                            },
+                            "recommendedInstanceTypes": {
+                                "type": "array",
+                                "items": {
+                                    "type": "string"
+                                }
+                            },
+                            "notes": {
+                                "type": "string"
+                            }
+                        },
+                        "additionalProperties": false
+                    }
+                },
+                "notes": {
+                    "type": "string"
+                }
+            },
+            "additionalProperties": false
+        }
+    },
+    "oneOf": [
+        {
+            "type": "array",
+            "minItems": 1,
+            "items": {
+                "$ref": "#/definitions/imageEntry"
+            }
+        },
+        {
+            "type": "object",
+            "minProperties": 1,
+            "additionalProperties": {
+                "type": "array",
+                "minItems": 1,
+                "items": {
+                    "$ref": "#/definitions/imageEntry"
+                }
+            }
+        }
+    ]
+}

package/servers/lib/schemas/instances.schema.json

@@ -0,0 +1,124 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "type": "object",
+    "required": [
+        "catalog",
+        "recommendations"
+    ],
+    "properties": {
+        "catalog": {
+            "type": "object",
+            "minProperties": 1,
+            "additionalProperties": {
+                "type": "object",
+                "required": [
+                    "category",
+                    "gpus",
+                    "vcpus",
+                    "memGb",
+                    "accelerator",
+                    "cudaVersions",
+                    "tags"
+                ],
+                "properties": {
+                    "category": {
+                        "type": "string",
+                        "enum": [
+                            "cpu",
+                            "gpu"
+                        ]
+                    },
+                    "gpus": {
+                        "type": "integer",
+                        "minimum": 0
+                    },
+                    "vcpus": {
+                        "type": "integer",
+                        "minimum": 1
+                    },
+                    "memGb": {
+                        "type": "number",
+                        "minimum": 0
+                    },
+                    "accelerator": {
+                        "type": "string"
+                    },
+                    "cudaVersions": {
+                        "oneOf": [
+                            {
+                                "type": "array",
+                                "items": {
+                                    "type": "string"
+                                }
+                            },
+                            {
+                                "type": "null"
+                            }
+                        ]
+                    },
+                    "tags": {
+                        "type": "array",
+                        "items": {
+                            "type": "string"
+                        }
+                    },
+                    "family": {
+                        "type": "string"
+                    },
+                    "acceleratorType": {
+                        "type": "string",
+                        "enum": [
+                            "cuda",
+                            "neuron",
+                            "cpu",
+                            "rocm"
+                        ]
+                    },
+                    "hardware": {
+                        "type": "string"
+                    },
+                    "gpuArchitecture": {
+                        "type": "string"
+                    },
+                    "defaultCudaVersion": {
+                        "oneOf": [
+                            {
+                                "type": "string"
+                            },
+                            {
+                                "type": "null"
+                            }
+                        ]
+                    },
+                    "notes": {
+                        "type": "string"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        "recommendations": {
+            "type": "object",
+            "required": [
+                "cpu",
+                "gpu"
+            ],
+            "properties": {
+                "cpu": {
+                    "type": "array",
+                    "items": {
+                        "type": "string"
+                    }
+                },
+                "gpu": {
+                    "type": "array",
+                    "items": {
+                        "type": "string"
+                    }
+                }
+            },
+            "additionalProperties": false
+        }
+    },
+    "additionalProperties": false
+}

package/servers/lib/schemas/manifest.schema.json

@@ -0,0 +1,64 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "type": "object",
+    "required": [
+        "name",
+        "version",
+        "description",
+        "modes",
+        "catalogs",
+        "tool"
+    ],
+    "properties": {
+        "name": {
+            "type": "string",
+            "minLength": 1
+        },
+        "version": {
+            "type": "string",
+            "pattern": "^\\d+\\.\\d+\\.\\d+"
+        },
+        "description": {
+            "type": "string"
+        },
+        "modes": {
+            "type": "object",
+            "required": [
+                "static",
+                "smart",
+                "discover"
+            ],
+            "properties": {
+                "static": {
+                    "type": "boolean"
+                },
+                "smart": {
+                    "type": "boolean"
+                },
+                "discover": {
+                    "type": "boolean"
+                }
+            },
+            "additionalProperties": false
+        },
+        "catalogs": {
+            "type": "object",
+            "additionalProperties": {
+                "type": "string"
+            }
+        },
+        "tool": {
+            "type": "object",
+            "required": [
+                "name"
+            ],
+            "properties": {
+                "name": {
+                    "type": "string"
+                }
+            },
+            "additionalProperties": false
+        }
+    },
+    "additionalProperties": false
+}

package/servers/lib/schemas/model-catalog.schema.json

@@ -0,0 +1,91 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "$id": "model-catalog.schema.json",
+    "type": "object",
+    "minProperties": 1,
+    "additionalProperties": {
+        "type": "object",
+        "required": [
+            "family",
+            "chat_template",
+            "gated",
+            "tags",
+            "architecture",
+            "framework_compatibility",
+            "validation_level",
+            "notes"
+        ],
+        "properties": {
+            "family": {
+                "type": "string"
+            },
+            "chat_template": {
+                "oneOf": [
+                    { "type": "string" },
+                    { "type": "null" }
+                ]
+            },
+            "gated": {
+                "type": "boolean"
+            },
+            "tags": {
+                "type": "array",
+                "items": {
+                    "type": "string"
+                }
+            },
+            "architecture": {
+                "oneOf": [
+                    { "type": "string" },
+                    { "type": "null" }
+                ]
+            },
+            "framework_compatibility": {
+                "type": "object",
+                "additionalProperties": {
+                    "type": "string"
+                }
+            },
+            "validation_level": {
+                "type": "string",
+                "enum": [
+                    "tested",
+                    "community-validated",
+                    "experimental",
+                    "untested"
+                ]
+            },
+            "notes": {
+                "type": "string"
+            },
+            "profiles": {
+                "type": "object",
+                "additionalProperties": {
+                    "type": "object",
+                    "required": [
+                        "displayName"
+                    ],
+                    "properties": {
+                        "displayName": {
+                            "type": "string"
+                        },
+                        "envVars": {
+                            "type": "object",
+                            "additionalProperties": {
+                                "type": "string"
+                            }
+                        },
+                        "recommendedInstanceTypes": {
+                            "type": "array",
+                            "items": {
+                                "type": "string"
+                            }
+                        }
+                    },
+                    "additionalProperties": false
+                }
+            }
+        },
+        "additionalProperties": false
+    }
+}

package/servers/lib/schemas/regions.schema.json

@@ -0,0 +1,26 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "type": "array",
+    "minItems": 1,
+    "items": {
+        "type": "object",
+        "required": [
+            "code",
+            "labels"
+        ],
+        "properties": {
+            "code": {
+                "type": "string",
+                "pattern": "^[a-z]{2,4}-[a-z]+-\\d+$"
+            },
+            "labels": {
+                "type": "array",
+                "minItems": 1,
+                "items": {
+                    "type": "string"
+                }
+            }
+        },
+        "additionalProperties": false
+    }
+}

package/servers/lib/schemas/triton-backends.schema.json

@@ -0,0 +1,51 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "$id": "triton-backends.schema.json",
+    "type": "object",
+    "minProperties": 1,
+    "additionalProperties": {
+        "type": "object",
+        "required": [
+            "requiresGpu",
+            "modelFormats",
+            "modelArtifactName",
+            "requiresModelName",
+            "supportsSampleModel"
+        ],
+        "properties": {
+            "requiresGpu": {
+                "type": "boolean"
+            },
+            "modelFormats": {
+                "oneOf": [
+                    {
+                        "type": "array",
+                        "items": {
+                            "type": "string"
+                        }
+                    },
+                    {
+                        "type": "null"
+                    }
+                ]
+            },
+            "modelArtifactName": {
+                "oneOf": [
+                    {
+                        "type": "string"
+                    },
+                    {
+                        "type": "null"
+                    }
+                ]
+            },
+            "requiresModelName": {
+                "type": "boolean"
+            },
+            "supportsSampleModel": {
+                "type": "boolean"
+            }
+        },
+        "additionalProperties": false
+    }
+}