Generated banner for SDK Getting Started
Generated logo for SDK Getting Started

SDK Getting Started

AWS LabsOfficialPublished by AWS Labs

  • Skill
  • Tool integration
  • Cloud and DevOps
  • Apache-2.0
  • Low risk

From the author

Validates the user's environment for SageMaker AI operations — checks SDK version, AWS region, and execution role. Use when the user says "set up", "getting started", "check my environment", "configure SDK", or as the first step in any plan involving SageMaker/Bedrock training, evaluation, or deployment.
— the description by AWS Labs

AutoGineer summary

Written by AutoGineer from the skill’s own files, not by the author

This skill performs essential preflight checks to ensure the user's environment is correctly configured for SageMaker AI operations. It verifies the SageMaker Python SDK version, detects the active AWS region, and validates the IAM execution role's trust policies. By automating these checks, it ensures that downstream tasks like training, evaluation, or deployment have the necessary prerequisites in place, storing the validated configuration in the conversation context for later use.

What it does

  • Verifies the installed version of the SageMaker Python SDK.
  • Automatically detects the current AWS region.
  • Resolves and validates the IAM execution role ARN.
  • Checks IAM role trust policies for required service principals.
  • Reports environment status and configuration details.
  • Stores validated environment data for use by other agent skills.

When to use it

  • At the beginning of any SageMaker or Bedrock project.
  • When a user requests help setting up or configuring their SDK.
  • When troubleshooting environment-related issues in SageMaker.
  • As the first step in a multi-step plan involving AI model training or deployment.

Advantages

  • Prevents runtime failures by catching configuration issues early.
  • Reduces manual effort by programmatically detecting environment settings.
  • Ensures consistent environment state across different agent tasks.
  • Provides clear feedback on missing permissions or trust policy requirements.

Usage examples

“I want to get started with SageMaker. Can you help me set up?”
The agent will verify the SDK version, detect the AWS region, and validate the execution role, reporting the environment status.
“Check my environment for SageMaker training.”
The agent performs a preflight check to confirm the SDK is up to date and the IAM role has the necessary trust policies.

Install

  1. Download the skill as a ZIP:

    https://mp-staging.autogineer.com/agent-skills/awslabs/sdk-getting-started/download
  2. Unzip it into the cross-client skills folder of your project:

    mkdir -p .agents/skills && unzip sdk-getting-started.zip -d .agents/skills

Documentation by AWS Labs

SDK Getting Started

Preflight checks to verify the user's environment can run SageMaker AI operations. The agent runs these checks directly (no code generation) and stores results in conversation context for downstream skills.

Principles

  1. Don't ask for what you can look up. Resolve region and role programmatically before asking the user.

Workflow

Read and follow references/sagemaker-python-sdk-setup.md.

References

  • references/sagemaker-python-sdk-setup.md - SageMaker Python SDK version, region, and execution role checks

  • references/execution-role-setup.md — Execution role resolution and validation

Files

references/sagemaker-python-sdk-setup.md

SageMaker Python SDK Setup

Workflow for validating the SageMaker Python SDK environment.

Step 1: Install/Verify SDK

First, check if the SDK is already installed:

python -c "from importlib.metadata import version; print(version('sagemaker'))"
  • If version ≥ 3.7.1 → report the version to the user. Offer to upgrade but do not force it.

  • If missing or < 3.7.1 → install:

pip install --upgrade 'sagemaker>=3.7.1' boto3 -q

Then re-run the version check to confirm.

If install fails

STOP. Do NOT proceed with the plan. Tell the user:

pip install failed — this is likely a system-level issue, not something I can fix by trying different install commands.

Show the exact error, then:

Common causes: missing C build tools (gcc/python3-devel), incompatible Python version, or network/proxy issues.

Do NOT retry with --no-deps, alternative package names, or extras like [core] or [train]. These result in a broken partial install that fails later with import errors.

Step 2: Check Region

If REGION is already stored in conversation context, skip this step — do not re-prompt the user.

Otherwise, run:

python -c "import boto3; print(boto3.session.Session().region_name)"
  • None → STOP. Tell user: "Set your region via export AWS_DEFAULT_REGION={region} or aws configure."

  • Set → store REGION in context, continue.

Step 3: Resolve and Validate Execution Role

Read and follow execution-role-setup.md.

Step 4: Summary

Print:

Environment ready:
  SDK:    sagemaker X.Y.Z ✅
  Region: <region> ✅
  Role:   <arn> ✅
          sagemaker trust ✅ | bedrock trust ⚠️ | lambda trust ✅

Downstream skills use REGION and ROLE_ARN from conversation context. They MUST NOT re-resolve these values.

Requirements and compatibility

Works with: Claude Code, Claude apps, OpenAI Codex, GitHub Copilot, Cursor, Gemini CLI, Kiro, OpenCode, Goose, Amp, Roo Code, JetBrains Junie, Factory Droid, Devin (ex-Windsurf), Mistral Vibe, OpenHands, Letta

“Works with” means the tool reads the open Agent Skills format; a skill written for one tool may use features others lack.

Needs: Needs Python

Security

What our scanner noticed· Low risk

  • Instructions that use the network

A static scan is not a guarantee. Skills are third-party content: review a skill before you let an agent use it.

Attribution and licence

Published by
AWS Labs
Publisher
AWS Labs (Official)
Source
awslabs/agent-plugins · plugins/sagemaker-ai/skills/sdk-getting-started · commit fbefb37
Licence
Apache License 2.0 (redistribution allowed)
  • LICENSE (added by AutoGineer from LICENSE)
  • NOTICE (added by AutoGineer from NOTICE)
  • ATTRIBUTION.md (written by AutoGineer)
Synced

AutoGineer is not affiliated with or endorsed by AWS Labs.

Download ZIP

The ZIP holds the skill’s files, its licence files and an ATTRIBUTION.md. Download sdk-getting-started.zip