proxysql

Commit Graph

Author	SHA1	Message	Date
Rene Cannao	3f44229e28	feat: Add MCP AI Tool Handler for NL2SQL with test script Phase 5: MCP Tool Implementation for NL2SQL This commit implements the AI Tool Handler for the MCP (Model Context Protocol) server, exposing NL2SQL functionality as an MCP tool. New Files: - include/AI_Tool_Handler.h: Header for AI_Tool_Handler class - Provides ai_nl2sql_convert tool via MCP protocol - Wraps NL2SQL_Converter and Anomaly_Detector - Inherits from MCP_Tool_Handler base class - lib/AI_Tool_Handler.cpp: Implementation - Implements ai_nl2sql_convert tool execution - Accepts parameters: natural_language (required), schema, context_tables, max_latency_ms, allow_cache - Returns JSON response with sql_query, confidence, explanation, cached, cache_id - scripts/mcp/test_nl2sql_tools.sh: Test script for NL2SQL MCP tool - Tests ai_nl2sql_convert via JSON-RPC over HTTPS - 10 test cases covering SELECT, WHERE, JOIN, aggregation, etc. - Includes error handling test for empty queries - Supports --verbose, --quiet options Modified Files: - include/MCP_Thread.h: Add AI_Tool_Handler forward declaration and pointer - lib/Makefile: Add AI_Tool_Handler.oo to _OBJ_CXX list - lib/ProxySQL_MCP_Server.cpp: Initialize and register AI tool handler - Creates AI_Tool_Handler with GloAI components - Registers /mcp/ai endpoint - Adds cleanup in destructor MCP Tool Details: - Endpoint: /mcp/ai - Tool: ai_nl2sql_convert - Parameters: - natural_language (string, required): Natural language query - schema (string, optional): Database schema name - context_tables (string, optional): Comma-separated table list - max_latency_ms (integer, optional): Max acceptable latency - allow_cache (boolean, optional): Check semantic cache (default: true) Testing: Run the test script with: ./scripts/mcp/test_nl2sql_tools.sh [--verbose] [--quiet] See scripts/mcp/test_nl2sql_tools.sh --help for usage. Related: Phase 1-4 (Documentation, Unit Tests, Integration Tests, E2E Tests) Related: Phase 6-8 (User Docs, Developer Docs, Test Docs)	6 months ago

Author

SHA1

Message

Date

Rene Cannao

3f44229e28

feat: Add MCP AI Tool Handler for NL2SQL with test script

Phase 5: MCP Tool Implementation for NL2SQL

This commit implements the AI Tool Handler for the MCP (Model Context
Protocol) server, exposing NL2SQL functionality as an MCP tool.

**New Files:**
- include/AI_Tool_Handler.h: Header for AI_Tool_Handler class
  - Provides ai_nl2sql_convert tool via MCP protocol
  - Wraps NL2SQL_Converter and Anomaly_Detector
  - Inherits from MCP_Tool_Handler base class

- lib/AI_Tool_Handler.cpp: Implementation
  - Implements ai_nl2sql_convert tool execution
  - Accepts parameters: natural_language (required), schema,
    context_tables, max_latency_ms, allow_cache
  - Returns JSON response with sql_query, confidence, explanation,
    cached, cache_id

- scripts/mcp/test_nl2sql_tools.sh: Test script for NL2SQL MCP tool
  - Tests ai_nl2sql_convert via JSON-RPC over HTTPS
  - 10 test cases covering SELECT, WHERE, JOIN, aggregation, etc.
  - Includes error handling test for empty queries
  - Supports --verbose, --quiet options

**Modified Files:**
- include/MCP_Thread.h: Add AI_Tool_Handler forward declaration and pointer
- lib/Makefile: Add AI_Tool_Handler.oo to _OBJ_CXX list
- lib/ProxySQL_MCP_Server.cpp: Initialize and register AI tool handler
  - Creates AI_Tool_Handler with GloAI components
  - Registers /mcp/ai endpoint
  - Adds cleanup in destructor

**MCP Tool Details:**
- Endpoint: /mcp/ai
- Tool: ai_nl2sql_convert
- Parameters:
  - natural_language (string, required): Natural language query
  - schema (string, optional): Database schema name
  - context_tables (string, optional): Comma-separated table list
  - max_latency_ms (integer, optional): Max acceptable latency
  - allow_cache (boolean, optional): Check semantic cache (default: true)

**Testing:**
Run the test script with:
  ./scripts/mcp/test_nl2sql_tools.sh [--verbose] [--quiet]

See scripts/mcp/test_nl2sql_tools.sh --help for usage.

Related: Phase 1-4 (Documentation, Unit Tests, Integration Tests, E2E Tests)
Related: Phase 6-8 (User Docs, Developer Docs, Test Docs)

1 Commits (feature/pgsql-native-backend-protocol)