fix: Add debug logging to Python request handlers

Author: clareliguori

src/python/src/mcp_lambda/handlers/api_gateway_proxy_event_handler.py

@@ -47,7 +47,7 @@ def parse_event(self, event: APIGatewayProxyEvent) -> ParsedHttpRequest:
         headers = event.get("headers") or {}
         # Ensure headers are Dict[str, Optional[str]] as expected by ParsedHttpRequest
         normalized_headers: Dict[str, Optional[str]] = dict(headers)
-        
+
         return ParsedHttpRequest(
             method=event["httpMethod"],
             headers=normalized_headers,
@@ -60,4 +60,5 @@ def format_response(self, response: HttpResponse) -> APIGatewayProxyResult:
             "statusCode": response.status_code,
             "headers": response.headers,
             "body": response.body,
+            "isBase64Encoded": False,
         }

src/python/src/mcp_lambda/handlers/streamable_http_handler.py

@@ -15,6 +15,7 @@
 
 import json
 import logging
+import os
 from abc import ABC, abstractmethod
 from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
 
@@ -36,6 +37,9 @@
 
 # Set up logging
 logger = logging.getLogger(__name__)
+log_level = os.getenv("LOG_LEVEL", "INFO").upper()
+logger.setLevel(getattr(logging, log_level))
+logger.addHandler(logging.StreamHandler())
 
 # Type variables for generic event and result types
 TEvent = TypeVar("TEvent")
@@ -97,7 +101,8 @@ def handle(self, event: TEvent, context: LambdaContext) -> TResult:
             http_response = self.process_http_request(http_request, context)
 
             # Format the response for the specific event type
-            return self.format_response(http_response)
+            response = self.format_response(http_response)
+            logger.debug("Response: %s", json.dumps(response, default=str, indent=2))
         except Exception as error:
             logger.error(
                 "Error processing MCP Streamable HTTP request: %s", error, exc_info=True

src/python/src/mcp_lambda/server_adapter/stdio_server_adapter_request_handler.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from typing import Union
 
 from aws_lambda_typing.context import Context as LambdaContext
@@ -16,6 +17,9 @@
 
 # Set up logging
 logger = logging.getLogger(__name__)
+log_level = os.getenv("LOG_LEVEL", "INFO").upper()
+logger.setLevel(getattr(logging, log_level))
+logger.addHandler(logging.StreamHandler())
 
 
 class StdioServerAdapterRequestHandler(RequestHandler):