awslabs.stepfunctions-tool-mcp-server 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

awslabs/__init__.py

@@ -1,2 +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.
+# 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.
+
 # This file is part of the awslabs namespace.
 # It is intentionally minimal to support PEP 420 namespace packages.

awslabs/stepfunctions_tool_mcp_server/__init__.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.
+
 """awslabs.stepfunctions-tool-mcp-server"""
 
 __version__ = '0.0.0'

awslabs/stepfunctions_tool_mcp_server/aws_helper.py

@@ -0,0 +1,71 @@
+# 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.
+
+"""AWS Helper for Step Functions Tool MCP Server."""
+
+import boto3
+import botocore.config
+import os
+from typing import Any, Optional
+
+
+class AwsHelper:
+    """Helper class for AWS operations."""
+
+    @staticmethod
+    def get_aws_region() -> Optional[str]:
+        """Get AWS region from environment variable.
+
+        Returns:
+            str: AWS region if set in environment, None otherwise
+        """
+        return os.environ.get('AWS_REGION', 'us-east-1')
+
+    @staticmethod
+    def get_aws_profile() -> Optional[str]:
+        """Get AWS profile from environment variable.
+
+        Returns:
+            str: AWS profile if set in environment, None otherwise
+        """
+        return os.environ.get('AWS_PROFILE')
+
+    @staticmethod
+    def create_boto3_client(service_name: str, region_name: Optional[str] = None) -> Any:
+        """Create a boto3 client with the appropriate configuration.
+
+        Args:
+            service_name: AWS service name (e.g., 'stepfunctions', 'schemas')
+            region_name: Optional region override
+
+        Returns:
+            boto3.client: Configured boto3 client
+        """
+        from awslabs.stepfunctions_tool_mcp_server.server import __version__
+
+        # Create config with user agent
+        config = botocore.config.Config(
+            user_agent_extra=f'awslabs/mcp/aws-stepfunctions-tool-mcp-server/{__version__}'
+        )
+
+        # Get profile and region
+        profile = AwsHelper.get_aws_profile()
+        region = region_name or AwsHelper.get_aws_region()
+
+        # Create client with or without profile
+        if profile:
+            session = boto3.Session(profile_name=profile)
+            return session.client(service_name, region_name=region, config=config)
+        else:
+            return boto3.client(service_name, region_name=region, config=config)

awslabs/stepfunctions_tool_mcp_server/server.py

@@ -1,11 +1,28 @@
+# 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.
+
 """awslabs Step Functions Tool MCP Server implementation."""
 
+# This version should match the version in pyproject.toml
+__version__ = '0.1.5'
+
 import asyncio
-import boto3
 import json
 import logging
 import os
 import re
+from awslabs.stepfunctions_tool_mcp_server.aws_helper import AwsHelper
 from mcp.server.fastmcp import Context, FastMCP
 from typing import Optional
 
@@ -14,11 +31,8 @@ from typing import Optional
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-AWS_PROFILE = os.environ.get('AWS_PROFILE', 'default')
-logger.info(f'AWS_PROFILE: {AWS_PROFILE}')
-
-AWS_REGION = os.environ.get('AWS_REGION', 'us-east-1')
-logger.info(f'AWS_REGION: {AWS_REGION}')
+logger.info(f'AWS_PROFILE: {AwsHelper.get_aws_profile()}')
+logger.info(f'AWS_REGION: {AwsHelper.get_aws_region()}')
 
 STATE_MACHINE_PREFIX = os.environ.get('STATE_MACHINE_PREFIX', '')
 logger.info(f'STATE_MACHINE_PREFIX: {STATE_MACHINE_PREFIX}')
@@ -40,9 +54,8 @@ STATE_MACHINE_INPUT_SCHEMA_ARN_TAG_KEY = os.environ.get('STATE_MACHINE_INPUT_SCH
 logger.info(f'STATE_MACHINE_INPUT_SCHEMA_ARN_TAG_KEY: {STATE_MACHINE_INPUT_SCHEMA_ARN_TAG_KEY}')
 
 # Initialize AWS clients
-session = boto3.Session(profile_name=AWS_PROFILE, region_name=AWS_REGION)
-sfn_client = session.client('stepfunctions')
-schemas_client = session.client('schemas')
+sfn_client = AwsHelper.create_boto3_client('stepfunctions')
+schemas_client = AwsHelper.create_boto3_client('schemas')
 
 mcp = FastMCP(
     'awslabs.stepfunctions-tool-mcp-server',

{awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info → awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.stepfunctions-tool-mcp-server
-Version: 0.1.4
+Version: 0.1.7
 Summary: An AWS Labs Model Context Protocol (MCP) server for AWS Step Functions
 Project-URL: Homepage, https://awslabs.github.io/mcp/
 Project-URL: Documentation, https://awslabs.github.io/mcp/servers/stepfunctions-tool-mcp-server/

awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=WuqxdDgUZylWNmVoPKiK7qGsTB_G4UmuXIrJ-VBwDew,731
+awslabs/stepfunctions_tool_mcp_server/__init__.py,sha256=4SROt2yOCZeQ1L008_6ngAffiudcpwQ8fRFSgCdanAg,683
+awslabs/stepfunctions_tool_mcp_server/aws_helper.py,sha256=TCwI6AoEguBG9tD8g2mf-4f2Tk0E-mnsiTUUnEOcI5o,2444
+awslabs/stepfunctions_tool_mcp_server/server.py,sha256=3mhfTltnMoo-lzqE7tI68OwF_1YcsEv3CB0IuVWE24U,16106
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/METADATA,sha256=cxC_9iabILzpObVvTXkicZzBHdVLWKzvx61urr_vI6k,9698
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/entry_points.txt,sha256=PVQOhNJ_2mgqvdXcqv2KS8Wa2-vZkjatVw0upZga7RU,108
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/licenses/NOTICE,sha256=9GfleTZWBLGtrNSl42TlT_fT7wPBajAx1OQdlIaXUkY,105
+awslabs_stepfunctions_tool_mcp_server-0.1.7.dist-info/RECORD,,

awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/RECORD

@@ -1,9 +0,0 @@
-awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
-awslabs/stepfunctions_tool_mcp_server/__init__.py,sha256=n0IrKZR35oICjSeI7vz7nq78OleFf8Rvo09My7d7SJY,67
-awslabs/stepfunctions_tool_mcp_server/server.py,sha256=B0yqqtbRS-sKEFaGEUKtTLQ_c2zrCvVDfC5Ry1wc_f8,15474
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/METADATA,sha256=tvSFFB0f8U23p7W9tbNlt4kq553TOozKuHkIOiaCMOA,9698
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/entry_points.txt,sha256=PVQOhNJ_2mgqvdXcqv2KS8Wa2-vZkjatVw0upZga7RU,108
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/licenses/NOTICE,sha256=9GfleTZWBLGtrNSl42TlT_fT7wPBajAx1OQdlIaXUkY,105
-awslabs_stepfunctions_tool_mcp_server-0.1.4.dist-info/RECORD,,