awslabs.healthlake-mcp-server 0.0.1__py3-none-any.whl

awslabs/healthlake_mcp_server/main.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+# 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.
+
+"""Main entry point for the AWS HealthLake MCP server."""
+
+# Standard library imports
+import argparse
+import asyncio
+import os
+import sys
+
+# Local imports
+from .server import create_healthlake_server
+
+# Third-party imports
+from loguru import logger
+from mcp.server.stdio import stdio_server
+
+
+# Configure logging
+logger.remove()
+logger.add(sys.stderr, level=os.getenv('MCP_LOG_LEVEL', 'WARNING'))
+
+
+def parse_args():
+    """Parse command line arguments."""
+    parser = argparse.ArgumentParser(description='AWS HealthLake MCP Server')
+    parser.add_argument(
+        '--readonly',
+        action='store_true',
+        help='Run server in read-only mode (prevents all mutating operations)',
+    )
+    return parser.parse_args()
+
+
+async def main() -> None:
+    """Main entry point for the server."""
+    try:
+        # Parse command line arguments
+        args = parse_args()
+
+        # Create the HealthLake MCP server with read-only mode
+        server = create_healthlake_server(read_only=args.readonly)
+
+        # Log server mode
+        if args.readonly:
+            logger.info('Server started in READ-ONLY mode - mutating operations disabled')
+        else:
+            logger.info('Server started in FULL ACCESS mode')
+
+        # Run the server using stdio transport
+        async with stdio_server() as (read_stream, write_stream):
+            await server.run(read_stream, write_stream, server.create_initialization_options())
+    except Exception as e:
+        logger.error(f'Server error: {e}')
+        raise
+
+
+def sync_main() -> None:
+    """Synchronous wrapper for the main function."""
+    asyncio.run(main())
+
+
+if __name__ == '__main__':
+    sync_main()

awslabs/healthlake_mcp_server/models.py

@@ -0,0 +1,120 @@
+# 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.
+
+"""Pydantic models for HealthLake MCP server."""
+
+# Standard library imports
+# Local imports
+from .fhir_operations import DATASTORE_ID_LENGTH
+
+# Third-party imports
+from pydantic import BaseModel, Field, field_validator
+from typing import Any, Dict, Optional
+
+
+class FHIRResource(BaseModel):
+    """Base FHIR resource model."""
+
+    resourceType: str
+    id: Optional[str] = None
+    meta: Optional[Dict[str, Any]] = None
+
+
+class SearchParameters(BaseModel):
+    """FHIR search parameters."""
+
+    parameters: Dict[str, str] = Field(default_factory=dict)
+    count: int = Field(default=100, ge=1, le=100)
+
+
+class CreateResourceRequest(BaseModel):
+    """Request to create a FHIR resource."""
+
+    datastore_id: str = Field(..., min_length=DATASTORE_ID_LENGTH, max_length=DATASTORE_ID_LENGTH)
+    resource_type: str
+    resource_data: Dict[str, Any]
+
+    @field_validator('datastore_id')
+    @classmethod
+    def validate_datastore_id(cls, v):
+        """Validate datastore ID is alphanumeric."""
+        if not v.isalnum():
+            raise ValueError('Datastore ID must be alphanumeric')
+        return v
+
+
+class UpdateResourceRequest(BaseModel):
+    """Request to update a FHIR resource."""
+
+    datastore_id: str = Field(..., min_length=DATASTORE_ID_LENGTH, max_length=DATASTORE_ID_LENGTH)
+    resource_type: str
+    resource_id: str
+    resource_data: Dict[str, Any]
+
+    @field_validator('datastore_id')
+    @classmethod
+    def validate_datastore_id(cls, v):
+        """Validate datastore ID is alphanumeric."""
+        if not v.isalnum():
+            raise ValueError('Datastore ID must be alphanumeric')
+        return v
+
+
+class DatastoreFilter(BaseModel):
+    """Filter for listing datastores."""
+
+    status: Optional[str] = Field(None, pattern='^(CREATING|ACTIVE|DELETING|DELETED)$')
+
+
+class ImportJobConfig(BaseModel):
+    """Configuration for FHIR import job."""
+
+    datastore_id: str = Field(..., min_length=DATASTORE_ID_LENGTH, max_length=DATASTORE_ID_LENGTH)
+    input_data_config: Dict[str, Any]
+    data_access_role_arn: str
+    job_name: Optional[str] = None
+
+    @field_validator('datastore_id')
+    @classmethod
+    def validate_datastore_id(cls, v):
+        """Validate datastore ID is alphanumeric."""
+        if not v.isalnum():
+            raise ValueError('Datastore ID must be alphanumeric')
+        return v
+
+
+class ExportJobConfig(BaseModel):
+    """Configuration for FHIR export job."""
+
+    datastore_id: str = Field(..., min_length=DATASTORE_ID_LENGTH, max_length=DATASTORE_ID_LENGTH)
+    output_data_config: Dict[str, Any]
+    data_access_role_arn: str
+    job_name: Optional[str] = None
+
+    @field_validator('datastore_id')
+    @classmethod
+    def validate_datastore_id(cls, v):
+        """Validate datastore ID is alphanumeric."""
+        if not v.isalnum():
+            raise ValueError('Datastore ID must be alphanumeric')
+        return v
+
+
+class JobFilter(BaseModel):
+    """Filter for listing jobs."""
+
+    job_status: Optional[str] = Field(
+        None, pattern='^(SUBMITTED|IN_PROGRESS|COMPLETED|FAILED|STOP_REQUESTED|STOPPED)$'
+    )
+    job_type: Optional[str] = Field(None, pattern='^(IMPORT|EXPORT)$')