[Sherpa] Add pre-packaged skills with persistence and subagent config for SMAI spaces

template/v4/Dockerfile

@@ -91,6 +91,16 @@ RUN apt-get update && apt-get upgrade -y && \
     unzip /tmp/kirocli.zip -d /tmp && \
     Q_INSTALL_GLOBAL=1 KIRO_CLI_SKIP_SETUP=1 /tmp/kirocli/install.sh && \
     rm -rf /tmp/kirocli.zip /tmp/kirocli && \
+    : && \
+    # Clone skills from manifest config
+    mkdir -p /etc/sagemaker/skills && \
+    jq -c '.[]' /etc/sagemaker/skills/skills-manifest.json | while read -r entry; do \
+        repo=$(echo "$entry" | jq -r '.repo'); \
+        spath=$(echo "$entry" | jq -r '.path'); \
+        git clone --depth 1 "$repo" /tmp/skill-repo && \
+        cp -r "/tmp/skill-repo/$spath"/* /etc/sagemaker/skills/ && \
+        rm -rf /tmp/skill-repo; \
+    done && \
 # CodeEditor - create server, user data dirs
     mkdir -p /opt/amazon/sagemaker/sagemaker-code-editor-server-data /opt/amazon/sagemaker/sagemaker-ui-code-editor-server-data /opt/amazon/sagemaker/sagemaker-code-editor-user-data \
     && chown $MAMBA_USER:$MAMBA_USER /opt/amazon/sagemaker/sagemaker-code-editor-server-data /opt/amazon/sagemaker/sagemaker-ui-code-editor-server-data /opt/amazon/sagemaker/sagemaker-code-editor-user-data && \

template/v4/dirs/etc/sagemaker/sagemaker-default-agent.json

@@ -0,0 +1,12 @@
+{
+  "name": "sagemaker_default",
+  "description": "SageMaker AI assistant with pre-installed skills for model customization, evaluation, deployment, and HyperPod operations.",
+  "tools": ["@builtin"],
+  "allowedTools": [
+    "mcp::jupyter-mcp::read_notebook",
+    "mcp::jupyter-mcp::read",
+    "mcp::jupyter-mcp::ls"
+  ],
+  "mcpServers": {},
+  "resources": []
+}

template/v4/dirs/etc/sagemaker/skills/skills-manifest.json

@@ -0,0 +1,6 @@
+[
+  {
+    "repo": "https://github.com/awslabs/agent-plugins.git",
+    "path": "plugins/sagemaker-ai/skills"
+  }
+]

template/v4/dirs/etc/sagemaker/skills/sync_skills.sh

@@ -0,0 +1,72 @@
+#!/bin/bash
+# Syncs pre-packaged SageMaker skills from image to user's EBS.
+set -eu
+
+IMAGE_SKILLS_DIR="/etc/sagemaker/skills"
+EBS_SKILLS_DIR="$HOME/.agent/skills"
+LOCK_FILE="$EBS_SKILLS_DIR/.sagemaker-lock"
+
+# Agent targets to symlink skills into (add new agents here)
+AGENT_SKILLS_DIRS=("$HOME/.kiro/skills")
+
+compute_checksum() {
+    (cd "$1" && find . -type f -print0 | sort -z | xargs -0 sha256sum | sha256sum | awk '{print $1}')
+}
+
+get_locked_checksum() {
+    [ -f "$LOCK_FILE" ] && jq -r --arg s "$1" '.skills[$s].checksum // empty' "$LOCK_FILE" 2>/dev/null
+}
+
+set_locked_checksum() {
+    [ -f "$LOCK_FILE" ] || echo '{"skills":{}}' > "$LOCK_FILE"
+    jq --arg s "$1" --arg c "$2" '.skills[$s].checksum = $c' "$LOCK_FILE" > "$LOCK_FILE.tmp"
+    mv "$LOCK_FILE.tmp" "$LOCK_FILE"
+}
+
+mkdir -p "$EBS_SKILLS_DIR"
+for dir in "${AGENT_SKILLS_DIRS[@]}"; do mkdir -p "$dir"; done
+
+if [ ! -d "$IMAGE_SKILLS_DIR" ]; then
+    echo "No bundled skills found at $IMAGE_SKILLS_DIR, skipping."
+    exit 0
+fi
+
+for skill_path in "$IMAGE_SKILLS_DIR"/*/; do
+    [ -d "$skill_path" ] || continue
+    skill_name=$(basename "$skill_path")
+    ebs_skill="$EBS_SKILLS_DIR/$skill_name"
+    image_checksum=$(compute_checksum "$skill_path")
+
+    if [ ! -d "$ebs_skill" ]; then
+        cp -r "$skill_path" "$ebs_skill"
+        set_locked_checksum "$skill_name" "$image_checksum"
+        echo "Installed skill '$skill_name'"
+    else
+        recorded_checksum=$(get_locked_checksum "$skill_name")
+        current_checksum=$(compute_checksum "$ebs_skill")
+
+        if [ "$current_checksum" = "$recorded_checksum" ]; then
+            if [ "$image_checksum" != "$recorded_checksum" ]; then
+                rm -rf "$ebs_skill"
+                cp -r "$skill_path" "$ebs_skill"
+                set_locked_checksum "$skill_name" "$image_checksum"
+                echo "Updated skill '$skill_name'"
+            else
+                echo "Skill '$skill_name' already current, skipping"
+            fi
+        else
+            echo "Skipping skill '$skill_name' — user modified"
+        fi
+    fi
+
+    # Create symlinks for all agent targets
+    for agent_dir in "${AGENT_SKILLS_DIRS[@]}"; do
+        link="$agent_dir/$skill_name"
+        if [ ! -e "$link" ]; then
+            ln -s "$ebs_skill" "$link"
+            echo "Created symlink $link -> $ebs_skill"
+        fi
+    done
+done
+
+echo "Skills sync complete."

template/v4/dirs/etc/sagemaker/sync_agent.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+# Syncs agent config (subagent, default agent settings) for SMAI spaces.
+set -eu
+
+AGENT_CONFIG_SRC="/etc/sagemaker/sagemaker-default-agent.json"
+AGENT_CONFIG_DST="$HOME/.kiro/agents/sagemaker-default-agent.json"
+KIRO_CLI_SETTINGS="$HOME/.kiro/settings/cli.json"
+
+# Install subagent config (always overwrite)
+mkdir -p "$HOME/.kiro/agents"
+cp "$AGENT_CONFIG_SRC" "$AGENT_CONFIG_DST"
+echo "Installed agent config to $AGENT_CONFIG_DST"
+
+# Set default agent only if user hasn't configured one
+mkdir -p "$HOME/.kiro/settings"
+if [ ! -f "$KIRO_CLI_SETTINGS" ]; then
+    echo '{"chat.defaultAgent":"sagemaker_default"}' > "$KIRO_CLI_SETTINGS"
+    echo "Created default agent settings"
+elif ! jq -e '."chat.defaultAgent"' "$KIRO_CLI_SETTINGS" >/dev/null 2>&1; then
+    jq '. + {"chat.defaultAgent":"sagemaker_default"}' "$KIRO_CLI_SETTINGS" > "$KIRO_CLI_SETTINGS.tmp"
+    mv "$KIRO_CLI_SETTINGS.tmp" "$KIRO_CLI_SETTINGS"
+    echo "Added default agent to existing settings"
+fi

template/v4/dirs/usr/local/bin/start-jupyter-server

@@ -14,6 +14,17 @@ else
     micromamba remove -y sagemaker-studio-dataengineering-sessions sagemaker-studio-dataengineering-extensions
     # Disable SMUS-specific extensions
     jupyter labextension disable sagemaker-data-explorer:plugin
+    # Disable old Q chat extension
+    jupyter labextension disable @amzn/amazon_sagemaker_jupyter_ai_q_developer
+fi
+
+# Setup skills and subagent for SMAI spaces
+if [ -n "$SAGEMAKER_APP_TYPE_LOWERCASE" ]; then
+    # Sync pre-packaged skills to EBS
+    bash /etc/sagemaker/skills/sync_skills.sh 2>&1 || echo "Warning: skills sync failed, continuing..."
+
+    # Sync agent config (subagent + default agent settings)
+    bash /etc/sagemaker/sync_agent.sh 2>&1 || echo "Warning: agent config sync failed, continuing..."
 fi
 
 # Enable S3AG plugin if TIP is enabled