Get value with default.

Log info message.

Log warning message.

Apply patches to the library. Args: mock_config: Mock configuration dictionary for this library (fixture paths must be absolute)

Log debug message.

Call the wrapped MCP tool function with proper parameter handling. Args: ctx: MCP context **kwargs: Parameter values

No-op error handler.

Convert to dictionary representation.

Decode a continuation token string back to a GlobalContinuationToken object.

Mock MCP ctx.error with the given message.

Create a mock ECR client with default PTC rules configured. This ensures that _is_pull_through_cache_repository doesn't hang waiting for a re

Convert model to dictionary.

Encode the continuation token to a string for client use.

Add a document to the search index. Args: doc: Document to add to the index Note: Extracts and weights diffe

Return a QBusinessClient instance for testing.

Update an existing entity with optimistic locking (prevents lost updates) Note: Uses PutItem with exclude_none=True to support sparse GSIs. T

Generic get operation with optional consistent read

Read all content.

Clear the entire cache.

Parse JSON dict into DataModel with validation. Args: data: Dictionary containing table definitions Returns:

Parse a CLI command string into an IRCommand object.

CloudWatch RUM – monitor real user experience across web and mobile apps. Use the ``action`` parameter to select an operation. All other paramet

Fixture for creating a BedrockModelImportClient instance with mocked AWS clients.

Get boto3 Q Business client. Returns: Boto3QBusinessClient: Boto3 Q Business client

Log error message.

Initialize the context with configuration options. Args: readonly: Whether to run in read-only mode (prevents mutations)

Log an info message to both MCP context and Loguru. Args: message: The info message to log

Log a message with the request ID from the context. Args: ctx: The MCP context containing the request ID level: The log level (fr

Handle AWS service errors in a standardized way. Args: ctx: The MCP context object error: The exception that was raised o

Ensure only one instance of K8sClientCache exists.

Format a standard API response. Args: status: Response status ("success" or "error") data: Response data payload message:

Handle tool errors by logging and returning error information to the agent. This ensures errors are communicated to the agent rather than being s

Get an AWS HealthOmics client. Args: region_name: Optional region override profile_name: Optional AWS profile override Retur

List all available tools.

Search the index and return ranked results. Args: query: Search query string k: Maximum number of results to return

Fixture that patches MongoClient with our mock. Args: monkeypatch: pytest monkeypatch fixture Returns: function: Function to

Create a new connection to DocumentDB. Args: connection_string: DocumentDB connection string Example: "mongodb://

Create a standardized AWS IoT SiteWise client with proper user agent. Args: region: AWS region name (default: us-east-1) Returns:

Create a new HealthOmics workflow. Args: ctx: MCP context for error reporting name: Name of the workflow definition_sourc

Create a new version of an existing workflow. Args: ctx: MCP context for error reporting workflow_id: ID of the workflow

Get a database connection from the map.

Handle IAM-specific errors and return standardized error responses. Args: error: The exception that was raised Returns: Stan

Helper function to run code generation for integration tests.

Build a mock ssm-sap client with configurable responses.

Helper method to validate a schema dictionary.

Check for potential SQL injection risks in sql query. Args: sql: query string Returns: dictionaries containing detected secu

List AWS Transform resources by type.

r"""Check for potential SQL injection risks in sql query. Comment-based bypasses are mitigated in two stages: 1. MySQL conditional comments

Return a list of mutating keywords found in the SQL (excluding comments and string literals).

Format log record with structured information. Args: record: Log record to format Returns: Formatted log mes

Parse the MCP text result envelope into a Python dict.

Get comprehensive health summary for SAP systems managed by SSM for SAP. Returns structured per-application health data including: -

Call FES using the stored connection config via boto3 client. Accepts a FESRequest Pydantic model, a plain mapping, or None. Pydantic models

Check for potential SQL injection risks in sql query. Args: sql: query string Returns: dictionaries containing detected secu

r"""Format Pydantic validation errors with location context and custom messages. Extracts location paths from Pydantic errors and prefixes ea

Load language configuration from JSON file.

Remove a database connection from the map.

Capture the accessibility tree and return formatted text. When a selector is provided, uses CDP Accessibility.getPartialAXTree for re

Get or create an AWS service client connection with retry capabilities. Returns: boto3.client: An AWS service client configured w

Lint a multi-file workflow bundle and return findings. Args: workflow_files: Dictionary mapping file paths to their content

Retrieve change events for AWS resources within specified time range. Args: start_time: Start time for change event query (ISO 8601 or Un

Check if a container image is available in ECR and accessible by HealthOmics. Queries ECR to determine if a specific container image exists in a

Create a pull-through cache rule configured for HealthOmics. Creates an ECR pull-through cache rule for the specified upstream registry and c

Retrieve a specific AWS Transform resource by type.

Return a list of mutating keywords found in the SQL (excluding comments). SQL inline comments (`/* ... */`, `-- ...`, `# ...`) are treated as

Manage AWS EMR EC2 clusters with comprehensive control over cluster lifecycle. This tool provides operations for managing Amazon EMR clusters

Manage AWS Glue ETL jobs and job runs with both read and write operations. This tool provides comprehensive operations for managing AWS Glue

Execute an ECS API operation with the provided parameters. Args: api_operation: The boto3 ECS API operation to execute (camelCase)

Manage AWS Glue workflows to orchestrate complex ETL activities. This tool allows you to create, delete, retrieve, list, and start AWS Glue w

Validate a complete filter expression block. Args: filter_expr: The filter_expression dict from the access pattern en

Get a value from the cache. Args: key: Cache key Returns: Any: Cached value or None if not found or expired

Gets an AWS service client. Parameters ---------- service_name : str The name of the AWS service (e.g., 'ecs', 's3', 'ec2')

Get a medical imaging client with proper user agent.

Mock run method.

Parse the MCP text result envelope into a Python dict.

Generic get operation with optional consistent read

Set a database connection in the map.

Execute and manage AWS Athena SQL queries. This tool provides comprehensive operations for AWS Athena query management, including sta

Manage AWS Glue Data Catalog connections with both read and write operations. Connections in AWS Glue store connection information for data s

Configure security group rules to allow access from EC2 instance to ElastiCache replication group. Args: replication_group_id (str): ID o

Helper to create a KBEntry with sensible defaults.

Call a specific tool with given arguments.

Check for potential SQL injection risks in sql query. Args: sql: query string readonly: when True, also checks for sensitive dict

Lint multi-file WDL or CWL workflow bundles and return validation findings. This tool validates multi-file workflow bundles using appropriate lin

Parse a JSON string into a Python object. Args: json_str: JSON string to parse parameter_name: Name of the parameter (for error m

Helper to create a ServiceGroups entry.

Helper method to create test GenomicsFile objects.

Mimic dict.get behavior.

Get details about a specific workflow. Args: ctx: MCP context for error reporting workflow_id: ID of the workflow to retrieve

Manage AWS EMR EC2 instances with both read and write operations. This tool provides comprehensive operations for managing Amazon EMR EC2 ins

Log a message with the request ID from the context. Args: ctx: The MCP context containing the request ID level: The log level (fr

Manage AWS Glue Interactive Sessions for running Spark and Ray workloads. This tool provides operations for creating and managing Glue Intera