cognitive-modules-cli 1.2.0 → 1.3.0

package/dist/modules/subagent.js

@@ -0,0 +1,185 @@
+/**
+ * Subagent - Orchestrate module calls with isolated execution contexts.
+ *
+ * Supports:
+ * - @call:module-name - Call another module
+ * - @call:module-name(args) - Call with arguments
+ * - context: fork - Isolated execution (no shared state)
+ * - context: main - Shared execution (default)
+ */
+import { findModule, getDefaultSearchPaths } from './loader.js';
+import { runModule } from './runner.js';
+// =============================================================================
+// Context Management
+// =============================================================================
+/**
+ * Create a new root context
+ */
+export function createContext(maxDepth = 5) {
+    return {
+        parentId: null,
+        depth: 0,
+        maxDepth,
+        results: {},
+        isolated: false
+    };
+}
+/**
+ * Fork context (isolated - no inherited results)
+ */
+export function forkContext(ctx, moduleName) {
+    return {
+        parentId: moduleName,
+        depth: ctx.depth + 1,
+        maxDepth: ctx.maxDepth,
+        results: {},
+        isolated: true
+    };
+}
+/**
+ * Extend context (shared - inherits results)
+ */
+export function extendContext(ctx, moduleName) {
+    return {
+        parentId: moduleName,
+        depth: ctx.depth + 1,
+        maxDepth: ctx.maxDepth,
+        results: { ...ctx.results },
+        isolated: false
+    };
+}
+// =============================================================================
+// Call Parsing
+// =============================================================================
+// Pattern to match @call:module-name or @call:module-name(args)
+const CALL_PATTERN = /@call:([a-zA-Z0-9_-]+)(?:\(([^)]*)\))?/g;
+/**
+ * Parse @call directives from text
+ */
+export function parseCalls(text) {
+    const calls = [];
+    let match;
+    // Reset regex state
+    CALL_PATTERN.lastIndex = 0;
+    while ((match = CALL_PATTERN.exec(text)) !== null) {
+        calls.push({
+            module: match[1],
+            args: match[2] || '',
+            match: match[0]
+        });
+    }
+    return calls;
+}
+/**
+ * Replace @call directives with their results
+ */
+export function substituteCallResults(text, callResults) {
+    let result = text;
+    for (const [callStr, callResult] of Object.entries(callResults)) {
+        const resultStr = typeof callResult === 'object'
+            ? JSON.stringify(callResult, null, 2)
+            : String(callResult);
+        result = result.replace(callStr, `[Result from ${callStr}]:\n${resultStr}`);
+    }
+    return result;
+}
+// =============================================================================
+// Orchestrator
+// =============================================================================
+export class SubagentOrchestrator {
+    provider;
+    running = new Set();
+    cwd;
+    constructor(provider, cwd = process.cwd()) {
+        this.provider = provider;
+        this.cwd = cwd;
+    }
+    /**
+     * Run a module with subagent support.
+     * Recursively resolves @call directives before final execution.
+     */
+    async run(moduleName, options = {}, context) {
+        const { input = {}, validateInput = true, validateOutput = true, maxDepth = 5 } = options;
+        // Initialize context
+        const ctx = context ?? createContext(maxDepth);
+        // Check depth limit
+        if (ctx.depth > ctx.maxDepth) {
+            throw new Error(`Max subagent depth (${ctx.maxDepth}) exceeded. Check for circular calls.`);
+        }
+        // Prevent circular calls
+        if (this.running.has(moduleName)) {
+            throw new Error(`Circular call detected: ${moduleName}`);
+        }
+        this.running.add(moduleName);
+        try {
+            // Find and load module
+            const searchPaths = getDefaultSearchPaths(this.cwd);
+            const module = await findModule(moduleName, searchPaths);
+            if (!module) {
+                throw new Error(`Module not found: ${moduleName}`);
+            }
+            // Check if this module wants isolated execution
+            const moduleContextMode = module.context ?? 'main';
+            // Parse @call directives from prompt
+            const calls = parseCalls(module.prompt);
+            const callResults = {};
+            // Resolve each @call directive
+            for (const call of calls) {
+                const childModule = call.module;
+                const childArgs = call.args;
+                // Prepare child input
+                const childInput = childArgs
+                    ? { query: childArgs, code: childArgs }
+                    : { ...input };
+                // Determine child context
+                const childContext = moduleContextMode === 'fork'
+                    ? forkContext(ctx, moduleName)
+                    : extendContext(ctx, moduleName);
+                // Recursively run child module
+                const childResult = await this.run(childModule, {
+                    input: childInput,
+                    validateInput: false, // Skip validation for @call args
+                    validateOutput
+                }, childContext);
+                // Store result
+                if (childResult.ok && 'data' in childResult) {
+                    callResults[call.match] = childResult.data;
+                }
+                else if ('error' in childResult) {
+                    callResults[call.match] = { error: childResult.error };
+                }
+            }
+            // Substitute call results into prompt
+            let modifiedModule = module;
+            if (Object.keys(callResults).length > 0) {
+                const modifiedPrompt = substituteCallResults(module.prompt, callResults);
+                modifiedModule = {
+                    ...module,
+                    prompt: modifiedPrompt + '\n\n## Subagent Results Available\nThe @call results have been injected above. Use them in your response.\n'
+                };
+            }
+            // Run the module
+            const result = await runModule(modifiedModule, this.provider, {
+                input,
+                verbose: false,
+                useV22: true
+            });
+            // Store result in context
+            if (result.ok && 'data' in result) {
+                ctx.results[moduleName] = result.data;
+            }
+            return result;
+        }
+        finally {
+            this.running.delete(moduleName);
+        }
+    }
+}
+/**
+ * Convenience function to run a module with subagent support
+ */
+export async function runWithSubagents(moduleName, provider, options = {}) {
+    const { cwd = process.cwd(), ...runOptions } = options;
+    const orchestrator = new SubagentOrchestrator(provider, cwd);
+    return orchestrator.run(moduleName, runOptions);
+}

package/dist/server/http.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Cognitive Modules HTTP API Server
+ *
+ * Provides RESTful API interface for workflow platform integration.
+ *
+ * Start with:
+ *   cog serve --port 8000
+ *
+ * Environment variables:
+ *   COGNITIVE_API_KEY - API Key authentication (optional)
+ *   OPENAI_API_KEY, ANTHROPIC_API_KEY, etc. - LLM provider keys
+ */
+import http from 'node:http';
+export interface ServeOptions {
+    host?: string;
+    port?: number;
+    cwd?: string;
+}
+export declare function createServer(options?: ServeOptions): http.Server;
+export declare function serve(options?: ServeOptions): Promise<void>;

package/dist/server/http.js

@@ -0,0 +1,243 @@
+/**
+ * Cognitive Modules HTTP API Server
+ *
+ * Provides RESTful API interface for workflow platform integration.
+ *
+ * Start with:
+ *   cog serve --port 8000
+ *
+ * Environment variables:
+ *   COGNITIVE_API_KEY - API Key authentication (optional)
+ *   OPENAI_API_KEY, ANTHROPIC_API_KEY, etc. - LLM provider keys
+ */
+import http from 'node:http';
+import { URL } from 'node:url';
+import { findModule, listModules, getDefaultSearchPaths } from '../modules/loader.js';
+import { runModule } from '../modules/runner.js';
+import { getProvider } from '../providers/index.js';
+// =============================================================================
+// Helpers
+// =============================================================================
+function jsonResponse(res, status, data) {
+    res.writeHead(status, {
+        'Content-Type': 'application/json',
+        'Access-Control-Allow-Origin': '*',
+        'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+        'Access-Control-Allow-Headers': 'Content-Type, Authorization',
+    });
+    res.end(JSON.stringify(data, null, 2));
+}
+function parseBody(req) {
+    return new Promise((resolve, reject) => {
+        let body = '';
+        req.on('data', (chunk) => (body += chunk));
+        req.on('end', () => resolve(body));
+        req.on('error', reject);
+    });
+}
+function verifyApiKey(req) {
+    const expectedKey = process.env.COGNITIVE_API_KEY;
+    if (!expectedKey)
+        return true; // No auth required
+    const authHeader = req.headers.authorization;
+    if (!authHeader)
+        return false;
+    const token = authHeader.startsWith('Bearer ') ? authHeader.slice(7) : authHeader;
+    return token === expectedKey;
+}
+// =============================================================================
+// Request Handlers
+// =============================================================================
+async function handleRoot(res) {
+    jsonResponse(res, 200, {
+        name: 'Cognitive Modules API',
+        version: '1.2.0',
+        docs: '/docs',
+        endpoints: {
+            run: 'POST /run',
+            modules: 'GET /modules',
+            module_info: 'GET /modules/{name}',
+            health: 'GET /health',
+        },
+    });
+}
+async function handleHealth(res) {
+    const providers = {
+        openai: Boolean(process.env.OPENAI_API_KEY),
+        anthropic: Boolean(process.env.ANTHROPIC_API_KEY),
+        minimax: Boolean(process.env.MINIMAX_API_KEY),
+        deepseek: Boolean(process.env.DEEPSEEK_API_KEY),
+        gemini: Boolean(process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY),
+        qwen: Boolean(process.env.QWEN_API_KEY || process.env.DASHSCOPE_API_KEY),
+    };
+    jsonResponse(res, 200, {
+        status: 'healthy',
+        version: '1.2.0',
+        providers,
+    });
+}
+async function handleModules(res, searchPaths) {
+    const modules = await listModules(searchPaths);
+    const moduleInfos = modules.map((m) => ({
+        name: m.name,
+        version: m.version,
+        description: m.responsibility,
+        format: m.format,
+        path: m.location,
+        responsibility: m.responsibility,
+        tier: m.tier,
+    }));
+    jsonResponse(res, 200, {
+        modules: moduleInfos,
+        count: moduleInfos.length,
+    });
+}
+async function handleModuleInfo(res, moduleName, searchPaths) {
+    const moduleData = await findModule(moduleName, searchPaths);
+    if (!moduleData) {
+        jsonResponse(res, 404, { error: `Module '${moduleName}' not found` });
+        return;
+    }
+    jsonResponse(res, 200, {
+        name: moduleData.name,
+        version: moduleData.version,
+        description: moduleData.responsibility,
+        format: moduleData.format,
+        path: moduleData.location,
+        responsibility: moduleData.responsibility,
+        tier: moduleData.tier,
+        inputSchema: moduleData.inputSchema,
+        outputSchema: moduleData.outputSchema,
+    });
+}
+async function handleRun(req, res, searchPaths) {
+    // Verify API key
+    if (!verifyApiKey(req)) {
+        jsonResponse(res, 401, {
+            error: 'Missing or invalid API Key. Use header: Authorization: Bearer <your-api-key>',
+        });
+        return;
+    }
+    // Parse request body
+    let request;
+    try {
+        const body = await parseBody(req);
+        request = JSON.parse(body);
+    }
+    catch {
+        jsonResponse(res, 400, { error: 'Invalid JSON body' });
+        return;
+    }
+    // Validate request
+    if (!request.module || !request.args) {
+        jsonResponse(res, 400, { error: 'Missing required fields: module, args' });
+        return;
+    }
+    // Find module
+    const moduleData = await findModule(request.module, searchPaths);
+    if (!moduleData) {
+        jsonResponse(res, 404, { error: `Module '${request.module}' not found` });
+        return;
+    }
+    try {
+        // Create provider
+        const provider = getProvider(request.provider, request.model);
+        // Run module
+        const result = await runModule(moduleData, provider, {
+            input: { query: request.args, code: request.args },
+            useV22: true,
+        });
+        const response = {
+            ok: result.ok,
+            module: request.module,
+            provider: request.provider || process.env.LLM_PROVIDER || 'openai',
+        };
+        if (result.ok) {
+            if ('meta' in result)
+                response.meta = result.meta;
+            if ('data' in result)
+                response.data = result.data;
+        }
+        else {
+            if ('error' in result)
+                response.error = result.error?.message;
+        }
+        jsonResponse(res, 200, response);
+    }
+    catch (error) {
+        jsonResponse(res, 500, {
+            ok: false,
+            error: error instanceof Error ? error.message : String(error),
+            module: request.module,
+        });
+    }
+}
+export function createServer(options = {}) {
+    const { cwd = process.cwd() } = options;
+    const searchPaths = getDefaultSearchPaths(cwd);
+    const server = http.createServer(async (req, res) => {
+        const url = new URL(req.url || '/', `http://${req.headers.host}`);
+        const path = url.pathname;
+        const method = req.method?.toUpperCase();
+        // Handle CORS preflight
+        if (method === 'OPTIONS') {
+            res.writeHead(204, {
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+                'Access-Control-Allow-Headers': 'Content-Type, Authorization',
+            });
+            res.end();
+            return;
+        }
+        try {
+            // Route requests
+            if (path === '/' && method === 'GET') {
+                await handleRoot(res);
+            }
+            else if (path === '/health' && method === 'GET') {
+                await handleHealth(res);
+            }
+            else if (path === '/modules' && method === 'GET') {
+                await handleModules(res, searchPaths);
+            }
+            else if (path.startsWith('/modules/') && method === 'GET') {
+                const moduleName = path.slice('/modules/'.length);
+                await handleModuleInfo(res, moduleName, searchPaths);
+            }
+            else if (path === '/run' && method === 'POST') {
+                await handleRun(req, res, searchPaths);
+            }
+            else {
+                jsonResponse(res, 404, { error: 'Not found' });
+            }
+        }
+        catch (error) {
+            console.error('Server error:', error);
+            jsonResponse(res, 500, {
+                error: error instanceof Error ? error.message : 'Internal server error',
+            });
+        }
+    });
+    return server;
+}
+export async function serve(options = {}) {
+    const { host = '0.0.0.0', port = 8000 } = options;
+    const server = createServer(options);
+    return new Promise((resolve, reject) => {
+        server.on('error', reject);
+        server.listen(port, host, () => {
+            console.log(`Cognitive Modules HTTP Server running at http://${host}:${port}`);
+            console.log('Endpoints:');
+            console.log('  GET  /          - API info');
+            console.log('  GET  /health    - Health check');
+            console.log('  GET  /modules   - List modules');
+            console.log('  GET  /modules/:name - Module info');
+            console.log('  POST /run       - Run module');
+            resolve();
+        });
+    });
+}
+// Allow running directly
+if (import.meta.url === `file://${process.argv[1]}`) {
+    serve().catch(console.error);
+}

package/dist/server/index.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Server - Re-export all server functionality
+ */
+export { serve, createServer } from './http.js';
+export type { ServeOptions } from './http.js';

package/dist/server/index.js

@@ -0,0 +1,4 @@
+/**
+ * Server - Re-export all server functionality
+ */
+export { serve, createServer } from './http.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cognitive-modules-cli",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "Cognitive Modules - Structured AI Task Execution with version management",
   "type": "module",
   "main": "dist/index.js",
@@ -39,6 +39,9 @@
   "dependencies": {
     "js-yaml": "^4.1.1"
   },
+  "optionalDependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0"
+  },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9",
     "@types/node": "^22.0.0",

package/src/cli.ts

@@ -19,7 +19,7 @@ import { getProvider, listProviders } from './providers/index.js';
 import { run, list, pipe, init, add, update, remove, versions } from './commands/index.js';
 import type { CommandContext } from './types.js';
 
-const VERSION = '1.0.1';
+const VERSION = '1.3.0';
 
 async function main() {
   const args = process.argv.slice(2);
@@ -52,6 +52,9 @@ async function main() {
       tag: { type: 'string', short: 't' },
       branch: { type: 'string', short: 'b' },
       limit: { type: 'string', short: 'l' },
+      // Server options
+      host: { type: 'string', short: 'H' },
+      port: { type: 'string', short: 'P' },
     },
     allowPositionals: true,
   });
@@ -288,6 +291,30 @@ async function main() {
         break;
       }
 
+      case 'serve': {
+        const { serve } = await import('./server/http.js');
+        const port = values.port ? parseInt(values.port as string, 10) : 8000;
+        const host = (values.host as string) || '0.0.0.0';
+        console.log('Starting Cognitive Modules HTTP Server...');
+        await serve({ host, port, cwd: ctx.cwd });
+        break;
+      }
+
+      case 'mcp': {
+        try {
+          const { serve: serveMcp } = await import('./mcp/server.js');
+          await serveMcp();
+        } catch (e) {
+          if (e instanceof Error && e.message.includes('Cannot find module')) {
+            console.error('MCP dependencies not installed.');
+            console.error('Install with: npm install @modelcontextprotocol/sdk');
+            process.exit(1);
+          }
+          throw e;
+        }
+        break;
+      }
+
       default:
         console.error(`Unknown command: ${command}`);
         console.error('Run "cog --help" for usage.');
@@ -319,6 +346,8 @@ COMMANDS:
   versions <url>    List available versions
   pipe              Pipe mode (stdin/stdout)
   init [name]       Initialize project or create module
+  serve             Start HTTP API server
+  mcp               Start MCP server (for Claude Code, Cursor)
   doctor            Check configuration
 
 OPTIONS:
@@ -333,6 +362,8 @@ OPTIONS:
   --pretty              Pretty-print JSON output
   -V, --verbose         Verbose output
   --no-validate         Skip schema validation
+  -H, --host <host>     Server host (default: 0.0.0.0)
+  -P, --port <port>     Server port (default: 8000)
   -v, --version         Show version
   -h, --help            Show this help
 
@@ -351,6 +382,10 @@ EXAMPLES:
   cog run code-reviewer --provider openai --model gpt-4o --args "..."
   cog list
 
+  # Servers
+  cog serve --port 8080
+  cog mcp
+
   # Other
   echo "review this code" | cog pipe --module code-reviewer
   cog init my-module

package/src/mcp/index.ts

@@ -0,0 +1,5 @@
+/**
+ * MCP Server - Re-export all MCP functionality
+ */
+
+export { serve } from './server.js';