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

awslabs/stepfunctions_tool_mcp_server/aws_helper.py

@@ -0,0 +1,57 @@
+"""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,14 @@
 """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 +17,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 +40,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.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.stepfunctions-tool-mcp-server
-Version: 0.1.4
+Version: 0.1.6
 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.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
+awslabs/stepfunctions_tool_mcp_server/__init__.py,sha256=n0IrKZR35oICjSeI7vz7nq78OleFf8Rvo09My7d7SJY,67
+awslabs/stepfunctions_tool_mcp_server/aws_helper.py,sha256=hVRCBeF49LMCpcO-RSfiIh5-AR9vBa8t1TuY5g6pz7I,1828
+awslabs/stepfunctions_tool_mcp_server/server.py,sha256=Dn6fH80ChAY7AKFO9lJPAlHLGVdnnH7flogTtjdiRuU,15490
+awslabs_stepfunctions_tool_mcp_server-0.1.6.dist-info/METADATA,sha256=ijHCCGyVbbseWvziFDyNTtQa6cL_KiUu19WJobm8veQ,9698
+awslabs_stepfunctions_tool_mcp_server-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_stepfunctions_tool_mcp_server-0.1.6.dist-info/entry_points.txt,sha256=PVQOhNJ_2mgqvdXcqv2KS8Wa2-vZkjatVw0upZga7RU,108
+awslabs_stepfunctions_tool_mcp_server-0.1.6.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+awslabs_stepfunctions_tool_mcp_server-0.1.6.dist-info/licenses/NOTICE,sha256=9GfleTZWBLGtrNSl42TlT_fT7wPBajAx1OQdlIaXUkY,105
+awslabs_stepfunctions_tool_mcp_server-0.1.6.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,,