@gethmy/mcp 2.3.4 → 2.4.0

package/src/server.ts

@@ -58,7 +58,12 @@ import {
 } from "./context-assembly.js";
 import { autoExpandGraph } from "./graph-expansion.js";
 import { runLifecycleMaintenance } from "./lifecycle-maintenance.js";
-import { runMemoryCleanup, purgeMemories, type PurgeFilters } from "./memory-cleanup.js";
+import { runMemoryAudit } from "./memory-audit.js";
+import {
+  type PurgeFilters,
+  purgeMemories,
+  runMemoryCleanup,
+} from "./memory-cleanup.js";
 import { onboardNewUser } from "./onboard.js";
 import type { PromptVariant } from "./prompt-builder.js";
 
@@ -1657,7 +1662,7 @@ const TOOLS = {
 
   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.",
+      "Run a unified memory cleanup: prune stale drafts, consolidate similar memories, detect orphans and duplicates, backfill embeddings, and optionally run a quality audit. Returns a health report. Dry-run by default — run with dryRun=false to execute.",
     inputSchema: {
       type: "object",
       properties: {
@@ -1678,10 +1683,17 @@ const TOOLS = {
           type: "array",
           items: {
             type: "string",
-            enum: ["prune", "consolidate", "orphans", "duplicates", "backfill"],
+            enum: [
+              "prune",
+              "consolidate",
+              "orphans",
+              "duplicates",
+              "backfill",
+              "audit",
+            ],
           },
           description:
-            "Which cleanup steps to run (default: all). Options: prune, consolidate, orphans, duplicates, backfill.",
+            "Which cleanup steps to run (default: all). Options: prune, consolidate, orphans, duplicates, backfill, audit.",
         },
         maxAgeDays: {
           type: "number",
@@ -1695,6 +1707,52 @@ const TOOLS = {
           type: "number",
           description: "Min age in days for orphan detection (default: 14)",
         },
+        auditArchiveBelow: {
+          type: "number",
+          description: "Audit: archive entities scoring below this (default: 40)",
+        },
+        auditDeleteBelow: {
+          type: "number",
+          description: "Audit: delete entities scoring below this (default: 20)",
+        },
+      },
+      required: [],
+    },
+  },
+  harmony_audit_memories: {
+    description:
+      "Rate every memory against state-of-the-art quality standards (confidence, decay, structural completeness, content, tier-age fit, access). Flags legacy entities from before recent optimizations (default confidence, missing embeddings, stuck drafts). Buckets: keep (≥70), review (40-69), archive (20-39), delete (<20). Dry-run by default.",
+    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 audit without flagging/archiving/deleting (default: true)",
+        },
+        archiveBelow: {
+          type: "number",
+          description:
+            "Score threshold below which entities are archived (confidence set to 0.25). Default: 40",
+        },
+        deleteBelow: {
+          type: "number",
+          description:
+            "Score threshold below which entities are hard-deleted. Default: 20. Set to 0 to never delete.",
+        },
+        limit: {
+          type: "number",
+          description:
+            "Max number of entities to audit (default: 500). Paginated fetch.",
+        },
       },
       required: [],
     },
@@ -4091,6 +4149,37 @@ async function handleToolCall(
       };
     }
 
+    case "harmony_audit_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 report = await runMemoryAudit(client, workspaceId, projectId, {
+        dryRun: args.dryRun as boolean | undefined,
+        archiveBelow: args.archiveBelow as number | undefined,
+        deleteBelow: args.deleteBelow as number | undefined,
+        limit: args.limit as number | undefined,
+      });
+
+      return {
+        success: report.success,
+        dryRun: report.dryRun,
+        summary: report.summary,
+        distribution: report.distribution,
+        legacyBreakdown: report.legacyBreakdown,
+        actionsTaken: report.actionsTaken,
+        lowest: report.lowest,
+        errors: report.errors,
+        healthReport: report.healthReport,
+      };
+    }
+
     case "harmony_cleanup_memories": {
       const workspaceId =
         (args.workspaceId as string) || deps.getActiveWorkspaceId();
@@ -4108,6 +4197,7 @@ async function handleToolCall(
         "orphans",
         "duplicates",
         "backfill",
+        "audit",
       ];
       const rawSteps = args.steps as string[] | undefined;
       const steps = rawSteps?.filter((s) => validSteps.includes(s));
@@ -4119,10 +4209,14 @@ async function handleToolCall(
 
       const report = await runMemoryCleanup(client, workspaceId, projectId, {
         dryRun: args.dryRun as boolean | undefined,
-        steps,
+        steps: steps as
+          | ("prune" | "consolidate" | "orphans" | "duplicates" | "backfill" | "audit")[]
+          | undefined,
         maxAgeDays: args.maxAgeDays as number | undefined,
         minClusterSize: args.minClusterSize as number | undefined,
         orphanAgeDays: args.orphanAgeDays as number | undefined,
+        auditArchiveBelow: args.auditArchiveBelow as number | undefined,
+        auditDeleteBelow: args.auditDeleteBelow as number | undefined,
       });
 
       return {
@@ -4142,8 +4236,7 @@ async function handleToolCall(
           "No workspace specified. Use harmony_set_workspace_context or provide workspaceId.",
         );
       }
-      const projectId =
-        (args.projectId as string) || deps.getActiveProjectId();
+      const projectId = (args.projectId as string) || deps.getActiveProjectId();
       if (!projectId) {
         throw new Error(
           "No project specified. Purge requires a project scope. Use harmony_set_project_context or provide projectId.",