Skip to content

fix: pass lifespan to starlette #51

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed

fix: pass lifespan to starlette #51

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions src/bedrock_agentcore/runtime/app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
from starlette.applications import Starlette
from starlette.responses import JSONResponse, Response, StreamingResponse
from starlette.routing import Route
from starlette.types import Lifespan

from .context import BedrockAgentCoreContext, RequestContext
from .models import (
Expand Down Expand Up @@ -49,11 +50,12 @@ def format(self, record):
class BedrockAgentCoreApp(Starlette):
"""Bedrock AgentCore application class that extends Starlette for AI agent deployment."""

def __init__(self, debug: bool = False):
def __init__(self, debug: bool = False, lifespan: Lifespan | None = None):
"""Initialize Bedrock AgentCore application.

Args:
debug: Enable debug actions for task management (default: False)
lifespan: used as the Starlette lifespan
"""
self.handlers: Dict[str, Callable] = {}
self._ping_handler: Optional[Callable] = None
Expand All @@ -66,7 +68,7 @@ def __init__(self, debug: bool = False):
Route("/invocations", self._handle_invocation, methods=["POST"]),
Route("/ping", self._handle_ping, methods=["GET"]),
]
super().__init__(routes=routes)
super().__init__(routes=routes, lifespan=lifespan)
self.debug = debug # Set after super().__init__ to avoid override

self.logger = logging.getLogger("bedrock_agentcore.app")
Expand Down
Loading