@umengfe/mcp-server-chart 0.0.2 → 0.0.3

package/README.md

@@ -68,7 +68,7 @@ To use with `Desktop APP`, such as Claude, VSCode, [Cline](https://cline.bot/mcp
   "mcpServers": {
     "mcp-server-chart": {
       "command": "npx",
-      "args": ["-y", "@antv/mcp-server-chart"]
+      "args": ["-y", "@umengfe/mcp-server-chart"]
     }
   }
 }
@@ -81,7 +81,7 @@ On Window system:
   "mcpServers": {
     "mcp-server-chart": {
       "command": "cmd",
-      "args": ["/c", "npx", "-y", "@antv/mcp-server-chart"]
+      "args": ["/c", "npx", "-y", "@umengfe/mcp-server-chart"]
     }
   }
 }
@@ -96,7 +96,7 @@ Also, you can use it on [aliyun](https://bailian.console.aliyun.com/?tab=mcp#/mc
 Install the package globally.
 
 ```bash
-npm install -g @antv/mcp-server-chart
+npm install -g @umengfe/mcp-server-chart
 ```
 
 Run the server with your preferred transport option:
@@ -167,7 +167,7 @@ Options:
   "mcpServers": {
     "mcp-server-chart": {
       "command": "npx",
-      "args": ["-y", "@antv/mcp-server-chart"],
+      "args": ["-y", "@umengfe/mcp-server-chart"],
       "env": {
         "VIS_REQUEST_SERVER": "<YOUR_VIS_REQUEST_SERVER>"
       }
@@ -203,7 +203,7 @@ Next, you need to add the `SERVICE_ID` environment variable to the MCP server co
   "mcpServers": {
     "AntV Map": {
       "command": "npx",
-      "args": ["-y", "@antv/mcp-server-chart"],
+      "args": ["-y", "@umengfe/mcp-server-chart"],
       "env": {
         "SERVICE_ID": "***********************************"
       }
@@ -225,7 +225,7 @@ You can disable specific chart generation tools using the `DISABLED_TOOLS` envir
   "mcpServers": {
     "mcp-server-chart": {
       "command": "npx",
-      "args": ["-y", "@antv/mcp-server-chart"],
+      "args": ["-y", "@umengfe/mcp-server-chart"],
       "env": {
         "DISABLED_TOOLS": "generate_fishbone_diagram,generate_mind_map"
       }

package/build/charts/heatmap.d.ts

@@ -0,0 +1,47 @@
+import { z } from "zod";
+export declare const heatmap: {
+    schema: {
+        data: z.ZodArray<z.ZodObject<{
+            x: z.ZodUnion<[z.ZodNumber, z.ZodString]>;
+            y: z.ZodUnion<[z.ZodNumber, z.ZodString]>;
+            value: z.ZodNumber;
+        }, "strip", z.ZodTypeAny, {
+            value: number;
+            x: string | number;
+            y: string | number;
+        }, {
+            value: number;
+            x: string | number;
+            y: string | number;
+        }>, "atleastone">;
+        style: z.ZodOptional<z.ZodObject<{
+            backgroundColor: z.ZodOptional<z.ZodString>;
+            palette: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            texture: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "rough"]>>>;
+        }, "strip", z.ZodTypeAny, {
+            texture: "default" | "rough";
+            backgroundColor?: string | undefined;
+            palette?: string[] | undefined;
+        }, {
+            backgroundColor?: string | undefined;
+            palette?: string[] | undefined;
+            texture?: "default" | "rough" | undefined;
+        }>>;
+        theme: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "academy", "dark"]>>>;
+        width: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        height: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        title: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+        axisXTitle: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+        axisYTitle: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    };
+    tool: {
+        name: string;
+        description: string;
+        inputSchema: import("zod-to-json-schema").JsonSchema7Type & {
+            $schema?: string | undefined;
+            definitions?: {
+                [key: string]: import("zod-to-json-schema").JsonSchema7Type;
+            } | undefined;
+        };
+    };
+};

package/build/charts/heatmap.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.heatmap = void 0;
+const zod_1 = require("zod");
+const utils_1 = require("../utils/index.js");
+const base_1 = require("./base.js");
+// Heatmap chart data schema
+const data = zod_1.z.object({
+    x: zod_1.z.union([zod_1.z.number(), zod_1.z.string()]).describe("X coordinate or category"),
+    y: zod_1.z.union([zod_1.z.number(), zod_1.z.string()]).describe("Y coordinate or category"),
+    value: zod_1.z.number().describe("Value for the heatmap cell"),
+});
+// Heatmap chart input schema
+const schema = {
+    data: zod_1.z
+        .array(data)
+        .describe("Data for heatmap chart, such as, [{ x: 1, y: 1, value: 10 }, { x: 1, y: 2, value: 20 }].")
+        .nonempty({ message: "Heatmap chart data cannot be empty." }),
+    style: zod_1.z
+        .object({
+        backgroundColor: base_1.BackgroundColorSchema,
+        palette: base_1.PaletteSchema,
+        texture: base_1.TextureSchema,
+    })
+        .optional()
+        .describe("Style configuration for the chart with a JSON object, optional."),
+    theme: base_1.ThemeSchema,
+    width: base_1.WidthSchema,
+    height: base_1.HeightSchema,
+    title: base_1.TitleSchema,
+    axisXTitle: base_1.AxisXTitleSchema,
+    axisYTitle: base_1.AxisYTitleSchema,
+};
+// Heatmap chart tool descriptor
+const tool = {
+    name: "generate_heatmap_chart",
+    description: "Generate a heatmap chart to visualize data through variations in coloring within a matrix. Suitable for showing patterns, correlations, or intensity in 2-dimensional data.",
+    inputSchema: (0, utils_1.zodToJsonSchema)(schema),
+};
+exports.heatmap = {
+    schema,
+    tool,
+};

package/build/charts/indented-tree.d.ts

@@ -0,0 +1,28 @@
+import { z } from "zod";
+import { type TreeDataType } from "../utils/validator";
+export declare const IndentedTreeNodeSchema: z.ZodType<TreeDataType>;
+export declare const indentedTree: {
+    schema: {
+        data: z.ZodEffects<z.ZodType<TreeDataType, z.ZodTypeDef, TreeDataType>, TreeDataType, TreeDataType>;
+        style: z.ZodOptional<z.ZodObject<{
+            texture: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "rough"]>>>;
+        }, "strip", z.ZodTypeAny, {
+            texture: "default" | "rough";
+        }, {
+            texture?: "default" | "rough" | undefined;
+        }>>;
+        theme: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "academy", "dark"]>>>;
+        width: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        height: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    };
+    tool: {
+        name: string;
+        description: string;
+        inputSchema: import("zod-to-json-schema").JsonSchema7Type & {
+            $schema?: string | undefined;
+            definitions?: {
+                [key: string]: import("zod-to-json-schema").JsonSchema7Type;
+            } | undefined;
+        };
+    };
+};

package/build/charts/indented-tree.js

@@ -0,0 +1,55 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.indentedTree = exports.IndentedTreeNodeSchema = void 0;
+const zod_1 = require("zod");
+const utils_1 = require("../utils/index.js");
+const validator_1 = require("../utils/validator.js");
+const base_1 = require("./base.js");
+// Indented tree node schema
+// The recursive schema is not supported by gemini, and other clients, so we use a non-recursive schema which can represent a tree structure with a fixed depth.
+// Ref: https://github.com/antvis/mcp-server-chart/issues/155
+// Ref: https://github.com/antvis/mcp-server-chart/issues/132
+exports.IndentedTreeNodeSchema = zod_1.z.object({
+    name: zod_1.z.string(),
+    children: zod_1.z
+        .array(zod_1.z.object({
+        name: zod_1.z.string(),
+        children: zod_1.z
+            .array(zod_1.z.object({
+            name: zod_1.z.string(),
+            children: zod_1.z
+                .array(zod_1.z.object({
+                name: zod_1.z.string(),
+            }))
+                .optional(),
+        }))
+            .optional(),
+    }))
+        .optional(),
+});
+// Indented tree chart input schema
+const schema = {
+    data: exports.IndentedTreeNodeSchema.describe("Data for indented tree chart which is a hierarchical structure, such as, { name: 'main topic', children: [{ name: 'topic 1', children: [{ name: 'subtopic 1-1' }] }] }, and the maximum depth is 3.").refine(validator_1.validatedTreeDataSchema, {
+        message: "Invalid parameters: node name is not unique.",
+        path: ["data"],
+    }),
+    style: zod_1.z
+        .object({
+        texture: base_1.TextureSchema,
+    })
+        .optional()
+        .describe("Style configuration for the chart with a JSON object, optional."),
+    theme: base_1.ThemeSchema,
+    width: base_1.WidthSchema,
+    height: base_1.HeightSchema,
+};
+// Indented tree chart tool descriptor
+const tool = {
+    name: "generate_indented_tree",
+    description: "Generate an indented tree chart to display hierarchical data in an indented outline format, suitable for showing organizational structures, file directories, or any tree-like data where nesting levels are represented by indentation.",
+    inputSchema: (0, utils_1.zodToJsonSchema)(schema),
+};
+exports.indentedTree = {
+    schema,
+    tool,
+};

package/build/charts/index.d.ts

@@ -11,7 +11,9 @@ export { fishboneDiagram as "fishbone-diagram" } from "./fishbone-diagram";
 export { flowDiagram as "flow-diagram" } from "./flow-diagram";
 export { funnel } from "./funnel";
 export { histogram } from "./histogram";
+export { indentedTree as "indented-tree" } from "./indented-tree";
 export { line } from "./line";
+export { heatmap } from "./heatmap";
 export { liquid } from "./liquid";
 export { mindMap as "mind-map" } from "./mind-map";
 export { networkGraph as "network-graph" } from "./network-graph";
@@ -22,6 +24,7 @@ export { pinMap as "pin-map" } from "./pin-map";
 export { radar } from "./radar";
 export { sankey } from "./sankey";
 export { scatter } from "./scatter";
+export { table } from "./table";
 export { treemap } from "./treemap";
 export { venn } from "./venn";
 export { violin } from "./violin";

package/build/charts/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports["word-cloud"] = exports.violin = exports.venn = exports.treemap = exports.scatter = exports.sankey = exports.radar = exports["pin-map"] = exports.pie = exports["path-map"] = exports["organization-chart"] = exports["network-graph"] = exports["mind-map"] = exports.liquid = exports.line = exports.histogram = exports.funnel = exports["flow-diagram"] = exports["fishbone-diagram"] = exports["dual-axes"] = exports["district-map"] = exports.column = exports.boxplot = exports.bar = exports.area = void 0;
+exports["word-cloud"] = exports.violin = exports.venn = exports.treemap = exports.table = exports.scatter = exports.sankey = exports.radar = exports["pin-map"] = exports.pie = exports["path-map"] = exports["organization-chart"] = exports["network-graph"] = exports["mind-map"] = exports.liquid = exports.heatmap = exports.line = exports["indented-tree"] = exports.histogram = exports.funnel = exports["flow-diagram"] = exports["fishbone-diagram"] = exports["dual-axes"] = exports["district-map"] = exports.column = exports.boxplot = exports.bar = exports.area = void 0;
 /**
  * export all charts as named exports to match the chart type
  */
@@ -24,8 +24,12 @@ var funnel_1 = require("./funnel.js");
 Object.defineProperty(exports, "funnel", { enumerable: true, get: function () { return funnel_1.funnel; } });
 var histogram_1 = require("./histogram.js");
 Object.defineProperty(exports, "histogram", { enumerable: true, get: function () { return histogram_1.histogram; } });
+var indented_tree_1 = require("./indented-tree.js");
+Object.defineProperty(exports, "indented-tree", { enumerable: true, get: function () { return indented_tree_1.indentedTree; } });
 var line_1 = require("./line.js");
 Object.defineProperty(exports, "line", { enumerable: true, get: function () { return line_1.line; } });
+var heatmap_1 = require("./heatmap.js");
+Object.defineProperty(exports, "heatmap", { enumerable: true, get: function () { return heatmap_1.heatmap; } });
 var liquid_1 = require("./liquid.js");
 Object.defineProperty(exports, "liquid", { enumerable: true, get: function () { return liquid_1.liquid; } });
 var mind_map_1 = require("./mind-map.js");
@@ -46,6 +50,8 @@ var sankey_1 = require("./sankey.js");
 Object.defineProperty(exports, "sankey", { enumerable: true, get: function () { return sankey_1.sankey; } });
 var scatter_1 = require("./scatter.js");
 Object.defineProperty(exports, "scatter", { enumerable: true, get: function () { return scatter_1.scatter; } });
+var table_1 = require("./table.js");
+Object.defineProperty(exports, "table", { enumerable: true, get: function () { return table_1.table; } });
 var treemap_1 = require("./treemap.js");
 Object.defineProperty(exports, "treemap", { enumerable: true, get: function () { return treemap_1.treemap; } });
 var venn_1 = require("./venn.js");

package/build/charts/table.d.ts

@@ -0,0 +1,33 @@
+import { z } from "zod";
+export declare const table: {
+    schema: {
+        data: z.ZodArray<z.ZodRecord<z.ZodString, z.ZodAny>, "atleastone">;
+        style: z.ZodOptional<z.ZodObject<{
+            backgroundColor: z.ZodOptional<z.ZodString>;
+            palette: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            texture: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "rough"]>>>;
+        }, "strip", z.ZodTypeAny, {
+            texture: "default" | "rough";
+            backgroundColor?: string | undefined;
+            palette?: string[] | undefined;
+        }, {
+            backgroundColor?: string | undefined;
+            palette?: string[] | undefined;
+            texture?: "default" | "rough" | undefined;
+        }>>;
+        theme: z.ZodDefault<z.ZodOptional<z.ZodEnum<["default", "academy", "dark"]>>>;
+        width: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        height: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        title: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    };
+    tool: {
+        name: string;
+        description: string;
+        inputSchema: import("zod-to-json-schema").JsonSchema7Type & {
+            $schema?: string | undefined;
+            definitions?: {
+                [key: string]: import("zod-to-json-schema").JsonSchema7Type;
+            } | undefined;
+        };
+    };
+};

package/build/charts/table.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.table = void 0;
+const zod_1 = require("zod");
+const utils_1 = require("../utils/index.js");
+const base_1 = require("./base.js");
+// Table chart data item schema - allows any string keys with any values
+const dataItem = zod_1.z.record(zod_1.z.string(), zod_1.z.any());
+// Table chart input schema
+const schema = {
+    data: zod_1.z
+        .array(dataItem)
+        .describe("Data for table chart, such as, [{ \"Name\": \"胡彦斌\", \"Age\": 32, \"Address\": \"西湖区湖底公园1号\" }, { \"Name\": \"吴彦祖\", \"Age\": 42, \"Address\": \"西湖区湖底公园1号\" }]. Each item in the array represents a row, and the keys of each item represent the column names.")
+        .nonempty({ message: "Table chart data cannot be empty." }),
+    style: zod_1.z
+        .object({
+        backgroundColor: base_1.BackgroundColorSchema,
+        palette: base_1.PaletteSchema,
+        texture: base_1.TextureSchema,
+    })
+        .optional()
+        .describe("Style configuration for the chart with a JSON object, optional."),
+    theme: base_1.ThemeSchema,
+    width: base_1.WidthSchema,
+    height: base_1.HeightSchema,
+    title: base_1.TitleSchema,
+};
+// Table chart tool descriptor
+const tool = {
+    name: "generate_table",
+    description: "Generate a table chart to display structured data in rows and columns, suitable for presenting detailed information and precise values.",
+    inputSchema: (0, utils_1.zodToJsonSchema)(schema),
+};
+exports.table = {
+    schema,
+    tool,
+};

package/build/server.js

@@ -59,7 +59,7 @@ const logger_1 = require("./utils/logger.js");
 function createServer() {
     const server = new index_js_1.Server({
         name: "umeng-mcp-server-chart",
-        version: "0.0.2",
+        version: "0.0.3",
         description: "本工具返回内容为 Markdown 格式，建议客户端按 Markdown 渲染",
     }, {
         capabilities: {

package/build/utils/callTool.d.ts

@@ -12,7 +12,7 @@ export declare function callTool(tool: string, args?: object): Promise<{
     _meta: {
         description: string;
         spec: {
-            type: "area" | "bar" | "boxplot" | "column" | "district-map" | "dual-axes" | "fishbone-diagram" | "flow-diagram" | "funnel" | "histogram" | "line" | "liquid" | "mind-map" | "network-graph" | "organization-chart" | "path-map" | "pie" | "pin-map" | "radar" | "sankey" | "scatter" | "treemap" | "venn" | "violin" | "word-cloud";
+            type: "area" | "bar" | "boxplot" | "column" | "district-map" | "dual-axes" | "fishbone-diagram" | "flow-diagram" | "funnel" | "histogram" | "line" | "liquid" | "mind-map" | "network-graph" | "organization-chart" | "path-map" | "pie" | "pin-map" | "radar" | "sankey" | "scatter" | "treemap" | "venn" | "violin" | "word-cloud" | "table" | "indented-tree" | "heatmap";
         };
     };
 }>;

package/build/utils/callTool.js

@@ -50,31 +50,34 @@ const logger_1 = require("./logger.js");
 const validator_1 = require("./validator.js");
 // Chart type mapping
 const CHART_TYPE_MAP = {
-    generate_area_chart: "area",
-    generate_bar_chart: "bar",
-    generate_boxplot_chart: "boxplot",
-    generate_column_chart: "column",
+    generate_area_chart: "area", //✅
+    generate_bar_chart: "bar", //✅
+    generate_boxplot_chart: "boxplot", //✅
+    generate_column_chart: "column", //✅
     generate_district_map: "district-map",
-    generate_dual_axes_chart: "dual-axes",
-    generate_fishbone_diagram: "fishbone-diagram",
-    generate_flow_diagram: "flow-diagram",
-    generate_funnel_chart: "funnel",
-    generate_histogram_chart: "histogram",
-    generate_line_chart: "line",
-    generate_liquid_chart: "liquid",
-    generate_mind_map: "mind-map",
-    generate_network_graph: "network-graph",
-    generate_organization_chart: "organization-chart",
-    generate_path_map: "path-map",
-    generate_pie_chart: "pie",
-    generate_pin_map: "pin-map",
-    generate_radar_chart: "radar",
-    generate_sankey_chart: "sankey",
-    generate_scatter_chart: "scatter",
-    generate_treemap_chart: "treemap",
-    generate_venn_chart: "venn",
-    generate_violin_chart: "violin",
-    generate_word_cloud_chart: "word-cloud",
+    generate_dual_axes_chart: "dual-axes", //✅
+    generate_fishbone_diagram: "fishbone-diagram", //✅
+    generate_flow_diagram: "flow-diagram", //✅
+    generate_funnel_chart: "funnel", //✅
+    generate_histogram_chart: "histogram", //✅
+    generate_line_chart: "line", //✅
+    generate_liquid_chart: "liquid", //✅
+    generate_mind_map: "mind-map", //✅
+    generate_network_graph: "network-graph", //✅
+    generate_organization_chart: "organization-chart", //✅
+    generate_path_map: "path-map", //✅
+    generate_pie_chart: "pie", //✅
+    generate_pin_map: "pin-map", //✅
+    generate_radar_chart: "radar", //✅
+    generate_sankey_chart: "sankey", //✅
+    generate_scatter_chart: "scatter", //✅
+    generate_treemap_chart: "treemap", //✅
+    generate_venn_chart: "venn", //✅
+    generate_violin_chart: "violin", //✅
+    generate_word_cloud_chart: "word-cloud", //✅
+    generate_table: "table", //✅
+    generate_indented_tree: "indented-tree", //✅
+    generate_heatmap: "heatmap", //✅
 };
 /**
  * Call a tool to generate a chart based on the provided name and arguments.
@@ -106,7 +109,7 @@ function callTool(tool_1) {
                 content: [
                     {
                         type: "text",
-                        text: `'\`\`\`vis-chart\n${JSON.stringify(Object.assign({ type: chartType }, args))}\n\`\`\`'`,
+                        text: `\`\`\`vis-chart\n${JSON.stringify(Object.assign({ type: chartType }, args))}\n\`\`\``,
                     },
                 ],
                 _meta: {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@umengfe/mcp-server-chart",
   "description": "A Model Context Protocol server for generating charts using AntV. This is a TypeScript-based MCP server that provides chart generation capabilities. It allows you to create various types of charts through MCP tools.",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "main": "build/index.js",
   "types": "build/index.d.ts",
   "exports": {