awslabs.syntheticdata-mcp-server 0.0.2__tar.gz → 1.0.1__tar.gz

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.syntheticdata-mcp-server
-Version: 0.0.2
+Version: 1.0.1
 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/

awslabs_syntheticdata_mcp_server-1.0.1/awslabs/__init__.py

@@ -0,0 +1,15 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# It is intentionally minimal to support PEP 420 namespace packages.

awslabs_syntheticdata_mcp_server-1.0.1/awslabs/syntheticdata_mcp_server/__init__.py

@@ -0,0 +1,17 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""SyntheticData MCP Server package."""
+
+__version__ = '0.0.0'

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/awslabs/syntheticdata_mcp_server/pandas_interpreter.py

@@ -1,13 +1,16 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
-#    http://www.apache.org/licenses/LICENSE-2.0
+#     http://www.apache.org/licenses/LICENSE-2.0
 #
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 
 import ast

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/awslabs/syntheticdata_mcp_server/server.py

@@ -1,17 +1,19 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
-#    http://www.apache.org/licenses/LICENSE-2.0
+#     http://www.apache.org/licenses/LICENSE-2.0
 #
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 """AWS syntheticdata MCP Server implementation."""
 
-import argparse
 import os
 import pandas as pd
 import re
@@ -756,20 +758,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-1.0.1/awslabs/syntheticdata_mcp_server/storage/__init__.py

@@ -0,0 +1,21 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Storage module for synthetic data loading."""
+
+from .base import DataTarget
+from .s3 import S3Target
+from .loader import UnifiedDataLoader
+
+__all__ = ['DataTarget', 'S3Target', 'UnifiedDataLoader']

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/awslabs/syntheticdata_mcp_server/storage/base.py

@@ -1,13 +1,16 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
-#    http://www.apache.org/licenses/LICENSE-2.0
+#     http://www.apache.org/licenses/LICENSE-2.0
 #
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 
 """Base classes for data storage targets."""

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/awslabs/syntheticdata_mcp_server/storage/loader.py

@@ -1,13 +1,16 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
-#    http://www.apache.org/licenses/LICENSE-2.0
+#     http://www.apache.org/licenses/LICENSE-2.0
 #
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 
 """Unified data loader implementation."""

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/awslabs/syntheticdata_mcp_server/storage/s3.py

@@ -1,3 +1,17 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """S3 storage target implementation."""
 
 import asyncio

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/pyproject.toml

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

awslabs_syntheticdata_mcp_server-1.0.1/tests/__init__.py

@@ -0,0 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Test package for syntheticdata MCP server."""

{awslabs_syntheticdata_mcp_server-0.0.2 → awslabs_syntheticdata_mcp_server-1.0.1}/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,
@@ -16,7 +15,6 @@ from awslabs.syntheticdata_mcp_server.server import (
     get_data_gen_instructions,
     load_to_storage,
     main,
-    mcp,
     validate_and_save_data,
 )
 from pytest import mark
@@ -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.2 → awslabs_syntheticdata_mcp_server-1.0.1}/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.1"
+version = "1.0.9223372036854775807"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },

awslabs_syntheticdata_mcp_server-0.0.2/awslabs/__init__.py

@@ -1,12 +0,0 @@
-# This file is part of the awslabs namespace.
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
-# It is intentionally minimal to support PEP 420 namespace packages.

awslabs_syntheticdata_mcp_server-0.0.2/awslabs/syntheticdata_mcp_server/__init__.py

@@ -1,16 +0,0 @@
-#
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
-
-
-"""SyntheticData MCP Server package."""
-
-__version__ = '0.0.0'

awslabs_syntheticdata_mcp_server-0.0.2/awslabs/syntheticdata_mcp_server/storage/__init__.py

@@ -1,7 +0,0 @@
-"""Storage module for synthetic data loading."""
-
-from .base import DataTarget
-from .s3 import S3Target
-from .loader import UnifiedDataLoader
-
-__all__ = ['DataTarget', 'S3Target', 'UnifiedDataLoader']

awslabs_syntheticdata_mcp_server-0.0.2/tests/__init__.py

@@ -1,11 +0,0 @@
-# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
-# with the License. A copy of the License is located at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
-# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
-# and limitations under the License.
-"""Test package for syntheticdata MCP server."""