Tooluck logoTooluck
SKILL.md

model-selection

Agent skill fromawslabs/agent-plugins

Selects a base model for the user's use case by querying SageMaker Hub. Use when the user asks which model to use, wants to select or change their base model, mentions a model name or family (e.g., "Llama", "Mistral", "Nova"), or wants to evaluate a base model — always activate even for known model names because the exact Hub model ID must be resolved. Queries available models, presents benchmarks and licenses, and confirms selection.

How to install model-selection

install model-selection
# install the agent-plugins pack with the skills CLI
npx skills add awslabs/agent-plugins

# or add only this skill to Claude Code: copy the folder

plugins/sagemaker-ai/skills/model-selection/

# into

~/.claude/skills/model-selection/ (all projects)

.claude/skills/model-selection/ (one project)

plugins/sagemaker-ai/skills/model-selection/SKILL.md

Apache-2.0Full file
1# Model Selection
2
3Guides the user through selecting a base model based on their use case.
4
5## When to Use
6
7- User asks which model to use
8- User wants to select or change their base model
9- User mentions a model name or family (e.g., "Llama", "Mistral", "Nova") — the exact Hub model ID still needs to be resolved
10- User wants to evaluate a base model before deciding whether to finetune
11
12## Prerequisites
13
14- A `use_case_spec.md` file exists. If not, activate the use-case-specification skill to generate it first.
15
16## Workflow
17
18### Step 1: Check Region
19
20Run:
21
22```
23python -c "import boto3; print(boto3.session.Session().region_name)"
24```
25
26- `None` → STOP. Tell user: "Set your region via `export AWS_DEFAULT_REGION=us-west-2` or `aws configure`."
27- Set → store REGION in context, continue.
28
29### Step 2: Discover Hub
30
311. List all available SageMaker Hubs in the user's region by calling the SageMaker `ListHubs` API using the `aws___call_aws` tool.
322. From the results, filter out any hub whose `HubDescription` contains "AI Registry" — these do not contain JumpStart models.
333. The remaining hubs are eligible (e.g., `SageMakerPublicHub` and any private hubs).
344. If exactly one eligible hub exists, use it automatically — do not ask the user.
355. If multiple eligible hubs exist, present them to the user and ask which one to use. Example:
36
37 ```
38 I found the following model hubs:
39 - SageMakerPublicHub — SageMaker Public Hub
40 - Private-Hub-XYZ — Private Hub models
41…

Excerpt shown under the repository's Apache-2.0 license.

More skills in agent-plugins

Newsletter

Join the Community

Subscribe to our newsletter for the latest news and updates