@zivis/mcp 0.1.11 → 0.1.12

package/dist/pattern-packs/zivis-public-0.2.0/manifest.json

@@ -3,7 +3,7 @@
   "pack_id": "zivis-public",
   "pack_name": "ZIVIS Public Pattern Pack",
   "version": "0.2.0",
-  "built_at": "2026-08-31T22:32:10.810Z",
+  "built_at": "2026-09-01T21:19:00.226Z",
   "tier": "customer_safe",
   "description": "ZIVIS-curated public pattern pack — capsules + inference prompts evaluated locally on the user's machine.",
   "capsules": [

package/dist/server.js

@@ -73,7 +73,6 @@ import { UPDATE_ENDPOINT_NAME, UPDATE_ENDPOINT_DESCRIPTION, UPDATE_ENDPOINT_SCHE
 import { MANAGE_APPLICATION_NAME, MANAGE_APPLICATION_DESCRIPTION, MANAGE_APPLICATION_SCHEMA, createManageApplicationHandler, } from "./tools/manage-application.js";
 import { MANAGE_ENDPOINT_LIFECYCLE_NAME, MANAGE_ENDPOINT_LIFECYCLE_DESCRIPTION, MANAGE_ENDPOINT_LIFECYCLE_SCHEMA, createManageEndpointLifecycleHandler, } from "./tools/manage-endpoint-lifecycle.js";
 import { DISCOVER_LOCAL_INFRA_NAME, DISCOVER_LOCAL_INFRA_DESCRIPTION, DISCOVER_LOCAL_INFRA_SCHEMA, createDiscoverLocalInfraHandler, } from "./tools/discover-local-infra.js";
-import { GENERATE_DIAGRAM_NAME, GENERATE_DIAGRAM_DESCRIPTION, GENERATE_DIAGRAM_SCHEMA, createGenerateDiagramHandler, } from "./tools/generate-diagram.js";
 import { LIST_DIAGRAMS_NAME, LIST_DIAGRAMS_DESCRIPTION, LIST_DIAGRAMS_SCHEMA, createListDiagramsHandler, } from "./tools/list-diagrams.js";
 import { GET_DIAGRAM_NAME, GET_DIAGRAM_DESCRIPTION, GET_DIAGRAM_SCHEMA, createGetDiagramHandler, } from "./tools/get-diagram.js";
 import { CREATE_DIAGRAM_NAME, CREATE_DIAGRAM_DESCRIPTION, CREATE_DIAGRAM_SCHEMA, createCreateDiagramHandler, } from "./tools/create-diagram.js";
@@ -215,10 +214,6 @@ export async function startServer(incoming = DEFAULT_CONFIG) {
         description: DISCOVER_LOCAL_INFRA_DESCRIPTION,
         inputSchema: DISCOVER_LOCAL_INFRA_SCHEMA,
     }, createDiscoverLocalInfraHandler());
-    server.registerTool(GENERATE_DIAGRAM_NAME, {
-        description: GENERATE_DIAGRAM_DESCRIPTION,
-        inputSchema: GENERATE_DIAGRAM_SCHEMA,
-    }, createGenerateDiagramHandler(apiClient));
     server.registerTool(LIST_DIAGRAMS_NAME, {
         description: LIST_DIAGRAMS_DESCRIPTION,
         inputSchema: LIST_DIAGRAMS_SCHEMA,

package/dist/tools/create-diagram.d.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import type { ApiClient } from "../api-client.js";
 export declare const CREATE_DIAGRAM_NAME = "zivis_create_diagram";
-export declare const CREATE_DIAGRAM_DESCRIPTION = "Create a new canvas diagram with nodes, connections, and boundaries in a single call.\n\nUse this to build system architecture diagrams, data flow diagrams, attack chains, etc. from code analysis.\n\n**Temp IDs:** When creating nodes and connections together, assign each node a temp_id (e.g., \"node_0\", \"node_1\") and reference them in connections via source_temp_id / target_temp_id. Similarly, assign boundaries a temp_id and reference them in nodes via boundary_temp_id.\n\n**Positioning:** Nodes default to (0,0). For readable layouts, space nodes apart (e.g., 250px horizontal, 150px vertical gaps). Default node size is 200x100px. Boundaries should be sized to contain their child nodes with padding.\n\n**Tags:** Use tags to annotate nodes with metadata (e.g., [\"database\", \"postgres\", \"port:5432\", \"pii\"]). These are searchable and displayed in the UI.\n\n**Colors:** Use hex colors (e.g., \"#336791\" for databases, \"#DC382D\" for caches, \"#009639\" for proxies).";
+export declare const CREATE_DIAGRAM_DESCRIPTION = "Create a new diagram from Mermaid source.\n\nUse this to record system architecture diagrams, data flow diagrams, attack chains, sequence diagrams, etc. from code analysis. Content is Mermaid syntax (flowchart, sequenceDiagram, etc.) \u2014 see zivis_update_mermaid_source for editing an existing diagram's source.";
 export declare const CREATE_DIAGRAM_SCHEMA: {
     name: z.ZodString;
     description: z.ZodOptional<z.ZodString>;
@@ -15,84 +15,14 @@ export declare const CREATE_DIAGRAM_SCHEMA: {
         trust_boundary: "trust_boundary";
         network: "network";
     }>;
-    nodes: z.ZodOptional<z.ZodArray<z.ZodObject<{
-        temp_id: z.ZodOptional<z.ZodString>;
-        name: z.ZodString;
-        description: z.ZodOptional<z.ZodString>;
-        tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
-        icon: z.ZodOptional<z.ZodString>;
-        color: z.ZodOptional<z.ZodString>;
-        position_x: z.ZodOptional<z.ZodNumber>;
-        position_y: z.ZodOptional<z.ZodNumber>;
-        width: z.ZodOptional<z.ZodNumber>;
-        height: z.ZodOptional<z.ZodNumber>;
-        step_order: z.ZodOptional<z.ZodNumber>;
-        boundary_temp_id: z.ZodOptional<z.ZodString>;
-        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    }, z.core.$strip>>>;
-    connections: z.ZodOptional<z.ZodArray<z.ZodObject<{
-        source_temp_id: z.ZodOptional<z.ZodString>;
-        target_temp_id: z.ZodOptional<z.ZodString>;
-        source_node_id: z.ZodOptional<z.ZodString>;
-        target_node_id: z.ZodOptional<z.ZodString>;
-        label: z.ZodOptional<z.ZodString>;
-        description: z.ZodOptional<z.ZodString>;
-        tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
-        bidirectional: z.ZodOptional<z.ZodBoolean>;
-        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    }, z.core.$strip>>>;
-    boundaries: z.ZodOptional<z.ZodArray<z.ZodObject<{
-        temp_id: z.ZodOptional<z.ZodString>;
-        label: z.ZodString;
-        description: z.ZodOptional<z.ZodString>;
-        color: z.ZodOptional<z.ZodString>;
-        position_x: z.ZodOptional<z.ZodNumber>;
-        position_y: z.ZodOptional<z.ZodNumber>;
-        width: z.ZodOptional<z.ZodNumber>;
-        height: z.ZodOptional<z.ZodNumber>;
-    }, z.core.$strip>>>;
+    content: z.ZodString;
     linked_threat_model_id: z.ZodOptional<z.ZodString>;
 };
 export declare function createCreateDiagramHandler(apiClient: ApiClient): (params: {
     name: string;
     description?: string;
     diagram_type: string;
-    nodes?: Array<{
-        temp_id?: string;
-        name: string;
-        description?: string;
-        tags?: string[];
-        icon?: string;
-        color?: string;
-        position_x?: number;
-        position_y?: number;
-        width?: number;
-        height?: number;
-        step_order?: number;
-        boundary_temp_id?: string;
-        metadata?: Record<string, unknown>;
-    }>;
-    connections?: Array<{
-        source_temp_id?: string;
-        target_temp_id?: string;
-        source_node_id?: string;
-        target_node_id?: string;
-        label?: string;
-        description?: string;
-        tags?: string[];
-        bidirectional?: boolean;
-        metadata?: Record<string, unknown>;
-    }>;
-    boundaries?: Array<{
-        temp_id?: string;
-        label: string;
-        description?: string;
-        color?: string;
-        position_x?: number;
-        position_y?: number;
-        width?: number;
-        height?: number;
-    }>;
+    content: string;
     linked_threat_model_id?: string;
 }) => Promise<{
     content: {

package/dist/tools/create-diagram.js

@@ -1,17 +1,9 @@
 import { z } from "zod";
 import { sanitizeResponse } from "../sanitize.js";
 export const CREATE_DIAGRAM_NAME = "zivis_create_diagram";
-export const CREATE_DIAGRAM_DESCRIPTION = `Create a new canvas diagram with nodes, connections, and boundaries in a single call.
+export const CREATE_DIAGRAM_DESCRIPTION = `Create a new diagram from Mermaid source.
 
-Use this to build system architecture diagrams, data flow diagrams, attack chains, etc. from code analysis.
-
-**Temp IDs:** When creating nodes and connections together, assign each node a temp_id (e.g., "node_0", "node_1") and reference them in connections via source_temp_id / target_temp_id. Similarly, assign boundaries a temp_id and reference them in nodes via boundary_temp_id.
-
-**Positioning:** Nodes default to (0,0). For readable layouts, space nodes apart (e.g., 250px horizontal, 150px vertical gaps). Default node size is 200x100px. Boundaries should be sized to contain their child nodes with padding.
-
-**Tags:** Use tags to annotate nodes with metadata (e.g., ["database", "postgres", "port:5432", "pii"]). These are searchable and displayed in the UI.
-
-**Colors:** Use hex colors (e.g., "#336791" for databases, "#DC382D" for caches, "#009639" for proxies).`;
+Use this to record system architecture diagrams, data flow diagrams, attack chains, sequence diagrams, etc. from code analysis. Content is Mermaid syntax (flowchart, sequenceDiagram, etc.) — see zivis_update_mermaid_source for editing an existing diagram's source.`;
 export const CREATE_DIAGRAM_SCHEMA = {
     name: z
         .string()
@@ -23,51 +15,9 @@ export const CREATE_DIAGRAM_SCHEMA = {
     diagram_type: z
         .enum(["architecture", "data_flow", "attack_chain", "sequence", "trust_boundary", "network", "deployment", "custom"])
         .describe("Type of diagram"),
-    nodes: z
-        .array(z.object({
-        temp_id: z.string().optional().describe("Temporary ID for referencing in connections (e.g., 'node_0')"),
-        name: z.string().describe("Node name (e.g., 'PostgreSQL', 'API Gateway')"),
-        description: z.string().optional().describe("Node description"),
-        tags: z.array(z.string()).optional().describe("Tags (e.g., ['database', 'port:5432'])"),
-        icon: z.string().optional().describe("Icon name (e.g., 'database', 'server', 'shield', 'globe')"),
-        color: z.string().optional().describe("Hex color (e.g., '#336791')"),
-        position_x: z.number().optional().describe("X position on canvas (default 0)"),
-        position_y: z.number().optional().describe("Y position on canvas (default 0)"),
-        width: z.number().optional().describe("Node width (default 200)"),
-        height: z.number().optional().describe("Node height (default 100)"),
-        step_order: z.number().optional().describe("Step order for attack chain diagrams"),
-        boundary_temp_id: z.string().optional().describe("Temp ID of boundary this node belongs to"),
-        metadata: z.record(z.string(), z.unknown()).optional().describe("Custom metadata"),
-    }))
-        .optional()
-        .describe("Nodes to create"),
-    connections: z
-        .array(z.object({
-        source_temp_id: z.string().optional().describe("Source node temp ID"),
-        target_temp_id: z.string().optional().describe("Target node temp ID"),
-        source_node_id: z.string().optional().describe("Source node UUID (if referencing existing node)"),
-        target_node_id: z.string().optional().describe("Target node UUID (if referencing existing node)"),
-        label: z.string().optional().describe("Connection label (e.g., 'HTTPS', 'depends on')"),
-        description: z.string().optional().describe("Connection description"),
-        tags: z.array(z.string()).optional().describe("Connection tags"),
-        bidirectional: z.boolean().optional().describe("Whether connection goes both ways (default false)"),
-        metadata: z.record(z.string(), z.unknown()).optional().describe("Custom metadata"),
-    }))
-        .optional()
-        .describe("Connections between nodes"),
-    boundaries: z
-        .array(z.object({
-        temp_id: z.string().optional().describe("Temporary ID for referencing in nodes (e.g., 'boundary_0')"),
-        label: z.string().describe("Boundary label (e.g., 'DMZ', 'Internal Network')"),
-        description: z.string().optional().describe("Boundary description"),
-        color: z.string().optional().describe("Hex color"),
-        position_x: z.number().optional().describe("X position (default 0)"),
-        position_y: z.number().optional().describe("Y position (default 0)"),
-        width: z.number().optional().describe("Boundary width (default 400)"),
-        height: z.number().optional().describe("Boundary height (default 300)"),
-    }))
-        .optional()
-        .describe("Boundary groups (trust boundaries, network segments)"),
+    content: z
+        .string()
+        .describe("Mermaid diagram source (e.g., 'flowchart TD\\n  A[Client] --> B[Server]')"),
     linked_threat_model_id: z
         .string()
         .optional()
@@ -76,51 +26,13 @@ export const CREATE_DIAGRAM_SCHEMA = {
 export function createCreateDiagramHandler(apiClient) {
     return async (params) => {
         try {
-            const apiNodes = params.nodes?.map((n) => ({
-                tempId: n.temp_id,
-                name: n.name,
-                description: n.description,
-                tags: n.tags,
-                icon: n.icon,
-                color: n.color,
-                positionX: n.position_x,
-                positionY: n.position_y,
-                width: n.width,
-                height: n.height,
-                stepOrder: n.step_order,
-                boundaryTempId: n.boundary_temp_id,
-                metadata: n.metadata,
-            }));
-            const apiConnections = params.connections?.map((c) => ({
-                sourceTempId: c.source_temp_id,
-                targetTempId: c.target_temp_id,
-                sourceNodeId: c.source_node_id,
-                targetNodeId: c.target_node_id,
-                label: c.label,
-                description: c.description,
-                tags: c.tags,
-                bidirectional: c.bidirectional,
-                metadata: c.metadata,
-            }));
-            const apiBoundaries = params.boundaries?.map((b) => ({
-                tempId: b.temp_id,
-                label: b.label,
-                description: b.description,
-                color: b.color,
-                positionX: b.position_x,
-                positionY: b.position_y,
-                width: b.width,
-                height: b.height,
-            }));
             const data = await apiClient.post("/api/diagrams", {
                 name: params.name,
                 description: params.description,
                 diagramType: params.diagram_type,
-                contentType: "canvas",
+                contentType: "mermaid",
+                content: params.content,
                 sourceType: "manual",
-                nodes: apiNodes,
-                connections: apiConnections,
-                boundaries: apiBoundaries,
             });
             const sanitized = sanitizeResponse(data);
             const diagramId = sanitized.id;
@@ -135,16 +47,12 @@ export function createCreateDiagramHandler(apiClient) {
                 catch {
                 }
             }
-            const nodeCount = Array.isArray(sanitized.nodes) ? sanitized.nodes.length : 0;
-            const connCount = Array.isArray(sanitized.connections) ? sanitized.connections.length : 0;
-            const boundaryCount = Array.isArray(sanitized.boundaries) ? sanitized.boundaries.length : 0;
             const result = {
                 diagram_id: diagramId,
                 name: sanitized.name,
                 diagram_type: sanitized.diagramType,
-                stats: { nodes: nodeCount, connections: connCount, boundaries: boundaryCount },
                 linked_threat_model_id: linkedThreatModelId,
-                _instruction: "Diagram created. Use zivis_get_diagram to see full content, or zivis_manage_diagram to modify.",
+                _instruction: "Diagram created. Use zivis_get_diagram to see full content, or zivis_manage_diagram to modify metadata/links, or zivis_update_mermaid_source to change the diagram source.",
             };
             return {
                 content: [

package/dist/tools/get-diagram.d.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import type { ApiClient } from "../api-client.js";
 export declare const GET_DIAGRAM_NAME = "zivis_get_diagram";
-export declare const GET_DIAGRAM_DESCRIPTION = "Get the full content of a diagram including all nodes, connections, boundaries, and Mermaid source.\n\nReturns node IDs, names, tags, positions, boundary assignments, and metadata. Connection IDs include source/target node references and labels. Boundary IDs include labels and positions.\n\nFor mermaid-type diagrams, the `content` field contains the full Mermaid source string \u2014 use it to read or render the diagram. For canvas-type diagrams, `content` is null and the structure is expressed via nodes/connections/boundaries.\n\nUse this before modifying a diagram \u2014 you need node/connection/boundary IDs for zivis_manage_diagram (canvas) or the Mermaid source for zivis_update_mermaid_source (mermaid).";
+export declare const GET_DIAGRAM_DESCRIPTION = "Get the full content of a diagram, including its Mermaid source.\n\nUse this before modifying a diagram \u2014 you need the current Mermaid source for zivis_update_mermaid_source.";
 export declare const GET_DIAGRAM_SCHEMA: {
     diagram_id: z.ZodString;
 };

package/dist/tools/get-diagram.js

@@ -1,13 +1,9 @@
 import { z } from "zod";
 import { sanitizeResponse } from "../sanitize.js";
 export const GET_DIAGRAM_NAME = "zivis_get_diagram";
-export const GET_DIAGRAM_DESCRIPTION = `Get the full content of a diagram including all nodes, connections, boundaries, and Mermaid source.
+export const GET_DIAGRAM_DESCRIPTION = `Get the full content of a diagram, including its Mermaid source.
 
-Returns node IDs, names, tags, positions, boundary assignments, and metadata. Connection IDs include source/target node references and labels. Boundary IDs include labels and positions.
-
-For mermaid-type diagrams, the \`content\` field contains the full Mermaid source string — use it to read or render the diagram. For canvas-type diagrams, \`content\` is null and the structure is expressed via nodes/connections/boundaries.
-
-Use this before modifying a diagram — you need node/connection/boundary IDs for zivis_manage_diagram (canvas) or the Mermaid source for zivis_update_mermaid_source (mermaid).`;
+Use this before modifying a diagram — you need the current Mermaid source for zivis_update_mermaid_source.`;
 export const GET_DIAGRAM_SCHEMA = {
     diagram_id: z
         .string()
@@ -18,49 +14,6 @@ export function createGetDiagramHandler(apiClient) {
         try {
             const data = await apiClient.get(`/api/diagrams/${params.diagram_id}`);
             const sanitized = sanitizeResponse(data);
-            const nodes = Array.isArray(sanitized.nodes)
-                ? sanitized.nodes.map((n) => ({
-                    node_id: n.id,
-                    name: n.name,
-                    description: n.description || null,
-                    tags: n.tags || [],
-                    icon: n.icon || null,
-                    color: n.color || null,
-                    position_x: n.positionX,
-                    position_y: n.positionY,
-                    width: n.width,
-                    height: n.height,
-                    step_order: n.stepOrder || null,
-                    boundary_id: n.boundaryId || null,
-                    metadata: n.metadata || null,
-                    linked_finding_ids: n.linkedFindingIds || [],
-                    linked_evidence_ids: n.linkedEvidenceIds || [],
-                }))
-                : [];
-            const connections = Array.isArray(sanitized.connections)
-                ? sanitized.connections.map((c) => ({
-                    connection_id: c.id,
-                    source_node_id: c.sourceNodeId,
-                    target_node_id: c.targetNodeId,
-                    label: c.label || null,
-                    description: c.description || null,
-                    tags: c.tags || [],
-                    bidirectional: c.bidirectional || false,
-                    metadata: c.metadata || null,
-                }))
-                : [];
-            const boundaries = Array.isArray(sanitized.boundaries)
-                ? sanitized.boundaries.map((b) => ({
-                    boundary_id: b.id,
-                    label: b.label,
-                    description: b.description || null,
-                    color: b.color || null,
-                    position_x: b.positionX,
-                    position_y: b.positionY,
-                    width: b.width,
-                    height: b.height,
-                }))
-                : [];
             const linkedThreatModelIds = Array.isArray(sanitized.threatModelLinks)
                 ? sanitized.threatModelLinks.map((l) => l.threatModelId)
                 : [];
@@ -74,9 +27,6 @@ export function createGetDiagramHandler(apiClient) {
                 source_type: sanitized.sourceType,
                 created_at: sanitized.createdAt,
                 updated_at: sanitized.updatedAt,
-                nodes,
-                connections,
-                boundaries,
                 linked_threat_model_ids: linkedThreatModelIds,
             };
             return {

package/dist/tools/list-diagrams.d.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import type { ApiClient } from "../api-client.js";
 export declare const LIST_DIAGRAMS_NAME = "zivis_list_diagrams";
-export declare const LIST_DIAGRAMS_DESCRIPTION = "List diagrams for your organization. Returns summary info (name, type, node/connection counts) for each diagram.\n\nUse this tool to find existing diagrams before modifying them. To get the full content (nodes, connections, boundaries), use zivis_get_diagram with the diagram_id from this response.\n\nDiagrams can be linked to threat models. Use linked_threat_model_id to find diagrams for a specific threat model.";
+export declare const LIST_DIAGRAMS_DESCRIPTION = "List diagrams for your organization. Returns summary info (name, type, description) for each diagram.\n\nUse this tool to find existing diagrams before modifying them. To get the full content (Mermaid source), use zivis_get_diagram with the diagram_id from this response.\n\nDiagrams can be linked to threat models. Use linked_threat_model_id to find diagrams for a specific threat model.";
 export declare const LIST_DIAGRAMS_SCHEMA: {
     diagram_type: z.ZodOptional<z.ZodEnum<{
         custom: "custom";

package/dist/tools/list-diagrams.js

@@ -1,9 +1,9 @@
 import { z } from "zod";
 import { sanitizeResponse } from "../sanitize.js";
 export const LIST_DIAGRAMS_NAME = "zivis_list_diagrams";
-export const LIST_DIAGRAMS_DESCRIPTION = `List diagrams for your organization. Returns summary info (name, type, node/connection counts) for each diagram.
+export const LIST_DIAGRAMS_DESCRIPTION = `List diagrams for your organization. Returns summary info (name, type, description) for each diagram.
 
-Use this tool to find existing diagrams before modifying them. To get the full content (nodes, connections, boundaries), use zivis_get_diagram with the diagram_id from this response.
+Use this tool to find existing diagrams before modifying them. To get the full content (Mermaid source), use zivis_get_diagram with the diagram_id from this response.
 
 Diagrams can be linked to threat models. Use linked_threat_model_id to find diagrams for a specific threat model.`;
 export const LIST_DIAGRAMS_SCHEMA = {
@@ -52,7 +52,6 @@ export function createListDiagramsHandler(apiClient) {
                     description: d.description || null,
                     created_at: d.createdAt,
                     updated_at: d.updatedAt,
-                    stats: d._count || null,
                 }))
                 : sanitized.diagrams;
             return {

package/dist/tools/manage-diagram.d.ts

@@ -1,21 +1,12 @@
 import { z } from "zod";
 import type { ApiClient } from "../api-client.js";
 export declare const MANAGE_DIAGRAM_NAME = "zivis_manage_diagram";
-export declare const MANAGE_DIAGRAM_DESCRIPTION = "Modify an existing diagram: update metadata, add/update/delete nodes, connections, and boundaries, or link/unlink threat models.\n\n**Before modifying:** Use zivis_get_diagram to fetch current content and get node/connection/boundary IDs.\n\nActions:\n- update_metadata: Update diagram name, description, or type\n- delete_diagram: Delete the entire diagram\n- add_node: Add a new node (requires node_name)\n- update_node: Update an existing node (requires node_id)\n- delete_node: Remove a node (requires node_id)\n- add_connection: Add a connection between two nodes (requires source_node_id, target_node_id)\n- update_connection: Update a connection (requires connection_id)\n- delete_connection: Remove a connection (requires connection_id)\n- add_boundary: Add a boundary group (requires boundary_label)\n- update_boundary: Update a boundary (requires boundary_id)\n- delete_boundary: Remove a boundary (requires boundary_id)\n- link_threat_model: Link diagram to a threat model (requires threat_model_id)\n- unlink_threat_model: Unlink from a threat model (requires threat_model_id)";
+export declare const MANAGE_DIAGRAM_DESCRIPTION = "Modify an existing diagram: update metadata, delete it, or link/unlink it to a threat model.\n\nTo change a diagram's Mermaid source, use zivis_update_mermaid_source instead.\n\nActions:\n- update_metadata: Update diagram name, description, or type\n- delete_diagram: Delete the entire diagram\n- link_threat_model: Link diagram to a threat model (requires threat_model_id)\n- unlink_threat_model: Unlink from a threat model (requires threat_model_id)";
 export declare const MANAGE_DIAGRAM_SCHEMA: {
     diagram_id: z.ZodString;
     action: z.ZodEnum<{
         update_metadata: "update_metadata";
         delete_diagram: "delete_diagram";
-        add_node: "add_node";
-        update_node: "update_node";
-        delete_node: "delete_node";
-        add_connection: "add_connection";
-        update_connection: "update_connection";
-        delete_connection: "delete_connection";
-        add_boundary: "add_boundary";
-        update_boundary: "update_boundary";
-        delete_boundary: "delete_boundary";
         link_threat_model: "link_threat_model";
         unlink_threat_model: "unlink_threat_model";
     }>;
@@ -31,33 +22,6 @@ export declare const MANAGE_DIAGRAM_SCHEMA: {
         trust_boundary: "trust_boundary";
         network: "network";
     }>>;
-    node_id: z.ZodOptional<z.ZodString>;
-    node_name: z.ZodOptional<z.ZodString>;
-    node_description: z.ZodOptional<z.ZodString>;
-    node_tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
-    node_icon: z.ZodOptional<z.ZodString>;
-    node_color: z.ZodOptional<z.ZodString>;
-    position_x: z.ZodOptional<z.ZodNumber>;
-    position_y: z.ZodOptional<z.ZodNumber>;
-    width: z.ZodOptional<z.ZodNumber>;
-    height: z.ZodOptional<z.ZodNumber>;
-    step_order: z.ZodOptional<z.ZodNumber>;
-    boundary_id: z.ZodOptional<z.ZodString>;
-    metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    connection_id: z.ZodOptional<z.ZodString>;
-    source_node_id: z.ZodOptional<z.ZodString>;
-    target_node_id: z.ZodOptional<z.ZodString>;
-    label: z.ZodOptional<z.ZodString>;
-    connection_description: z.ZodOptional<z.ZodString>;
-    connection_tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
-    bidirectional: z.ZodOptional<z.ZodBoolean>;
-    boundary_label: z.ZodOptional<z.ZodString>;
-    boundary_description: z.ZodOptional<z.ZodString>;
-    boundary_color: z.ZodOptional<z.ZodString>;
-    boundary_position_x: z.ZodOptional<z.ZodNumber>;
-    boundary_position_y: z.ZodOptional<z.ZodNumber>;
-    boundary_width: z.ZodOptional<z.ZodNumber>;
-    boundary_height: z.ZodOptional<z.ZodNumber>;
     threat_model_id: z.ZodOptional<z.ZodString>;
 };
 type ManageDiagramParams = {
@@ -66,33 +30,6 @@ type ManageDiagramParams = {
     name?: string;
     description?: string;
     diagram_type?: string;
-    node_id?: string;
-    node_name?: string;
-    node_description?: string;
-    node_tags?: string[];
-    node_icon?: string;
-    node_color?: string;
-    position_x?: number;
-    position_y?: number;
-    width?: number;
-    height?: number;
-    step_order?: number;
-    boundary_id?: string;
-    metadata?: Record<string, unknown>;
-    connection_id?: string;
-    source_node_id?: string;
-    target_node_id?: string;
-    label?: string;
-    connection_description?: string;
-    connection_tags?: string[];
-    bidirectional?: boolean;
-    boundary_label?: string;
-    boundary_description?: string;
-    boundary_color?: string;
-    boundary_position_x?: number;
-    boundary_position_y?: number;
-    boundary_width?: number;
-    boundary_height?: number;
     threat_model_id?: string;
 };
 export declare function createManageDiagramHandler(apiClient: ApiClient): (params: ManageDiagramParams) => Promise<{

package/dist/tools/manage-diagram.js

@@ -1,22 +1,13 @@
 import { z } from "zod";
 import { sanitizeResponse } from "../sanitize.js";
 export const MANAGE_DIAGRAM_NAME = "zivis_manage_diagram";
-export const MANAGE_DIAGRAM_DESCRIPTION = `Modify an existing diagram: update metadata, add/update/delete nodes, connections, and boundaries, or link/unlink threat models.
+export const MANAGE_DIAGRAM_DESCRIPTION = `Modify an existing diagram: update metadata, delete it, or link/unlink it to a threat model.
 
-**Before modifying:** Use zivis_get_diagram to fetch current content and get node/connection/boundary IDs.
+To change a diagram's Mermaid source, use zivis_update_mermaid_source instead.
 
 Actions:
 - update_metadata: Update diagram name, description, or type
 - delete_diagram: Delete the entire diagram
-- add_node: Add a new node (requires node_name)
-- update_node: Update an existing node (requires node_id)
-- delete_node: Remove a node (requires node_id)
-- add_connection: Add a connection between two nodes (requires source_node_id, target_node_id)
-- update_connection: Update a connection (requires connection_id)
-- delete_connection: Remove a connection (requires connection_id)
-- add_boundary: Add a boundary group (requires boundary_label)
-- update_boundary: Update a boundary (requires boundary_id)
-- delete_boundary: Remove a boundary (requires boundary_id)
 - link_threat_model: Link diagram to a threat model (requires threat_model_id)
 - unlink_threat_model: Unlink from a threat model (requires threat_model_id)`;
 export const MANAGE_DIAGRAM_SCHEMA = {
@@ -27,15 +18,6 @@ export const MANAGE_DIAGRAM_SCHEMA = {
         .enum([
         "update_metadata",
         "delete_diagram",
-        "add_node",
-        "update_node",
-        "delete_node",
-        "add_connection",
-        "update_connection",
-        "delete_connection",
-        "add_boundary",
-        "update_boundary",
-        "delete_boundary",
         "link_threat_model",
         "unlink_threat_model",
     ])
@@ -46,33 +28,6 @@ export const MANAGE_DIAGRAM_SCHEMA = {
         .enum(["architecture", "data_flow", "attack_chain", "sequence", "trust_boundary", "network", "deployment", "custom"])
         .optional()
         .describe("Diagram type (update_metadata)"),
-    node_id: z.string().optional().describe("Node UUID (required for update_node, delete_node)"),
-    node_name: z.string().optional().describe("Node name (required for add_node, optional for update_node)"),
-    node_description: z.string().optional().describe("Node description"),
-    node_tags: z.array(z.string()).optional().describe("Node tags (e.g., ['database', 'port:5432'])"),
-    node_icon: z.string().optional().describe("Node icon"),
-    node_color: z.string().optional().describe("Hex color (e.g., '#336791')"),
-    position_x: z.number().optional().describe("X position on canvas"),
-    position_y: z.number().optional().describe("Y position on canvas"),
-    width: z.number().optional().describe("Width in pixels"),
-    height: z.number().optional().describe("Height in pixels"),
-    step_order: z.number().optional().describe("Step order (attack chain diagrams)"),
-    boundary_id: z.string().optional().describe("Boundary UUID to assign node to, or boundary to update/delete"),
-    metadata: z.record(z.string(), z.unknown()).optional().describe("Custom metadata JSON"),
-    connection_id: z.string().optional().describe("Connection UUID (required for update_connection, delete_connection)"),
-    source_node_id: z.string().optional().describe("Source node UUID (required for add_connection)"),
-    target_node_id: z.string().optional().describe("Target node UUID (required for add_connection)"),
-    label: z.string().optional().describe("Connection label"),
-    connection_description: z.string().optional().describe("Connection description"),
-    connection_tags: z.array(z.string()).optional().describe("Connection tags"),
-    bidirectional: z.boolean().optional().describe("Whether connection goes both ways"),
-    boundary_label: z.string().optional().describe("Boundary label (required for add_boundary)"),
-    boundary_description: z.string().optional().describe("Boundary description"),
-    boundary_color: z.string().optional().describe("Boundary hex color"),
-    boundary_position_x: z.number().optional().describe("Boundary X position"),
-    boundary_position_y: z.number().optional().describe("Boundary Y position"),
-    boundary_width: z.number().optional().describe("Boundary width"),
-    boundary_height: z.number().optional().describe("Boundary height"),
     threat_model_id: z.string().optional().describe("Threat model UUID (link/unlink)"),
 };
 function errorResult(message) {
@@ -107,170 +62,6 @@ export function createManageDiagramHandler(apiClient) {
                     await apiClient.del(base);
                     return successResult({ action: "deleted", diagram_id });
                 }
-                case "add_node": {
-                    if (!params.node_name)
-                        return errorResult("node_name is required for add_node");
-                    const body = {
-                        name: params.node_name,
-                    };
-                    if (params.node_description)
-                        body.description = params.node_description;
-                    if (params.node_tags)
-                        body.tags = params.node_tags;
-                    if (params.node_icon)
-                        body.icon = params.node_icon;
-                    if (params.node_color)
-                        body.color = params.node_color;
-                    if (params.position_x !== undefined)
-                        body.positionX = params.position_x;
-                    if (params.position_y !== undefined)
-                        body.positionY = params.position_y;
-                    if (params.width !== undefined)
-                        body.width = params.width;
-                    if (params.height !== undefined)
-                        body.height = params.height;
-                    if (params.step_order !== undefined)
-                        body.stepOrder = params.step_order;
-                    if (params.boundary_id)
-                        body.boundaryId = params.boundary_id;
-                    if (params.metadata)
-                        body.metadata = params.metadata;
-                    const data = await apiClient.post(`${base}/nodes`, body);
-                    return successResult(data);
-                }
-                case "update_node": {
-                    if (!params.node_id)
-                        return errorResult("node_id is required for update_node");
-                    const body = {};
-                    if (params.node_name)
-                        body.name = params.node_name;
-                    if (params.node_description !== undefined)
-                        body.description = params.node_description;
-                    if (params.node_tags)
-                        body.tags = params.node_tags;
-                    if (params.node_icon !== undefined)
-                        body.icon = params.node_icon;
-                    if (params.node_color !== undefined)
-                        body.color = params.node_color;
-                    if (params.position_x !== undefined)
-                        body.positionX = params.position_x;
-                    if (params.position_y !== undefined)
-                        body.positionY = params.position_y;
-                    if (params.width !== undefined)
-                        body.width = params.width;
-                    if (params.height !== undefined)
-                        body.height = params.height;
-                    if (params.step_order !== undefined)
-                        body.stepOrder = params.step_order;
-                    if (params.boundary_id !== undefined)
-                        body.boundaryId = params.boundary_id;
-                    if (params.metadata)
-                        body.metadata = params.metadata;
-                    const data = await apiClient.patch(`${base}/nodes/${params.node_id}`, body);
-                    return successResult(data);
-                }
-                case "delete_node": {
-                    if (!params.node_id)
-                        return errorResult("node_id is required for delete_node");
-                    await apiClient.del(`${base}/nodes/${params.node_id}`);
-                    return successResult({ action: "deleted", node_id: params.node_id });
-                }
-                case "add_connection": {
-                    if (!params.source_node_id || !params.target_node_id) {
-                        return errorResult("source_node_id and target_node_id are required for add_connection");
-                    }
-                    const body = {
-                        sourceNodeId: params.source_node_id,
-                        targetNodeId: params.target_node_id,
-                    };
-                    if (params.label)
-                        body.label = params.label;
-                    if (params.connection_description)
-                        body.description = params.connection_description;
-                    if (params.connection_tags)
-                        body.tags = params.connection_tags;
-                    if (params.bidirectional !== undefined)
-                        body.bidirectional = params.bidirectional;
-                    if (params.metadata)
-                        body.metadata = params.metadata;
-                    const data = await apiClient.post(`${base}/connections`, body);
-                    return successResult(data);
-                }
-                case "update_connection": {
-                    if (!params.connection_id)
-                        return errorResult("connection_id is required for update_connection");
-                    const body = {};
-                    if (params.source_node_id)
-                        body.sourceNodeId = params.source_node_id;
-                    if (params.target_node_id)
-                        body.targetNodeId = params.target_node_id;
-                    if (params.label !== undefined)
-                        body.label = params.label;
-                    if (params.connection_description !== undefined)
-                        body.description = params.connection_description;
-                    if (params.connection_tags)
-                        body.tags = params.connection_tags;
-                    if (params.bidirectional !== undefined)
-                        body.bidirectional = params.bidirectional;
-                    if (params.metadata)
-                        body.metadata = params.metadata;
-                    const data = await apiClient.patch(`${base}/connections/${params.connection_id}`, body);
-                    return successResult(data);
-                }
-                case "delete_connection": {
-                    if (!params.connection_id)
-                        return errorResult("connection_id is required for delete_connection");
-                    await apiClient.del(`${base}/connections/${params.connection_id}`);
-                    return successResult({ action: "deleted", connection_id: params.connection_id });
-                }
-                case "add_boundary": {
-                    if (!params.boundary_label)
-                        return errorResult("boundary_label is required for add_boundary");
-                    const body = {
-                        label: params.boundary_label,
-                    };
-                    if (params.boundary_description)
-                        body.description = params.boundary_description;
-                    if (params.boundary_color)
-                        body.color = params.boundary_color;
-                    if (params.boundary_position_x !== undefined)
-                        body.positionX = params.boundary_position_x;
-                    if (params.boundary_position_y !== undefined)
-                        body.positionY = params.boundary_position_y;
-                    if (params.boundary_width !== undefined)
-                        body.width = params.boundary_width;
-                    if (params.boundary_height !== undefined)
-                        body.height = params.boundary_height;
-                    const data = await apiClient.post(`${base}/boundaries`, body);
-                    return successResult(data);
-                }
-                case "update_boundary": {
-                    if (!params.boundary_id)
-                        return errorResult("boundary_id is required for update_boundary");
-                    const body = {};
-                    if (params.boundary_label)
-                        body.label = params.boundary_label;
-                    if (params.boundary_description !== undefined)
-                        body.description = params.boundary_description;
-                    if (params.boundary_color !== undefined)
-                        body.color = params.boundary_color;
-                    if (params.boundary_position_x !== undefined)
-                        body.positionX = params.boundary_position_x;
-                    if (params.boundary_position_y !== undefined)
-                        body.positionY = params.boundary_position_y;
-                    if (params.boundary_width !== undefined)
-                        body.width = params.boundary_width;
-                    if (params.boundary_height !== undefined)
-                        body.height = params.boundary_height;
-                    const data = await apiClient.patch(`${base}/boundaries/${params.boundary_id}`, body);
-                    return successResult(data);
-                }
-                case "delete_boundary": {
-                    if (!params.boundary_id)
-                        return errorResult("boundary_id is required for delete_boundary");
-                    await apiClient.del(`${base}/boundaries/${params.boundary_id}`);
-                    return successResult({ action: "deleted", boundary_id: params.boundary_id });
-                }
                 case "link_threat_model": {
                     if (!params.threat_model_id)
                         return errorResult("threat_model_id is required for link_threat_model");

package/dist/tools/update-mermaid-source.d.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import type { ApiClient } from "../api-client.js";
 export declare const UPDATE_MERMAID_SOURCE_NAME = "zivis_update_mermaid_source";
-export declare const UPDATE_MERMAID_SOURCE_DESCRIPTION = "Replace the Mermaid source of a diagram with new Mermaid syntax.\n\n**Use this when:** the diagram's contentType is \"mermaid\" (text-based) and you want to add, rename, delete, or reconnect nodes in the diagram via Mermaid syntax.\n\n**Workflow:**\n1. Call zivis_get_diagram first to read the current Mermaid source.\n2. Reason about the requested change (rename a node, add a connection, restructure a subgraph, etc.).\n3. Produce a COMPLETE, valid Mermaid source string.\n4. Call this tool with the new source \u2014 it replaces the entire content atomically.\n\n**Do not use for canvas diagrams** (structured nodes/connections). Use zivis_manage_diagram for those.\n\n**Keep the overall structure.** Only change what the user asked for. Preserve unrelated nodes, edges, labels, subgraphs, styles.";
+export declare const UPDATE_MERMAID_SOURCE_DESCRIPTION = "Replace the Mermaid source of a diagram with new Mermaid syntax.\n\n**Use this when:** you want to add, rename, delete, or reconnect nodes in a diagram via Mermaid syntax.\n\n**Workflow:**\n1. Call zivis_get_diagram first to read the current Mermaid source.\n2. Reason about the requested change (rename a node, add a connection, restructure a subgraph, etc.).\n3. Produce a COMPLETE, valid Mermaid source string.\n4. Call this tool with the new source \u2014 it replaces the entire content atomically.\n\n**Keep the overall structure.** Only change what the user asked for. Preserve unrelated nodes, edges, labels, subgraphs, styles.";
 export declare const UPDATE_MERMAID_SOURCE_SCHEMA: {
     diagram_id: z.ZodString;
     mermaid_source: z.ZodString;

package/dist/tools/update-mermaid-source.js

@@ -3,7 +3,7 @@ import { sanitizeResponse } from "../sanitize.js";
 export const UPDATE_MERMAID_SOURCE_NAME = "zivis_update_mermaid_source";
 export const UPDATE_MERMAID_SOURCE_DESCRIPTION = `Replace the Mermaid source of a diagram with new Mermaid syntax.
 
-**Use this when:** the diagram's contentType is "mermaid" (text-based) and you want to add, rename, delete, or reconnect nodes in the diagram via Mermaid syntax.
+**Use this when:** you want to add, rename, delete, or reconnect nodes in a diagram via Mermaid syntax.
 
 **Workflow:**
 1. Call zivis_get_diagram first to read the current Mermaid source.
@@ -11,8 +11,6 @@ export const UPDATE_MERMAID_SOURCE_DESCRIPTION = `Replace the Mermaid source of
 3. Produce a COMPLETE, valid Mermaid source string.
 4. Call this tool with the new source — it replaces the entire content atomically.
 
-**Do not use for canvas diagrams** (structured nodes/connections). Use zivis_manage_diagram for those.
-
 **Keep the overall structure.** Only change what the user asked for. Preserve unrelated nodes, edges, labels, subgraphs, styles.`;
 export const UPDATE_MERMAID_SOURCE_SCHEMA = {
     diagram_id: z.string().describe("UUID of the diagram to update"),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zivis/mcp",
-  "version": "0.1.11",
+  "version": "0.1.12",
   "description": "ZIVIS MCP server — threat modeling, security scans, and AI red team tools for IDE integration",
   "license": "Apache-2.0",
   "homepage": "https://zivis.ai",
@@ -59,7 +59,7 @@
     "@ast-grep/lang-python": "^0.0.6",
     "@ast-grep/napi": "^0.42.1",
     "@modelcontextprotocol/sdk": "^1.26.0",
-    "js-yaml": "^4.1.0",
+    "js-yaml": "^4.3.2",
     "open": "^10.1.0",
     "zod": "^4.0.0"
   },

package/dist/tools/generate-diagram.d.ts

@@ -1,30 +0,0 @@
-import { z } from "zod";
-import type { ApiClient } from "../api-client.js";
-export declare const GENERATE_DIAGRAM_NAME = "zivis_generate_diagram";
-export declare const GENERATE_DIAGRAM_DESCRIPTION = "Generate a visual architecture diagram from a Docker Compose file.\n\nReads a local docker-compose.yml and sends it to the ZIVIS platform to generate\na positioned canvas diagram with:\n- Services mapped to color-coded nodes (database, cache, app, proxy, etc.)\n- Docker networks mapped to boundary groups\n- depends_on relationships mapped to connections\n- Auto-positioned grid layout grouped by network\n\nThe diagram is saved to your organization and can be viewed/edited in the diagram editor.\n\nSupported formats: docker-compose (docker-compose.yml, compose.yml)\nComing soon: Kubernetes manifests, Terraform configs";
-export declare const GENERATE_DIAGRAM_SCHEMA: {
-    file_path: z.ZodOptional<z.ZodString>;
-    source_format: z.ZodDefault<z.ZodEnum<{
-        "docker-compose": "docker-compose";
-    }>>;
-    name: z.ZodOptional<z.ZodString>;
-    project_dir: z.ZodOptional<z.ZodString>;
-};
-export declare function createGenerateDiagramHandler(apiClient: ApiClient): (params: {
-    file_path?: string;
-    source_format?: string;
-    name?: string;
-    project_dir?: string;
-}) => Promise<{
-    content: {
-        type: "text";
-        text: string;
-    }[];
-    isError: boolean;
-} | {
-    content: {
-        type: "text";
-        text: string;
-    }[];
-    isError?: undefined;
-}>;

package/dist/tools/generate-diagram.js

@@ -1,161 +0,0 @@
-import { z } from "zod";
-import { readFile } from "fs/promises";
-import { join, relative, resolve } from "path";
-import { sanitizeResponse } from "../sanitize.js";
-function resolveWithinProjectDir(projectDir, candidate) {
-    const base = resolve(projectDir);
-    const target = resolve(base, candidate);
-    const rel = relative(base, target);
-    if (rel.startsWith(".."))
-        return null;
-    return target;
-}
-export const GENERATE_DIAGRAM_NAME = "zivis_generate_diagram";
-export const GENERATE_DIAGRAM_DESCRIPTION = `Generate a visual architecture diagram from a Docker Compose file.
-
-Reads a local docker-compose.yml and sends it to the ZIVIS platform to generate
-a positioned canvas diagram with:
-- Services mapped to color-coded nodes (database, cache, app, proxy, etc.)
-- Docker networks mapped to boundary groups
-- depends_on relationships mapped to connections
-- Auto-positioned grid layout grouped by network
-
-The diagram is saved to your organization and can be viewed/edited in the diagram editor.
-
-Supported formats: docker-compose (docker-compose.yml, compose.yml)
-Coming soon: Kubernetes manifests, Terraform configs`;
-export const GENERATE_DIAGRAM_SCHEMA = {
-    file_path: z
-        .string()
-        .optional()
-        .describe("Path to the infrastructure file (e.g., docker-compose.yml). " +
-        "If not provided, searches the current directory for docker-compose.yml, " +
-        "docker-compose.yaml, compose.yml, or compose.yaml."),
-    source_format: z
-        .enum(["docker-compose"])
-        .default("docker-compose")
-        .describe("The format of the source file. Currently only docker-compose is supported."),
-    name: z
-        .string()
-        .optional()
-        .describe("Custom name for the generated diagram. Defaults to 'Docker Compose Architecture'."),
-    project_dir: z
-        .string()
-        .optional()
-        .describe("Project directory to search for compose files. Defaults to current working directory."),
-};
-const COMPOSE_FILE_CANDIDATES = [
-    "docker-compose.yml",
-    "docker-compose.yaml",
-    "compose.yml",
-    "compose.yaml",
-];
-export function createGenerateDiagramHandler(apiClient) {
-    return async (params) => {
-        const projectDir = params.project_dir || process.cwd();
-        const sourceFormat = params.source_format || "docker-compose";
-        let filePath = params.file_path;
-        let content = null;
-        if (filePath) {
-            const resolved = resolveWithinProjectDir(projectDir, filePath);
-            if (!resolved) {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: `Error: file_path must resolve inside project_dir (got: ${filePath})`,
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-            try {
-                content = await readFile(resolved, "utf-8");
-            }
-            catch {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: `Error: File not found: ${resolved}`,
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-        }
-        else {
-            for (const candidate of COMPOSE_FILE_CANDIDATES) {
-                try {
-                    content = await readFile(join(projectDir, candidate), "utf-8");
-                    filePath = candidate;
-                    break;
-                }
-                catch {
-                }
-            }
-            if (!content) {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: "Error: No docker-compose file found. Searched for: " +
-                                COMPOSE_FILE_CANDIDATES.join(", "),
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-        }
-        if (content.length > 500 * 1024) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: "Error: File too large (max 500KB)",
-                    },
-                ],
-                isError: true,
-            };
-        }
-        try {
-            const data = await apiClient.post("/api/diagrams/generate", {
-                content,
-                sourceFormat,
-                name: params.name,
-            });
-            const summary = {
-                diagram_id: data.id,
-                name: data.name,
-                type: data.diagramType,
-                source_file: filePath,
-                stats: {
-                    nodes: data.nodes?.length || 0,
-                    connections: data.connections?.length || 0,
-                    boundaries: data.boundaries?.length || 0,
-                },
-                nodes: (data.nodes || []).map((n) => ({
-                    name: n.name,
-                    tags: n.tags,
-                })),
-                boundaries: (data.boundaries || []).map((b) => b.label),
-            };
-            const sanitized = sanitizeResponse(summary);
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify(sanitized, null, 2),
-                    },
-                ],
-            };
-        }
-        catch (err) {
-            const message = err instanceof Error ? err.message : "Failed to generate diagram";
-            return {
-                content: [{ type: "text", text: `Error: ${message}` }],
-                isError: true,
-            };
-        }
-    };
-}