diff --git a/reflex/middleware/hydrate_middleware.py b/reflex/middleware/hydrate_middleware.py index 062ff33b3..3992919a2 100644 --- a/reflex/middleware/hydrate_middleware.py +++ b/reflex/middleware/hydrate_middleware.py @@ -13,9 +13,6 @@ if TYPE_CHECKING: from reflex.app import App -State.add_var(constants.CompileVars.IS_HYDRATED, type_=bool, default_value=False) - - class HydrateMiddleware(Middleware): """Middleware to handle initial app hydration.""" @@ -57,7 +54,6 @@ class HydrateMiddleware(Middleware): # Get the route for on_load events. route = event.router_data.get(constants.RouteVar.PATH, "") - # Add the on_load events and set is_hydrated to True. events = [*app.get_load_events(route), type(state).set_is_hydrated(True)] # type: ignore events = fix_events(events, event.token, router_data=event.router_data) diff --git a/reflex/state.py b/reflex/state.py index 99d907d3d..11bb5e9a4 100644 --- a/reflex/state.py +++ b/reflex/state.py @@ -194,6 +194,9 @@ class State(Base, ABC, extra=pydantic.Extra.allow): # The router data for the current page router: RouterData = RouterData() + # The hydrated bool. + is_hydrated: bool = False + def __init__(self, *args, parent_state: State | None = None, **kwargs): """Initialize the state.