Add deleteAgent to GovernanceStorage and bump to 0.7.2

Adds deleteAgent(id) to the storage interface, with implementations
for memory and PostgreSQL adapters. Audit events are preserved.
Tests cover successful delete, isolation between agents, audit
preservation, and not-found errors.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: scotty595

packages/governance/package.json

@@ -1,6 +1,6 @@
 {
   "name": "governance-sdk",
-  "version": "0.7.1",
+  "version": "0.7.2",
   "description": "AI Agent Governance for TypeScript — policy enforcement, scoring, compliance, and audit for AI agents",
   "type": "module",
   "main": "./dist/index.js",

packages/governance/src/storage-postgres.test.ts

@@ -21,10 +21,11 @@ function createMockPool(): PgPoolLike & { queries: string[] } {
         return { rows: [], rowCount: 0 };
       }
 
-      // Parse table name from INSERT/SELECT/UPDATE
+      // Parse table name from INSERT/SELECT/UPDATE/DELETE
       const insertMatch = text.match(/INSERT INTO (\S+)/);
       const selectMatch = text.match(/SELECT .* FROM (\S+)/);
       const updateMatch = text.match(/UPDATE (\S+)/);
+      const deleteMatch = text.match(/DELETE FROM (\S+)/);
 
       if (insertMatch) {
         const table = insertMatch[1];
@@ -105,6 +106,19 @@ function createMockPool(): PgPoolLike & { queries: string[] } {
         return { rows: [row], rowCount: 1 };
       }
 
+      if (deleteMatch) {
+        const table = deleteMatch[1];
+        const rows = tables.get(table) ?? [];
+        const before = rows.length;
+        const filtered = applyWhere(rows, text, values ?? []);
+        // Remove matched rows in place
+        for (const row of filtered) {
+          const idx = rows.indexOf(row);
+          if (idx >= 0) rows.splice(idx, 1);
+        }
+        return { rows: [], rowCount: before - rows.length };
+      }
+
       return { rows: [], rowCount: 0 };
     },
     async end() {
@@ -258,6 +272,34 @@ describe("PostgreSQL Storage Adapter", () => {
     assert.ok(updated);
   });
 
+  test("deleteAgent removes agent and throws if not found", async () => {
+    const pool = createMockPool();
+    const storage = await createPostgresStorage({ pool });
+
+    await storage.createAgent({
+      id: "del-1",
+      name: "doomed",
+      framework: "mastra",
+      owner: "team",
+      version: "1.0.0",
+      channels: [],
+      tools: [],
+      compositeScore: 50,
+      governanceLevel: 2,
+      status: "registered",
+      registeredAt: new Date().toISOString(),
+      updatedAt: new Date().toISOString(),
+    });
+
+    await storage.deleteAgent("del-1");
+    assert.equal(await storage.getAgent("del-1"), null);
+
+    await assert.rejects(
+      () => storage.deleteAgent("del-1"),
+      { message: "Agent del-1 not found" },
+    );
+  });
+
   test("createAuditEvent and queryAuditEvents work", async () => {
     const pool = createMockPool();
     const storage = await createPostgresStorage({ pool });

packages/governance/src/storage-postgres.ts

@@ -154,6 +154,12 @@ export async function createPostgresStorage(
     return rowToAgent(result.rows[0]);
   }
 
+  async function deleteAgent(id: string): Promise<void> {
+    await ensureMigrated();
+    const result = await pool.query(`DELETE FROM ${prefix}_agents WHERE id = $1`, [id]);
+    if (result.rowCount === 0) throw new Error(`Agent ${id} not found`);
+  }
+
   async function createAuditEvent(event: AuditEvent): Promise<AuditEvent> {
     await ensureMigrated();
     await pool.query(
@@ -188,7 +194,7 @@ export async function createPostgresStorage(
 
   if (autoMigrate) await migrate();
 
-  return { createAgent, getAgent, getAgentByName, listAgents, updateAgent, createAuditEvent, queryAuditEvents, countAuditEvents, migrate, close: () => pool.end() };
+  return { createAgent, getAgent, getAgentByName, listAgents, updateAgent, deleteAgent, createAuditEvent, queryAuditEvents, countAuditEvents, migrate, close: () => pool.end() };
 }
 
 function buildAuditWhere(filters: AuditQueryFilters): { clauses: string[]; values: unknown[]; paramIdx: number } {

packages/governance/src/storage.test.ts

@@ -91,6 +91,40 @@ describe("createMemoryStorage", () => {
     );
   });
 
+  test("deletes an agent", async () => {
+    const storage = createMemoryStorage();
+    await storage.createAgent(makeAgent({ id: "del-1" }));
+    await storage.deleteAgent("del-1");
+    assert.equal(await storage.getAgent("del-1"), null);
+  });
+
+  test("delete preserves other agents", async () => {
+    const storage = createMemoryStorage();
+    await storage.createAgent(makeAgent({ id: "keep" }));
+    await storage.createAgent(makeAgent({ id: "drop" }));
+    await storage.deleteAgent("drop");
+    const remaining = await storage.listAgents();
+    assert.equal(remaining.length, 1);
+    assert.equal(remaining[0].id, "keep");
+  });
+
+  test("delete preserves audit events", async () => {
+    const storage = createMemoryStorage();
+    await storage.createAgent(makeAgent({ id: "a-with-events" }));
+    await storage.createAuditEvent(makeEvent({ agentId: "a-with-events" }));
+    await storage.deleteAgent("a-with-events");
+    const events = await storage.queryAuditEvents({ agentId: "a-with-events" });
+    assert.equal(events.length, 1);
+  });
+
+  test("throws when deleting nonexistent agent", async () => {
+    const storage = createMemoryStorage();
+    await assert.rejects(
+      () => storage.deleteAgent("nope"),
+      { message: "Agent nope not found" },
+    );
+  });
+
   test("creates and counts audit events", async () => {
     const storage = createMemoryStorage();
     await storage.createAuditEvent(makeEvent());

packages/governance/src/storage.ts

@@ -14,6 +14,8 @@ export interface GovernanceStorage {
   getAgentByName(name: string, owner: string): Promise<StoredAgent | null>;
   listAgents(organizationId?: string): Promise<StoredAgent[]>;
   updateAgent(id: string, data: Partial<StoredAgent>): Promise<StoredAgent>;
+  /** Delete an agent by id. Audit events are preserved. Throws if not found. */
+  deleteAgent(id: string): Promise<void>;
   createAuditEvent(event: AuditEvent): Promise<AuditEvent>;
   queryAuditEvents(filters: AuditQueryFilters): Promise<AuditEvent[]>;
   countAuditEvents(filters?: AuditQueryFilters): Promise<number>;
@@ -112,6 +114,10 @@ export function createMemoryStorage(): GovernanceStorage {
       agents.set(id, updated);
       return updated;
     },
+    async deleteAgent(id) {
+      if (!agents.has(id)) throw new Error(`Agent ${id} not found`);
+      agents.delete(id);
+    },
     async createAuditEvent(event) {
       events.push(event);
       // Evict oldest events when exceeding capacity