feat: add library (#14)

* feat: add library

* fix: fix ci

* fix: fix ci

Author: interstellarmt

README.md

@@ -4,7 +4,7 @@
 
 <img width="768" alt="mcp-server-antv Technical Architecture" src="https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*WHSOR7L8U0YAAAAATjAAAAgAemJ7AQ/fmt.webp" />
 
-Supports **G2**, **G6**, and **F2** libraries for declarative visualization workflows, and **S2**, **X6**, and **L7** on the way～
+Supports **G2**, **G6**, and **F2** libraries for declarative visualization workflows, with **S2**, **X6**, **L7**, and more (including **AVA**, **ADC**, and **G**) coming soon.
 
 ## ✨ Features
 

__tests__/constant.spec.ts

@@ -1,6 +1,5 @@
 import { describe, expect, it } from 'vitest';
 import {
-  getLibraryConfig,
   ANTV_LIBRARY_META,
   CONTEXT7_TOKENS,
 } from '../src/constant';
@@ -18,11 +17,14 @@ describe('context7', () => {
     expect(isValidLibrary('g6')).toBe(true);
     expect(isValidLibrary('x6')).toBe(true);
     expect(isValidLibrary('l7')).toBe(true);
+    expect(isValidLibrary('ava')).toBe(true);
+    expect(isValidLibrary('adc')).toBe(true);
+    expect(isValidLibrary('g')).toBe(true);
     expect(isValidLibrary('unknown')).toBe(false);
   });
 
   it('ANTV_LIBRARY_META', () => {
-    expect(Object.values(ANTV_LIBRARY_META).length).toBe(6);
+    expect(Object.values(ANTV_LIBRARY_META).length).toBe(9);
   });
 
   it('CONTEXT7_TOKENS', () => {

__tests__/tools/extract_antv_topic.json

@@ -1,6 +1,6 @@
 {
   "name": "extract_antv_topic",
-  "description": "AntV Intelligent Assistant Preprocessing Tool - Specifically designed to handle any user queries related to AntV visualization libraries.\n  This tool is the first step in processing AntV technology stack issues, responsible for intelligently identifying, parsing, and structuring user visualization requirements.\n\n**MANDATORY: Must be called for ANY new AntV-related queries, including simple questions. Always precedes query_antv_document tool.**\n\nWhen to use this tool:\n- **AntV-related queries**: Questions about g2/g6/l7/x6/f2/s2 libraries.\n- **Visualization tasks**: Creating charts, graphs, maps, or other visualizations.\n- **Problem solving**: Debugging errors, performance issues, or compatibility problems.\n- **Learning & implementation**: Understanding concepts or requesting code examples.\n\nKey features:\n- **Smart Library Detection**: Scans installed AntV libraries and recommends the best fit based on query and project dependencies.\n- **Topic & Intent Extraction**: Intelligently extracts technical topics and determines user intent (learn/implement/solve).\n- **Task Complexity Handling**: Detects complex tasks and decomposes them into manageable subtasks.\n- **Seamless Integration**: Prepares structured data for the query_antv_document tool to provide precise solutions.",
+  "description": "AntV Intelligent Assistant Preprocessing Tool - Specifically designed to handle any user queries related to AntV visualization libraries.\n  This tool is the first step in processing AntV technology stack issues, responsible for intelligently identifying, parsing, and structuring user visualization requirements.\n\n**MANDATORY: Must be called for ANY new AntV-related queries, including simple questions. Always precedes query_antv_document tool.**\n\nWhen to use this tool:\n- **AntV-related queries**: Questions about g2/g6/l7/x6/f2/s2/g/ava/adc libraries.\n- **Visualization tasks**: Creating charts, graphs, maps, or other visualizations.\n- **Problem solving**: Debugging errors, performance issues, or compatibility problems.\n- **Learning & implementation**: Understanding concepts or requesting code examples.\n\nKey features:\n- **Smart Library Detection**: Scans installed AntV libraries and recommends the best fit based on query and project dependencies.\n- **Topic & Intent Extraction**: Intelligently extracts technical topics and determines user intent (learn/implement/solve).\n- **Task Complexity Handling**: Detects complex tasks and decomposes them into manageable subtasks.\n- **Seamless Integration**: Prepares structured data for the query_antv_document tool to provide precise solutions.",
   "inputSchema": {
     "$schema": "http://json-schema.org/draft-07/schema#",
     "type": "object",
@@ -12,7 +12,7 @@
       },
       "library": {
         "type": "string",
-        "enum": ["g2", "g6", "l7", "x6", "f2", "s2"],
+        "enum": ["g2", "g6", "l7", "x6", "f2", "s2", "g", "ava", "adc"],
         "description": "AntV library name (optional) - If not specified, tool will automatically detect project dependencies and intelligently recommend"
       },
       "maxTopics": {

__tests__/tools/query_antv_document.json

@@ -1,13 +1,13 @@
 {
   "name": "query_antv_document",
-  "description": "AntV Context Retrieval Assistant - Fetches relevant documentation, code examples, and best practices from official AntV resources. Supports g2, g6, l7, x6, f2, s2 libraries, and handles subtasks iterative queries.\n\n**MANDATORY: Must be called for ANY AntV-related query (g2, g6, l7, x6, f2, s2), regardless of task complexity. No exceptions for simple tasks.**\n\nWhen to use this tool:\n- **Implementation & Optimization**: To implement new features, modify styles, refactor code, or optimize performance in AntV solutions.\n- **Debugging & Problem Solving**: For troubleshooting errors, unexpected behaviors, or technical challenges in AntV projects.\n- **Learning & Best Practices**: To explore official documentation, code examples, design patterns, or advanced features.\n- **Complex Task Handling**: For multi-step tasks requiring subtask decomposition (e.g., \"Build a dashboard with interactive charts\").\n- **Simple modifications**: Even basic changes like \"Change the chart's color\" or \"Update legend position\" in AntV context.",
+  "description": "AntV Context Retrieval Assistant - Fetches relevant documentation, code examples, and best practices from official AntV resources. Supports g2, g6, l7, x6, f2, s2, g, ava, adc libraries, and handles subtasks iterative queries.\n\n**MANDATORY: Must be called for ANY AntV-related query (g2, g6, l7, x6, f2, s2, g, ava, adc), regardless of task complexity. No exceptions for simple tasks.**\n\nWhen to use this tool:\n- **Implementation & Optimization**: To implement new features, modify styles, refactor code, or optimize performance in AntV solutions.\n- **Debugging & Problem Solving**: For troubleshooting errors, unexpected behaviors, or technical challenges in AntV projects.\n- **Learning & Best Practices**: To explore official documentation, code examples, design patterns, or advanced features.\n- **Complex Task Handling**: For multi-step tasks requiring subtask decomposition (e.g., \"Build a dashboard with interactive charts\").\n- **Simple modifications**: Even basic changes like \"Change the chart's color\" or \"Update legend position\" in AntV context.",
   "inputSchema": {
     "$schema": "http://json-schema.org/draft-07/schema#",
     "type": "object",
     "properties": {
       "library": {
         "type": "string",
-        "enum": ["g2", "g6", "l7", "x6", "f2", "s2"],
+        "enum": ["g2", "g6", "l7", "x6", "f2", "s2", "g", "ava", "adc"],
         "description": "Specified AntV library type, intelligently identified based on user query"
       },
       "query": {

src/constant.ts

@@ -83,6 +83,29 @@ export const ANTV_LIBRARY_META = {
     keywords: '',
     codeStyle: '',
   },
+  g: {
+    id: 'g' as AntVLibrary,
+    name: 'G',
+    description: 'AntV Grammar of Graphics runtime and specification',
+    keywords: '',
+    codeStyle: '',
+  },
+  ava: {
+    id: 'ava' as AntVLibrary,
+    name: 'AVA',
+    description:
+      'Automated Visual Analytics: chart advisor, insight, narrative',
+    keywords: '',
+    codeStyle: '',
+  },
+  adc: {
+    id: 'adc' as AntVLibrary,
+    name: 'ADC',
+    description:
+      'Ant Design Charts built on G2 for React and modern frameworks',
+    keywords: '',
+    codeStyle: '',
+  },
 };
 
 export const CONTEXT7_TOKENS = {

src/types.ts

@@ -7,7 +7,7 @@
  * Defines the available libraries in the AntV ecosystem.
  * These libraries are used to fetch documentation and provide context for the AntV Assistant tool.
  */
-export type AntVLibrary = 'g2' | 'g6' | 'l7' | 'x6' | 'f2' | 's2';
+export type AntVLibrary = 'g2' | 'g6' | 'l7' | 'x6' | 'f2' | 's2' | 'g' | 'ava' | 'adc';
 
 export type AntVConfig = {
   id: AntVLibrary;

src/utils/context7.ts

@@ -62,6 +62,7 @@ async function fetchContext7Library(url: string): Promise<string | null> {
  * Get the Context7 library ID corresponding to the AntV organization.
  */
 export function getLibraryId(library: AntVLibrary): string {
+  if (library === 'adc') return '/ant-design/ant-design-charts';
   return `/antvis/${library}`;
 }