feat: add machine learning engineering workflow

(cherry picked from commit 4b0eeacd66b2f65b7b11d7f2c8bef056c50b08e4)
This commit is contained in:
Divyesh Thirukonda
2026-05-01 20:19:25 -05:00
committed by Affaan Mustafa
parent 54efa1a150
commit 240d52d27f
20 changed files with 1236 additions and 21 deletions

View File

@@ -98,6 +98,12 @@ function runTests() {
'Should include lang:c');
assert.ok(components.some(component => component.id === 'capability:security'),
'Should include capability:security');
assert.ok(components.some(component => component.id === 'capability:machine-learning'),
'Should include capability:machine-learning');
assert.ok(components.some(component => component.id === 'agent:mle-reviewer'),
'Should include agent:mle-reviewer');
assert.ok(components.some(component => component.id === 'skill:mle-workflow'),
'Should include skill:mle-workflow');
})) passed++; else failed++;
if (test('gets install component details and validates component IDs', () => {
@@ -271,6 +277,30 @@ function runTests() {
);
})) passed++; else failed++;
if (test('resolves machine-learning component with workflow dependencies', () => {
const plan = resolveInstallPlan({
includeComponentIds: ['capability:machine-learning'],
target: 'claude',
projectRoot: '/workspace/ml-app',
});
assert.ok(plan.selectedModuleIds.includes('machine-learning'),
'Should include machine-learning module');
assert.ok(plan.selectedModuleIds.includes('framework-language'),
'Should include Python and framework-language support');
assert.ok(plan.selectedModuleIds.includes('workflow-quality'),
'Should include eval and verification workflows');
assert.ok(plan.selectedModuleIds.includes('database'),
'Should include database/data persistence support');
assert.ok(plan.selectedModuleIds.includes('devops-infra'),
'Should include deployment and container support');
assert.ok(plan.selectedModuleIds.includes('security'),
'Should include security through machine-learning dependencies');
assert.ok(plan.operations.some(operation => (
operation.sourceRelativePath === 'skills/mle-workflow'
)), 'Should install the MLE workflow skill');
})) passed++; else failed++;
if (test('resolves explicit modules with dependency expansion', () => {
const plan = resolveInstallPlan({ moduleIds: ['security'] });
assert.ok(plan.selectedModuleIds.includes('security'), 'Should include requested module');