wiggum-cli 0.3.2 → 0.4.1

package/src/ai/enhancer.ts

@@ -8,7 +8,7 @@ import type { ScanResult, DetectedStack, DetectionResult } from '../scanner/type
 import { getModel, type AIProvider, hasApiKey, getApiKeyEnvVar, isReasoningModel } from './providers.js';
 import { SYSTEM_PROMPT, SYSTEM_PROMPT_AGENTIC, createAnalysisPrompt } from './prompts.js';
 import { createExplorationTools } from './tools.js';
-import { runMultiAgentAnalysis, type MultiAgentAnalysis } from './agents/index.js';
+import { runMultiAgentAnalysis, type MultiAgentAnalysis, type ProgressCallback } from './agents/index.js';
 import { logger } from '../utils/logger.js';
 import { parseJsonSafe } from '../utils/json-repair.js';
 import { getTracedAI, traced } from '../utils/tracing.js';
@@ -116,6 +116,8 @@ export interface EnhancerOptions {
   tavilyApiKey?: string;
   /** Context7 API key for documentation lookup (optional) */
   context7ApiKey?: string;
+  /** Progress callback for phase updates */
+  onProgress?: ProgressCallback;
 }
 
 /**
@@ -192,6 +194,7 @@ export class AIEnhancer {
   private agentic: boolean;
   private tavilyApiKey?: string;
   private context7ApiKey?: string;
+  private onProgress?: ProgressCallback;
 
   constructor(options: EnhancerOptions = {}) {
     this.provider = options.provider || 'anthropic';
@@ -200,6 +203,7 @@ export class AIEnhancer {
     this.agentic = options.agentic || false;
     this.tavilyApiKey = options.tavilyApiKey;
     this.context7ApiKey = options.context7ApiKey;
+    this.onProgress = options.onProgress;
   }
 
   /**
@@ -327,6 +331,7 @@ export class AIEnhancer {
         tavilyApiKey: this.tavilyApiKey,
         context7ApiKey: this.context7ApiKey,
         verbose: this.verbose,
+        onProgress: this.onProgress,
       }
     );
 

package/src/ai/index.ts

@@ -44,19 +44,40 @@ export {
   canUseContext7,
 } from './tools/index.js';
 
-// Agents
+// Agents - New 4-phase architecture
 export {
+  // Main orchestration
   runMultiAgentAnalysis,
-  runCodebaseAnalyst,
-  runStackResearcher,
-  runOrchestrator,
-  mergeAgentResults,
+  // Phase 1: Planning
+  runPlanningOrchestrator,
+  // Phase 2: Parallel workers
+  runContextEnricher,
+  runTechResearcher,
+  runTechResearchPool,
+  // Phase 3: Synthesis + MCP detection
+  runSynthesisAgent,
+  detectRalphMcpServers,
+  convertToLegacyMcpRecommendations,
+  // Phase 4: QA loop
+  runEvaluatorOptimizer,
+  // New types
+  type AnalysisPlan,
+  type EnrichedContext,
+  type TechResearchResult,
+  type RalphMcpServers,
+  type EvaluationResult,
+  // Legacy types (backward compatibility)
   type CodebaseAnalysis,
   type StackResearch,
   type McpRecommendations,
   type MultiAgentAnalysis,
   type AgentCapabilities,
   type AgentOptions,
+  // Legacy agents (deprecated, kept for backward compatibility)
+  runCodebaseAnalyst,
+  runStackResearcher,
+  runOrchestrator,
+  mergeAgentResults,
 } from './agents/index.js';
 
 // AI enhancer

package/src/commands/init.ts

@@ -26,6 +26,7 @@ import pc from 'picocolors';
 import fs from 'fs';
 import path from 'path';
 import { simpson, sectionHeader, drawLine } from '../utils/colors.js';
+import { flushTracing } from '../utils/tracing.js';
 
 export interface InitOptions {
   provider?: AIProvider;
@@ -109,8 +110,8 @@ async function collectApiKeys(
     const providerChoice = await prompts.select({
       message: 'Select your AI provider:',
       options: [
-        { value: 'anthropic', label: 'Anthropic (Claude)', hint: 'recommended' },
-        { value: 'openai', label: 'OpenAI (GPT-4/5)' },
+        { value: 'anthropic', label: 'Anthropic', hint: 'recommended' },
+        { value: 'openai', label: 'OpenAI' },
         { value: 'openrouter', label: 'OpenRouter', hint: 'multiple providers' },
       ],
     });
@@ -261,6 +262,7 @@ export async function initCommand(options: InitOptions): Promise<void> {
   if (!apiKeys) {
     // In --yes mode, null means missing API key (hard failure)
     // In interactive mode, null means user cancelled
+    await flushTracing();
     if (options.yes) {
       process.exit(1);
     }
@@ -281,6 +283,7 @@ export async function initCommand(options: InitOptions): Promise<void> {
   } catch (error) {
     spinner.stop('Scan failed');
     logger.error(`Failed to scan project: ${error instanceof Error ? error.message : String(error)}`);
+    await flushTracing();
     process.exit(1);
   }
 
@@ -297,15 +300,23 @@ export async function initCommand(options: InitOptions): Promise<void> {
   console.log(pc.dim(`Capabilities: ${capabilities.join(' • ')}`));
   console.log('');
 
-  spinner.start('Running AI analysis...');
+  spinner.start('Starting AI analysis...');
 
   const aiEnhancer = new AIEnhancer({
     provider: apiKeys.provider,
     model: apiKeys.model,
-    verbose: true,
+    verbose: false, // Disable verbose logging when using progress callback
     agentic: true, // Always use agentic mode for deeper analysis
     tavilyApiKey: apiKeys.tavilyKey,
     context7ApiKey: apiKeys.context7Key,
+    onProgress: (phase, detail) => {
+      // Update spinner with current phase
+      if (detail) {
+        spinner.message(`${phase} - ${detail}`);
+      } else {
+        spinner.message(phase);
+      }
+    },
   });
 
   let enhancedResult: EnhancedScanResult;
@@ -342,6 +353,7 @@ export async function initCommand(options: InitOptions): Promise<void> {
     });
 
     if (prompts.isCancel(shouldContinue) || !shouldContinue) {
+      await flushTracing();
       logger.info('Initialization cancelled');
       return;
     }
@@ -365,6 +377,9 @@ export async function initCommand(options: InitOptions): Promise<void> {
     console.log(simpson.yellow('─── Generation Results ───'));
     console.log(formatGenerationResult(generationResult));
 
+    // Flush tracing spans before completing
+    await flushTracing();
+
     if (generationResult.success) {
       console.log('');
       logger.success('Ralph initialized successfully!');
@@ -380,6 +395,7 @@ export async function initCommand(options: InitOptions): Promise<void> {
   } catch (error) {
     spinner.stop('Generation failed');
     logger.error(`Failed to generate files: ${error instanceof Error ? error.message : String(error)}`);
+    await flushTracing();
     process.exit(1);
   }
 }

package/src/utils/tracing.ts

@@ -3,7 +3,7 @@
  * Provides AI call tracing for debugging and analysis
  */
 
-import { initLogger, wrapAISDK } from 'braintrust';
+import { initLogger, wrapAISDK, flush as braintrustFlush } from 'braintrust';
 import * as ai from 'ai';
 
 // Re-export traced utilities
@@ -13,6 +13,7 @@ export { traced, currentSpan, wrapTraced } from 'braintrust';
  * Initialize Braintrust logger if API key is available
  */
 let loggerInitialized = false;
+let exitHandlersRegistered = false;
 
 export function initTracing(): void {
   if (loggerInitialized) return;
@@ -29,11 +30,53 @@ export function initTracing(): void {
       projectName: process.env.BRAINTRUST_PROJECT_NAME || 'wiggum-cli',
     });
     loggerInitialized = true;
+
+    // Register exit handlers to flush spans before process exits
+    registerExitHandlers();
   } catch {
     // Silently fail if tracing can't be initialized
   }
 }
 
+/**
+ * Register process exit handlers to flush tracing spans
+ */
+function registerExitHandlers(): void {
+  if (exitHandlersRegistered) return;
+  exitHandlersRegistered = true;
+
+  // Flush on normal exit
+  process.on('beforeExit', async () => {
+    await flushTracing();
+  });
+
+  // Flush on SIGINT (Ctrl+C)
+  process.on('SIGINT', async () => {
+    await flushTracing();
+    process.exit(0);
+  });
+
+  // Flush on SIGTERM
+  process.on('SIGTERM', async () => {
+    await flushTracing();
+    process.exit(0);
+  });
+}
+
+/**
+ * Flush all pending tracing spans to Braintrust
+ * Call this before process exit to ensure all spans are recorded
+ */
+export async function flushTracing(): Promise<void> {
+  if (!loggerInitialized) return;
+
+  try {
+    await braintrustFlush();
+  } catch {
+    // Silently fail if flush fails
+  }
+}
+
 /**
  * Check if tracing is enabled
  */

package/tsconfig.json

@@ -15,5 +15,5 @@
     "resolveJsonModule": true
   },
   "include": ["src/**/*"],
-  "exclude": ["node_modules", "dist"]
+  "exclude": ["node_modules", "dist", "**/*.test.ts"]
 }

package/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    include: ['src/**/*.test.ts'],
+  },
+});