wiggum-cli 0.2.6 → 0.3.1

package/src/ai/agents/orchestrator.ts

@@ -0,0 +1,218 @@
+/**
+ * Orchestrator Agent
+ * Coordinates the multi-agent analysis and merges results
+ */
+
+import type { LanguageModel } from 'ai';
+import type {
+  CodebaseAnalysis,
+  StackResearch,
+  MultiAgentAnalysis,
+  McpRecommendations,
+  OrchestratorInput,
+} from './types.js';
+import type { DetectedStack } from '../../scanner/types.js';
+import { isReasoningModel } from '../providers.js';
+import { logger } from '../../utils/logger.js';
+import { parseJsonSafe } from '../../utils/json-repair.js';
+import { getTracedAI } from '../../utils/tracing.js';
+
+/**
+ * System prompt for the Orchestrator
+ */
+const ORCHESTRATOR_SYSTEM_PROMPT = `You are an Orchestrator agent that merges analysis results into final recommendations.
+
+Your job is to:
+1. Review the codebase analysis and stack research
+2. Recommend appropriate MCP servers for the stack
+3. Produce a final merged recommendation
+
+## MCP Server Selection Guidelines
+Based on the detected stack, recommend these MCP servers:
+
+**Always Essential:**
+- filesystem: For file operations (all projects)
+- git: For version control (all projects)
+
+**By Project Type:**
+- MCP Server: memory (for context)
+- REST API: fetch (external APIs), postgres/sqlite (databases)
+- React/Next.js: fetch (data fetching), memory (state persistence)
+- CLI Tool: filesystem (primary), memory (caching)
+- Library: git (versioning)
+
+**By Technology:**
+- Docker in use → docker MCP server
+- PostgreSQL → postgres MCP server
+- SQLite → sqlite MCP server
+- AWS services → aws-kb-retrieval MCP server
+- GitHub workflows → github MCP server
+
+## Output Format
+Output ONLY valid JSON:
+{
+  "mcpServers": {
+    "essential": ["filesystem", "git"],
+    "recommended": ["docker", "postgres"]
+  }
+}`;
+
+/**
+ * Run the Orchestrator to merge results and recommend MCP servers
+ */
+export async function runOrchestrator(
+  model: LanguageModel,
+  modelId: string,
+  input: OrchestratorInput,
+  verbose: boolean = false
+): Promise<McpRecommendations> {
+  const prompt = createOrchestratorPrompt(input);
+
+  try {
+    const { generateText } = getTracedAI();
+
+    const result = await generateText({
+      model,
+      system: ORCHESTRATOR_SYSTEM_PROMPT,
+      prompt,
+      maxOutputTokens: 1000,
+      ...(isReasoningModel(modelId) ? {} : { temperature: 0.2 }),
+      experimental_telemetry: {
+        isEnabled: true,
+        metadata: { agent: 'orchestrator', projectType: input.codebaseAnalysis.projectContext.projectType },
+      },
+    });
+
+    const mcpServers = parseMcpRecommendations(result.text, input.stack, verbose);
+    return mcpServers;
+  } catch (error) {
+    if (verbose) {
+      logger.error(`Orchestrator error: ${error instanceof Error ? error.message : String(error)}`);
+    }
+    // Return default recommendations based on project type
+    return getDefaultMcpRecommendations(input.codebaseAnalysis.projectContext.projectType, input.stack);
+  }
+}
+
+/**
+ * Create the orchestrator prompt
+ */
+function createOrchestratorPrompt(input: OrchestratorInput): string {
+  const { codebaseAnalysis, stackResearch, stack } = input;
+
+  // Summarize the stack
+  const stackSummary: string[] = [];
+  if (stack.framework) stackSummary.push(`Framework: ${stack.framework.name}`);
+  if (stack.database) stackSummary.push(`Database: ${stack.database.name}`);
+  if (stack.orm) stackSummary.push(`ORM: ${stack.orm.name}`);
+  if (stack.deployment?.length) stackSummary.push(`Deployment: ${stack.deployment.map(d => d.name).join(', ')}`);
+  if (stack.mcp?.isProject) stackSummary.push('This is an MCP Server project');
+
+  return `Analyze these results and recommend MCP servers:
+
+## Project Type
+${codebaseAnalysis.projectContext.projectType}
+
+## Detected Stack
+${stackSummary.join('\n') || 'Unknown'}
+
+## Testing Tools Identified
+${stackResearch.testingTools.join(', ') || 'None identified'}
+
+## Debugging Tools
+${stackResearch.debuggingTools.join(', ') || 'None identified'}
+
+Based on this analysis, recommend essential and optional MCP servers.
+Output as JSON with "mcpServers" containing "essential" and "recommended" arrays.`;
+}
+
+/**
+ * Parse MCP recommendations from orchestrator response
+ */
+function parseMcpRecommendations(
+  text: string,
+  stack: DetectedStack,
+  verbose: boolean
+): McpRecommendations {
+  if (!text || text.trim() === '') {
+    return getDefaultMcpRecommendations('Unknown', stack);
+  }
+
+  // Use safe JSON parser with repair capabilities
+  const parsed = parseJsonSafe<{ mcpServers?: McpRecommendations }>(text);
+
+  if (!parsed) {
+    if (verbose) {
+      logger.warn('Orchestrator: Failed to parse JSON response');
+    }
+    return getDefaultMcpRecommendations('Unknown', stack);
+  }
+
+  if (parsed.mcpServers) {
+    return {
+      essential: parsed.mcpServers.essential || ['filesystem', 'git'],
+      recommended: parsed.mcpServers.recommended || [],
+    };
+  }
+
+  return getDefaultMcpRecommendations('Unknown', stack);
+}
+
+/**
+ * Get default MCP recommendations based on project type and stack
+ */
+function getDefaultMcpRecommendations(projectType: string, stack: DetectedStack): McpRecommendations {
+  const essential: string[] = ['filesystem', 'git'];
+  const recommended: string[] = [];
+
+  // Add based on project type
+  const lowerType = projectType.toLowerCase();
+
+  if (lowerType.includes('mcp')) {
+    recommended.push('memory');
+  }
+
+  if (lowerType.includes('api') || lowerType.includes('server')) {
+    recommended.push('fetch');
+  }
+
+  // Add based on detected stack
+  if (stack.database) {
+    const dbName = stack.database.name.toLowerCase();
+    if (dbName.includes('postgres')) {
+      recommended.push('postgres');
+    } else if (dbName.includes('sqlite')) {
+      recommended.push('sqlite');
+    }
+  }
+
+  if (stack.deployment?.some(d => d.name.toLowerCase().includes('docker'))) {
+    recommended.push('docker');
+  }
+
+  // Add from existing MCP recommendations in stack
+  if (stack.mcp?.recommended) {
+    for (const rec of stack.mcp.recommended) {
+      if (!essential.includes(rec) && !recommended.includes(rec)) {
+        recommended.push(rec);
+      }
+    }
+  }
+
+  return { essential, recommended };
+}
+
+/**
+ * Merge all agent results into a final MultiAgentAnalysis
+ */
+export function mergeAgentResults(
+  codebaseAnalysis: CodebaseAnalysis,
+  stackResearch: StackResearch,
+  mcpServers: McpRecommendations
+): MultiAgentAnalysis {
+  return {
+    codebaseAnalysis,
+    stackResearch,
+    mcpServers,
+  };
+}

package/src/ai/agents/stack-researcher.ts

@@ -0,0 +1,294 @@
+/**
+ * Stack Researcher Agent
+ * Researches best practices and tools for the detected stack
+ * Gracefully degrades when optional services are unavailable
+ */
+
+import { stepCountIs, type LanguageModel, type Tool } from 'ai';
+import type { StackResearch, StackResearcherInput, AgentCapabilities } from './types.js';
+import type { DetectedStack } from '../../scanner/types.js';
+import { createTavilySearchTool } from '../tools/tavily.js';
+import { createContext7Tool } from '../tools/context7.js';
+import { isReasoningModel } from '../providers.js';
+import { logger } from '../../utils/logger.js';
+import { parseJsonSafe } from '../../utils/json-repair.js';
+import { getTracedAI } from '../../utils/tracing.js';
+
+/**
+ * System prompt for Stack Researcher with tools
+ */
+const STACK_RESEARCHER_WITH_TOOLS_PROMPT = `You are a Stack Researcher agent with access to web search and documentation lookup tools.
+
+## Your Mission
+Research the detected technology stack to find:
+1. Current best practices
+2. Common anti-patterns to avoid
+3. Testing tools and patterns
+4. Debugging approaches
+5. Useful documentation links
+
+## Tools Available
+- tavilySearch: Search the web for current best practices and patterns
+- context7Lookup: Look up library documentation
+
+## Research Strategy
+1. Search for "[technology] best practices 2024"
+2. Search for "[project type] testing patterns"
+3. Look up documentation for key dependencies
+4. Search for "[framework] anti-patterns"
+
+## Output Format
+After research, output ONLY valid JSON with this structure:
+{
+  "bestPractices": [
+    "Use TypeScript strict mode",
+    "Implement proper error boundaries"
+  ],
+  "antiPatterns": [
+    "Don't use any type",
+    "Avoid prop drilling"
+  ],
+  "testingTools": [
+    "npx vitest",
+    "npx playwright test"
+  ],
+  "debuggingTools": [
+    "React DevTools",
+    "DEBUG=* npm run dev"
+  ],
+  "documentationHints": [
+    "React docs: react.dev",
+    "Vitest: vitest.dev"
+  ],
+  "researchMode": "full"
+}
+
+Keep each item concise (5-15 words max). Max 5 items per array.`;
+
+/**
+ * System prompt for Stack Researcher without tools (knowledge-only)
+ */
+const STACK_RESEARCHER_KNOWLEDGE_ONLY_PROMPT = `You are a Stack Researcher agent. You don't have web access, so rely on your training knowledge.
+
+## Your Mission
+Based on your knowledge, provide:
+1. Best practices for the detected technologies
+2. Common anti-patterns to avoid
+3. Testing tools commonly used with this stack
+4. Debugging approaches
+5. Documentation hints
+
+## Important Notes
+- Be explicit about what you're confident about vs uncertain
+- Focus on well-established practices from your training
+- Mention if something might be outdated
+
+## Output Format
+Output ONLY valid JSON with this structure:
+{
+  "bestPractices": [
+    "Use TypeScript strict mode",
+    "Implement proper error boundaries"
+  ],
+  "antiPatterns": [
+    "Don't use any type",
+    "Avoid prop drilling"
+  ],
+  "testingTools": [
+    "npm test",
+    "npx vitest"
+  ],
+  "debuggingTools": [
+    "console.log debugging",
+    "Node.js inspector"
+  ],
+  "documentationHints": [
+    "Check official docs for updates",
+    "Framework docs: [URL]"
+  ],
+  "researchMode": "knowledge-only"
+}
+
+Keep each item concise (5-15 words max). Max 5 items per array.
+Note: Research mode should reflect that you're using training knowledge only.`;
+
+/**
+ * Create the research prompt based on stack and project type
+ */
+function createResearchPrompt(stack: DetectedStack, projectType: string, hasTools: boolean): string {
+  const technologies: string[] = [];
+
+  // Collect all detected technologies
+  if (stack.framework) technologies.push(stack.framework.name);
+  if (stack.testing?.unit) technologies.push(stack.testing.unit.name);
+  if (stack.testing?.e2e) technologies.push(stack.testing.e2e.name);
+  if (stack.orm) technologies.push(stack.orm.name);
+  if (stack.database) technologies.push(stack.database.name);
+  if (stack.stateManagement) technologies.push(stack.stateManagement.name);
+  if (stack.auth) technologies.push(stack.auth.name);
+  if (stack.mcp?.isProject) technologies.push('MCP Server');
+
+  const techList = technologies.length > 0 ? technologies.join(', ') : 'Unknown stack';
+
+  if (hasTools) {
+    return `Research best practices for this stack:
+
+Project Type: ${projectType}
+Technologies: ${techList}
+
+Use the available tools to search for:
+1. Current best practices for ${projectType} projects
+2. Testing patterns for ${techList}
+3. Common anti-patterns to avoid
+
+Then produce your research findings as JSON.`;
+  }
+
+  return `Based on your knowledge, provide best practices for this stack:
+
+Project Type: ${projectType}
+Technologies: ${techList}
+
+Provide:
+1. Best practices for ${projectType} projects
+2. Testing tools commonly used with ${techList}
+3. Anti-patterns to avoid
+4. Debugging approaches
+
+Output your findings as JSON. Be clear this is based on training knowledge.`;
+}
+
+/**
+ * Determine research mode based on capabilities
+ */
+function getResearchMode(capabilities: AgentCapabilities): StackResearch['researchMode'] {
+  if (capabilities.hasTavily && capabilities.hasContext7) return 'full';
+  if (capabilities.hasTavily) return 'web-only';
+  if (capabilities.hasContext7) return 'docs-only';
+  return 'knowledge-only';
+}
+
+/**
+ * Run the Stack Researcher agent
+ */
+export async function runStackResearcher(
+  model: LanguageModel,
+  modelId: string,
+  input: StackResearcherInput,
+  options: { tavilyApiKey?: string; context7ApiKey?: string },
+  verbose: boolean = false
+): Promise<StackResearch | null> {
+  const tools: Record<string, Tool> = {};
+
+  // Add tools based on available keys
+  if (options.tavilyApiKey) {
+    tools.tavilySearch = createTavilySearchTool(options.tavilyApiKey);
+  }
+  if (options.context7ApiKey) {
+    tools.context7Lookup = createContext7Tool(options.context7ApiKey);
+  }
+
+  const hasTools = Object.keys(tools).length > 0;
+  const researchMode = getResearchMode(input.capabilities);
+
+  if (verbose) {
+    logger.info(`Stack Researcher running in ${researchMode} mode`);
+  }
+
+  const systemPrompt = hasTools
+    ? STACK_RESEARCHER_WITH_TOOLS_PROMPT
+    : STACK_RESEARCHER_KNOWLEDGE_ONLY_PROMPT;
+
+  const prompt = createResearchPrompt(input.stack, input.projectType, hasTools);
+
+  try {
+    const { generateText } = getTracedAI();
+
+    const result = await generateText({
+      model,
+      system: systemPrompt,
+      prompt,
+      ...(hasTools ? { tools, stopWhen: stepCountIs(8) } : {}),
+      maxOutputTokens: 2000,
+      ...(isReasoningModel(modelId) ? {} : { temperature: 0.3 }),
+      experimental_telemetry: {
+        isEnabled: true,
+        metadata: { agent: 'stack-researcher', researchMode, projectType: input.projectType },
+      },
+    });
+
+    // Parse the response
+    const research = parseStackResearch(result.text, result.steps, researchMode, verbose);
+    return research;
+  } catch (error) {
+    if (verbose) {
+      logger.error(`Stack Researcher error: ${error instanceof Error ? error.message : String(error)}`);
+    }
+    return null;
+  }
+}
+
+/**
+ * Parse the stack research from agent response
+ */
+function parseStackResearch(
+  text: string,
+  steps: Array<{ text?: string }> | undefined,
+  researchMode: StackResearch['researchMode'],
+  verbose: boolean
+): StackResearch | null {
+  // Try to get text from the result or steps
+  let textToParse = text;
+
+  if (!textToParse || textToParse.trim() === '') {
+    const stepsList = steps || [];
+    for (let i = stepsList.length - 1; i >= 0; i--) {
+      const step = stepsList[i];
+      if (step.text && step.text.trim() !== '') {
+        textToParse = step.text;
+        break;
+      }
+    }
+  }
+
+  if (!textToParse || textToParse.trim() === '') {
+    if (verbose) {
+      logger.warn('Stack Researcher: No text output found');
+    }
+    return getDefaultStackResearch(researchMode);
+  }
+
+  // Use safe JSON parser with repair capabilities
+  const parsed = parseJsonSafe<Partial<StackResearch>>(textToParse);
+
+  if (!parsed) {
+    if (verbose) {
+      logger.warn('Stack Researcher: Failed to parse JSON response');
+    }
+    return getDefaultStackResearch(researchMode);
+  }
+
+  // Build result with defaults for missing fields
+  return {
+    bestPractices: parsed.bestPractices || [],
+    antiPatterns: parsed.antiPatterns || [],
+    testingTools: parsed.testingTools || [],
+    debuggingTools: parsed.debuggingTools || [],
+    documentationHints: parsed.documentationHints || [],
+    researchMode: researchMode,
+  };
+}
+
+/**
+ * Get default stack research when parsing fails
+ */
+function getDefaultStackResearch(researchMode: StackResearch['researchMode']): StackResearch {
+  return {
+    bestPractices: ['Follow project conventions', 'Write tests for new code'],
+    antiPatterns: ['Avoid skipping tests', 'Don\'t ignore type errors'],
+    testingTools: ['npm test'],
+    debuggingTools: ['console.log', 'debugger statement'],
+    documentationHints: ['Check package.json for dependencies'],
+    researchMode,
+  };
+}

package/src/ai/agents/types.ts

@@ -0,0 +1,132 @@
+/**
+ * Agent Types and Interfaces
+ * Defines the structure for multi-agent analysis
+ */
+
+import type { ScanResult, DetectedStack } from '../../scanner/types.js';
+
+/**
+ * Codebase analysis result from the Codebase Analyst agent
+ */
+export interface CodebaseAnalysis {
+  /** Project structure and context */
+  projectContext: {
+    /** Key entry point files */
+    entryPoints: string[];
+    /** Important directories and their purposes */
+    keyDirectories: Record<string, string>;
+    /** Naming conventions used */
+    namingConventions?: string;
+    /** The primary project type (MCP Server, REST API, React SPA, CLI, Library) */
+    projectType: string;
+  };
+  /** Detected commands from package.json */
+  commands: {
+    test?: string;
+    lint?: string;
+    typecheck?: string;
+    build?: string;
+    dev?: string;
+    format?: string;
+  };
+  /** Short, actionable implementation guidelines */
+  implementationGuidelines: string[];
+  /** Additional technologies that may have been missed */
+  possibleMissedTechnologies?: string[];
+}
+
+/**
+ * Stack research result from the Stack Researcher agent
+ */
+export interface StackResearch {
+  /** Best practices for the detected stack */
+  bestPractices: string[];
+  /** Anti-patterns to avoid */
+  antiPatterns: string[];
+  /** Technology-specific testing tools */
+  testingTools: string[];
+  /** Technology-specific debugging tools */
+  debuggingTools: string[];
+  /** Documentation hints and links */
+  documentationHints: string[];
+  /** Whether research was performed with tools or knowledge-only */
+  researchMode: 'full' | 'web-only' | 'docs-only' | 'knowledge-only';
+}
+
+/**
+ * MCP server recommendations
+ */
+export interface McpRecommendations {
+  /** Essential MCP servers for this stack */
+  essential: string[];
+  /** Recommended but optional MCP servers */
+  recommended: string[];
+}
+
+/**
+ * Combined result from all agents
+ */
+export interface MultiAgentAnalysis {
+  /** Codebase analysis from the Codebase Analyst */
+  codebaseAnalysis: CodebaseAnalysis;
+  /** Stack research from the Stack Researcher */
+  stackResearch: StackResearch;
+  /** MCP server recommendations (merged from both agents) */
+  mcpServers: McpRecommendations;
+}
+
+/**
+ * Agent capabilities based on available API keys
+ */
+export interface AgentCapabilities {
+  /** Tavily web search available */
+  hasTavily: boolean;
+  /** Context7 documentation lookup available */
+  hasContext7: boolean;
+}
+
+/**
+ * Options for running agents
+ */
+export interface AgentOptions {
+  /** Tavily API key (optional) */
+  tavilyApiKey?: string;
+  /** Context7 API key (optional) */
+  context7ApiKey?: string;
+  /** Enable verbose logging */
+  verbose?: boolean;
+}
+
+/**
+ * Input for the Codebase Analyst agent
+ */
+export interface CodebaseAnalystInput {
+  /** The scan result from the scanner */
+  scanResult: ScanResult;
+  /** Project root directory */
+  projectRoot: string;
+}
+
+/**
+ * Input for the Stack Researcher agent
+ */
+export interface StackResearcherInput {
+  /** The detected stack */
+  stack: DetectedStack;
+  /** The identified project type */
+  projectType: string;
+  /** Agent capabilities */
+  capabilities: AgentCapabilities;
+}
+
+/**
+ * Input for the Orchestrator agent
+ */
+export interface OrchestratorInput {
+  /** Codebase analysis result */
+  codebaseAnalysis: CodebaseAnalysis;
+  /** Stack research result */
+  stackResearch: StackResearch;
+  /** The detected stack */
+  stack: DetectedStack;
+}