Add missing Layer 1, 2, and 3 MCP tools

semanticintent · claude · semanticintent · commit 79f98d2f82fb · 2025-10-17T10:46:32.000-04:00
Expose all 9 missing Wake Intelligence tools as MCP endpoints: Layer 1 (Causality Engine): - reconstruct_reasoning: Explain WHY contexts were created - build_causal_chain: Trace decision history backwards - get_causality_stats: Analytics on causal relationships Layer 2 (Memory Manager): - get_memory_stats: View memory tier distribution - recalculate_memory_tiers: Update tier classifications - prune_expired_contexts: Clean up expired contexts Layer 3 (Propagation Engine): - update_predictions: Refresh prediction scores - get_high_value_contexts: Get contexts likely to be accessed next - get_propagation_stats: Analytics on prediction quality Fixes: - Fixed property name mismatches in stats handlers - Fixed CausalChainNode property access (node.snapshot.*) - All 12 tools now working and tested 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
diff --git a/src/application/handlers/MCPProtocolHandler.ts b/src/application/handlers/MCPProtocolHandler.ts
@@ -52,6 +52,111 @@ const TOOL_DEFINITIONS = [
       },
       required: ["query"]
     }
+  },
+  // Layer 1: Causality Engine (Past - WHY)
+  {
+    name: "reconstruct_reasoning",
+    description: "Explain WHY a context was created by reconstructing the reasoning chain",
+    inputSchema: {
+      type: "object",
+      properties: {
+        snapshotId: { type: "string", description: "ID of the context snapshot to analyze" }
+      },
+      required: ["snapshotId"]
+    }
+  },
+  {
+    name: "build_causal_chain",
+    description: "Trace decision history backwards through time to see how contexts influenced each other",
+    inputSchema: {
+      type: "object",
+      properties: {
+        snapshotId: { type: "string", description: "Starting snapshot ID to trace backwards from" }
+      },
+      required: ["snapshotId"]
+    }
+  },
+  {
+    name: "get_causality_stats",
+    description: "Get analytics on causal relationships for a project",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project identifier" }
+      },
+      required: ["project"]
+    }
+  },
+  // Layer 2: Memory Manager (Present - HOW)
+  {
+    name: "get_memory_stats",
+    description: "View memory tier distribution and access patterns for a project",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project identifier" }
+      },
+      required: ["project"]
+    }
+  },
+  {
+    name: "recalculate_memory_tiers",
+    description: "Update tier classifications based on current time",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project to recalculate (optional, processes all if omitted)" }
+      },
+      required: []
+    }
+  },
+  {
+    name: "prune_expired_contexts",
+    description: "Clean up old, unused contexts that have expired",
+    inputSchema: {
+      type: "object",
+      properties: {
+        limit: { type: "number", description: "Maximum number of contexts to prune (optional)" }
+      },
+      required: []
+    }
+  },
+  // Layer 3: Propagation Engine (Future - WHAT)
+  {
+    name: "update_predictions",
+    description: "Refresh prediction scores for a project's contexts",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project identifier" },
+        staleThreshold: { type: "number", description: "Hours threshold for stale predictions (default: 24)" }
+      },
+      required: ["project"]
+    }
+  },
+  {
+    name: "get_high_value_contexts",
+    description: "Retrieve contexts most likely to be accessed next (predicted high-value)",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project identifier" },
+        minScore: { type: "number", description: "Minimum prediction score (default: 0.6)" },
+        limit: { type: "number", description: "Maximum contexts to return (default: 5)" }
+      },
+      required: ["project"]
+    }
+  },
+  {
+    name: "get_propagation_stats",
+    description: "Get analytics on prediction quality and patterns for a project",
+    inputSchema: {
+      type: "object",
+      properties: {
+        project: { type: "string", description: "Project identifier" }
+      },
+      required: ["project"]
+    }
   }
 ];
 
diff --git a/src/application/handlers/ToolExecutionHandler.ts b/src/application/handlers/ToolExecutionHandler.ts
@@ -48,6 +48,36 @@ export class ToolExecutionHandler {
       case 'search_context':
         return this.handleSearchContext(args as SearchContextInput);
 
+      // Layer 1: Causality Engine (Past - WHY)
+      case 'reconstruct_reasoning':
+        return this.handleReconstructReasoning(args as { snapshotId: string });
+
+      case 'build_causal_chain':
+        return this.handleBuildCausalChain(args as { snapshotId: string });
+
+      case 'get_causality_stats':
+        return this.handleGetCausalityStats(args as { project: string });
+
+      // Layer 2: Memory Manager (Present - HOW)
+      case 'get_memory_stats':
+        return this.handleGetMemoryStats(args as { project: string });
+
+      case 'recalculate_memory_tiers':
+        return this.handleRecalculateMemoryTiers(args as { project?: string });
+
+      case 'prune_expired_contexts':
+        return this.handlePruneExpiredContexts(args as { limit?: number });
+
+      // Layer 3: Propagation Engine (Future - WHAT)
+      case 'update_predictions':
+        return this.handleUpdatePredictions(args as { project: string; staleThreshold?: number });
+
+      case 'get_high_value_contexts':
+        return this.handleGetHighValueContexts(args as { project: string; minScore?: number; limit?: number });
+
+      case 'get_propagation_stats':
+        return this.handleGetPropagationStats(args as { project: string });
+
       default:
         throw new Error(`Unknown tool: ${toolName}`);
     }
@@ -138,4 +168,229 @@ export class ToolExecutionHandler {
       }]
     };
   }
+
+  // =============================================================================
+  // Layer 1: Causality Engine (Past - WHY)
+  // =============================================================================
+
+  /**
+   * 🎯 LAYER 1 TOOL: reconstruct_reasoning
+   *
+   * PURPOSE: Explain WHY a context was created
+   */
+  private async handleReconstructReasoning(args: { snapshotId: string }): Promise<ToolResult> {
+    const reasoning = await this.contextService.reconstructReasoning(args.snapshotId);
+
+    return {
+      content: [{
+        type: "text",
+        text: reasoning
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 1 TOOL: build_causal_chain
+   *
+   * PURPOSE: Trace decision history backwards through time
+   */
+  private async handleBuildCausalChain(args: { snapshotId: string }): Promise<ToolResult> {
+    const chain = await this.contextService.buildCausalChain(args.snapshotId);
+
+    if (chain.length === 0) {
+      return {
+        content: [{
+          type: "text",
+          text: `No causal chain found for snapshot: ${args.snapshotId}`
+        }]
+      };
+    }
+
+    const chainText = chain
+      .map((node, idx) => {
+        const depth = '  '.repeat(idx);
+        return `${depth}${idx + 1}. **${node.snapshot.causality?.actionType || 'unknown'}** (${node.snapshot.timestamp})\n${depth}   ${node.snapshot.summary}`;
+      })
+      .join('\n\n');
+
+    return {
+      content: [{
+        type: "text",
+        text: `Causal Chain (${chain.length} steps):\n\n${chainText}`
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 1 TOOL: get_causality_stats
+   *
+   * PURPOSE: Analytics on causal relationships
+   */
+  private async handleGetCausalityStats(args: { project: string }): Promise<ToolResult> {
+    const stats = await this.contextService.getCausalityStats(args.project);
+
+    const statsText = `**Causality Statistics for ${args.project}**
+
+📊 **Action Type Distribution:**
+${Object.entries(stats.actionTypeCounts)
+  .map(([type, count]) => `  - ${type}: ${count}`)
+  .join('\n')}
+
+🔗 **Causal Analysis:**
+  - Total contexts with causality: ${stats.totalWithCausality}
+  - Root causes (no parent): ${stats.rootCauses}
+  - Average chain length: ${stats.averageChainLength.toFixed(2)}`;
+
+    return {
+      content: [{
+        type: "text",
+        text: statsText
+      }]
+    };
+  }
+
+  // =============================================================================
+  // Layer 2: Memory Manager (Present - HOW)
+  // =============================================================================
+
+  /**
+   * 🎯 LAYER 2 TOOL: get_memory_stats
+   *
+   * PURPOSE: View memory tier distribution and access patterns
+   */
+  private async handleGetMemoryStats(args: { project: string }): Promise<ToolResult> {
+    const stats = await this.contextService.getMemoryStats(args.project);
+
+    const statsText = `**Memory Statistics for ${args.project}**
+
+📊 **Memory Tier Distribution:**
+  - 🔥 ACTIVE (< 1 hour): ${stats.active}
+  - ⚡ RECENT (1-24 hours): ${stats.recent}
+  - 📦 ARCHIVED (1-30 days): ${stats.archived}
+  - ❄️  EXPIRED (> 30 days): ${stats.expired}
+
+📈 **Total Contexts:** ${stats.total}`;
+
+    return {
+      content: [{
+        type: "text",
+        text: statsText
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 2 TOOL: recalculate_memory_tiers
+   *
+   * PURPOSE: Update tier classifications based on current time
+   */
+  private async handleRecalculateMemoryTiers(args: { project?: string }): Promise<ToolResult> {
+    const updatedCount = await this.contextService.recalculateMemoryTiers(args.project);
+
+    return {
+      content: [{
+        type: "text",
+        text: `✅ Recalculated memory tiers\n\nUpdated ${updatedCount} context(s)${args.project ? ` for project: ${args.project}` : ' across all projects'}`
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 2 TOOL: prune_expired_contexts
+   *
+   * PURPOSE: Clean up old, unused contexts
+   */
+  private async handlePruneExpiredContexts(args: { limit?: number }): Promise<ToolResult> {
+    const deletedCount = await this.contextService.pruneExpiredContexts(args.limit);
+
+    return {
+      content: [{
+        type: "text",
+        text: `🗑️ Pruned expired contexts\n\nDeleted ${deletedCount} expired context(s)`
+      }]
+    };
+  }
+
+  // =============================================================================
+  // Layer 3: Propagation Engine (Future - WHAT)
+  // =============================================================================
+
+  /**
+   * 🎯 LAYER 3 TOOL: update_predictions
+   *
+   * PURPOSE: Refresh prediction scores for a project
+   */
+  private async handleUpdatePredictions(args: { project: string; staleThreshold?: number }): Promise<ToolResult> {
+    const updatedCount = await this.contextService.updatePredictions(args.project, args.staleThreshold);
+
+    return {
+      content: [{
+        type: "text",
+        text: `🔮 Updated predictions for ${args.project}\n\nRefreshed ${updatedCount} prediction(s) (stale threshold: ${args.staleThreshold || 24} hours)`
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 3 TOOL: get_high_value_contexts
+   *
+   * PURPOSE: Retrieve contexts most likely to be accessed next
+   */
+  private async handleGetHighValueContexts(args: { project: string; minScore?: number; limit?: number }): Promise<ToolResult> {
+    const contexts = await this.contextService.getHighValueContexts(args.project, args.minScore, args.limit);
+
+    if (contexts.length === 0) {
+      return {
+        content: [{
+          type: "text",
+          text: `No high-value contexts found for ${args.project} (min score: ${args.minScore || 0.6})`
+        }]
+      };
+    }
+
+    const contextList = contexts
+      .map(ctx => {
+        const score = ctx.propagation?.predictionScore || 0;
+        const reasons = ctx.propagation?.propagationReason.join(', ') || 'none';
+        return `**${ctx.project}** (Score: ${score.toFixed(3)})\n${ctx.summary}\nReasons: ${reasons}`;
+      })
+      .join('\n\n');
+
+    return {
+      content: [{
+        type: "text",
+        text: `🎯 High-Value Contexts (${contexts.length} found):\n\n${contextList}`
+      }]
+    };
+  }
+
+  /**
+   * 🎯 LAYER 3 TOOL: get_propagation_stats
+   *
+   * PURPOSE: Analytics on prediction quality and patterns
+   */
+  private async handleGetPropagationStats(args: { project: string }): Promise<ToolResult> {
+    const stats = await this.contextService.getPropagationStats(args.project);
+
+    const reasonsText = Object.entries(stats.reasonFrequency)
+      .map(([reason, count]) => `  - ${reason}: ${count}`)
+      .join('\n');
+
+    const statsText = `**Propagation Statistics for ${args.project}**
+
+🔮 **Predictions:**
+  - Total contexts: ${stats.totalContexts}
+  - Contexts predicted: ${stats.totalPredicted}
+  - Average prediction score: ${stats.averagePredictionScore.toFixed(3)}
+
+📊 **Prediction Reasons:**
+${reasonsText || '  - No predictions yet'}`;
+
+    return {
+      content: [{
+        type: "text",
+        text: statsText
+      }]
+    };
+  }
 }
