@kubun/mcp 0.4.1 → 0.4.2

package/lib/tools/cluster.js

@@ -64,7 +64,8 @@ export function createClusterTools(client) {
             ],
             structuredContent: {
                 id: result.id,
-                models: result.models
+                record: result.record,
+                aliases: result.aliases
             }
         };
     });

package/lib/tools/graph.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../../src/tools/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAGhD,OAAO,EAAE,UAAU,EAAe,MAAM,uBAAuB,CAAA;AAG/D,KAAK,cAAc,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,CAAA;AA8BnD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CA+KpF"}
+{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../../src/tools/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAGhD,OAAO,EAAE,UAAU,EAAe,MAAM,uBAAuB,CAAA;AAG/D,KAAK,cAAc,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,CAAA;AA0BnD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAsMpF"}

package/lib/tools/graph.js

@@ -2,11 +2,8 @@ import { createSchema } from '@kubun/graphql';
 import { createTool } from '@mokei/context-server';
 import { printSchema } from 'graphql';
 function getModelsInfo(result) {
-    if (result.models == null) {
-        return {};
-    }
     const aliases = result.aliases ?? {};
-    return Object.entries(result.models).reduce((acc, [id, model])=>{
+    return Object.entries(result.record).reduce((acc, [id, model])=>{
         const alias = aliases[id] ?? model.name;
         acc[alias] = {
             id,
@@ -51,9 +48,6 @@ export function createGraphTools(client) {
             const result = await client.loadGraph({
                 id: ctx.arguments.id
             });
-            if (result.models == null) {
-                throw new Error(`Models not found for graph: ${ctx.arguments.id}`);
-            }
             const info = getModelsInfo(result);
             return {
                 content: [
@@ -76,11 +70,35 @@ export function createGraphTools(client) {
             };
         }
     });
-    const graph_schema = createTool('Get the GraphQL schema for the given graph ID', {
+    const graph_schema = createTool('Get the GraphQL schema for the given graph ID. Supports selective generation to reduce schema size for LLM context.', {
         type: 'object',
         properties: {
             id: {
-                type: 'string'
+                type: 'string',
+                description: 'The graph ID'
+            },
+            onlyModels: {
+                type: 'array',
+                items: {
+                    type: 'string'
+                },
+                description: 'Optional array of model IDs or aliases to include in schema. If not provided, all models are included.'
+            },
+            includeInterfaceDependencies: {
+                type: 'boolean',
+                description: 'Whether to include interface dependencies (default: true)'
+            },
+            includeRelationDependencies: {
+                type: 'boolean',
+                description: 'Whether to include relation dependencies (default: true)'
+            },
+            includeMutations: {
+                type: 'boolean',
+                description: 'Whether to include mutations in the schema (default: true)'
+            },
+            includeSubscriptions: {
+                type: 'boolean',
+                description: 'Whether to include subscriptions in the schema (default: true)'
             }
         },
         required: [
@@ -92,10 +110,15 @@ export function createGraphTools(client) {
             const result = await client.loadGraph({
                 id: ctx.arguments.id
             });
-            if (result.models == null) {
-                throw new Error(`Models not found for graph: ${ctx.arguments.id}`);
-            }
-            const schema = createSchema(result.models, result.aliases);
+            const schema = createSchema({
+                record: result.record,
+                aliases: result.aliases,
+                onlyModels: ctx.arguments.onlyModels,
+                includeInterfaceDependencies: ctx.arguments.includeInterfaceDependencies,
+                includeRelationDependencies: ctx.arguments.includeRelationDependencies,
+                includeMutations: ctx.arguments.includeMutations,
+                includeSubscriptions: ctx.arguments.includeSubscriptions
+            });
             return {
                 content: [
                     {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kubun/mcp",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "description": "MCP server for Kubun",
   "keywords": [],
   "type": "module",
@@ -21,19 +21,19 @@
     "@enkaku/token": "0.12.3",
     "@mokei/context-server": "^0.5.0",
     "graphql": "^16.12.0",
-    "@kubun/db": "^0.4.0",
     "@kubun/db-postgres": "^0.4.0",
     "@kubun/db-sqlite": "^0.4.0",
-    "@kubun/graphql": "^0.4.4",
-    "@kubun/http-client": "^0.4.0",
-    "@kubun/protocol": "^0.4.0",
-    "@kubun/server": "^0.4.2"
+    "@kubun/db": "^0.4.0",
+    "@kubun/server": "^0.4.3",
+    "@kubun/protocol": "^0.4.1",
+    "@kubun/graphql": "^0.4.5",
+    "@kubun/http-client": "^0.4.1"
   },
   "devDependencies": {
     "@enkaku/transport": "0.12.0",
     "@mokei/context-client": "^0.5.0",
     "@mokei/context-protocol": "^0.5.0",
-    "@kubun/client": "^0.4.0"
+    "@kubun/client": "^0.4.1"
   },
   "scripts": {
     "build:clean": "del lib",