@mentagen/mcp 0.2.0 → 0.4.0

package/dist/tools/create-graph.js

@@ -1,10 +1,14 @@
 import { z } from 'zod';
 import { formatError } from '../utils/errors.js';
-import { generateId } from './create.js';
+import { applyExtension, generateId } from './create.js';
 import { calculateNodeSize } from './size-calc.js';
 const nodeInputSchema = z.object({
     tempId: z.string().describe('Temporary ID for referencing in edges'),
     name: z.string().describe('The node title/name'),
+    extension: z
+        .string()
+        .optional()
+        .describe('File extension to append to name (e.g., ".ts", ".py"). Only applies to code nodes. Defaults to .js.'),
     content: z.string().optional().describe('Node content'),
     type: z
         .enum(['text', 'markdown', 'code', 'url'])
@@ -43,8 +47,9 @@ const DEFAULT_NODE_COLORS = {
 async function createNode(client, boardId, nodeInput, idMap) {
     const realId = generateId();
     const type = nodeInput.type ?? 'text';
+    const finalName = applyExtension(nodeInput.name, type, nodeInput.extension);
     const color = nodeInput.color ?? DEFAULT_NODE_COLORS[type];
-    const autoSize = calculateNodeSize(nodeInput.name, type);
+    const autoSize = calculateNodeSize(finalName, type);
     const x = nodeInput.x ?? 100;
     const y = nodeInput.y ?? 100;
     const width = nodeInput.width ?? autoSize.width;
@@ -57,7 +62,7 @@ async function createNode(client, boardId, nodeInput, idMap) {
             _id: realId,
             boardId,
             node: {
-                name: nodeInput.name,
+                name: finalName,
                 ...(isCodeNode ? { code: nodeContent } : { content: nodeContent }),
                 type,
                 color,

package/dist/tools/create.js

@@ -27,13 +27,32 @@ const DEFAULT_NODE_COLORS = {
     excalidraw: 'teal',
     teleprompter: 'rose',
 };
+const DEFAULT_CODE_EXTENSION = '.js';
+/**
+ * Append extension to name for code nodes only.
+ * Defaults to .js if no extension provided.
+ * Avoids double extensions if name already ends with the extension.
+ */
+export function applyExtension(name, type, extension) {
+    if (type !== 'code')
+        return name;
+    const ext = extension ?? DEFAULT_CODE_EXTENSION;
+    if (name.endsWith(ext))
+        return name;
+    return `${name}${ext}`;
+}
 export const createSchema = z.object({
     boardId: z.string().describe('The board ID to create the node on'),
     name: z.string().describe('The node title/name'),
-    content: z
+    extension: z
+        .string()
+        .optional()
+        .describe('File extension to append to name (e.g., ".ts", ".py"). Only applies to code nodes. Defaults to .js.'),
+    content: z.string().optional().describe('Content for text/markdown nodes.'),
+    code: z
         .string()
         .optional()
-        .describe('Content for text/markdown nodes. Use "url" field for url-type nodes.'),
+        .describe('Code content for code-type nodes. Takes precedence over content.'),
     url: z
         .string()
         .optional()
@@ -77,6 +96,7 @@ export function generateId() {
 }
 /**
  * Build node fields object with content mapped to the correct field based on type.
+ * Expects params.name to already have extension applied.
  */
 function buildNodeFields(params, color, position) {
     const nodeFields = {
@@ -86,7 +106,7 @@ function buildNodeFields(params, color, position) {
         ...position,
     };
     if (params.type === 'code') {
-        nodeFields.code = params.content || '';
+        nodeFields.code = params.code ?? params.content ?? '';
     }
     else if (params.type === 'url') {
         if (!params.url) {
@@ -105,15 +125,16 @@ function buildNodeFields(params, color, position) {
 export async function handleCreate(client, params, mentagenUrl) {
     try {
         const _id = generateId();
+        const finalName = applyExtension(params.name, params.type, params.extension);
         const color = params.color || DEFAULT_NODE_COLORS[params.type];
-        const autoSize = calculateNodeSize(params.name, params.type);
+        const autoSize = calculateNodeSize(finalName, params.type);
         const position = {
             x: snapToGrid(params.x ?? 96),
             y: snapToGrid(params.y ?? 96),
             width: snapToGrid(params.width ?? autoSize.width),
             height: snapToGrid(params.height ?? autoSize.height),
         };
-        const nodeFields = buildNodeFields(params, color, position);
+        const nodeFields = buildNodeFields({ ...params, name: finalName }, color, position);
         const node = await client.addNode({
             _id,
             boardId: params.boardId,

package/dist/tools/index.js

@@ -231,7 +231,11 @@ export function registerTools(server, client, config) {
                         },
                         name: {
                             type: 'string',
-                            description: 'The node title/name (used for auto-sizing). For code nodes, use a filename with extension (e.g., "utils.ts", "app.py") for syntax highlighting.',
+                            description: 'The node title/name (used for auto-sizing). Extension is appended automatically via the extension parameter.',
+                        },
+                        extension: {
+                            type: 'string',
+                            description: 'File extension to append to name (e.g., ".ts", ".py"). Only applies to code nodes. Defaults to .js.',
                         },
                         content: {
                             type: 'string',
@@ -240,7 +244,7 @@ export function registerTools(server, client, config) {
                         type: {
                             type: 'string',
                             enum: ['text', 'markdown', 'code', 'url'],
-                            description: 'Node type: text (default), markdown, code, or url. Code nodes get syntax highlighting based on the file extension in the name.',
+                            description: 'Node type: text (default), markdown, code, or url.',
                         },
                         color: {
                             type: 'string',
@@ -478,7 +482,11 @@ export function registerTools(server, client, config) {
                                     },
                                     name: {
                                         type: 'string',
-                                        description: 'Node title/name (used for auto-sizing). For code nodes, use a filename with extension for syntax highlighting.',
+                                        description: 'Node title/name (used for auto-sizing). Extension is appended automatically via the extension parameter.',
+                                    },
+                                    extension: {
+                                        type: 'string',
+                                        description: 'File extension to append to name (e.g., ".ts", ".py"). Only applies to code nodes. Defaults to .js.',
                                     },
                                     content: {
                                         type: 'string',
@@ -487,7 +495,7 @@ export function registerTools(server, client, config) {
                                     type: {
                                         type: 'string',
                                         enum: ['text', 'markdown', 'code', 'url'],
-                                        description: 'Node type (default: text). Code nodes get syntax highlighting based on file extension in name.',
+                                        description: 'Node type (default: text).',
                                     },
                                     color: { type: 'string', description: 'Node color' },
                                     x: { type: 'number', description: 'X position' },

package/dist/tools/update.js

@@ -7,10 +7,11 @@ export const updateSchema = z.object({
     boardId: z.string().describe('The board ID containing the node'),
     nodeId: z.string().describe('The node ID to update'),
     name: z.string().optional().describe('New name for the node'),
-    content: z
+    content: z.string().optional().describe('Content for text/markdown nodes.'),
+    code: z
         .string()
         .optional()
-        .describe('Content for text/markdown nodes. Use "url" field for url-type nodes.'),
+        .describe('Code content for code-type nodes. Takes precedence over content.'),
     url: z
         .string()
         .optional()
@@ -46,8 +47,11 @@ const POSITION_FIELDS = ['x', 'y', 'width', 'height'];
  * Add content fields to update data based on node type.
  */
 function addContentFields(data, params, nodeType) {
-    if (nodeType === NodeType.Code && params.content !== undefined) {
-        data.code = params.content;
+    if (nodeType === NodeType.Code) {
+        const codeContent = params.code ?? params.content;
+        if (codeContent !== undefined) {
+            data.code = codeContent;
+        }
     }
     else if (nodeType === NodeType.URL) {
         if (params.url !== undefined) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mentagen/mcp",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "MCP server for Mentagen knowledge base integration with Cursor",
   "type": "module",
   "bin": "dist/index.js",