@iyulab/mloop-mcp 0.1.0

package/build/tools/promote.js

@@ -0,0 +1,72 @@
+/**
+ * mloop_promote tool - Promote experiment to production
+ */
+import { z } from 'zod';
+import { executeMloop } from '../executor.js';
+import { parseCliOutput } from '../utils/parser.js';
+import { formatError } from '../utils/errors.js';
+export const promoteSchema = z.object({
+    projectPath: z.string().describe('Path to MLoop project directory'),
+    experimentId: z.string().describe('Experiment ID to promote (e.g., exp-003)'),
+    modelName: z.string().optional().describe('Model name'),
+    force: z.boolean().optional().describe('Skip confirmation'),
+});
+export async function promote(params) {
+    try {
+        const { projectPath, experimentId, modelName, force } = params;
+        const args = ['promote', experimentId];
+        if (modelName) {
+            args.push('--name', modelName);
+        }
+        if (force) {
+            args.push('--force');
+        }
+        const result = await executeMloop(args, {
+            cwd: projectPath,
+        });
+        const output = parseCliOutput(result.stdout);
+        return {
+            content: [{
+                    type: 'text',
+                    text: output || `Experiment ${experimentId} promoted to production successfully.`,
+                }],
+        };
+    }
+    catch (error) {
+        return {
+            content: [{
+                    type: 'text',
+                    text: formatError(error),
+                }],
+            isError: true,
+        };
+    }
+}
+export const promoteTool = {
+    name: 'mloop_promote',
+    description: 'Promote an experiment to production. The experiment\'s model will be used for predictions.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            projectPath: {
+                type: 'string',
+                description: 'Path to MLoop project directory',
+            },
+            experimentId: {
+                type: 'string',
+                description: 'Experiment ID to promote (e.g., exp-003)',
+            },
+            modelName: {
+                type: 'string',
+                description: 'Model name for namespacing',
+            },
+            force: {
+                type: 'boolean',
+                description: 'Skip confirmation prompt',
+            },
+        },
+        required: ['projectPath', 'experimentId'],
+    },
+    handler: promote,
+};
+//# sourceMappingURL=promote.js.map

package/build/tools/promote.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"promote.js","sourceRoot":"","sources":["../../src/tools/promote.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACnE,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;IAC7E,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;IACvD,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;CAC5D,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,MAAqB;IACjD,IAAI,CAAC;QACH,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAE/D,MAAM,IAAI,GAAa,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QACjD,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE;YACtC,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM,IAAI,cAAc,YAAY,uCAAuC;iBAClF,CAAC;SACH,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC;iBACzB,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,4FAA4F;IACzG,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;aAC/C;YACD,YAAY,EAAE;gBACZ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0CAA0C;aACxD;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4BAA4B;aAC1C;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,0BAA0B;aACxC;SACF;QACD,QAAQ,EAAE,CAAC,aAAa,EAAE,cAAc,CAAC;KAC1C;IACD,OAAO,EAAE,OAAO;CACjB,CAAC"}

package/build/tools/serve.d.ts

@@ -0,0 +1,49 @@
+/**
+ * mloop_serve tool - Start REST API server
+ */
+import { z } from 'zod';
+export declare const serveSchema: z.ZodObject<{
+    projectPath: z.ZodString;
+    port: z.ZodOptional<z.ZodNumber>;
+    host: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    projectPath: string;
+    port?: number | undefined;
+    host?: string | undefined;
+}, {
+    projectPath: string;
+    port?: number | undefined;
+    host?: string | undefined;
+}>;
+export type ServeParams = z.infer<typeof serveSchema>;
+export declare function serve(params: ServeParams): Promise<{
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    isError?: boolean;
+}>;
+export declare const serveTool: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            projectPath: {
+                type: string;
+                description: string;
+            };
+            port: {
+                type: string;
+                description: string;
+            };
+            host: {
+                type: string;
+                description: string;
+            };
+        };
+        required: string[];
+    };
+    handler: typeof serve;
+};
+//# sourceMappingURL=serve.d.ts.map

package/build/tools/serve.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"serve.d.ts","sourceRoot":"","sources":["../../src/tools/serve.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,WAAW;;;;;;;;;;;;EAItB,CAAC;AAEH,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,WAAW,CAAC,CAAC;AAEtD,wBAAsB,KAAK,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC,CA8C/H;AAED,eAAO,MAAM,SAAS;;;;;;;;;;;;;;;;;;;;;;CAsBrB,CAAC"}

package/build/tools/serve.js

@@ -0,0 +1,79 @@
+/**
+ * mloop_serve tool - Start REST API server
+ */
+import { z } from 'zod';
+import { executeMloop } from '../executor.js';
+import { parseCliOutput } from '../utils/parser.js';
+import { formatError } from '../utils/errors.js';
+export const serveSchema = z.object({
+    projectPath: z.string().describe('Path to MLoop project directory'),
+    port: z.number().optional().describe('Port number (default: 5000)'),
+    host: z.string().optional().describe('Host to bind to'),
+});
+export async function serve(params) {
+    try {
+        const { projectPath, port, host } = params;
+        const args = ['serve'];
+        if (port) {
+            args.push('--port', String(port));
+        }
+        if (host) {
+            args.push('--host', host);
+        }
+        // Note: serve is a long-running command, so we use a short timeout
+        // and expect it to "fail" when it starts successfully
+        const result = await executeMloop(args, {
+            cwd: projectPath,
+            timeout: 5000, // 5 seconds - just enough to start
+        });
+        const output = parseCliOutput(result.stdout);
+        return {
+            content: [{
+                    type: 'text',
+                    text: output || `API server starting on port ${port ?? 5000}...`,
+                }],
+        };
+    }
+    catch (error) {
+        // If it's a timeout, the server likely started successfully
+        if (error instanceof Error && error.name === 'TimeoutError') {
+            return {
+                content: [{
+                        type: 'text',
+                        text: `API server started on port ${params.port ?? 5000}. The server is running in the background.`,
+                    }],
+            };
+        }
+        return {
+            content: [{
+                    type: 'text',
+                    text: formatError(error),
+                }],
+            isError: true,
+        };
+    }
+}
+export const serveTool = {
+    name: 'mloop_serve',
+    description: 'Start the MLoop REST API server for serving predictions via HTTP.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            projectPath: {
+                type: 'string',
+                description: 'Path to MLoop project directory',
+            },
+            port: {
+                type: 'number',
+                description: 'Port number to listen on (default: 5000)',
+            },
+            host: {
+                type: 'string',
+                description: 'Host address to bind to',
+            },
+        },
+        required: ['projectPath'],
+    },
+    handler: serve,
+};
+//# sourceMappingURL=serve.js.map

package/build/tools/serve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"serve.js","sourceRoot":"","sources":["../../src/tools/serve.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IACnE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;CACxD,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,MAAmB;IAC7C,IAAI,CAAC;QACH,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;QAE3C,MAAM,IAAI,GAAa,CAAC,OAAO,CAAC,CAAC;QACjC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,mEAAmE;QACnE,sDAAsD;QACtD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE;YACtC,GAAG,EAAE,WAAW;YAChB,OAAO,EAAE,IAAI,EAAE,mCAAmC;SACnD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM,IAAI,+BAA+B,IAAI,IAAI,IAAI,KAAK;iBACjE,CAAC;SACH,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,4DAA4D;QAC5D,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAC5D,OAAO;gBACL,OAAO,EAAE,CAAC;wBACR,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,8BAA8B,MAAM,CAAC,IAAI,IAAI,IAAI,4CAA4C;qBACpG,CAAC;aACH,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC;iBACzB,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,mEAAmE;IAChF,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;aAC/C;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0CAA0C;aACxD;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yBAAyB;aACvC;SACF;QACD,QAAQ,EAAE,CAAC,aAAa,CAAC;KAC1B;IACD,OAAO,EAAE,KAAK;CACf,CAAC"}

package/build/tools/status.d.ts

@@ -0,0 +1,35 @@
+/**
+ * mloop_status tool - Project status overview
+ */
+import { z } from 'zod';
+export declare const statusSchema: z.ZodObject<{
+    projectPath: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    projectPath: string;
+}, {
+    projectPath: string;
+}>;
+export type StatusParams = z.infer<typeof statusSchema>;
+export declare function status(params: StatusParams): Promise<{
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    isError?: boolean;
+}>;
+export declare const statusTool: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            projectPath: {
+                type: string;
+                description: string;
+            };
+        };
+        required: string[];
+    };
+    handler: typeof status;
+};
+//# sourceMappingURL=status.d.ts.map

package/build/tools/status.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"status.d.ts","sourceRoot":"","sources":["../../src/tools/status.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,YAAY;;;;;;EAEvB,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAExD,wBAAsB,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC,CA2BjI;AAED,eAAO,MAAM,UAAU;;;;;;;;;;;;;;CActB,CAAC"}

package/build/tools/status.js

@@ -0,0 +1,51 @@
+/**
+ * mloop_status tool - Project status overview
+ */
+import { z } from 'zod';
+import { executeMloop } from '../executor.js';
+import { parseCliOutput } from '../utils/parser.js';
+import { formatError } from '../utils/errors.js';
+export const statusSchema = z.object({
+    projectPath: z.string().describe('Path to MLoop project directory'),
+});
+export async function status(params) {
+    try {
+        const { projectPath } = params;
+        const args = ['status'];
+        const result = await executeMloop(args, {
+            cwd: projectPath,
+        });
+        const output = parseCliOutput(result.stdout);
+        return {
+            content: [{
+                    type: 'text',
+                    text: output || 'No status information available.',
+                }],
+        };
+    }
+    catch (error) {
+        return {
+            content: [{
+                    type: 'text',
+                    text: formatError(error),
+                }],
+            isError: true,
+        };
+    }
+}
+export const statusTool = {
+    name: 'mloop_status',
+    description: 'Show MLoop project status including models, experiments, and production deployments.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            projectPath: {
+                type: 'string',
+                description: 'Path to MLoop project directory',
+            },
+        },
+        required: ['projectPath'],
+    },
+    handler: status,
+};
+//# sourceMappingURL=status.js.map

package/build/tools/status.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"status.js","sourceRoot":"","sources":["../../src/tools/status.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;CACpE,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,MAAM,CAAC,MAAoB;IAC/C,IAAI,CAAC;QACH,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAE/B,MAAM,IAAI,GAAa,CAAC,QAAQ,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE;YACtC,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM,IAAI,kCAAkC;iBACnD,CAAC;SACH,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC;iBACzB,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,IAAI,EAAE,cAAc;IACpB,WAAW,EAAE,sFAAsF;IACnG,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;aAC/C;SACF;QACD,QAAQ,EAAE,CAAC,aAAa,CAAC;KAC1B;IACD,OAAO,EAAE,MAAM;CAChB,CAAC"}

package/build/tools/train.d.ts

@@ -0,0 +1,78 @@
+/**
+ * mloop_train tool - Train ML models with AutoML
+ */
+import { z } from 'zod';
+export declare const trainSchema: z.ZodObject<{
+    projectPath: z.ZodString;
+    dataFile: z.ZodOptional<z.ZodString>;
+    label: z.ZodOptional<z.ZodString>;
+    task: z.ZodOptional<z.ZodEnum<["binary-classification", "multiclass-classification", "regression"]>>;
+    time: z.ZodOptional<z.ZodNumber>;
+    metric: z.ZodOptional<z.ZodString>;
+    modelName: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    projectPath: string;
+    dataFile?: string | undefined;
+    label?: string | undefined;
+    task?: "binary-classification" | "multiclass-classification" | "regression" | undefined;
+    time?: number | undefined;
+    metric?: string | undefined;
+    modelName?: string | undefined;
+}, {
+    projectPath: string;
+    dataFile?: string | undefined;
+    label?: string | undefined;
+    task?: "binary-classification" | "multiclass-classification" | "regression" | undefined;
+    time?: number | undefined;
+    metric?: string | undefined;
+    modelName?: string | undefined;
+}>;
+export type TrainParams = z.infer<typeof trainSchema>;
+export declare function train(params: TrainParams): Promise<{
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    isError?: boolean;
+}>;
+export declare const trainTool: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            projectPath: {
+                type: string;
+                description: string;
+            };
+            dataFile: {
+                type: string;
+                description: string;
+            };
+            label: {
+                type: string;
+                description: string;
+            };
+            task: {
+                type: string;
+                enum: string[];
+                description: string;
+            };
+            time: {
+                type: string;
+                description: string;
+            };
+            metric: {
+                type: string;
+                description: string;
+            };
+            modelName: {
+                type: string;
+                description: string;
+            };
+        };
+        required: string[];
+    };
+    handler: typeof train;
+};
+//# sourceMappingURL=train.d.ts.map

package/build/tools/train.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"train.d.ts","sourceRoot":"","sources":["../../src/tools/train.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;EAUtB,CAAC;AAEH,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,WAAW,CAAC,CAAC;AAEtD,wBAAsB,KAAK,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC,CAwC/H;AAED,eAAO,MAAM,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCrB,CAAC"}

package/build/tools/train.js

@@ -0,0 +1,101 @@
+/**
+ * mloop_train tool - Train ML models with AutoML
+ */
+import { z } from 'zod';
+import { executeMloop, buildArgsWithPositional } from '../executor.js';
+import { parseCliOutput } from '../utils/parser.js';
+import { formatError } from '../utils/errors.js';
+export const trainSchema = z.object({
+    projectPath: z.string().describe('Path to MLoop project directory'),
+    dataFile: z.string().optional().describe('Training data file (default: datasets/train.csv)'),
+    label: z.string().optional().describe('Label column name'),
+    task: z.enum(['binary-classification', 'multiclass-classification', 'regression'])
+        .optional()
+        .describe('ML task type'),
+    time: z.number().positive().optional().describe('Training time in seconds'),
+    metric: z.string().optional().describe('Optimization metric'),
+    modelName: z.string().optional().describe('Model name (default: "default")'),
+});
+export async function train(params) {
+    try {
+        const { projectPath, dataFile, label, ...options } = params;
+        // Build positional arguments
+        const positional = [];
+        if (dataFile)
+            positional.push(dataFile);
+        if (label)
+            positional.push(label);
+        // Map parameters to CLI flags
+        const cliParams = {};
+        if (options.task)
+            cliParams['task'] = options.task;
+        if (options.time)
+            cliParams['time'] = options.time;
+        if (options.metric)
+            cliParams['metric'] = options.metric;
+        if (options.modelName)
+            cliParams['name'] = options.modelName;
+        const args = buildArgsWithPositional('train', positional, cliParams);
+        const result = await executeMloop(args, {
+            cwd: projectPath,
+            timeout: 600_000, // 10 minutes for training
+        });
+        const output = parseCliOutput(result.stdout);
+        return {
+            content: [{
+                    type: 'text',
+                    text: output || 'Training completed successfully.',
+                }],
+        };
+    }
+    catch (error) {
+        return {
+            content: [{
+                    type: 'text',
+                    text: formatError(error),
+                }],
+            isError: true,
+        };
+    }
+}
+export const trainTool = {
+    name: 'mloop_train',
+    description: 'Train an ML model using AutoML. Runs mloop train command with specified parameters.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            projectPath: {
+                type: 'string',
+                description: 'Path to MLoop project directory',
+            },
+            dataFile: {
+                type: 'string',
+                description: 'Training data file path (default: datasets/train.csv)',
+            },
+            label: {
+                type: 'string',
+                description: 'Label column name for training',
+            },
+            task: {
+                type: 'string',
+                enum: ['binary-classification', 'multiclass-classification', 'regression'],
+                description: 'ML task type',
+            },
+            time: {
+                type: 'number',
+                description: 'Training time limit in seconds',
+            },
+            metric: {
+                type: 'string',
+                description: 'Optimization metric (e.g., Accuracy, F1Score, RSquared)',
+            },
+            modelName: {
+                type: 'string',
+                description: 'Model name for namespacing (default: "default")',
+            },
+        },
+        required: ['projectPath'],
+    },
+    handler: train,
+};
+//# sourceMappingURL=train.js.map

package/build/tools/train.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"train.js","sourceRoot":"","sources":["../../src/tools/train.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACnE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;IAC5F,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAC1D,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,uBAAuB,EAAE,2BAA2B,EAAE,YAAY,CAAC,CAAC;SAC/E,QAAQ,EAAE;SACV,QAAQ,CAAC,cAAc,CAAC;IAC3B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IAC3E,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IAC7D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;CAC7E,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,MAAmB;IAC7C,IAAI,CAAC;QACH,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,MAAM,CAAC;QAE5D,6BAA6B;QAC7B,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,IAAI,QAAQ;YAAE,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,KAAK;YAAE,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAElC,8BAA8B;QAC9B,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,IAAI,OAAO,CAAC,IAAI;YAAE,SAAS,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;QACnD,IAAI,OAAO,CAAC,IAAI;YAAE,SAAS,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;QACnD,IAAI,OAAO,CAAC,MAAM;YAAE,SAAS,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACzD,IAAI,OAAO,CAAC,SAAS;YAAE,SAAS,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;QAE7D,MAAM,IAAI,GAAG,uBAAuB,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE;YACtC,GAAG,EAAE,WAAW;YAChB,OAAO,EAAE,OAAO,EAAE,0BAA0B;SAC7C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM,IAAI,kCAAkC;iBACnD,CAAC;SACH,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC;iBACzB,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,qFAAqF;IAClG,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;aAC/C;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,uDAAuD;aACrE;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gCAAgC;aAC9C;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,uBAAuB,EAAE,2BAA2B,EAAE,YAAY,CAAC;gBAC1E,WAAW,EAAE,cAAc;aAC5B;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gCAAgC;aAC9C;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yDAAyD;aACvE;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iDAAiD;aAC/D;SACF;QACD,QAAQ,EAAE,CAAC,aAAa,CAAC;KAC1B;IACD,OAAO,EAAE,KAAK;CACf,CAAC"}

package/build/utils/errors.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Error handling utilities for mloop-mcp
+ */
+/**
+ * Remove ANSI escape codes from string
+ */
+export declare function stripAnsi(text: string): string;
+/**
+ * Custom error for CLI execution failures
+ */
+export declare class CliExecutionError extends Error {
+    readonly exitCode: number;
+    readonly stdout: string;
+    readonly stderr: string;
+    constructor(message: string, exitCode: number, stdout: string, stderr: string);
+}
+/**
+ * Custom error for timeout
+ */
+export declare class TimeoutError extends Error {
+    readonly timeoutMs: number;
+    constructor(message: string, timeoutMs: number);
+}
+/**
+ * Format error for MCP response
+ */
+export declare function formatError(error: unknown): string;
+//# sourceMappingURL=errors.d.ts.map

package/build/utils/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../src/utils/errors.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;GAEG;AACH,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAG9C;AAED;;GAEG;AACH,qBAAa,iBAAkB,SAAQ,KAAK;aAGxB,QAAQ,EAAE,MAAM;aAChB,MAAM,EAAE,MAAM;aACd,MAAM,EAAE,MAAM;gBAH9B,OAAO,EAAE,MAAM,EACC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,MAAM;CAKjC;AAED;;GAEG;AACH,qBAAa,YAAa,SAAQ,KAAK;aAGnB,SAAS,EAAE,MAAM;gBADjC,OAAO,EAAE,MAAM,EACC,SAAS,EAAE,MAAM;CAKpC;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAuBlD"}

package/build/utils/errors.js

@@ -0,0 +1,59 @@
+/**
+ * Error handling utilities for mloop-mcp
+ */
+/**
+ * Remove ANSI escape codes from string
+ */
+export function stripAnsi(text) {
+    // eslint-disable-next-line no-control-regex
+    return text.replace(/\x1B\[[0-9;]*[a-zA-Z]/g, '');
+}
+/**
+ * Custom error for CLI execution failures
+ */
+export class CliExecutionError extends Error {
+    exitCode;
+    stdout;
+    stderr;
+    constructor(message, exitCode, stdout, stderr) {
+        super(message);
+        this.exitCode = exitCode;
+        this.stdout = stdout;
+        this.stderr = stderr;
+        this.name = 'CliExecutionError';
+    }
+}
+/**
+ * Custom error for timeout
+ */
+export class TimeoutError extends Error {
+    timeoutMs;
+    constructor(message, timeoutMs) {
+        super(message);
+        this.timeoutMs = timeoutMs;
+        this.name = 'TimeoutError';
+    }
+}
+/**
+ * Format error for MCP response
+ */
+export function formatError(error) {
+    if (error instanceof CliExecutionError) {
+        const parts = [`CLI Error (exit code ${error.exitCode})`];
+        if (error.stderr) {
+            parts.push(`\nStderr: ${stripAnsi(error.stderr)}`);
+        }
+        if (error.stdout) {
+            parts.push(`\nStdout: ${stripAnsi(error.stdout)}`);
+        }
+        return parts.join('');
+    }
+    if (error instanceof TimeoutError) {
+        return `Timeout: Command did not complete within ${error.timeoutMs}ms`;
+    }
+    if (error instanceof Error) {
+        return error.message;
+    }
+    return String(error);
+}
+//# sourceMappingURL=errors.js.map

package/build/utils/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/utils/errors.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,IAAY;IACpC,4CAA4C;IAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;AACpD,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,iBAAkB,SAAQ,KAAK;IAGxB;IACA;IACA;IAJlB,YACE,OAAe,EACC,QAAgB,EAChB,MAAc,EACd,MAAc;QAE9B,KAAK,CAAC,OAAO,CAAC,CAAC;QAJC,aAAQ,GAAR,QAAQ,CAAQ;QAChB,WAAM,GAAN,MAAM,CAAQ;QACd,WAAM,GAAN,MAAM,CAAQ;QAG9B,IAAI,CAAC,IAAI,GAAG,mBAAmB,CAAC;IAClC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,KAAK;IAGnB;IAFlB,YACE,OAAe,EACC,SAAiB;QAEjC,KAAK,CAAC,OAAO,CAAC,CAAC;QAFC,cAAS,GAAT,SAAS,CAAQ;QAGjC,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC7B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,CAAC,wBAAwB,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC;QAE1D,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,aAAa,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,aAAa,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,YAAY,YAAY,EAAE,CAAC;QAClC,OAAO,4CAA4C,KAAK,CAAC,SAAS,IAAI,CAAC;IACzE,CAAC;IAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IAED,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC"}

package/build/utils/parser.d.ts

@@ -0,0 +1,23 @@
+/**
+ * CLI output parsing utilities
+ */
+/**
+ * Parse CLI output and clean it for MCP response
+ */
+export declare function parseCliOutput(output: string): string;
+/**
+ * Try to parse JSON from CLI output
+ * Returns null if parsing fails
+ */
+export declare function tryParseJson<T>(output: string): T | null;
+/**
+ * Extract table data from CLI output
+ * Parses simple table format with headers
+ */
+export declare function parseTable(output: string): Array<Record<string, string>>;
+/**
+ * Extract key-value pairs from CLI output
+ * Handles formats like "Key: Value" or "Key = Value"
+ */
+export declare function parseKeyValues(output: string): Record<string, string>;
+//# sourceMappingURL=parser.d.ts.map

package/build/utils/parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parser.d.ts","sourceRoot":"","sources":["../../src/utils/parser.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH;;GAEG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAErD;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,GAAG,CAAC,GAAG,IAAI,CAOxD;AAED;;;GAGG;AACH,wBAAgB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAgCxE;AA+BD;;;GAGG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAerE"}