ReddRadar
Agent skill
bentoml-model-packager
BentoML skill for model packaging, serving, and containerization.
Install this agent skill to your Project
npx add-skill https://github.com/a5c-ai/babysitter/tree/main/library/specializations/data-science-ml/skills/bentoml-model-packager
SKILL.md
bentoml-model-packager
Overview
BentoML skill for model packaging, serving, and containerization with support for multiple ML frameworks.
Capabilities
- Bento creation and versioning
- Multi-framework model support (sklearn, PyTorch, TensorFlow, etc.)
- API endpoint definition with validation
- Docker containerization
- Kubernetes deployment YAML generation
- Adaptive batching configuration
- Model signatures and runners
- Service composition
Target Processes
- Model Deployment Pipeline with Canary Release
- Model Training Pipeline
- ML System Integration Testing
Tools and Libraries
- BentoML
- Docker
- Kubernetes
Input Schema
{
"type": "object",
"required": ["action"],
"properties": {
"action": {
"type": "string",
"enum": ["save", "build", "serve", "containerize", "push", "list"],
"description": "BentoML action to perform"
},
"modelConfig": {
"type": "object",
"properties": {
"name": { "type": "string" },
"framework": { "type": "string" },
"modelPath": { "type": "string" },
"signatures": { "type": "object" }
}
},
"serviceConfig": {
"type": "object",
"properties": {
"servicePath": { "type": "string" },
"port": { "type": "integer" },
"workers": { "type": "integer" },
"batchConfig": {
"type": "object",
"properties": {
"maxBatchSize": { "type": "integer" },
"maxLatencyMs": { "type": "integer" }
}
}
}
},
"buildConfig": {
"type": "object",
"properties": {
"bentoName": { "type": "string" },
"version": { "type": "string" },
"includeFiles": { "type": "array", "items": { "type": "string" } },
"pythonRequirements": { "type": "string" }
}
},
"containerConfig": {
"type": "object",
"properties": {
"imageName": { "type": "string" },
"registry": { "type": "string" },
"dockerOptions": { "type": "object" }
}
}
}
}
Output Schema
{
"type": "object",
"required": ["status", "action"],
"properties": {
"status": {
"type": "string",
"enum": ["success", "error"]
},
"action": {
"type": "string"
},
"modelTag": {
"type": "string"
},
"bentoTag": {
"type": "string"
},
"imageTag": {
"type": "string"
},
"endpoint": {
"type": "string"
},
"kubernetesYaml": {
"type": "string"
}
}
}
Usage Example
{
kind: 'skill',
title: 'Package and containerize model',
skill: {
name: 'bentoml-model-packager',
context: {
action: 'containerize',
modelConfig: {
name: 'fraud_classifier',
framework: 'sklearn',
modelPath: 'models/fraud_model.pkl'
},
buildConfig: {
bentoName: 'fraud-service',
version: '1.0.0',
pythonRequirements: 'requirements.txt'
},
containerConfig: {
imageName: 'fraud-service',
registry: 'gcr.io/my-project'
}
}
}
}
Recommended Agent Skills
Expand your agent's capabilities with these related and highly-rated skills.
a5c-ai/babysitter
gsd-tools
Central utility skill for GSD operations. Provides config parsing, slug generation, timestamps, path operations, and orchestrates calls to other specialized skills. Acts as the unified entry point that the original gsd-tools.cjs provided via its lib/ modules (commands, config, core, init).
a5c-ai/babysitter
model-profile-resolution
Resolve model profile (quality/balanced/budget) at orchestration start and map agents to specific models. Enables cost/quality tradeoffs by selecting appropriate AI models for each agent role.
a5c-ai/babysitter
verification-suite
Plan structure validation, phase completeness checks, reference integrity verification, and artifact existence confirmation. Provides the structured verification layer ensuring GSD artifacts are well-formed and complete.
a5c-ai/babysitter
state-management
STATE.md reading, writing, and field-level updates. Provides cross-session state persistence via .planning/STATE.md with structured fields for current task, completed phases, blockers, decisions, and quick tasks.
a5c-ai/babysitter
git-integration
Git commit patterns, formats, and conventions for GSD methodology. Provides atomic commits per task, structured commit messages, planning file commits, branch management, and milestone tag operations.
a5c-ai/babysitter
frontmatter-parsing
YAML frontmatter parsing and manipulation for .planning/ documents. Provides read, write, update, query, and validation operations on frontmatter blocks in GSD markdown artifacts.
Didn't find tool you were looking for?