fix: correct FastAPI with new state

airtai · Dec 12, 2024 · cddf8de · cddf8de
1 parent b84c032
commit cddf8de
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/faststream/_internal/application.py b/faststream/_internal/application.py
@@ -16,9 +16,10 @@
 from faststream._internal.constants import EMPTY
 from faststream._internal.context import ContextRepo
 from faststream._internal.log import logger
-from faststream._internal.state import (
+from faststream._internal.state import DIState
+from faststream._internal.state.application import (
+    ApplicationState,
     BasicApplicationState,
-    DIState,
     RunningApplicationState,
 )
 from faststream._internal.state.broker import OuterBrokerState
@@ -103,7 +104,7 @@ def _init_setupable_(  # noqa: PLW3201
 
             serializer = PydanticSerializer()
 
-        self._state = BasicApplicationState(
+        self._state: ApplicationState = BasicApplicationState(
             di_state=DIState(
                 use_fastdepends=True,
                 get_dependent=None,
@@ -171,6 +172,10 @@ def __init__(
         else:
             self.lifespan_context = fake_context
 
+    @property
+    def running(self) -> bool:
+        return self._state.running
+
     @abstractmethod
     def exit(self) -> None:
         """Stop application manually."""
@@ -278,7 +283,7 @@ async def _shutdown_logging(
             "FastStream app shut down gracefully.",
         )
 
-    # Setvice methods
+    # Service methods
 
     def _log(self, level: int, message: str) -> None:
         if self.logger is not None:
@@ -328,7 +333,3 @@ def after_shutdown(
             apply_types(to_async(func), context__=self.context)
         )
         return func
-
-    @property
-    def running(self) -> bool:
-        return self._state.running
diff --git a/faststream/_internal/fastapi/router.py b/faststream/_internal/fastapi/router.py
@@ -235,7 +235,7 @@ def wrapper(
                 response_model_exclude_defaults=response_model_exclude_defaults,
                 response_model_exclude_none=response_model_exclude_none,
                 provider_factory=self._get_dependencies_overides_provider,
-                state=self._state,
+                state=self._state.di_state,
             )
 
         return wrapper