chore(ava): optimise ckb folder structure (#844)

* chore(ava): optimise ckb folder structure

* chore(ava): update eslint rule and fix test ts problem

* fix(ava): fix auto ci problem

---------

Co-authored-by: xueren.dt <xueren.dt@antgroup.com>
Co-authored-by: hustcc <i@hust.cc>

Author: 3 people

.eslintrc.js

@@ -79,33 +79,7 @@ module.exports = {
     '@typescript-eslint/no-explicit-any': 0,
     '@typescript-eslint/type-annotation-spacing': 0,
     'no-await-in-loop': 0,
-    'import/order': [
-      2,
-      {
-        groups: ['builtin', 'external', 'internal', 'parent', 'sibling', 'index', 'object', 'type', 'unknown'],
-        pathGroups: [
-          {
-            pattern: 'react',
-            group: 'external',
-            position: 'before',
-          },
-          {
-            pattern: '@/**',
-            group: 'internal',
-            position: 'before',
-          },
-          {
-            pattern: '*.{less,css}',
-            patternOptions: { matchBase: true },
-            group: 'unknown',
-            position: 'after',
-          },
-        ],
-        pathGroupsExcludedImportTypes: ['builtin', 'type'],
-        warnOnUnassignedImports: true,
-        'newlines-between': 'always',
-      },
-    ],
+    'import/order': 0,
     'no-param-reassign': 1,
     'no-continue': 1,
     'no-bitwise': 1,
@@ -123,5 +97,7 @@ module.exports = {
         project: ['./tsconfig.json']
       },
     },
+    // 兼容 exports 子路径导入，避免旧版插件误报为非依赖
+    'import/core-modules': ['@antv/mcp-server-chart/sdk'],
   },
 };

.fatherrc.ts

@@ -5,9 +5,14 @@ export default defineConfig({
   umd: {
     name: 'AVA',
     output: 'dist',
+    externals: {
+      '@antv/mcp-server-chart/sdk': '@antv/mcp-server-chart/sdk',
+    },
   },
   alias: {
-    '@ava': path.resolve(__dirname, 'src'),
-    '@advisor': path.resolve(__dirname, 'src/advisor'),
+    '@antv/mcp-server-chart/sdk': path.resolve(
+      __dirname,
+      'node_modules/@antv/mcp-server-chart/build/sdk.js'
+    ),
   }
 });

__tests__/global.d.ts

@@ -0,0 +1,10 @@
+declare global {
+  namespace jest {
+    interface Matchers<R, _T = {}> {
+      toBeCloseToArray(expected: number[] | number[][]): R;
+      toBeCloseToObject(expected: Record<string, any>): R;
+    }
+  }
+}
+
+export {};

__tests__/unit/utils/statictics.test.ts

@@ -98,7 +98,7 @@ test.skip('statistics', () => {
   expect(valueMap([1, null])).toEqual({ 1: 1, null: 1 });
   expect(median([1, 3, 2, 4, 5])).toBe(3);
   expect(median([1, 2, 3, 4])).toBe(2.5);
-  expect(() => quartile([1])).toThrowError('The length of value cannot be less than 3.');
+  expect(() => quartile([1])).toThrow('The length of value cannot be less than 3.');
   expect(quartile([1, 2, 3, 4, 5, 6, 7, 8, 9])).toEqual([2.5, 5, 7.5]);
   expect(quartile([1, 2, 3, 4, 5, 6, 7, 8, 9], true)).toEqual([2.5, 5, 7.5]);
   expect(quartile([1, 2, 3, 4, 5, 6, 7, 8])).toEqual([2.5, 4.5, 6.5]);

package.json

@@ -41,6 +41,7 @@
     "@antv/algorithm": "^0.1.26",
     "@antv/color-schema": "^0.2.3",
     "@antv/gpt-vis": "^0.5.9",
+    "@antv/mcp-server-chart": "^0.9.3",
     "@antv/smart-color": "^0.2.1",
     "axios": "^1.12.2",
     "bayesian-changepoint": "^1.0.1",

src/advisor/advise-chart-pipeline/plugins/advisePlugin.ts

@@ -10,7 +10,6 @@ import {
 import { AdviseChartPluginEnum } from '../../../constants/pipeline';
 import { DATA_SHAPE } from '../../../extract/constants';
 import { getChartAdvisePrompt, getSpecGeneratePrompt } from '../../chartAdvise/prompt';
-import { CHART_ID_MAP } from '../../../ckb';
 import { Spec } from '../../../bind';
 
 export class AdvisePlugin implements AdvisorPlugin<AdviseChartParams> {
@@ -53,7 +52,7 @@ export class AdvisePlugin implements AdvisorPlugin<AdviseChartParams> {
         const adviseInputs = getChartAdvisePrompt(params);
         const recommendationStr = await requestLLM({ config: llm, prompt: adviseInputs });
         const recommendation = safeJsonParse(recommendationStr, []);
-        const bestCharts = recommendation.map((item) => CHART_ID_MAP[item[0]]) as string[];
+        const bestCharts = recommendation.map((item) => item[0]) as string[];
         const specGenerateInputs = dataShards.map((shard, index) => {
           const { data } = shard;
           return {

src/advisor/chartAdvise/prompt/index.ts

@@ -1,8 +1,7 @@
 import { CKB, VISUAL_CHANNEL_DESCRIPTION } from '../../../ckb/ckb-v2';
 import { CHART_NAME, CHART_PURPOSE_NAME_MAP, FULL_AND_ABBR_CHART_NAME_MAP } from '../../../constants';
-import { ChartConfig, ChartKnowledgeMap, Data, Meta, PlainLikeDataType } from '../../../types';
-import { CHART_ID_LIST } from '../../../ckb';
-import * as CHARTS from '../../../ckb/charts';
+import { ChartConfig, Data, Meta, PlainLikeDataType } from '../../../types';
+import { CHARTS } from '../../../ckb';
 
 const ROLE_CONTEXT = `
 # 角色设定
@@ -181,31 +180,34 @@ export const getGraphAdvisePrompt = (params: {
 };
 
 export const getChartAdvisePrompt = (params: { metas: Meta[]; data: PlainLikeDataType; purpose: string }[]) => {
-  const chartDescriptions = Object.entries(CHARTS as ChartKnowledgeMap)
-    .map(([chartId, item]) => `${chartId}: ${item.description}`)
+  const chartIds = Object.keys(CHARTS);
+  const chartDescriptions = Object.entries(CHARTS)
+    .map(([chartId, item]) => {
+      return `${chartId}: ${item.tool.description}`;
+    })
     .join('\n\n');
   return `
 # Role
-You are a chart recommendation and configuration generation expert, capable of selecting the most suitable chart type from the given Chart Knowledge Base (CKB) based on data and requirements.
+You are a chart recommendation and configuration generation expert, capable of selecting the most suitable chart type from the given Chart Knowledge Base based on data and requirements.
 
 # Objective
-Output the “best chart type (chartId)” with the rationale for selection, and provide 1–2 alternative chart types with reasons why they are not chosen as the primary chart. Only choose types from the provided CKB.
+Output the “best chart type (chartId)” with the rationale for selection, and provide 1–2 alternative chart types with reasons why they are not chosen as the primary chart. Only choose types from the provided Chart Knowledge Base.
 
 # Inputs
 - data: raw data (array) used to draw charts.
 - meta: field metadata (array), each item includes id, name, dataType (number/string/date/geo).
 - purpose: visualization intent (a sentence or several bullet points).
-- CKB: the collection of knowledge definitions for all available charts (including chart names, usage descriptions, etc.).
+- Chart Knowledge Base: the collection of knowledge definitions for all available charts (including chart names, usage descriptions, etc.).
 - Batch input support: The input may be an array containing multiple items, each with data, meta, and purpose. When the input is an array, you must make a recommendation for each item independently and output results in the same order as the input.
 
 # Output
-- chartCodesList: a two-dimensional array. Each item is a string array whose elements are chart “short codes” (see “Chart Types and Codes”), ordered from best to worst match, up to 3 items.
+- chartIdsList: a two-dimensional array. Each item is a string array whose elements are chartId (from the provided CKB list), ordered from best to worst match, up to 3 items.
 - If candidates are fewer than 3, output the actual number; do not exceed 3.
-- When the input is a single item, still output a two-dimensional array (e.g., [["l", "a", "c"]]).
+- When the input is a single item, still output a two-dimensional array (e.g., [["line", "area", "column"]]).
 - Only output a JSON two-dimensional array, without any explanatory text, object keys, or code block markers.
 
 # Strict Constraints (Must Follow)
-- Only select chart types from the provided CKB; do not add or remove types.
+- Only select chart types from the provided Chart Knowledge Base; do not add or remove types.
 - The output must be a strict JSON string; do not include any extra text, explanations, prefixes/suffixes, or code block markers.
 - Determine fitness based on data and meta; avoid subjective guessing; do not fabricate fields or change data.
 - If multiple chart types satisfy the needs, prefer the one that “clearly expresses the core intent with lower cognitive load”.
@@ -235,12 +237,12 @@ Output the “best chart type (chartId)” with the rationale for selection, and
 4. Selection and ranking: sort by evaluation dimensions, determine the primary chart and alternatives, and explain trade-offs (readability, order preservation, intent alignment, data match).
 
 # Response Format (JSON)
-- Only output a JSON two-dimensional array of “short codes”. Each item is a string array (up to 3, ordered from highest to lowest match). Do not output any extra text, e.g., [["l", "a", "c"], ["b", "c"]].
+- Only output a JSON two-dimensional array of chartId. Each item is a string array (up to 3, ordered from highest to lowest match). Do not output any extra text, e.g., [["line", "area", "column"], ["bar", "scatter"]].
 - Output MUST be a plain JSON string; do not use Markdown code fences (e.g., \`\`\`JSON).
 
-# Chart Knowledge Base (CKB)
-## Chart Types and Codes (object array)
-${JSON.stringify(CHART_ID_LIST)}
+# Chart Knowledge Base
+## Chart Types (chartId list)
+${JSON.stringify(chartIds)}
 
 ## Chart Function Descriptions
 ${chartDescriptions}
@@ -253,9 +255,7 @@ ${JSON.stringify(params)}
 export const getSpecGeneratePrompt = (params: { chartId: string; data: PlainLikeDataType }[]) => {
   const inputSchema = params
     .map((item) => {
-      return `ChartId: ${item.chartId}\nInputSchema: ${JSON.stringify(
-        (CHARTS as ChartKnowledgeMap)[item.chartId].inputSchema
-      )}`;
+      return `ChartId: ${item.chartId}\nInputSchema: ${JSON.stringify(CHARTS[item.chartId].tool.inputSchema)}`;
     })
     .join('\n\n');
   return `