@aws/nx-plugin 0.79.1 → 0.80.1

package/src/py/fast-api/__snapshots__/generator.spec.ts.snap

@@ -4,18 +4,19 @@ exports[`fastapi project generator > should match snapshot > main-snapshot 1`] =
 {
   "apps/test_api/proj_test_api/__init__.py": """"Automatically generated by Nx."""
 ",
-  "apps/test_api/proj_test_api/init.py": "import os
+  "apps/test_api/proj_test_api/init.py": "import json
+import os
 import uuid
-from collections.abc import Callable
+from collections.abc import AsyncIterator, Callable
+from typing import Any
 from urllib.parse import urlparse
 
 from aws_lambda_powertools import Logger, Metrics, Tracer
 from aws_lambda_powertools.metrics import MetricUnit
 from fastapi import FastAPI, Request, Response
 from fastapi.openapi.utils import get_openapi
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, StreamingResponse
 from fastapi.routing import APIRoute
-from mangum import Mangum
 from pydantic import BaseModel
 from starlette.middleware.exceptions import ExceptionMiddleware
 
@@ -31,16 +32,54 @@ class InternalServerErrorDetails(BaseModel):
     detail: str
 
 
-app = FastAPI(title="TestApi", responses={500: {"model": InternalServerErrorDetails}})
-lambda_handler = Mangum(app)
+class JsonStreamingResponse(StreamingResponse):
+    """A streaming response that serializes items to JSON Lines format."""
+
+    media_type = "application/jsonl"
+
+    def __init__(
+        self,
+        content: AsyncIterator[BaseModel],
+        status_code: int = 200,
+        headers: dict[str, str] | None = None,
+        **kwargs: Any,
+    ) -> None:
+        """Stream json lines from an async iterator yielding Pydantic models"""
+        super().__init__(
+            content=self._serialize(content),
+            status_code=status_code,
+            headers=headers,
+            media_type=self.media_type,
+            **kwargs,
+        )
+
+    @staticmethod
+    async def _serialize(
+        content: AsyncIterator[BaseModel],
+    ) -> AsyncIterator[bytes]:
+        """Serialize Pydantic models to JSON Lines format."""
+        async for item in content:
+            yield (item.model_dump_json() + "\\n").encode("utf-8")
+
+    @staticmethod
+    def openapi_response(
+        item_model: type[BaseModel],
+        description: str = "Streaming response",
+    ) -> dict[str, Any]:
+        """Generate an OpenAPI application/jsonl response for a stream of the given model"""
+        return {
+            "description": description,
+            "content": {
+                "application/jsonl": {
+                    "itemSchema": {"$ref": f"#/components/schemas/{item_model.__name__}"},
+                }
+            },
+            # Include the model so FastAPI registers the schema in components/schemas
+            "model": item_model,
+        }
 
-# Add tracing
-lambda_handler.__name__ = "handler"  # tracer requires __name__ to be set
-lambda_handler = tracer.capture_lambda_handler(lambda_handler)
-# Add logging
-lambda_handler = logger.inject_lambda_context(lambda_handler, clear_state=True)
-# Add metrics last to properly flush metrics.
-lambda_handler = metrics.log_metrics(lambda_handler, capture_cold_start_metric=True)
+
+app = FastAPI(title="TestApi", responses={500: {"model": InternalServerErrorDetails}})
 
 
 # Add cors middleware
@@ -98,10 +137,16 @@ async def metrics_handler(request: Request, call_next):
 async def add_correlation_id(request: Request, call_next):
     # Get correlation id from X-Correlation-Id header
     corr_id = request.headers.get("x-correlation-id")
-    if not corr_id and "aws.context" in request.scope:
-        # If empty, use request id from aws context
-        corr_id = request.scope["aws.context"].aws_request_id
-    elif not corr_id:
+    if not corr_id:
+        # Try to get request id from Lambda context (forwarded by Lambda Web Adapter)
+        lambda_context_header = request.headers.get("x-amzn-lambda-context")
+        if lambda_context_header:
+            try:
+                lambda_context = json.loads(lambda_context_header)
+                corr_id = lambda_context.get("request_id")
+            except (json.JSONDecodeError, KeyError):
+                pass
+    if not corr_id:
         # If still empty, use uuid
         corr_id = uuid.uuid4().hex
 
@@ -156,11 +201,10 @@ def custom_openapi():
 
 app.openapi = custom_openapi
 ",
-  "apps/test_api/proj_test_api/main.py": "from pydantic import BaseModel
-
-from .init import app, lambda_handler, tracer
+  "apps/test_api/proj_test_api/main.py": "import uvicorn
+from pydantic import BaseModel
 
-handler = lambda_handler
+from .init import app, tracer
 
 
 class EchoOutput(BaseModel):
@@ -171,6 +215,10 @@ class EchoOutput(BaseModel):
 @tracer.capture_method
 def echo(message: str) -> EchoOutput:
     return EchoOutput(message=f"{message}")
+
+
+if __name__ == "__main__":
+    uvicorn.run("proj_test_api.main:app", port=8000)
 ",
   "apps/test_api/scripts/generate_open_api.py": "from proj_test_api.main import app
 import json, os, sys
@@ -342,8 +390,10 @@ import {
   Function,
   FunctionProps,
   Tracing,
+  LayerVersion,
+  SnapStartConf,
 } from 'aws-cdk-lib/aws-lambda';
-import { Duration } from 'aws-cdk-lib';
+import { Duration, Stack } from 'aws-cdk-lib';
 import { CorsHttpMethod, CfnApi } from 'aws-cdk-lib/aws-apigatewayv2';
 import { HttpIamAuthorizer } from 'aws-cdk-lib/aws-apigatewayv2-authorizers';
 import { HttpLambdaIntegration } from 'aws-cdk-lib/aws-apigatewayv2-integrations';
@@ -392,7 +442,7 @@ export class TestApi<
       operations: OPERATION_DETAILS,
       defaultIntegrationOptions: {
         runtime: Runtime.PYTHON_3_12,
-        handler: 'proj_test_api.main.handler',
+        handler: 'run.sh',
         code: Code.fromAsset(
           url.fileURLToPath(
             new URL(
@@ -403,17 +453,29 @@ export class TestApi<
         ),
         timeout: Duration.seconds(30),
         tracing: Tracing.ACTIVE,
+        snapStart: SnapStartConf.ON_PUBLISHED_VERSIONS,
         environment: {
           AWS_CONNECTION_REUSE_ENABLED: '1',
+          PORT: '8000',
+          AWS_LWA_INVOKE_MODE: 'buffered',
+          AWS_LAMBDA_EXEC_WRAPPER: '/opt/bootstrap',
         },
       } satisfies FunctionProps,
       buildDefaultIntegration: (op, props: FunctionProps) => {
         const handler = new Function(scope, \`TestApi\${op}Handler\`, props);
+        const stack = Stack.of(scope);
+        handler.addLayers(
+          LayerVersion.fromLayerVersionArn(
+            scope,
+            \`TestApi\${op}LWALayer\`,
+            \`arn:aws:lambda:\${stack.region}:753240598075:layer:LambdaAdapterLayerX86:24\`,
+          ),
+        );
         return {
           handler,
           integration: new HttpLambdaIntegration(
             \`TestApi\${op}Integration\`,
-            handler,
+            handler.currentVersion,
           ),
         };
       },
@@ -875,13 +937,16 @@ import {
   Function,
   FunctionProps,
   Tracing,
+  LayerVersion,
+  SnapStartConf,
 } from 'aws-cdk-lib/aws-lambda';
 import {
   AuthorizationType,
   Cors,
   LambdaIntegration,
+  ResponseTransferMode,
 } from 'aws-cdk-lib/aws-apigateway';
-import { Duration } from 'aws-cdk-lib';
+import { Duration, Stack } from 'aws-cdk-lib';
 import {
   PolicyDocument,
   PolicyStatement,
@@ -934,7 +999,7 @@ export class TestApi<
       operations: OPERATION_DETAILS,
       defaultIntegrationOptions: {
         runtime: Runtime.PYTHON_3_12,
-        handler: 'proj_test_api.main.handler',
+        handler: 'run.sh',
         code: Code.fromAsset(
           url.fileURLToPath(
             new URL(
@@ -945,15 +1010,29 @@ export class TestApi<
         ),
         timeout: Duration.seconds(30),
         tracing: Tracing.ACTIVE,
+        snapStart: SnapStartConf.ON_PUBLISHED_VERSIONS,
         environment: {
           AWS_CONNECTION_REUSE_ENABLED: '1',
+          PORT: '8000',
+          AWS_LWA_INVOKE_MODE: 'response_stream',
+          AWS_LAMBDA_EXEC_WRAPPER: '/opt/bootstrap',
         },
       } satisfies FunctionProps,
       buildDefaultIntegration: (op, props: FunctionProps) => {
         const handler = new Function(scope, \`TestApi\${op}Handler\`, props);
+        const stack = Stack.of(scope);
+        handler.addLayers(
+          LayerVersion.fromLayerVersionArn(
+            scope,
+            \`TestApi\${op}LWALayer\`,
+            \`arn:aws:lambda:\${stack.region}:753240598075:layer:LambdaAdapterLayerX86:24\`,
+          ),
+        );
         return {
           handler,
-          integration: new LambdaIntegration(handler),
+          integration: new LambdaIntegration(handler.currentVersion, {
+            responseTransferMode: ResponseTransferMode.STREAM,
+          }),
         };
       },
     });
@@ -1658,10 +1737,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -1670,9 +1751,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "buffered"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -1732,6 +1821,15 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 # Cognito User Pool Authorizer
 resource "aws_apigatewayv2_authorizer" "cognito_authorizer" {
   api_id           = module.http_api.api_id
@@ -1749,12 +1847,12 @@ resource "aws_apigatewayv2_authorizer" "cognito_authorizer" {
 resource "aws_apigatewayv2_integration" "lambda_integration" {
   api_id           = module.http_api.api_id
   integration_type = "AWS_PROXY"
-  integration_uri  = aws_lambda_function.api_lambda.invoke_arn
+  integration_uri  = aws_lambda_alias.live.invoke_arn
 
   payload_format_version = "2.0"
   timeout_milliseconds   = 30000
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Route for proxy integration (catches all requests)
@@ -1783,15 +1881,16 @@ module "add_url_to_runtime_config" {
   depends_on = [module.http_api]
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.http_api.api_execution_arn}/*/*"
 
-  depends_on = [module.http_api, aws_lambda_function.api_lambda]
+  depends_on = [module.http_api, aws_lambda_alias.live]
 }
 
 # Outputs
@@ -2278,10 +2377,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -2290,9 +2391,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "buffered"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -2352,17 +2461,26 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 
 # Lambda integration for HTTP API
 resource "aws_apigatewayv2_integration" "lambda_integration" {
   api_id           = module.http_api.api_id
   integration_type = "AWS_PROXY"
-  integration_uri  = aws_lambda_function.api_lambda.invoke_arn
+  integration_uri  = aws_lambda_alias.live.invoke_arn
 
   payload_format_version = "2.0"
   timeout_milliseconds   = 30000
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Route for proxy integration (catches all requests)
@@ -2390,15 +2508,16 @@ module "add_url_to_runtime_config" {
   depends_on = [module.http_api]
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.http_api.api_execution_arn}/*/*"
 
-  depends_on = [module.http_api, aws_lambda_function.api_lambda]
+  depends_on = [module.http_api, aws_lambda_alias.live]
 }
 
 # Outputs
@@ -2885,10 +3004,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -2897,9 +3018,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "buffered"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -2959,17 +3088,26 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 
 # Lambda integration for HTTP API
 resource "aws_apigatewayv2_integration" "lambda_integration" {
   api_id           = module.http_api.api_id
   integration_type = "AWS_PROXY"
-  integration_uri  = aws_lambda_function.api_lambda.invoke_arn
+  integration_uri  = aws_lambda_alias.live.invoke_arn
 
   payload_format_version = "2.0"
   timeout_milliseconds   = 30000
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Route for proxy integration (catches all requests)
@@ -2999,15 +3137,16 @@ module "add_url_to_runtime_config" {
   depends_on = [module.http_api]
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.http_api.api_execution_arn}/*/*"
 
-  depends_on = [module.http_api, aws_lambda_function.api_lambda]
+  depends_on = [module.http_api, aws_lambda_alias.live]
 }
 
 # Outputs
@@ -3382,10 +3521,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -3394,9 +3535,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "response_stream"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -3456,6 +3605,15 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 # Cognito User Pool Authorizer
 resource "aws_api_gateway_authorizer" "cognito_authorizer" {
   name                   = "TestApiAuthorizer"
@@ -3480,9 +3638,11 @@ resource "aws_api_gateway_integration" "lambda_integration" {
 
   integration_http_method = "POST"
   type                    = "AWS_PROXY"
-  uri                     = aws_lambda_function.api_lambda.invoke_arn
+  # Use the response streaming invocation path with the alias ARN for SnapStart
+  uri                     = "arn:aws:apigateway:\${data.aws_region.current.name}:lambda:path/2021-11-15/functions/\${aws_lambda_alias.live.arn}/response-streaming-invocations"
+  response_transfer_mode  = "STREAM"
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Method for proxy integration
@@ -3614,15 +3774,28 @@ resource "aws_api_gateway_rest_api_policy" "api_policy" {
   })
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
 
-  depends_on = [module.rest_api, aws_lambda_function.api_lambda]
+  depends_on = [module.rest_api, aws_lambda_alias.live]
+}
+
+# Lambda permission for API Gateway to invoke with response streaming via alias
+resource "aws_lambda_permission" "api_gateway_invoke_streaming" {
+  statement_id  = "AllowStreamingFromAPIGateway"
+  action        = "lambda:InvokeFunctionWithResponseStream"
+  function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
+  principal     = "apigateway.amazonaws.com"
+  source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
+
+  depends_on = [module.rest_api, aws_lambda_alias.live]
 }
 
 # Add API url to runtime config
@@ -4008,10 +4181,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -4020,9 +4195,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "response_stream"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -4082,6 +4265,15 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 
 # Create proxy resource (captures all paths)
 resource "aws_api_gateway_resource" "proxy_resource" {
@@ -4098,9 +4290,11 @@ resource "aws_api_gateway_integration" "lambda_integration" {
 
   integration_http_method = "POST"
   type                    = "AWS_PROXY"
-  uri                     = aws_lambda_function.api_lambda.invoke_arn
+  # Use the response streaming invocation path with the alias ARN for SnapStart
+  uri                     = "arn:aws:apigateway:\${data.aws_region.current.name}:lambda:path/2021-11-15/functions/\${aws_lambda_alias.live.arn}/response-streaming-invocations"
+  response_transfer_mode  = "STREAM"
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Method for proxy integration
@@ -4242,15 +4436,28 @@ resource "aws_api_gateway_rest_api_policy" "api_policy" {
   })
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
 
-  depends_on = [module.rest_api, aws_lambda_function.api_lambda]
+  depends_on = [module.rest_api, aws_lambda_alias.live]
+}
+
+# Lambda permission for API Gateway to invoke with response streaming via alias
+resource "aws_lambda_permission" "api_gateway_invoke_streaming" {
+  statement_id  = "AllowStreamingFromAPIGateway"
+  action        = "lambda:InvokeFunctionWithResponseStream"
+  function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
+  principal     = "apigateway.amazonaws.com"
+  source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
+
+  depends_on = [module.rest_api, aws_lambda_alias.live]
 }
 
 # Add API url to runtime config
@@ -4636,10 +4843,12 @@ resource "aws_lambda_function" "api_lambda" {
   filename         = data.archive_file.lambda_zip.output_path
   function_name    = "TestApiHandler"
   role            = aws_iam_role.lambda_execution_role.arn
-  handler         = "proj_test_api.main.handler"
+  handler         = "run.sh"
   runtime         = "python3.12"
+  layers          = ["arn:aws:lambda:\${data.aws_region.current.name}:753240598075:layer:LambdaAdapterLayerX86:24"]
   timeout         = 30
   memory_size     = 128
+  publish         = true
 
   source_code_hash = data.archive_file.lambda_zip.output_base64sha256
 
@@ -4648,9 +4857,17 @@ resource "aws_lambda_function" "api_lambda" {
     mode = "Active"
   }
 
+  # Enable SnapStart for faster cold starts
+  snap_start {
+    apply_on = "PublishedVersions"
+  }
+
   environment {
     variables = merge({
       AWS_CONNECTION_REUSE_ENABLED = "1"
+      PORT                        = "8000"
+      AWS_LWA_INVOKE_MODE         = "response_stream"
+      AWS_LAMBDA_EXEC_WRAPPER     = "/opt/bootstrap"
     }, var.env)
   }
 
@@ -4710,6 +4927,15 @@ resource "aws_cloudwatch_log_group" "lambda_logs" {
   tags              = var.tags
 }
 
+# Lambda alias pointing to the latest published version for SnapStart
+resource "aws_lambda_alias" "live" {
+  name             = "live"
+  function_name    = aws_lambda_function.api_lambda.function_name
+  function_version = aws_lambda_function.api_lambda.version
+
+  depends_on = [aws_lambda_function.api_lambda]
+}
+
 
 # Create proxy resource (captures all paths)
 resource "aws_api_gateway_resource" "proxy_resource" {
@@ -4726,9 +4952,11 @@ resource "aws_api_gateway_integration" "lambda_integration" {
 
   integration_http_method = "POST"
   type                    = "AWS_PROXY"
-  uri                     = aws_lambda_function.api_lambda.invoke_arn
+  # Use the response streaming invocation path with the alias ARN for SnapStart
+  uri                     = "arn:aws:apigateway:\${data.aws_region.current.name}:lambda:path/2021-11-15/functions/\${aws_lambda_alias.live.arn}/response-streaming-invocations"
+  response_transfer_mode  = "STREAM"
 
-  depends_on = [aws_lambda_function.api_lambda]
+  depends_on = [aws_lambda_alias.live]
 }
 
 # Method for proxy integration
@@ -4861,15 +5089,28 @@ resource "aws_api_gateway_rest_api_policy" "api_policy" {
   })
 }
 
-# Lambda permission for API Gateway to invoke the function
+# Lambda permission for API Gateway to invoke the function via alias
 resource "aws_lambda_permission" "api_gateway_invoke" {
   statement_id  = "AllowExecutionFromAPIGateway"
   action        = "lambda:InvokeFunction"
   function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
+  principal     = "apigateway.amazonaws.com"
+  source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
+
+  depends_on = [module.rest_api, aws_lambda_alias.live]
+}
+
+# Lambda permission for API Gateway to invoke with response streaming via alias
+resource "aws_lambda_permission" "api_gateway_invoke_streaming" {
+  statement_id  = "AllowStreamingFromAPIGateway"
+  action        = "lambda:InvokeFunctionWithResponseStream"
+  function_name = aws_lambda_function.api_lambda.function_name
+  qualifier     = aws_lambda_alias.live.name
   principal     = "apigateway.amazonaws.com"
   source_arn    = "\${module.rest_api.api_execution_arn}/*/*"
 
-  depends_on = [module.rest_api, aws_lambda_function.api_lambda]
+  depends_on = [module.rest_api, aws_lambda_alias.live]
 }
 
 # Add API url to runtime config