agent-enderun 0.5.0 → 0.5.1

package/packages/framework-mcp/dist/schemas.js

@@ -29,6 +29,8 @@ export const SEND_AGENT_MESSAGE_ARGS_SCHEMA = z.object({
     to: z.string().min(1),
     message: z.string().min(1),
     traceId: z.string().min(1),
+    category: z.enum(["ACTION", "DELEGATION", "INFO", "ALERT"]).default("INFO"),
+    priority: z.enum(["LOW", "MEDIUM", "HIGH", "URGENT"]).default("MEDIUM"),
 });
 export const SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA = z.object({
     query: z.string().min(1),
@@ -82,3 +84,15 @@ export const GET_AGENT_AUDIT_REPORT_ARGS_SCHEMA = z.object({
     agent: z.string().min(1),
     days: z.number().default(7),
 });
+export const GET_AGENT_INBOX_STATS_ARGS_SCHEMA = z.object({
+    agent: z.string().min(1),
+});
+export const GET_KNOWLEDGE_GRAPH_ARGS_SCHEMA = z.object({
+    tag: z.string().optional(),
+});
+export const SYNC_CONTRACT_HASH_ARGS_SCHEMA = z.object({
+    force: z.boolean().default(false),
+});
+export const VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA = z.object({
+    detailed: z.boolean().default(false),
+});

package/packages/framework-mcp/dist/tools/framework.js

@@ -1,6 +1,8 @@
 import fs from "fs";
 import path from "path";
+import { execSync } from "child_process";
 import { getFrameworkDir, collectMarkdownArtifacts, FRAMEWORK_VERSION } from "../utils.js";
+import { VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA } from "../schemas.js";
 export const frameworkTools = [
     {
         name: "get_framework_status",
@@ -37,6 +39,16 @@ export const frameworkTools = [
         description: "Analyzes a pre-existing codebase to automatically generate the initial project memory, learning its tech stack and architecture.",
         inputSchema: { type: "object", properties: {} },
     },
+    {
+        name: "verify_framework_health",
+        description: "Runs the framework's internal 'check' command and reports system health status via MCP.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                detailed: { type: "boolean", default: false, description: "Include detailed issues if any." },
+            },
+        },
+    },
 ];
 export const frameworkHandlers = {
     bootstrap_legacy_memory: async (args, projectRoot) => {
@@ -158,4 +170,17 @@ export const frameworkHandlers = {
     get_system_time: async () => {
         return { content: [{ type: "text", text: new Date().toISOString() }] };
     },
+    verify_framework_health: async (args, projectRoot) => {
+        const parsed = VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA.safeParse(args ?? {});
+        try {
+            // Run the internal CLI check command via tsx to ensure we test current state
+            // If the CLI is not yet linked, we might need a relative path
+            const cliPath = path.join(projectRoot, "bin/cli.js");
+            const output = execSync(`node ${cliPath} check`, { cwd: projectRoot, encoding: "utf-8", stdio: "pipe" });
+            return { content: [{ type: "text", text: `### FRAMEWORK HEALTH CHECK\n\n${output}` }] };
+        }
+        catch (error) {
+            return { content: [{ type: "text", text: `### FRAMEWORK HEALTH CHECK (FAILED)\n\n${error.stdout || error.message}` }] };
+        }
+    }
 };

package/packages/framework-mcp/dist/tools/knowledge.js

@@ -5,28 +5,52 @@ import { SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA, UPDATE_KNOWLEDGE_BASE_ARGS_SCHEMA }
 export const knowledgeTools = [
     {
         name: "search_knowledge_base",
-        description: "Searches the Academy's internal knowledge base for architectural patterns, troubleshooting guides, and FAQs.",
+        description: "Searches the Academy's Obsidian-style knowledge base using keywords or tags.",
         inputSchema: {
             type: "object",
             properties: {
-                query: { type: "string", description: "Search query or topic" },
+                query: { type: "string", description: "Search query, topic, or tag (e.g. #security)" },
             },
             required: ["query"],
         },
     },
     {
         name: "update_knowledge_base",
-        description: "Adds or updates an entry in the Academy's internal knowledge base.",
+        description: "Adds or updates an entry with mandatory YAML frontmatter for Obsidian compatibility.",
         inputSchema: {
             type: "object",
             properties: {
-                topic: { type: "string", description: "The topic or title of the entry" },
-                content: { type: "string", description: "The technical content or guide" },
+                topic: { type: "string", description: "The topic or title" },
+                content: { type: "string", description: "Content (YAML frontmatter will be auto-generated if missing)" },
+                tags: { type: "array", items: { type: "string" }, description: "Tags for the entry" },
             },
             required: ["topic", "content"],
         },
     },
+    {
+        name: "get_knowledge_graph",
+        description: "Generates a Mermaid diagram showing relationships between knowledge entries based on 'related' metadata.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                tag: { type: "string", description: "Filter by tag" },
+            },
+        },
+    },
 ];
+function parseFrontmatter(content) {
+    const match = content.match(/^---\r?\n([\s\S]*?)\r?\n---/);
+    if (!match)
+        return { metadata: {}, body: content };
+    const yaml = match[1];
+    const metadata = {};
+    yaml.split("\n").forEach(line => {
+        const [key, ...val] = line.split(":");
+        if (key && val.length > 0)
+            metadata[key.trim()] = val.join(":").trim();
+    });
+    return { metadata, body: content.replace(match[0], "").trim() };
+}
 export const knowledgeHandlers = {
     search_knowledge_base: async (args, projectRoot) => {
         const parsed = SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA.safeParse(args ?? {});
@@ -37,14 +61,20 @@ export const knowledgeHandlers = {
             const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
             if (!fs.existsSync(kbDir))
                 return { content: [{ type: "text", text: "Knowledge base is empty." }] };
+            const query = parsed.data.query.toLowerCase();
             const results = fs.readdirSync(kbDir).filter(f => f.endsWith(".md")).map(file => {
                 const content = fs.readFileSync(path.join(kbDir, file), "utf-8");
-                if (content.toLowerCase().includes(parsed.data.query.toLowerCase()) || file.toLowerCase().includes(parsed.data.query.toLowerCase())) {
-                    return `### ${file.replace(".md", "")}\n\n${content.slice(0, 300)}...`;
+                const { metadata, body } = parseFrontmatter(content);
+                const matchesQuery = body.toLowerCase().includes(query) ||
+                    file.toLowerCase().includes(query) ||
+                    (metadata.tags && metadata.tags.toLowerCase().includes(query)) ||
+                    (metadata.title && metadata.title.toLowerCase().includes(query));
+                if (matchesQuery) {
+                    return `### ${metadata.title || file.replace(".md", "")}\n**Tags:** ${metadata.tags || "None"}\n\n${body.slice(0, 300)}...`;
                 }
                 return null;
             }).filter(Boolean);
-            return { content: [{ type: "text", text: results.length > 0 ? `### KNOWLEDGE BASE SEARCH RESULTS\n\n${results.join("\n\n---\n\n")}` : "No matching knowledge base entries found." }] };
+            return { content: [{ type: "text", text: results.length > 0 ? `### KNOWLEDGE BASE SEARCH RESULTS\n\n${results.join("\n\n---\n\n")}` : "No matching knowledge entries found." }] };
         }
         catch (error) {
             return { content: [{ type: "text", text: "Knowledge base search failed." }] };
@@ -53,17 +83,51 @@ export const knowledgeHandlers = {
     update_knowledge_base: async (args, projectRoot) => {
         const parsed = UPDATE_KNOWLEDGE_BASE_ARGS_SCHEMA.safeParse(args ?? {});
         if (!parsed.success)
-            return { content: [{ type: "text", text: "Invalid topic or content." }] };
+            return { content: [{ type: "text", text: "Invalid arguments." }] };
         try {
             const frameworkDir = getFrameworkDir(projectRoot);
             const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
             if (!fs.existsSync(kbDir))
                 fs.mkdirSync(kbDir, { recursive: true });
-            fs.writeFileSync(path.join(kbDir, parsed.data.topic.replace(/[^a-z0-9]/gi, "_").toLowerCase() + ".md"), parsed.data.content);
-            return { content: [{ type: "text", text: `Knowledge base updated: ${parsed.data.topic}` }] };
+            const fileName = parsed.data.topic.replace(/[^a-z0-9]/gi, "_").toLowerCase() + ".md";
+            const tags = parsed.data.tags || [];
+            let finalContent = parsed.data.content;
+            if (!finalContent.startsWith("---")) {
+                const frontmatter = `---\ntitle: ${parsed.data.topic}\ntags: [${tags.join(", ")}]\nlast_updated: ${new Date().toISOString()}\n---\n\n`;
+                finalContent = frontmatter + finalContent;
+            }
+            fs.writeFileSync(path.join(kbDir, fileName), finalContent);
+            return { content: [{ type: "text", text: `Obsidian Wiki updated: ${parsed.data.topic}` }] };
         }
         catch (error) {
             return { content: [{ type: "text", text: "Failed to update knowledge base." }] };
         }
     },
+    get_knowledge_graph: async (args, projectRoot) => {
+        try {
+            const frameworkDir = getFrameworkDir(projectRoot);
+            const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
+            if (!fs.existsSync(kbDir))
+                return { content: [{ type: "text", text: "Knowledge base empty." }] };
+            const files = fs.readdirSync(kbDir).filter(f => f.endsWith(".md"));
+            let mermaid = "graph TD\n";
+            files.forEach(file => {
+                const content = fs.readFileSync(path.join(kbDir, file), "utf-8");
+                const { metadata } = parseFrontmatter(content);
+                const id = file.replace(".md", "");
+                const label = metadata.title || id;
+                mermaid += `  ${id}["${label}"]\n`;
+                if (metadata.related) {
+                    const related = metadata.related.replace(/[\[\]]/g, "").split(",");
+                    related.forEach((r) => {
+                        mermaid += `  ${id} --> ${r.trim().replace(".md", "")}\n`;
+                    });
+                }
+            });
+            return { content: [{ type: "text", text: `### KNOWLEDGE GRAPH (Mermaid)\n\n\`\`\`mermaid\n${mermaid}\n\`\`\`` }] };
+        }
+        catch (error) {
+            return { content: [{ type: "text", text: "Failed to generate knowledge graph." }] };
+        }
+    }
 };

package/packages/framework-mcp/dist/tools/messages.js

@@ -1,28 +1,42 @@
 import fs from "fs";
 import path from "path";
 import { getFrameworkDir } from "../utils.js";
-import { SEND_AGENT_MESSAGE_ARGS_SCHEMA, READ_AGENT_MESSAGES_ARGS_SCHEMA } from "../schemas.js";
+import { SEND_AGENT_MESSAGE_ARGS_SCHEMA, READ_AGENT_MESSAGES_ARGS_SCHEMA, GET_AGENT_INBOX_STATS_ARGS_SCHEMA } from "../schemas.js";
 export const messageTools = [
     {
         name: "send_agent_message",
-        description: "Sends a message to another specialized agent. Useful for collaboration and delegation.",
+        description: "Sends a message to another specialized agent following the Hermes Protocol.",
         inputSchema: {
             type: "object",
             properties: {
                 to: { type: "string", description: "Recipient agent name" },
                 message: { type: "string", description: "The message content" },
                 traceId: { type: "string", description: "The active Trace ID" },
+                category: { type: "string", enum: ["ACTION", "DELEGATION", "INFO", "ALERT"], default: "INFO" },
+                priority: { type: "string", enum: ["LOW", "MEDIUM", "HIGH", "URGENT"], default: "MEDIUM" },
             },
             required: ["to", "message", "traceId"],
         },
     },
     {
         name: "read_agent_messages",
-        description: "Reads messages sent to the current agent.",
+        description: "Reads messages sent to the current agent. Filters by Trace ID if provided.",
         inputSchema: {
             type: "object",
             properties: {
                 agent: { type: "string", description: "Current agent name reading their messages" },
+                traceId: { type: "string", description: "Optional Trace ID to filter messages" },
+            },
+            required: ["agent"],
+        },
+    },
+    {
+        name: "get_agent_inbox_stats",
+        description: "Returns statistics about the agent's inbox (total, unread, priority distribution).",
+        inputSchema: {
+            type: "object",
+            properties: {
+                agent: { type: "string", description: "Agent name" },
             },
             required: ["agent"],
         },
@@ -41,12 +55,20 @@ export const messageHandlers = {
                 fs.mkdirSync(messagesDir, { recursive: true });
             const messagePath = path.join(messagesDir, `${recipient}.json`);
             const messages = (fs.existsSync(messagePath) ? JSON.parse(fs.readFileSync(messagePath, "utf-8")) : []);
-            messages.push({ timestamp: new Date().toISOString(), from: "manager", traceId: parsed.data.traceId, content: parsed.data.message, read: false });
+            messages.push({
+                timestamp: new Date().toISOString(),
+                from: "manager",
+                traceId: parsed.data.traceId,
+                category: parsed.data.category,
+                priority: parsed.data.priority,
+                content: parsed.data.message,
+                read: false
+            });
             fs.writeFileSync(messagePath, JSON.stringify(messages, null, 2));
-            return { content: [{ type: "text", text: `Message sent to @${recipient}.` }] };
+            return { content: [{ type: "text", text: `Hermes: Message sent to @${recipient} [${parsed.data.category} | ${parsed.data.priority}].` }] };
         }
         catch (error) {
-            return { content: [{ type: "text", text: "Failed to send message." }] };
+            return { content: [{ type: "text", text: "Failed to send message via Hermes." }] };
         }
     },
     read_agent_messages: async (args, projectRoot) => {
@@ -61,11 +83,42 @@ export const messageHandlers = {
                 return { content: [{ type: "text", text: "No messages found." }] };
             const messages = JSON.parse(fs.readFileSync(messagePath, "utf-8"));
             const unread = messages.filter((m) => !m.read);
+            // Mark as read
             fs.writeFileSync(messagePath, JSON.stringify(messages.map((m) => ({ ...m, read: true })), null, 2));
-            return { content: [{ type: "text", text: unread.length === 0 ? "No new messages." : `### INBOX: @${agentName}\n\n` + unread.map((m) => `- **From:** ${m.from}\n  **Message:** ${m.content}`).join("\n\n") }] };
+            if (unread.length === 0)
+                return { content: [{ type: "text", text: "No new messages." }] };
+            const formatted = unread.map((m) => `- [${m.priority}] **${m.from}** (${m.category}): ${m.content} *(Trace: ${m.traceId})*`).join("\n");
+            return { content: [{ type: "text", text: `### HERMES INBOX: @${agentName}\n\n${formatted}` }] };
         }
         catch (error) {
             return { content: [{ type: "text", text: "Failed to read messages." }] };
         }
     },
+    get_agent_inbox_stats: async (args, projectRoot) => {
+        const parsed = GET_AGENT_INBOX_STATS_ARGS_SCHEMA.safeParse(args ?? {});
+        if (!parsed.success)
+            return { content: [{ type: "text", text: "Invalid agent name." }] };
+        try {
+            const frameworkDir = getFrameworkDir(projectRoot);
+            const agentName = parsed.data.agent.replace(/^@/, "");
+            const messagePath = path.join(projectRoot, frameworkDir, "messages", `${agentName}.json`);
+            if (!fs.existsSync(messagePath))
+                return { content: [{ type: "text", text: "Inbox empty." }] };
+            const messages = JSON.parse(fs.readFileSync(messagePath, "utf-8"));
+            const unread = messages.filter(m => !m.read);
+            const priorityDist = messages.reduce((acc, m) => {
+                acc[m.priority] = (acc[m.priority] || 0) + 1;
+                return acc;
+            }, {});
+            return {
+                content: [{
+                        type: "text",
+                        text: `### INBOX STATS: @${agentName}\n- Total: ${messages.length}\n- Unread: ${unread.length}\n- Priority Distribution: ${JSON.stringify(priorityDist)}`
+                    }]
+            };
+        }
+        catch (error) {
+            return { content: [{ type: "text", text: "Failed to get inbox stats." }] };
+        }
+    }
 };

package/packages/framework-mcp/dist/utils.js

@@ -1,6 +1,6 @@
 import path from "path";
 import fs from "fs";
-export const FRAMEWORK_VERSION = "0.5.0";
+export const FRAMEWORK_VERSION = "0.5.1";
 export function getFrameworkDir(projectRoot) {
     const adapters = [".gemini", ".claude", ".cursor", ".codex", ".enderun"];
     for (const adp of adapters) {

package/packages/framework-mcp/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-enderun-mcp",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "Enterprise-grade MCP Server for AI Agent Framework",
   "author": "Yusuf BEKAR",
   "license": "MIT",

package/packages/framework-mcp/src/schemas.ts

@@ -35,6 +35,8 @@ export const SEND_AGENT_MESSAGE_ARGS_SCHEMA = z.object({
     to: z.string().min(1),
     message: z.string().min(1),
     traceId: z.string().min(1),
+    category: z.enum(["ACTION", "DELEGATION", "INFO", "ALERT"]).default("INFO"),
+    priority: z.enum(["LOW", "MEDIUM", "HIGH", "URGENT"]).default("MEDIUM"),
 });
 
 export const SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA = z.object({
@@ -104,3 +106,19 @@ export const GET_AGENT_AUDIT_REPORT_ARGS_SCHEMA = z.object({
     agent: z.string().min(1),
     days: z.number().default(7),
 });
+
+export const GET_AGENT_INBOX_STATS_ARGS_SCHEMA = z.object({
+    agent: z.string().min(1),
+});
+
+export const GET_KNOWLEDGE_GRAPH_ARGS_SCHEMA = z.object({
+    tag: z.string().optional(),
+});
+
+export const SYNC_CONTRACT_HASH_ARGS_SCHEMA = z.object({
+    force: z.boolean().default(false),
+});
+
+export const VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA = z.object({
+    detailed: z.boolean().default(false),
+});

package/packages/framework-mcp/src/tools/framework.ts

@@ -1,10 +1,12 @@
 import fs from "fs";
 import path from "path";
+import { execSync } from "child_process";
 import { 
     getFrameworkDir, 
     collectMarkdownArtifacts, 
     FRAMEWORK_VERSION 
 } from "../utils.js";
+import { VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA } from "../schemas.js";
 
 export const frameworkTools = [
     {
@@ -42,6 +44,16 @@ export const frameworkTools = [
         description: "Analyzes a pre-existing codebase to automatically generate the initial project memory, learning its tech stack and architecture.",
         inputSchema: { type: "object", properties: {} },
     },
+    {
+        name: "verify_framework_health",
+        description: "Runs the framework's internal 'check' command and reports system health status via MCP.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                detailed: { type: "boolean", default: false, description: "Include detailed issues if any." },
+            },
+        },
+    },
 ];
 
 export const frameworkHandlers = {
@@ -154,4 +166,16 @@ export const frameworkHandlers = {
     get_system_time: async () => {
         return { content: [{ type: "text", text: new Date().toISOString() }] };
     },
+    verify_framework_health: async (args: unknown, projectRoot: string) => {
+        const parsed = VERIFY_FRAMEWORK_HEALTH_ARGS_SCHEMA.safeParse(args ?? {});
+        try {
+            // Run the internal CLI check command via tsx to ensure we test current state
+            // If the CLI is not yet linked, we might need a relative path
+            const cliPath = path.join(projectRoot, "bin/cli.js");
+            const output = execSync(`node ${cliPath} check`, { cwd: projectRoot, encoding: "utf-8", stdio: "pipe" });
+            return { content: [{ type: "text", text: `### FRAMEWORK HEALTH CHECK\n\n${output}` }] };
+        } catch (error: any) {
+            return { content: [{ type: "text", text: `### FRAMEWORK HEALTH CHECK (FAILED)\n\n${error.stdout || error.message}` }] };
+        }
+    }
 };

package/packages/framework-mcp/src/tools/knowledge.ts

@@ -3,35 +3,59 @@ import path from "path";
 import { getFrameworkDir } from "../utils.js";
 import { 
     SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA, 
-    UPDATE_KNOWLEDGE_BASE_ARGS_SCHEMA 
+    UPDATE_KNOWLEDGE_BASE_ARGS_SCHEMA,
+    GET_KNOWLEDGE_GRAPH_ARGS_SCHEMA
 } from "../schemas.js";
 
 export const knowledgeTools = [
     {
         name: "search_knowledge_base",
-        description: "Searches the Academy's internal knowledge base for architectural patterns, troubleshooting guides, and FAQs.",
+        description: "Searches the Academy's Obsidian-style knowledge base using keywords or tags.",
         inputSchema: {
             type: "object",
             properties: {
-                query: { type: "string", description: "Search query or topic" },
+                query: { type: "string", description: "Search query, topic, or tag (e.g. #security)" },
             },
             required: ["query"],
         },
     },
     {
         name: "update_knowledge_base",
-        description: "Adds or updates an entry in the Academy's internal knowledge base.",
+        description: "Adds or updates an entry with mandatory YAML frontmatter for Obsidian compatibility.",
         inputSchema: {
             type: "object",
             properties: {
-                topic: { type: "string", description: "The topic or title of the entry" },
-                content: { type: "string", description: "The technical content or guide" },
+                topic: { type: "string", description: "The topic or title" },
+                content: { type: "string", description: "Content (YAML frontmatter will be auto-generated if missing)" },
+                tags: { type: "array", items: { type: "string" }, description: "Tags for the entry" },
             },
             required: ["topic", "content"],
         },
     },
+    {
+        name: "get_knowledge_graph",
+        description: "Generates a Mermaid diagram showing relationships between knowledge entries based on 'related' metadata.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                tag: { type: "string", description: "Filter by tag" },
+            },
+        },
+    },
 ];
 
+function parseFrontmatter(content: string) {
+    const match = content.match(/^---\r?\n([\s\S]*?)\r?\n---/);
+    if (!match) return { metadata: {}, body: content };
+    const yaml = match[1];
+    const metadata: Record<string, any> = {};
+    yaml.split("\n").forEach(line => {
+        const [key, ...val] = line.split(":");
+        if (key && val.length > 0) metadata[key.trim()] = val.join(":").trim();
+    });
+    return { metadata, body: content.replace(match[0], "").trim() };
+}
+
 export const knowledgeHandlers = {
     search_knowledge_base: async (args: unknown, projectRoot: string) => {
         const parsed = SEARCH_KNOWLEDGE_BASE_ARGS_SCHEMA.safeParse(args ?? {});
@@ -40,29 +64,79 @@ export const knowledgeHandlers = {
             const frameworkDir = getFrameworkDir(projectRoot);
             const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
             if (!fs.existsSync(kbDir)) return { content: [{ type: "text", text: "Knowledge base is empty." }] };
+            
+            const query = parsed.data.query.toLowerCase();
             const results = fs.readdirSync(kbDir).filter(f => f.endsWith(".md")).map(file => {
                 const content = fs.readFileSync(path.join(kbDir, file), "utf-8");
-                if (content.toLowerCase().includes(parsed.data.query.toLowerCase()) || file.toLowerCase().includes(parsed.data.query.toLowerCase())) {
-                    return `### ${file.replace(".md", "")}\n\n${content.slice(0, 300)}...`;
+                const { metadata, body } = parseFrontmatter(content);
+                
+                const matchesQuery = body.toLowerCase().includes(query) || 
+                                   file.toLowerCase().includes(query) ||
+                                   (metadata.tags && metadata.tags.toLowerCase().includes(query)) ||
+                                   (metadata.title && metadata.title.toLowerCase().includes(query));
+
+                if (matchesQuery) {
+                    return `### ${metadata.title || file.replace(".md", "")}\n**Tags:** ${metadata.tags || "None"}\n\n${body.slice(0, 300)}...`;
                 }
                 return null;
             }).filter(Boolean);
-            return { content: [{ type: "text", text: results.length > 0 ? `### KNOWLEDGE BASE SEARCH RESULTS\n\n${results.join("\n\n---\n\n")}` : "No matching knowledge base entries found." }] };
+
+            return { content: [{ type: "text", text: results.length > 0 ? `### KNOWLEDGE BASE SEARCH RESULTS\n\n${results.join("\n\n---\n\n")}` : "No matching knowledge entries found." }] };
         } catch (error) {
             return { content: [{ type: "text", text: "Knowledge base search failed." }] };
         }
     },
     update_knowledge_base: async (args: unknown, projectRoot: string) => {
         const parsed = UPDATE_KNOWLEDGE_BASE_ARGS_SCHEMA.safeParse(args ?? {});
-        if (!parsed.success) return { content: [{ type: "text", text: "Invalid topic or content." }] };
+        if (!parsed.success) return { content: [{ type: "text", text: "Invalid arguments." }] };
         try {
             const frameworkDir = getFrameworkDir(projectRoot);
             const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
             if (!fs.existsSync(kbDir)) fs.mkdirSync(kbDir, { recursive: true });
-            fs.writeFileSync(path.join(kbDir, parsed.data.topic.replace(/[^a-z0-9]/gi, "_").toLowerCase() + ".md"), parsed.data.content);
-            return { content: [{ type: "text", text: `Knowledge base updated: ${parsed.data.topic}` }] };
+            
+            const fileName = parsed.data.topic.replace(/[^a-z0-9]/gi, "_").toLowerCase() + ".md";
+            const tags = (parsed as any).data.tags || [];
+            
+            let finalContent = parsed.data.content;
+            if (!finalContent.startsWith("---")) {
+                const frontmatter = `---\ntitle: ${parsed.data.topic}\ntags: [${tags.join(", ")}]\nlast_updated: ${new Date().toISOString()}\n---\n\n`;
+                finalContent = frontmatter + finalContent;
+            }
+
+            fs.writeFileSync(path.join(kbDir, fileName), finalContent);
+            return { content: [{ type: "text", text: `Obsidian Wiki updated: ${parsed.data.topic}` }] };
         } catch (error) {
             return { content: [{ type: "text", text: "Failed to update knowledge base." }] };
         }
     },
+    get_knowledge_graph: async (args: unknown, projectRoot: string) => {
+        try {
+            const frameworkDir = getFrameworkDir(projectRoot);
+            const kbDir = path.join(projectRoot, frameworkDir, "knowledge");
+            if (!fs.existsSync(kbDir)) return { content: [{ type: "text", text: "Knowledge base empty." }] };
+            
+            const files = fs.readdirSync(kbDir).filter(f => f.endsWith(".md"));
+            let mermaid = "graph TD\n";
+            
+            files.forEach(file => {
+                const content = fs.readFileSync(path.join(kbDir, file), "utf-8");
+                const { metadata } = parseFrontmatter(content);
+                const id = file.replace(".md", "");
+                const label = metadata.title || id;
+                
+                mermaid += `  ${id}["${label}"]\n`;
+                
+                if (metadata.related) {
+                    const related = metadata.related.replace(/[\[\]]/g, "").split(",");
+                    related.forEach((r: string) => {
+                        mermaid += `  ${id} --> ${r.trim().replace(".md", "")}\n`;
+                    });
+                }
+            });
+
+            return { content: [{ type: "text", text: `### KNOWLEDGE GRAPH (Mermaid)\n\n\`\`\`mermaid\n${mermaid}\n\`\`\`` }] };
+        } catch (error) {
+            return { content: [{ type: "text", text: "Failed to generate knowledge graph." }] };
+        }
+    }
 };