@tonycasey/lisa 0.5.13

package/dist/templates/agents/skills/init-review/scripts/ai-enrich.js

@@ -0,0 +1,258 @@
+#!/usr/bin/env node
+"use strict";
+/**
+ * ai-enrich.ts - Background AI enrichment worker for init-review
+ *
+ * This script runs in the background after static analysis completes.
+ * It submits the codebase summary to Graphiti MCP for AI-powered entity extraction,
+ * generating richer context about the project architecture and patterns.
+ *
+ * Usage (spawned by init-review.ts or postinstall.js):
+ *   node ai-enrich.js <projectRoot> <agentsDir>
+ *
+ * The worker:
+ * 1. Reads static analysis from .agents/.init-review-static.json
+ * 2. Submits to Graphiti MCP for enrichment
+ * 3. Updates memory with enriched summary
+ * 4. Updates marker file with enriched: true
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs = require('fs');
+const path = require('path');
+// ============================================================================
+// Configuration
+// ============================================================================
+const args = process.argv.slice(2);
+const projectRoot = args[0] || process.cwd();
+const agentsDir = args[1] || path.join(projectRoot, '.agents');
+const STATIC_ANALYSIS_FILE = path.join(agentsDir, '.init-review-static.json');
+const MARKER_FILE = path.join(agentsDir, '.init-review-done');
+const LOG_FILE = path.join(agentsDir, '.init-review-enrich.log');
+// ============================================================================
+// Logging
+// ============================================================================
+function log(message) {
+    const timestamp = new Date().toISOString();
+    const line = `[${timestamp}] ${message}\n`;
+    fs.appendFileSync(LOG_FILE, line);
+}
+function loadConfig() {
+    const envPath = path.join(agentsDir, 'skills', '.env');
+    const config = {
+        endpoint: 'http://localhost:8010/mcp/',
+        groupId: path.basename(projectRoot).toLowerCase().replace(/[^a-z0-9-]/g, '-'),
+        zepApiKey: '',
+    };
+    try {
+        if (fs.existsSync(envPath)) {
+            const content = fs.readFileSync(envPath, 'utf8');
+            const lines = content.split('\n');
+            for (const line of lines) {
+                const [key, value] = line.split('=').map((s) => s.trim());
+                if (key === 'GRAPHITI_ENDPOINT')
+                    config.endpoint = value;
+                if (key === 'GRAPHITI_GROUP_ID')
+                    config.groupId = value;
+                if (key === 'ZEP_API_KEY')
+                    config.zepApiKey = value;
+            }
+        }
+    }
+    catch (_) {
+        // Use defaults
+    }
+    // Also check environment variable
+    if (!config.zepApiKey) {
+        config.zepApiKey = process.env.ZEP_API_KEY || '';
+    }
+    return config;
+}
+async function initializeMCP(endpoint, apiKey) {
+    try {
+        const body = {
+            jsonrpc: '2.0',
+            id: 'init',
+            method: 'initialize',
+            params: {
+                protocolVersion: '2024-11-05',
+                capabilities: {},
+                clientInfo: { name: 'init-review-enrich', version: '1.0.0' },
+            },
+        };
+        const headers = {
+            'Content-Type': 'application/json',
+            Accept: 'application/json, text/event-stream',
+        };
+        // Add Authorization header for Zep Cloud endpoints
+        if (apiKey && endpoint.includes('getzep.com')) {
+            headers['Authorization'] = `Api-Key ${apiKey}`;
+        }
+        const resp = await fetch(endpoint, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(body),
+            signal: AbortSignal.timeout(10000),
+        });
+        if (!resp.ok) {
+            log(`MCP init failed: HTTP ${resp.status}`);
+            return null;
+        }
+        const sessionId = resp.headers.get('mcp-session-id');
+        return sessionId;
+    }
+    catch (err) {
+        log(`MCP init error: ${err instanceof Error ? err.message : err}`);
+        return null;
+    }
+}
+async function addEnrichedMemory(endpoint, sessionId, summary, groupId, apiKey) {
+    try {
+        const payload = {
+            jsonrpc: '2.0',
+            id: '1',
+            method: 'tools/call',
+            params: {
+                name: 'add_memory',
+                arguments: {
+                    name: 'INIT-REVIEW (AI Enriched): ' + summary.slice(0, 60),
+                    episode_body: `INIT-REVIEW: ${summary}`,
+                    source: 'skill:init-review-enrich',
+                    group_id: groupId,
+                    tags: ['type:init-review', 'scope:codebase', 'ai:enriched'],
+                },
+            },
+        };
+        const headers = {
+            'Content-Type': 'application/json',
+            'MCP-SESSION-ID': sessionId,
+            Accept: 'application/json, text/event-stream',
+        };
+        // Add Authorization header for Zep Cloud endpoints
+        if (apiKey && endpoint.includes('getzep.com')) {
+            headers['Authorization'] = `Api-Key ${apiKey}`;
+        }
+        const resp = await fetch(endpoint, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(payload),
+            signal: AbortSignal.timeout(30000),
+        });
+        if (!resp.ok) {
+            log(`Add memory failed: HTTP ${resp.status}`);
+            return false;
+        }
+        return true;
+    }
+    catch (err) {
+        log(`Add memory error: ${err instanceof Error ? err.message : err}`);
+        return false;
+    }
+}
+function generateEnrichedSummary(analysis) {
+    const r = analysis.result;
+    const parts = [];
+    // Project identity
+    parts.push(`Project: ${r.project.name}`);
+    // Tech stack
+    const stack = [r.codebase.language];
+    if (r.codebase.framework)
+        stack.push(r.codebase.framework);
+    parts.push(`Stack: ${stack.join(' + ')}`);
+    // Architecture
+    if (r.patterns.architecture) {
+        parts.push(`Architecture: ${r.patterns.architecture}`);
+    }
+    // Key modules
+    if (r.structure.mainModules.length > 0) {
+        parts.push(`Core modules: ${r.structure.mainModules.slice(0, 5).join(', ')}`);
+    }
+    // Entry points
+    if (r.structure.entryPoints.length > 0) {
+        parts.push(`Entry: ${r.structure.entryPoints.slice(0, 3).join(', ')}`);
+    }
+    // Key dependencies
+    if (r.dependencies.noteworthy.length > 0) {
+        parts.push(`Key deps: ${r.dependencies.noteworthy.slice(0, 5).join(', ')}`);
+    }
+    // Build & test
+    if (r.codebase.buildTools.length > 0) {
+        parts.push(`Build: ${r.codebase.buildTools.join(', ')}`);
+    }
+    if (r.patterns.testing) {
+        parts.push(`Testing: ${r.patterns.testing}`);
+    }
+    // Metrics
+    parts.push(`Size: ${r.metrics.fileCount} files${r.metrics.hasTests ? ', has tests' : ''}`);
+    return parts.join('. ') + '.';
+}
+// ============================================================================
+// Marker File
+// ============================================================================
+function updateMarker(enriched) {
+    try {
+        let content = {};
+        if (fs.existsSync(MARKER_FILE)) {
+            content = JSON.parse(fs.readFileSync(MARKER_FILE, 'utf8'));
+        }
+        content.enriched = enriched;
+        content.enrichedAt = new Date().toISOString();
+        fs.writeFileSync(MARKER_FILE, JSON.stringify(content, null, 2));
+    }
+    catch (err) {
+        log(`Failed to update marker: ${err instanceof Error ? err.message : err}`);
+    }
+}
+// ============================================================================
+// Main
+// ============================================================================
+async function main() {
+    log('AI enrichment worker started');
+    // Check for static analysis file
+    if (!fs.existsSync(STATIC_ANALYSIS_FILE)) {
+        log('No static analysis file found, exiting');
+        return;
+    }
+    // Read static analysis
+    let analysis;
+    try {
+        analysis = JSON.parse(fs.readFileSync(STATIC_ANALYSIS_FILE, 'utf8'));
+    }
+    catch (err) {
+        log(`Failed to read static analysis: ${err instanceof Error ? err.message : err}`);
+        return;
+    }
+    // Generate enriched summary
+    const enrichedSummary = generateEnrichedSummary(analysis);
+    log(`Generated enriched summary: ${enrichedSummary.slice(0, 100)}...`);
+    // Load config
+    const config = loadConfig();
+    log(`Using endpoint: ${config.endpoint}, group: ${config.groupId}`);
+    // Initialize MCP (pass API key for Zep Cloud auth)
+    const sessionId = await initializeMCP(config.endpoint, config.zepApiKey);
+    if (!sessionId) {
+        log('Could not initialize MCP, skipping enrichment');
+        // Still mark as done but not enriched
+        return;
+    }
+    // Add enriched memory (pass API key for Zep Cloud auth)
+    const success = await addEnrichedMemory(config.endpoint, sessionId, enrichedSummary, config.groupId, config.zepApiKey);
+    if (success) {
+        log('Successfully stored enriched memory');
+        updateMarker(true);
+        // Clean up static analysis file
+        try {
+            fs.unlinkSync(STATIC_ANALYSIS_FILE);
+        }
+        catch (_) {
+            // Ignore cleanup errors
+        }
+    }
+    else {
+        log('Failed to store enriched memory');
+    }
+    log('AI enrichment worker completed');
+}
+main().catch((err) => {
+    log(`Worker error: ${err instanceof Error ? err.message : err}`);
+    process.exit(1);
+});