@gethmy/mcp 2.2.4 → 2.3.1

package/src/server.ts

@@ -58,6 +58,7 @@ import {
 } from "./context-assembly.js";
 import { autoExpandGraph } from "./graph-expansion.js";
 import { runLifecycleMaintenance } from "./lifecycle-maintenance.js";
+import { runMemoryCleanup } from "./memory-cleanup.js";
 import { onboardNewUser } from "./onboard.js";
 import type { PromptVariant } from "./prompt-builder.js";
 
@@ -1637,6 +1638,51 @@ const TOOLS = {
       required: [],
     },
   },
+
+  harmony_cleanup_memories: {
+    description:
+      "Run a unified memory cleanup: prune stale drafts, consolidate similar memories, detect orphans and duplicates, and backfill embeddings. Returns a health report. Dry-run by default — run with dryRun=false to execute.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        workspaceId: {
+          type: "string",
+          description: "Workspace ID (optional if context set)",
+        },
+        projectId: {
+          type: "string",
+          description: "Project ID (optional)",
+        },
+        dryRun: {
+          type: "boolean",
+          description:
+            "Preview cleanup without executing changes (default: true)",
+        },
+        steps: {
+          type: "array",
+          items: {
+            type: "string",
+            enum: ["prune", "consolidate", "orphans", "duplicates", "backfill"],
+          },
+          description:
+            "Which cleanup steps to run (default: all). Options: prune, consolidate, orphans, duplicates, backfill.",
+        },
+        maxAgeDays: {
+          type: "number",
+          description: "Max age in days for stale draft pruning (default: 30)",
+        },
+        minClusterSize: {
+          type: "number",
+          description: "Min cluster size for consolidation (default: 3)",
+        },
+        orphanAgeDays: {
+          type: "number",
+          description: "Min age in days for orphan detection (default: 14)",
+        },
+      },
+      required: [],
+    },
+  },
 };
 
 // Resource URIs
@@ -3967,6 +4013,49 @@ async function handleToolCall(
       };
     }
 
+    case "harmony_cleanup_memories": {
+      const workspaceId =
+        (args.workspaceId as string) || deps.getActiveWorkspaceId();
+      if (!workspaceId) {
+        throw new Error(
+          "No workspace specified. Use harmony_set_workspace_context or provide workspaceId.",
+        );
+      }
+      const projectId =
+        (args.projectId as string) || deps.getActiveProjectId() || undefined;
+
+      const validSteps = [
+        "prune",
+        "consolidate",
+        "orphans",
+        "duplicates",
+        "backfill",
+      ];
+      const rawSteps = args.steps as string[] | undefined;
+      const steps = rawSteps?.filter((s) => validSteps.includes(s));
+      if (rawSteps && steps && steps.length < rawSteps.length) {
+        const invalid = rawSteps.filter((s) => !validSteps.includes(s));
+        // Will appear in report.errors via the healthReport
+        console.warn(`Unknown cleanup steps ignored: ${invalid.join(", ")}`);
+      }
+
+      const report = await runMemoryCleanup(client, workspaceId, projectId, {
+        dryRun: args.dryRun as boolean | undefined,
+        steps,
+        maxAgeDays: args.maxAgeDays as number | undefined,
+        minClusterSize: args.minClusterSize as number | undefined,
+        orphanAgeDays: args.orphanAgeDays as number | undefined,
+      });
+
+      return {
+        success: report.success,
+        dryRun: report.dryRun,
+        summary: report.summary,
+        errors: report.errors,
+        healthReport: report.healthReport,
+      };
+    }
+
     default:
       throw new Error(`Unknown tool: ${name}`);
   }