Adding MySQL Support/Tools [POSTFIX]

Update README with suggested MCP server usage suggestions and additional
supported MCP tool.

Signed-off-by: Gregory Howe <[email protected]>

Author: GregorySHowe

src/dbtools-mcp-server/README.md

@@ -35,6 +35,8 @@ dbtools-mcp-server.py is a FastMCP-based server that provides various tools for
   - HeatWave similarity search and RAG on existing vector stores
   - HeatWave AutoML syntax reference helper
   - Create HeatWave vector store from objects in Object Store
+  - [NOTE] MySQL AI is not compatible with DBTools MCP Server
+  - [NOTE] MySQL MCP Server relies on MySQL connector python and is recommended for new MySQL HeatWave or MySQL AI users
 
 - **Additional Features**
   - OCI Object Store list buckets and objects
@@ -145,11 +147,11 @@ python dbtools-mcp-server.py
 16. `list_reports(dbtools_connection_display_name)`: Lists all reports for a connection
 17. `find_matching_reports(dbtools_connection_display_name, search_text, limit=5)`: Finds similar reports using vector similarity search
 18. `ragify_column(dbtools_connection_display_name, table_name, column_names, vector_column_name)`: Creates and populates a vector column for RAG integration
-19. `call_nl2ml(dbtools_connection_display_name: string, question: string)`: Ask natural language questions to machine learning help tool to get answers about heatwave ML (AutoML)
-20. `ask_ml_rag(dbtools_connection_display_name: string, question: string)`: Ask ml_rag - retrieval augmented generation tool a question. This is the preferred tool for answering questions using vector stores.
-21. `call_load_vector_store(dbtools_connection_display_name: string, namespace: string, bucket_name: string, document_prefix: string, schema_name: string, table_name: string)`: Load documents from object storage into a vector store for similarity search and RAG. Path can be file name, prefix, or full path.
-22. `call_list_buckets(dbtools_connection_display_name: string, compartment_id: string)`: List all accessible object store buckets
-23. `call_list_objects(dbtools_connection_display_name: string, namespace: string, bucket_name: string)`: List objects/files stored in a given object store bucket
+19. `heatwave_ask_help(dbtools_connection_display_name: string, question: string)`: Ask natural language questions to machine learning help tool to get answers about heatwave ML (AutoML)
+20. `heatwave_ask_ml_rag(dbtools_connection_display_name: string, question: string)`: Ask ml_rag - retrieval augmented generation tool a question. This is the preferred tool for answering questions using vector stores.
+21. `heatwave_load_vector_store(dbtools_connection_display_name: string, namespace: string, bucket_name: string, document_prefix: string, schema_name: string, table_name: string)`: Load documents from object storage into a vector store for similarity search and RAG. Path can be file name, prefix, or full path.
+22. `object_storage_list_buckets(compartment_id: string)`: List all accessible object store buckets
+23. `object_storage_list_objects(namespace: string, bucket_name: string)`: List objects/files stored in a given object store bucket
 
 ## Security
 

src/mysql-mcp-server/README.md

@@ -4,22 +4,23 @@ A Python-based MCP (Model Context Protocol) server that provides a suite of tool
 
 ## Overview
 
-`mysql_mcp_server.py` is a FastMCP-based server that provides tools for managing MySQL connections, executing SQL, using MySQL AI ML/AI features, and working with OCI Object Storage.
+`mysql_mcp_server.py` is a FastMCP-based server that provides tools for managing MySQL connections, executing SQL, using MySQL AI or MySQL HeatWave ML/GenAI features, and working with OCI Object Storage. MySQL MCP Server is the recommended server to get started with MySQL HeatWave or MySQL AI and the only server in this repo that supports MySQL AI.
 
 ## Features
 
 - **Database Connection Management**
   - Load connection configs from JSON or environment variables
   - List all configured database connections
-  - Validate connectivity and resolve provider mode (MySQL AI vs. OCI)
+  - Validate connectivity and resolve provider mode (MySQL AI vs. MySQL HeatWave)
 
 - **Database Operations**
   - Execute SQL queries
 
-- **MySQL AI ML and AI Tools**
-  - `ml_generate`: Text generation with MySQL AI GenAI
+- **MySQL AI and MySQL HeatWave ML and GenAI Tools**
+  - `ml_generate`: Text generation with GenAI
   - `ragify_column`: Create/populate vector columns for embeddings
   - `ask_ml_rag`: Retrieval-augmented generation from vector stores
+  - `heatwave_ask_help`: Answers questions about how to use HeatWave ML
 
 - **Vector Store Management**
   - List files in `secure_file_priv` (local mode)
@@ -62,14 +63,13 @@ see the [OCI SDK documentation](https://docs.oracle.com/en-us/iaas/Content/API/C
 ## Required Python Packages
 
 - `oci`
-- `requests`
 - `fastmcp`
 - `mysql-connector-python`
 
 ## Supported Database Modes
 
-- **MYSQL_AI** (local MySQL AI AI mode)
-- **OCI** (Oracle Cloud Infrastructure-managed databases)
+- `MySQL AI`
+- `HeatWave`
 
 ## MCP Server Configuration
 
@@ -202,16 +202,17 @@ python mysql_mcp_server.py
 
 1. `list_all_connections()`: List configured database connections and modes
 2. `execute_sql_tool_by_connection_id(connection_id, sql, params)`: Execute SQL on a database connection
-3. `ml_generate(connection_id, question)`: Generate text via MySQL AI ML
+3. `ml_generate(connection_id, question)`: Generate text
 4. `ragify_column(connection_id, table, input_col, embedding_col)`: Embed text into a VECTOR column
 5. `list_vector_store_files_local(connection_id)`: List available files in `secure_file_priv`
 6. `load_vector_store_local(connection_id, file_path)`: Load documents from local filesystem
 7. `load_vector_store_oci(connection_id, namespace, bucket, prefix, schema, table)`: Load documents from OCI Object Storage
 8. `ask_ml_rag_vector_store(connection_id, question)`: RAG query on default vector store
 9. `ask_ml_rag_innodb(connection_id, question, segment_col, embedding_col)`: RAG query restricted to InnoDB tables
-10. `list_all_compartments()`: List OCI compartments
-11. `object_storage_list_buckets(compartment_name | compartment_id)`: List buckets in a compartment
-12. `object_storage_list_objects(namespace, bucket_name)`: List objects in a bucket
+10. `heatwave_ask_help(connection_id, question)`: Ask natural language questions about MySQL HeatWave AutoML via NL2ML
+11. `list_all_compartments()`: List OCI compartments
+12. `object_storage_list_buckets(compartment_name | compartment_id)`: List buckets in a compartment
+13. `object_storage_list_objects(namespace, bucket_name)`: List objects in a bucket
 
 ## Security
 
@@ -230,10 +231,10 @@ Here are example prompts you can use to interact with the MCP server, note that
 "Add embeddings for 'body' column into 'embedding' column in docs table"
 ```
 
-### 2. MySQL AI AI
+### 2. MySQL AI/MySQL HeatWave
 
 ```
-"Generate a summary of error logs using MySQL AI ML"
+"Generate a summary of error logs"
 "Ask ml_rag: Show me refund policy from the vector store"
 ```
 

src/mysql-mcp-server/test_mysql_mcp_server.py

@@ -1,3 +1,8 @@
+"""
+Copyright (c) 2025, Oracle and/or its affiliates.
+Licensed under the Universal Permissive License v1.0 as shown at http://oss.oracle.com/licenses/upl.
+"""
+
 import contextlib
 import importlib.util
 import json