awslabs.syntheticdata-mcp-server 0.0.1__tar.gz → 1.0.0__tar.gz

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.syntheticdata-mcp-server
-Version: 0.0.1
+Version: 1.0.0
 Summary: An AWS Labs Model Context Protocol (MCP) server for syntheticdata
 Project-URL: homepage, https://awslabs.github.io/mcp/
 Project-URL: docs, https://awslabs.github.io/mcp/servers/syntheticdata-mcp-server/
@@ -97,7 +97,7 @@ The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment v
 ### Getting Data Generation Instructions
 
 ```python
-response = await server.get_data_generation_instructions(
+response = await server.get_data_gen_instructions(
     business_description="An e-commerce platform with customers, orders, and products"
 )
 ```

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/README.md

@@ -66,7 +66,7 @@ The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment v
 ### Getting Data Generation Instructions
 
 ```python
-response = await server.get_data_generation_instructions(
+response = await server.get_data_gen_instructions(
     business_description="An e-commerce platform with customers, orders, and products"
 )
 ```

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/awslabs/syntheticdata_mcp_server/server.py

@@ -11,7 +11,6 @@
 
 """AWS syntheticdata MCP Server implementation."""
 
-import argparse
 import os
 import pandas as pd
 import re
@@ -143,8 +142,8 @@ mcp = FastMCP(
 )
 
 
-@mcp.tool(name='get_data_generation_instructions')
-async def get_data_generation_instructions(
+@mcp.tool(name='get_data_gen_instructions')
+async def get_data_gen_instructions(
     business_description: str = Field(
         ...,
         description='A detailed description of the business domain and use case. The more specific and comprehensive the description, the better the data generation instructions will be.',
@@ -756,20 +755,7 @@ def _validate_table_data(table_name: str, records: List[Dict]) -> Dict:
 
 def main():
     """Run the MCP server with CLI argument support."""
-    parser = argparse.ArgumentParser(
-        description='MCP server for generating synthetic data based on business use cases'
-    )
-    parser.add_argument('--sse', action='store_true', help='Use SSE transport')
-    parser.add_argument('--port', type=int, default=8888, help='Port to run the server on')
-
-    args = parser.parse_args()
-
-    # Run server with appropriate transport
-    if args.sse:
-        mcp.settings.port = args.port
-        mcp.run(transport='sse')
-    else:
-        mcp.run()
+    mcp.run()
 
 
 if __name__ == '__main__':

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/pyproject.toml

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

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/tests/test_server.py

@@ -1,7 +1,6 @@
 """Tests for syntheticdata MCP server functionality."""
 
 import os
-import pytest
 from awslabs.syntheticdata_mcp_server.server import (
     ExecutePandasCodeInput,
     LoadToStorageInput,
@@ -13,17 +12,16 @@ from awslabs.syntheticdata_mcp_server.server import (
     _get_recommended_record_counts,
     _validate_table_data,
     execute_pandas_code,
-    get_data_generation_instructions,
+    get_data_gen_instructions,
     load_to_storage,
     main,
-    mcp,
     validate_and_save_data,
 )
 from pytest import mark
 
 
 @mark.asyncio
-async def test_get_data_generation_instructions() -> None:
+async def test_get_data_gen_instructions() -> None:
     """Test generation of data generation instructions."""
     business_description = """
     An e-commerce platform that sells electronics. We need customer data with their
@@ -31,7 +29,7 @@ async def test_get_data_generation_instructions() -> None:
     including payment status.
     """
 
-    result = await get_data_generation_instructions(business_description)
+    result = await get_data_gen_instructions(business_description)
 
     assert result['success'] is True
     assert 'instructions' in result
@@ -58,9 +56,9 @@ async def test_get_data_generation_instructions() -> None:
 
 
 @mark.asyncio
-async def test_get_data_generation_instructions_empty() -> None:
+async def test_get_data_gen_instructions_empty() -> None:
     """Test generation of data generation instructions with empty input."""
-    result = await get_data_generation_instructions('')
+    result = await get_data_gen_instructions('')
 
     assert result['success'] is False
     assert 'error' in result
@@ -68,9 +66,9 @@ async def test_get_data_generation_instructions_empty() -> None:
 
 
 @mark.asyncio
-async def test_get_data_generation_instructions_invalid() -> None:
+async def test_get_data_gen_instructions_invalid() -> None:
     """Test generation of data generation instructions with invalid input."""
-    result = await get_data_generation_instructions('   ')
+    result = await get_data_gen_instructions('   ')
 
     assert result['success'] is False
     assert 'error' in result
@@ -349,34 +347,20 @@ def test_get_entity_example_data() -> None:
     assert all('description' in record for record in custom_data)
 
 
-@pytest.mark.parametrize(
-    'args,expected_port,expected_sse',
-    [
-        ([], 8888, False),  # Default values
-        (['--port', '9999'], 9999, False),  # Custom port
-        (['--sse'], 8888, True),  # SSE enabled
-        (['--sse', '--port', '7777'], 7777, True),  # Both custom
-    ],
-)
-def test_main_cli_arguments(mock_cli_args, monkeypatch, args, expected_port, expected_sse) -> None:
-    """Test CLI argument handling."""
-    # Update mock CLI args
-    mock_cli_args.extend(args)
-
-    # Mock FastMCP.run to capture arguments
-    run_args = {}
+def test_main_cli_arguments(monkeypatch) -> None:
+    """Test that main() calls mcp.run() without arguments."""
+    # Mock FastMCP.run to verify it's called
+    run_called = False
 
     def mock_run(self, **kwargs):
-        run_args.update(kwargs)
+        nonlocal run_called
+        run_called = True
+        assert not kwargs  # Verify no arguments are passed
 
     monkeypatch.setattr('mcp.server.fastmcp.FastMCP.run', mock_run)
 
     # Run main
     main()
 
-    # Verify settings
-    if expected_sse:
-        assert run_args.get('transport') == 'sse'
-        assert mcp.settings.port == expected_port
-    else:
-        assert not run_args  # Default stdio transport
+    # Verify run was called
+    assert run_called

{awslabs_syntheticdata_mcp_server-0.0.1 → awslabs_syntheticdata_mcp_server-1.0.0}/uv.lock

@@ -1,4 +1,5 @@
 version = 1
+revision = 1
 requires-python = ">=3.10"
 resolution-markers = [
     "python_full_version >= '3.12'",
@@ -41,7 +42,7 @@ wheels = [
 
 [[package]]
 name = "awslabs-syntheticdata-mcp-server"
-version = "0.0.0"
+version = "1.0.9223372036854775807"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },