micro-models-agent 0.28.9 → 0.29.0

package/dist/modules/mcp/module.js

@@ -0,0 +1,146 @@
+import { MCPClient } from "./client";
+import { MCPRegistry } from "./registry";
+function sanitizeToolName(server, tool) {
+    const safeServer = server.replace(/[^a-zA-Z0-9_-]/g, "_");
+    const safeTool = tool.replace(/[^a-zA-Z0-9_-]/g, "_");
+    return `mcp__${safeServer}__${safeTool}`;
+}
+function convertInputSchema(schema) {
+    if (schema && typeof schema === "object" && "properties" in schema) {
+        const required = Array.isArray(schema.required)
+            ? schema.required
+            : [];
+        return {
+            type: "object",
+            properties: schema.properties,
+            required: required.length > 0 ? required : undefined,
+        };
+    }
+    return { type: "object", properties: {} };
+}
+export class MCPModule {
+    config;
+    name = "mcp";
+    connections = new Map();
+    discovered = [];
+    initialized = false;
+    initError = null;
+    constructor(config) {
+        this.config = config;
+    }
+    async initialize() {
+        if (this.initialized)
+            return;
+        this.initialized = true;
+        const servers = this.config.mcpServers || {};
+        const serverEntries = Object.entries(servers).filter(([, cfg]) => cfg.enabled !== false);
+        if (serverEntries.length === 0)
+            return;
+        const registry = new MCPRegistry();
+        for (const [key, cfg] of serverEntries) {
+            const serverConfig = {
+                name: cfg.name || key,
+                command: cfg.command,
+                args: cfg.args,
+                env: cfg.env,
+                transport: cfg.transport,
+                url: cfg.url,
+                headers: cfg.headers,
+                timeout: cfg.timeout,
+            };
+            registry.register(serverConfig);
+        }
+        for (const serverName of registry.list()) {
+            const serverConfig = registry.get(serverName);
+            const client = new MCPClient(serverConfig);
+            try {
+                await client.connect();
+                const tools = await client.listTools();
+                this.connections.set(serverName, client);
+                for (const tool of tools) {
+                    this.discovered.push({
+                        serverName,
+                        toolName: tool.name,
+                        description: tool.description || `Tool on MCP server "${serverName}"`,
+                        inputSchema: tool.inputSchema || {},
+                    });
+                }
+            }
+            catch (e) {
+                this.initError = `MCP server "${serverName}" init failed: ${e.message}`;
+                await client.disconnect().catch(() => { });
+            }
+        }
+    }
+    getSystemPromptBlock() {
+        if (this.discovered.length === 0)
+            return null;
+        const lines = ["---", "MCP servers available:"];
+        const byServer = new Map();
+        for (const d of this.discovered) {
+            const list = byServer.get(d.serverName) || [];
+            list.push(d);
+            byServer.set(d.serverName, list);
+        }
+        for (const [server, tools] of byServer) {
+            lines.push(`  [${server}]`);
+            for (const t of tools) {
+                const toolRef = sanitizeToolName(server, t.toolName);
+                lines.push(`    - ${toolRef}: ${t.description}`);
+            }
+        }
+        lines.push("---");
+        return {
+            content: lines.join("\n"),
+            priority: "high",
+            essential: false,
+            estimatedTokens: this.discovered.length * 20 + 20,
+        };
+    }
+    getToolDefinitions() {
+        return this.discovered.map((d) => ({
+            name: sanitizeToolName(d.serverName, d.toolName),
+            description: `${d.description} [MCP server: ${d.serverName}]`,
+            parameters: convertInputSchema(d.inputSchema),
+            tags: ["code", "research"],
+            handler: async (_ctx, args) => {
+                const client = this.connections.get(d.serverName);
+                if (!client) {
+                    return {
+                        success: false,
+                        output: `MCP server "${d.serverName}" not connected`,
+                    };
+                }
+                try {
+                    if (!client.isConnected()) {
+                        await client.connect();
+                    }
+                    const result = await client.callTool(d.toolName, args);
+                    return {
+                        success: true,
+                        output: typeof result === "string"
+                            ? result
+                            : JSON.stringify(result, null, 2),
+                    };
+                }
+                catch (e) {
+                    return { success: false, output: `MCP call failed: ${e.message}` };
+                }
+            },
+        }));
+    }
+    getPlugin() {
+        return {
+            name: "mcp_module",
+            priority: 0,
+            onSessionEnd: async () => {
+                for (const client of this.connections.values()) {
+                    try {
+                        await client.disconnect();
+                    }
+                    catch { }
+                }
+            },
+        };
+    }
+}

package/dist/modules/mcp/registry.js

@@ -0,0 +1,15 @@
+export class MCPRegistry {
+    servers = new Map();
+    register(config) {
+        this.servers.set(config.name, config);
+    }
+    get(name) {
+        return this.servers.get(name);
+    }
+    list() {
+        return Array.from(this.servers.keys()).sort();
+    }
+    remove(name) {
+        this.servers.delete(name);
+    }
+}

package/dist/modules/memory/index.js

@@ -0,0 +1 @@
+export { MemoryStore } from './store';

package/dist/modules/memory/module.js

@@ -0,0 +1,48 @@
+import { homedir } from 'os';
+import { join } from 'path';
+import { MemoryStore } from './store';
+import { rememberTool } from '../../tools/remember';
+import { recallTool } from '../../tools/recall';
+export class MemoryModule {
+    name = 'memory';
+    store;
+    constructor(memoryDir) {
+        const dir = memoryDir || join(homedir(), '.mma', 'memory');
+        this.store = new MemoryStore(dir);
+    }
+    getSystemPromptBlock() {
+        const prefs = this.store.getPreferences();
+        if (Object.keys(prefs).length === 0)
+            return null;
+        const prefStr = Object.entries(prefs)
+            .map(([k, v]) => `${k}=${v}`)
+            .join(', ');
+        return {
+            content: `User preferences: ${prefStr}`,
+            priority: 'normal',
+            essential: false,
+            estimatedTokens: Math.ceil(prefStr.length / 4) + 10,
+        };
+    }
+    getToolDefinitions() {
+        return [rememberTool, recallTool];
+    }
+    getPlugin() {
+        return {
+            name: 'memory',
+            isBuiltin: true,
+            onBuildPrompt: () => {
+                const prefs = this.store.getPreferences();
+                if (Object.keys(prefs).length === 0)
+                    return null;
+                const prefStr = Object.entries(prefs)
+                    .map(([k, v]) => `${k}=${v}`)
+                    .join(', ');
+                return `User preferences: ${prefStr}`;
+            },
+        };
+    }
+    getStore() {
+        return this.store;
+    }
+}

package/dist/modules/memory/search.js

@@ -0,0 +1,40 @@
+import { readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+const MEMORY_FILES = ['conventions', 'decisions', 'errors', 'facts'];
+export class MemorySearch {
+    memoryDir;
+    constructor(memoryDir) {
+        this.memoryDir = memoryDir;
+    }
+    query(query) {
+        const results = [];
+        const lowerQuery = query.toLowerCase();
+        for (const name of MEMORY_FILES) {
+            const path = join(this.memoryDir, `${name}.md`);
+            if (!existsSync(path))
+                continue;
+            const content = readFileSync(path, 'utf-8');
+            const lines = content.split('\n');
+            for (const line of lines) {
+                if (line.toLowerCase().includes(lowerQuery)) {
+                    results.push({ file: name, match: line.trim() });
+                }
+            }
+        }
+        // Search preferences.json
+        const prefsPath = join(this.memoryDir, 'preferences.json');
+        if (existsSync(prefsPath)) {
+            try {
+                const prefs = JSON.parse(readFileSync(prefsPath, 'utf-8'));
+                for (const [key, value] of Object.entries(prefs)) {
+                    const searchStr = `${key}=${value}`;
+                    if (searchStr.toLowerCase().includes(lowerQuery)) {
+                        results.push({ file: 'preferences', match: `${key} = ${value}` });
+                    }
+                }
+            }
+            catch { /* skip */ }
+        }
+        return results;
+    }
+}

package/dist/modules/memory/store.js

@@ -0,0 +1,65 @@
+import { readFileSync, writeFileSync, appendFileSync, existsSync, mkdirSync } from 'fs';
+import { join } from 'path';
+import { MemorySearch } from './search';
+const MEMORY_FILES = ['conventions', 'decisions', 'errors', 'facts'];
+export class MemoryStore {
+    memoryDir;
+    constructor(memoryDir) {
+        this.memoryDir = memoryDir;
+        this.ensureDir();
+        for (const name of MEMORY_FILES) {
+            const path = join(this.memoryDir, `${name}.md`);
+            if (!existsSync(path)) {
+                writeFileSync(path, `# ${name.charAt(0).toUpperCase() + name.slice(1)}\n\n`, 'utf-8');
+            }
+        }
+    }
+    ensureDir() {
+        if (!existsSync(this.memoryDir)) {
+            mkdirSync(this.memoryDir, { recursive: true });
+        }
+    }
+    read(name) {
+        const path = join(this.memoryDir, `${name}.md`);
+        if (!existsSync(path))
+            return '';
+        return readFileSync(path, 'utf-8');
+    }
+    append(name, entry) {
+        const path = join(this.memoryDir, `${name}.md`);
+        const timestamp = new Date().toISOString().replace('T', ' ').slice(0, 19);
+        const formatted = `- **${timestamp}** — ${entry}\n`;
+        appendFileSync(path, formatted, 'utf-8');
+    }
+    search(query) {
+        const searchModule = new MemorySearch(this.memoryDir);
+        return searchModule.query(query);
+    }
+    prefsPath() {
+        return join(this.memoryDir, 'preferences.json');
+    }
+    getPreferences() {
+        const path = this.prefsPath();
+        if (!existsSync(path))
+            return {};
+        try {
+            return JSON.parse(readFileSync(path, 'utf-8'));
+        }
+        catch {
+            return {};
+        }
+    }
+    setPreference(key, value) {
+        const prefs = this.getPreferences();
+        prefs[key] = value;
+        writeFileSync(this.prefsPath(), JSON.stringify(prefs, null, 2), 'utf-8');
+    }
+    deletePreference(key) {
+        const prefs = this.getPreferences();
+        if (!(key in prefs))
+            return false;
+        delete prefs[key];
+        writeFileSync(this.prefsPath(), JSON.stringify(prefs, null, 2), 'utf-8');
+        return true;
+    }
+}

package/dist/modules/pipelines/engine.js

@@ -0,0 +1,60 @@
+import { t } from '../../i18n/index';
+export class PipelineEngine {
+    stepStatus = new Map();
+    resolveDependencies(steps) {
+        const visited = new Set();
+        const inStack = new Set();
+        const order = [];
+        function visit(stepId, stepMap) {
+            if (inStack.has(stepId))
+                throw new Error(t('pipeline.circular', { stepId }));
+            if (visited.has(stepId))
+                return;
+            inStack.add(stepId);
+            const step = stepMap.get(stepId);
+            if (step?.depends_on) {
+                for (const dep of step.depends_on) {
+                    visit(dep, stepMap);
+                }
+            }
+            inStack.delete(stepId);
+            visited.add(stepId);
+            order.push(stepId);
+        }
+        const stepMap = new Map(steps.map(s => [s.id, s]));
+        for (const step of steps) {
+            if (!visited.has(step.id)) {
+                visit(step.id, stepMap);
+            }
+        }
+        return order;
+    }
+    getReadySteps(steps, completed) {
+        return steps.filter(s => {
+            if (completed.has(s.id))
+                return false;
+            if (this.stepStatus.get(s.id) === 'running')
+                return false;
+            if (this.stepStatus.get(s.id) === 'failed')
+                return false;
+            if (!s.depends_on || s.depends_on.length === 0)
+                return true;
+            return s.depends_on.every(d => completed.has(d));
+        });
+    }
+    setStepStatus(stepId, status) {
+        this.stepStatus.set(stepId, status);
+    }
+    getStepStatus(stepId) {
+        return this.stepStatus.get(stepId) || 'pending';
+    }
+    isComplete(stepIds) {
+        return stepIds.every(id => {
+            const s = this.stepStatus.get(id);
+            return s === 'done' || s === 'skipped';
+        });
+    }
+    reset() {
+        this.stepStatus.clear();
+    }
+}

package/dist/modules/pipelines/index.js

@@ -0,0 +1,3 @@
+export { PipelineParser } from './parser';
+export { PipelineEngine } from './engine';
+export { TemplateEngine } from './template';

package/dist/modules/pipelines/parser.js

@@ -0,0 +1,53 @@
+import { t } from '../../i18n/index';
+export class PipelineParser {
+    static parse(yaml) {
+        const lines = yaml.split('\n').filter(l => l.trim());
+        const result = {};
+        let currentSection = null;
+        let currentSteps = [];
+        for (const line of lines) {
+            if (line.match(/^\w+:/) && !line.startsWith(' ') && !line.startsWith('  -')) {
+                const [key, ...valParts] = line.split(':');
+                const val = valParts.join(':').trim();
+                if (key === 'steps') {
+                    currentSection = 'steps';
+                }
+                else {
+                    result[key] = val;
+                    currentSection = null;
+                }
+            }
+            else if (currentSection === 'steps') {
+                const stepMatch = line.match(/^\s+- id:\s*(\S+)/);
+                if (stepMatch) {
+                    currentSteps.push({ id: stepMatch[1] });
+                }
+                else if (currentSteps.length > 0) {
+                    const lastStep = currentSteps[currentSteps.length - 1];
+                    const kvMatch = line.match(/^\s+(\w+):\s*(.+)$/);
+                    if (kvMatch) {
+                        const key = kvMatch[1];
+                        let val = kvMatch[2].trim();
+                        if (val === 'true')
+                            val = true;
+                        else if (val === 'false')
+                            val = false;
+                        else if (val.startsWith('[') && val.endsWith(']')) {
+                            val = val.slice(1, -1).split(',').map((s) => s.trim().replace(/^['"]|['"]$/g, ''));
+                        }
+                        lastStep[key] = val;
+                    }
+                }
+            }
+        }
+        if (!result.name || currentSteps.length === 0) {
+            throw new Error(t('pipeline.invalid'));
+        }
+        for (const step of currentSteps) {
+            if (!step.id || !step.agent || !step.prompt) {
+                throw new Error(t('pipeline.step_missing_fields', { step: JSON.stringify(step) }));
+            }
+        }
+        return { name: result.name, steps: currentSteps };
+    }
+}

package/dist/modules/pipelines/template.js

@@ -0,0 +1,14 @@
+export class TemplateEngine {
+    static render(template, params, stepOutputs) {
+        let result = template;
+        for (const [key, val] of Object.entries(params)) {
+            result = result.replace(new RegExp(`\\{${key}\\}`, 'g'), val);
+        }
+        if (stepOutputs) {
+            result = result.replace(/\{\{steps\.(\w+)\.output\}\}/g, (_, stepId) => {
+                return stepOutputs[stepId]?.output || `{{steps.${stepId}.output}}`;
+            });
+        }
+        return result;
+    }
+}

package/dist/modules/plugins/builtin/lint-on-write.js

@@ -0,0 +1,121 @@
+import { execSync } from 'child_process';
+import { existsSync, readFileSync } from 'fs';
+import { resolve, extname, join } from 'path';
+let projectTypeCheckPromise = null;
+let projectTypeCheckTimestamp = 0;
+const TYPE_CHECK_DEBOUNCE_MS = 2000;
+export class LintOnWritePlugin {
+    name = 'lint-on-write';
+    priority = 10;
+    async onAfterTool(ctx, call, result) {
+        if (call.name !== 'write_file' && call.name !== 'edit_file') {
+            return;
+        }
+        if (!result.success) {
+            return;
+        }
+        const path = String(call.arguments.path || '');
+        if (!path)
+            return;
+        const fullPath = resolve(ctx.baseDir, path);
+        if (!existsSync(fullPath))
+            return;
+        const ext = extname(fullPath);
+        const syntaxError = this.checkSyntax(fullPath, ext, ctx.baseDir);
+        if (syntaxError) {
+            result.output += `\n\n[Syntax check failed]: ${syntaxError}`;
+            return;
+        }
+        await this.runProjectLint(ctx, result);
+        await this.runProjectTypeCheck(ctx, result);
+    }
+    checkSyntax(filePath, ext, baseDir) {
+        if (ext === '.ts' || ext === '.tsx') {
+            try {
+                execSync(`npx tsc --noEmit --skipLibCheck "${filePath}"`, { cwd: baseDir, stdio: 'pipe', timeout: 10000 });
+                return null;
+            }
+            catch (err) {
+                if (err.status === 127 || err.message.includes('not found') || err.message.includes('ENOENT')) {
+                    return null;
+                }
+                const stderr = err.stderr?.toString() || err.stdout?.toString() || '';
+                if (stderr.includes('error TS')) {
+                    const firstError = stderr.split('\n').find((line) => line.includes('error TS')) || 'TypeScript syntax error';
+                    return firstError.trim();
+                }
+                return null;
+            }
+        }
+        if (ext === '.js' || ext === '.jsx') {
+            try {
+                execSync(`node --check "${filePath}"`, { cwd: baseDir, stdio: 'pipe', timeout: 5000 });
+                return null;
+            }
+            catch (err) {
+                const stderr = err.stderr?.toString() || '';
+                const firstError = stderr.split('\n').find((line) => line.trim()) || 'JavaScript syntax error';
+                return firstError.trim();
+            }
+        }
+        return null;
+    }
+    async runProjectLint(ctx, result) {
+        try {
+            const packageJsonPath = join(ctx.baseDir, 'package.json');
+            if (!existsSync(packageJsonPath)) {
+                return;
+            }
+            const packageJson = JSON.parse(readFileSync(packageJsonPath, 'utf-8'));
+            const lintScript = packageJson.scripts?.lint;
+            if (!lintScript) {
+                return;
+            }
+            ctx.logger.debug(`Running lint: ${lintScript}`);
+            execSync(lintScript, { cwd: ctx.baseDir, stdio: 'pipe' });
+            ctx.logger.debug('Lint passed');
+        }
+        catch (err) {
+            ctx.logger.warn(`Lint failed: ${err.message}`);
+            result.output += `\n\n[Project lint failed]: ${err.message}`;
+        }
+    }
+    async runProjectTypeCheck(ctx, result) {
+        const tsconfigPath = join(ctx.baseDir, 'tsconfig.json');
+        if (!existsSync(tsconfigPath)) {
+            return;
+        }
+        const now = Date.now();
+        if (projectTypeCheckPromise && now - projectTypeCheckTimestamp < TYPE_CHECK_DEBOUNCE_MS) {
+            const error = await projectTypeCheckPromise;
+            if (error) {
+                result.output += `\n\n[Project typecheck failed]: ${error}`;
+            }
+            return;
+        }
+        projectTypeCheckTimestamp = now;
+        projectTypeCheckPromise = this.runTscCheck(ctx.baseDir);
+        const error = await projectTypeCheckPromise;
+        if (error) {
+            result.output += `\n\n[Project typecheck failed]: ${error}`;
+        }
+    }
+    async runTscCheck(baseDir) {
+        try {
+            execSync(`npx tsc --noEmit --skipLibCheck`, { cwd: baseDir, stdio: 'pipe', timeout: 30000 });
+            return null;
+        }
+        catch (err) {
+            if (err.status === 127 || err.message.includes('not found') || err.message.includes('ENOENT')) {
+                return null;
+            }
+            const stderr = err.stderr?.toString() || err.stdout?.toString() || '';
+            if (stderr.includes('error TS')) {
+                const firstError = stderr.split('\n').find((line) => line.includes('error TS')) || 'TypeScript type error';
+                return firstError.trim();
+            }
+            return null;
+        }
+    }
+}
+export const plugin = new LintOnWritePlugin();

package/dist/modules/plugins/builtin/notify.js

@@ -0,0 +1,8 @@
+export class NotifyPlugin {
+    name = 'notify';
+    priority = 5;
+    onSessionEnd(ctx) {
+        ctx.logger.info('Session ended');
+    }
+}
+export const plugin = new NotifyPlugin();

package/dist/modules/plugins/index.js

@@ -0,0 +1 @@
+export { PluginManager } from './manager';

package/dist/modules/plugins/loader.js

@@ -0,0 +1,28 @@
+import { readdirSync, existsSync, statSync } from 'fs';
+import { join } from 'path';
+import { t } from '../../i18n/index';
+export class PluginLoader {
+    loadFromDir(dirPath, pluginManager, logger) {
+        if (!existsSync(dirPath))
+            return;
+        const entries = readdirSync(dirPath);
+        for (const entry of entries) {
+            const fullPath = join(dirPath, entry);
+            if (!statSync(fullPath).isFile())
+                continue;
+            if (!entry.endsWith('.ts') && !entry.endsWith('.js'))
+                continue;
+            try {
+                const mod = require(fullPath);
+                const plugin = mod.default || mod.plugin;
+                if (plugin && plugin.name) {
+                    pluginManager.register(plugin);
+                    logger?.warn(t('plugin.loaded', { name: plugin.name }));
+                }
+            }
+            catch (e) {
+                logger?.warn(t('plugin.skipped', { entry, message: e.message }));
+            }
+        }
+    }
+}