1
- from typing import Any , Awaitable , Callable , Mapping , Union
1
+ from typing import Any , Awaitable , Callable , Mapping , Optional , Union
2
2
3
3
from fastapi import FastAPI , Request
4
4
from starlette .requests import HTTPConnection
@@ -39,7 +39,7 @@ async def startup(state: TaskiqState) -> None:
39
39
await app .router .startup ()
40
40
state .lf_ctx = app .router .lifespan_context (app )
41
41
asgi_state = await state .lf_ctx .__aenter__ ()
42
- populate_dependency_context (broker , app , asgi_state or {} )
42
+ populate_dependency_context (broker , app , asgi_state )
43
43
44
44
return startup
45
45
@@ -94,7 +94,7 @@ def init(broker: AsyncBroker, app_or_path: Union[str, FastAPI]) -> None:
94
94
def populate_dependency_context (
95
95
broker : AsyncBroker ,
96
96
app : FastAPI ,
97
- asgi_state : Mapping [str , Any ],
97
+ asgi_state : Optional [ Mapping [str , Any ]] = None ,
98
98
) -> None :
99
99
"""
100
100
Populate dependency context.
@@ -108,6 +108,7 @@ def populate_dependency_context(
108
108
:param app: current application.
109
109
:param kwargs: additional state args.
110
110
"""
111
+ asgi_state = asgi_state or {}
111
112
broker .dependency_overrides .update (
112
113
{
113
114
Request : lambda : Request (
0 commit comments