mcpField
M
model-selection

model-selection

github.com/awslabs开发93273 次安装3 周前更新
代码仓库

关于 model-selection

引导用户从 SageMaker Hub 中选择可用基础模型的技能。它会检查 AWS 区域和可用 Hub，再展示模型及许可证信息，帮助根据用途缩小选择范围。

主要功能

  1. 01检查已配置的 AWS 区域
  2. 02发现可用的 SageMaker Hub
  3. 03列出可用模型及许可证
  4. 04按使用场景引导选型
  5. 05确认选定的基础模型

适用场景

  1. 01为项目选择基础模型
  2. 02检查指定模型是否可用
  3. 03比较可用模型与许可证
  4. 04微调前评估基础模型

安装

复制下面的命令，在终端中运行。

$npx skills add https://github.com/awslabs/agent-plugins --skill model-selection

安装后在 Claude Code 中输入/model-selection确认已加载。

包含的文件

除 SKILL.md 外，安装时一起下载的参考资料和脚本。

references/
model-licenses.md
model-selection.md
references/benchmarks/
agenticIndex.md
codingIndex.md
gpqa.md
hle.md
ifbench.md
intelligenceIndex.md
mmmuPro.md
tau2.md
scripts/
get_model_names.py

详细介绍

模型选择

根据用户的使用场景，引导用户选择基础模型。

适用场景

  • 用户询问该使用哪个模型
  • 用户想选择或更换基础模型
  • 用户提到模型名称或系列（例如，“Llama”“Mistral”“Nova”）——仍需确定确切的 Hub 模型 ID
  • 用户想在决定是否进行微调前评估基础模型

前置条件

  • 存在 use_case_spec.md 文件。如果不存在，请先启用 use-case-specification 技能来生成该文件。

工作流

第 1 步：检查区域

运行：

python -c "import boto3; print(boto3.session.Session().region_name)"
  • None → 停止。告知用户：“通过 export AWS_DEFAULT_REGION=us-west-2 或 aws configure 设置区域。”
  • 已设置 → 在上下文中保存 REGION，然后继续。
第 2 步：发现 Hub

  1. 使用 aws___call_aws 工具调用 SageMaker ListHubs API，列出用户所在区域内所有可用的 SageMaker Hubs。

  2. 从结果中过滤掉 HubDescription 中包含“AI Registry”的 Hub——这些 Hub 不包含 JumpStart 模型。

  3. 剩余的 Hub 均符合条件（例如，SageMakerPublicHub 和任何私有 Hub）。

  4. 如果恰好只有一个符合条件的 Hub，则自动使用它——无需询问用户。

  5. 如果有多个符合条件的 Hub，则向用户展示并询问要使用哪一个。示例：

    I found the following model hubs:
- SageMakerPublicHub — SageMaker Public Hub
- Private-Hub-XYZ — Private Hub models
Which hub would you like to use?

  6. 保存所选 Hub 名称，供后续步骤使用。

第 3 步：选择基础模型

首先，运行 python model-selection/scripts/get_model_names.py <hub-name>，检索 SageMaker Hub 中所有可用的模型名称。

在提出任何建议前，先向用户展示所有可用模型及其许可证。将模型列表与 references/model-licenses.md 交叉核对，并按 <model name> - [<license>](<url>) 的格式显示每个模型。例如：“Qwen3-4B - Apache 2.0”

如果你已从对话上下文或规划文件中得知用户想使用哪个模型，请确认该模型在列表中，显示其许可证，然后继续。否则，按照 references/model-selection.md 中的说明帮助用户选择模型。 重要： 帮助用户选择模型时，请记住这份可用模型列表。不要推荐用户无法使用的模型。

第 4 步：确认选择

向用户展示摘要：

Here's what we've selected:
- Base model: [model name]

询问用户是否要继续使用此模型。

参考资料

  • references/model-selection.md — 模型选择说明和基准测试描述
  • references/model-licenses.md — 模型选择过程中用于展示的模型许可证信息

评论

支持 Markdown