@aws/ml-container-creator 0.10.0 → 0.12.1

package/templates/do/submit

@@ -9,6 +9,13 @@ set -o pipefail
 # Source configuration
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 source "${SCRIPT_DIR}/config"
+source "${SCRIPT_DIR}/lib/profile.sh"
+
+# ── Profile-resolved variables (env var > profile > default) ──────────────────
+ECR_REPOSITORY_NAME="${ECR_REPOSITORY_NAME:-${_PROFILE[ecrRepositoryName]:-ml-container-creator}}"
+
+# ── Derived variables (env var > computed default) ────────────────────────────
+CODEBUILD_PROJECT_NAME="${CODEBUILD_PROJECT_NAME:-${PROJECT_NAME}-build-$(date +%Y%m%d)}"
 
 echo "🚀 Submitting CodeBuild job for ${PROJECT_NAME}"
 echo "   Deployment config: ${DEPLOYMENT_CONFIG}"

package/templates/do/test

@@ -9,6 +9,7 @@ set -o pipefail
 # Source configuration
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 source "${SCRIPT_DIR}/config"
+source "${SCRIPT_DIR}/lib/profile.sh"
 
 <% if (deploymentTarget === 'realtime-inference') { %>
 # ============================================================
@@ -214,6 +215,7 @@ else
     # Resolve inference component name
     # Precedence: do/adapters/ → do/ic/ → legacy config
     IC_NAME=""
+    ADAPTER_MODEL_NAME=""
     if [ -n "${IC_ARG}" ] && [ -f "${SCRIPT_DIR}/adapters/${IC_ARG}.conf" ]; then
         # Argument matches an adapter name — use adapter IC
         ADAPTER_IC_NAME=""
@@ -223,6 +225,7 @@ else
             exit 1
         fi
         IC_NAME="${ADAPTER_IC_NAME}"
+        ADAPTER_MODEL_NAME="${IC_ARG}"
     elif [ -n "${IC_ARG}" ]; then
         # Explicit IC name provided as argument
         IC_CONF="${SCRIPT_DIR}/ic/${IC_ARG}.conf"
@@ -258,6 +261,17 @@ else
         IC_NAME="${INFERENCE_COMPONENT_NAME:-}"
     fi
 
+    # If testing an adapter, override the model name in the payload
+    if [ -n "${ADAPTER_MODEL_NAME}" ] && [ "${FRAMEWORK}" = "transformers" ]; then
+        case "${MODEL_SERVER}" in
+            vllm|sglang)
+                TEST_PAYLOAD='{"model": "'"${ADAPTER_MODEL_NAME}"'", "messages": [{"role": "user", "content": "What is machine learning?"}], "max_tokens": 50, "temperature": 0.7}'
+                echo "${TEST_PAYLOAD}" > "${TEMP_PAYLOAD}"
+                echo "   (Using adapter model name: ${ADAPTER_MODEL_NAME})"
+                ;;
+        esac
+    fi
+
     INVOKE_ARGS=(
         --endpoint-name "${ENDPOINT_NAME}"
         --region "${AWS_REGION}"