diff --git a/slowapi/extension.py b/slowapi/extension.py index 050f882..d35f87f 100644 --- a/slowapi/extension.py +++ b/slowapi/extension.py @@ -32,6 +32,7 @@ from starlette.datastructures import MutableHeaders from starlette.requests import Request from starlette.responses import JSONResponse, Response +from starlette.applications import Starlette from typing_extensions import Literal from .errors import RateLimitExceeded @@ -326,7 +327,7 @@ def emit(*_): self._fallback_storage = MemoryStorage() self._fallback_limiter = STRATEGIES[strategy](self._fallback_storage) - def slowapi_startup(self) -> None: + def slowapi_startup(self, app: Starlette) -> None: """ Starlette startup event handler that links the app with the Limiter instance. """