awslabs.mysql-mcp-server 1.0.8__tar.gz → 1.0.9__tar.gz

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.mysql-mcp-server
-Version: 1.0.8
+Version: 1.0.9
 Summary: An AWS Labs Model Context Protocol (MCP) server for mysql
 Project-URL: homepage, https://awslabs.github.io/mcp/
 Project-URL: docs, https://awslabs.github.io/mcp/servers/mysql-mcp-server/

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/awslabs/mysql_mcp_server/__init__.py

@@ -14,4 +14,4 @@
 
 """awslabs.mysql_mcp_server"""
 
-__version__ = '1.0.8'
+__version__ = '1.0.9'

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/awslabs/mysql_mcp_server/connection/asyncmy_pool_connection.py

@@ -132,7 +132,7 @@ class AsyncmyPoolConnection(AbstractDBConnection):
                         await cursor.execute('SET TRANSACTION READ ONLY')
                     # Execute the query
                     if parameters:
-                        params = _convert_parameters(self, parameters).values()
+                        params = list(_convert_parameters(self, parameters).values())
                         await cursor.execute(sql, params)
                     else:
                         await cursor.execute(sql)
@@ -146,7 +146,7 @@ class AsyncmyPoolConnection(AbstractDBConnection):
                         rows = await cursor.fetchall()
 
                         # Structure the response to match the interface contract required by server.py
-                        column_metadata = [{'name': col} for col in columns]
+                        column_metadata = [{'label': col} for col in columns]
                         records = []
 
                         # Convert each row to the expected format

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/awslabs/mysql_mcp_server/server.py

@@ -18,6 +18,7 @@ import argparse
 import asyncio
 import sys
 from awslabs.mysql_mcp_server.connection import DBConnectionSingleton
+from awslabs.mysql_mcp_server.connection.asyncmy_pool_connection import AsyncmyPoolConnection
 from awslabs.mysql_mcp_server.mutable_sql_detector import (
     check_sql_injection_risk,
     detect_mutating_keywords,
@@ -193,9 +194,16 @@ async def get_table_schema(
         ORDER BY
             ORDINAL_POSITION
     """
+    db_connection = DBConnectionSingleton.get().db_connection
+
+    if isinstance(db_connection, AsyncmyPoolConnection):
+        # Convert to positional parameters for asyncmy
+        sql = sql.replace(':database_name', '%s').replace(':table_name', '%s')
+
+    # Use consistent parameter order matching SQL placeholders
     params = [
-        {'name': 'table_name', 'value': {'stringValue': table_name}},
         {'name': 'database_name', 'value': {'stringValue': database_name}},
+        {'name': 'table_name', 'value': {'stringValue': table_name}},
     ]
 
     return await run_query(sql=sql, ctx=ctx, query_parameters=params)

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "awslabs.mysql-mcp-server"
-version = "1.0.8"
+version = "1.0.9"
 description = "An AWS Labs Model Context Protocol (MCP) server for mysql"
 readme = "README.md"
 requires-python = ">=3.10"

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/tests/test_asyncmy_pool_connection.py

@@ -97,7 +97,7 @@ async def test_execute_query_returns_results():
         conn.pool = fake_pool
         result = await conn.execute_query('SELECT id, name FROM users')
 
-        assert result['columnMetadata'] == [{'name': 'id'}, {'name': 'name'}]
+        assert result['columnMetadata'] == [{'label': 'id'}, {'label': 'name'}]
         assert result['records'] == [
             [{'longValue': 1}, {'stringValue': 'Alice'}],
             [{'longValue': 2}, {'stringValue': 'Bob'}],
@@ -400,13 +400,13 @@ async def test_execute_query_readonly_and_parameter_types_mapped_boolean_as_long
 
         fake_cursor.execute.assert_any_await('SET TRANSACTION READ ONLY')
         assert result['columnMetadata'] == [
-            {'name': 'n'},
-            {'name': 's'},
-            {'name': 'i'},
-            {'name': 'f'},
-            {'name': 'b'},
-            {'name': 'blob'},
-            {'name': 'odd'},
+            {'label': 'n'},
+            {'label': 's'},
+            {'label': 'i'},
+            {'label': 'f'},
+            {'label': 'b'},
+            {'label': 'blob'},
+            {'label': 'odd'},
         ]
         # Note: bool mapped via int branch -> {'longValue': True}
         assert result['records'] == [

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/tests/test_server.py

@@ -20,6 +20,7 @@ import json
 import pytest
 import sys
 import uuid
+from awslabs.mysql_mcp_server.connection.asyncmy_pool_connection import AsyncmyPoolConnection
 from awslabs.mysql_mcp_server.server import (
     DBConnectionSingleton,
     client_error_code_key,
@@ -30,6 +31,7 @@ from awslabs.mysql_mcp_server.server import (
     write_query_prohibited_key,
 )
 from conftest import DummyCtx, Mock_DBConnection, MockException
+from unittest.mock import AsyncMock, Mock, patch
 
 
 SAFE_READONLY_QUERIES = [
@@ -834,6 +836,46 @@ def test_main_with_valid_asyncmy_parameters(monkeypatch, capsys):
     main()
 
 
+@pytest.mark.asyncio
+async def test_get_table_schema_asyncmy_connection():
+    """Test get_table_schema with asyncmy connection type detection and SQL conversion."""
+    with patch(
+        'awslabs.mysql_mcp_server.connection.asyncmy_pool_connection._get_credentials_from_secret',
+        return_value=('user', 'pass'),
+    ):
+        DBConnectionSingleton.initialize(
+            'mock', 'mock', 'mock', hostname='mock', port=3306, readonly=False, is_test=True
+        )
+
+    # Replace the real asyncmy connection with a mock to avoid actual DB connection
+    mock_asyncmy_connection = Mock(spec=AsyncmyPoolConnection)
+    mock_asyncmy_connection.execute_query = AsyncMock(
+        return_value=get_mock_normal_query_response()
+    )
+    DBConnectionSingleton._instance._db_connection = mock_asyncmy_connection  # type: ignore
+
+    ctx = DummyCtx()
+    tool_response = await get_table_schema(table_name='table_name', database_name='mysql', ctx=ctx)
+
+    # Verify SQL was converted from :name to %s for asyncmy
+    call_args = mock_asyncmy_connection.execute_query.call_args
+    sql_used = call_args[0][0]  # First positional argument is the SQL
+
+    assert '%s' in sql_used
+    assert ':database_name' not in sql_used
+    assert ':table_name' not in sql_used
+
+    # validate tool_response
+    assert (
+        isinstance(tool_response, (list, tuple))
+        and len(tool_response) == 1
+        and isinstance(tool_response[0], dict)
+        and 'error' not in tool_response[0]
+    )
+    column_records = tool_response[0]
+    validate_normal_query_response(column_records)
+
+
 if __name__ == '__main__':
     DBConnectionSingleton.initialize(
         'mock', 'mock', 'mock', resource_arn='mock', readonly=True, is_test=True

awslabs_mysql_mcp_server-1.0.9/uv-requirements.txt

@@ -0,0 +1,27 @@
+#
+# This file is autogenerated by pip-compile with Python 3.12
+# by the following command:
+#
+#    pip-compile --generate-hashes --output-file=uv-requirements.txt --strip-extras uv-requirements.in
+#
+uv==0.9.6 \
+    --hash=sha256:0169a85d3ba5ef1c37089d64ff26de573439ca84ecf549276a2eee42d7f833f2 \
+    --hash=sha256:0fde18c22376c8b02954c7db3847bc75ac42619932c44b43f49d056e5cfb05f9 \
+    --hash=sha256:166175ba952d2ad727e1dbd57d7cfc1782dfe7b8d79972174a46a7aa33ddceec \
+    --hash=sha256:3c2c2b2b093330e603d838fec26941ab6f62e8d62a012f9fa0d5ed88da39d907 \
+    --hash=sha256:538716ec97f8d899baa7e1c427f4411525459c0ef72ea9b3625ce9610c9976e6 \
+    --hash=sha256:547fd27ab5da7cd1a833288a36858852451d416a056825f162ecf2af5be6f8b8 \
+    --hash=sha256:62e3f057a9ae5e5003a7cd56b617e940f519f6dabcbb22d36cdd0149df25d409 \
+    --hash=sha256:6403176b55388cf94fb8737e73b26ee2a7b1805a9139da5afa951210986d4fcd \
+    --hash=sha256:7e89c964f614fa3f0481060cac709d6da50feac553e1e11227d6c4c81c87af7c \
+    --hash=sha256:86e05782f9b75d39ab1c0af98bf11e87e646a36a61d425021d5b284073e56315 \
+    --hash=sha256:90122a76e6441b8c580fc9faf06bd8c4dbe276cb1c185ad91eceb2afa78e492a \
+    --hash=sha256:95a62c1f668272555ad0c446bf44a9924dee06054b831d04c162e0bad736dc28 \
+    --hash=sha256:a7c6067919d87208c4a6092033c3bc9799cb8be1c8bc6ef419a1f6d42a755329 \
+    --hash=sha256:b2f934737c93f88c906b6a47bcc083170210fe5d66565e80a7c139599e5cbf2f \
+    --hash=sha256:b31377ebf2d0499afc5abe3fe1abded5ca843f3a1161b432fe26eb0ce15bab8e \
+    --hash=sha256:d1072db92cc9525febdf9d113c23916dfc20ca03e21218cc7beefe7185a90631 \
+    --hash=sha256:e700b2098f9d365061c572d0729b4e8bc71c6468d83dfaae2537cd66e3cb1b98 \
+    --hash=sha256:ea67369918af24ea7e01991dfc8b8988d1b0b7c49cb39d9e5bc0c409930a0a3f \
+    --hash=sha256:f0ba311b3ca49d246f36d444d3ee81571619ef95e5f509eb694a81defcbed262
+    # via -r uv-requirements.in (contents of `uv==0.9.6`)

{awslabs_mysql_mcp_server-1.0.8 → awslabs_mysql_mcp_server-1.0.9}/uv.lock

@@ -83,7 +83,7 @@ wheels = [
 
 [[package]]
 name = "awslabs-mysql-mcp-server"
-version = "1.0.8"
+version = "1.0.9"
 source = { editable = "." }
 dependencies = [
     { name = "asyncmy" },

awslabs_mysql_mcp_server-1.0.8/uv-requirements.txt

@@ -1,24 +0,0 @@
-# This file was autogenerated by uv via the following command:
-#    echo "uv==0.8.10" > uv-requirements.in
-#    uv pip compile --generate-hashes --output-file=uv-requirements.txt --strip-extras --python=3.10 uv-requirements.in
-uv==0.8.10 \
-    --hash=sha256:31e4fc37ee94b94c032384a0957ad32ba7dce4ce6c04b4880fd3e31e25e51a82 \
-    --hash=sha256:36a5ce708d52388c37043e7335f9eb3fea5a19a56166a2cc6adb365179a1cd77 \
-    --hash=sha256:38286d230daad82388469c8dc7a1d2f5dc279c11178319c886d1a88d7938e513 \
-    --hash=sha256:3e190cee3bb2b4f574a419eef87ae8e33f713e9cd6f856b83277ece70ad9ca9b \
-    --hash=sha256:3fdf89fc40af9902141c39ed943bcfca15664623363335eb032a44f22001e2b4 \
-    --hash=sha256:4cc190d403a89e46d13cec83b6f8e8d7d07aaf1e5a996eac9a3f0c2a8cd92537 \
-    --hash=sha256:57b71dc79eff25a5419d3fe4a563d3b9397f55d789f685ef27f43f033b31f482 \
-    --hash=sha256:86fe044c2be43977566a0d184a487edd7aace2febb757fd95927684b629ef50b \
-    --hash=sha256:88df34c32555064fae459cce665757619fd1af7deb2dc393352b15d909d2d131 \
-    --hash=sha256:9ad21eeaa4156a1bf5ed85903f80db06e2c02badd3a587ba98d3171517960555 \
-    --hash=sha256:a5495b5a6e3111c03cf5e4dbdd598bc8fd1da887e3920d58cd5a2d4c8bc9a473 \
-    --hash=sha256:ab072cd3bf2f9dc264659a1ff48ad91a910ac4830bcfe965e2d3f89c86646f46 \
-    --hash=sha256:af8a5526b0e331775a264fa0dbccfd53c183cb974f269a208af136d7561f9eb2 \
-    --hash=sha256:b00637c63d5dfc9f879281c5c91db2bb909ab1f9ab275dab015e7fb6cac6be5b \
-    --hash=sha256:b3ff3c451fcd23ea78356d8c18e802d0e423cbe655273601e3ec039a51b33286 \
-    --hash=sha256:c4a493cd4b15b3aef11523531aff96a77a586666a63e842fa437966b7b7ee62d \
-    --hash=sha256:defc50bb319be2d58be74a680710cd4b7697e88d5f79974eacd354df95f0b6b0 \
-    --hash=sha256:e0a02bcec766eb0862b7082ab746b204add7d9fcaa62322502d159b5a7ccc54a \
-    --hash=sha256:eb79a46d8099f563ef58237bf4e9009f876a40145e757ea883a92b24b724d01e
-    # via -r uv-requirements.in