refactor(telemetry): derive feature counts from compile-time walks

Move TelemetryContext into reflex_base and drop the scattered
`increment_feature` call sites in Upload, lifespan, shared state,
storage classes, and Model. Compile-event collection now walks the
live app (states, pages, config, ModelRegistry) at compile end so the
counters reflect actual final state rather than import-time markers
that could double-count under reload or get out of sync.

Author: FarhanAliRaza

reflex/utils/telemetry_context.py …ase/src/reflex_base/telemetry_context.pyreflex/utils/telemetry_context.py renamed to packages/reflex-base/src/reflex_base/telemetry_context.py reflex/utils/telemetry_context.py renamed to packages/reflex-base/src/reflex_base/telemetry_context.py

@@ -1,4 +1,10 @@
-"""Compile-time telemetry context."""
+"""Compile-time telemetry context.
+
+Lives in ``reflex_base`` (not ``reflex``) so deep packages like
+``reflex_components_core`` — which depend on ``reflex_base`` but not
+``reflex`` — can read the active context without inverting the
+dependency hierarchy.
+"""
 
 from __future__ import annotations
 
@@ -31,33 +37,12 @@
 
 _KNOWN_FEATURES: tuple[FeatureName, ...] = get_args(FeatureName)
 
-# Counters bumped outside an active compile context (import-time class
-# definitions, decorators, registrations) accumulate here so they survive
-# into the next compile event.
-_recorded_features: dict[FeatureName, int] = {}
-
-
-def increment_feature(name: FeatureName, by: int = 1) -> None:
-    """Bump a feature invocation counter.
-
-    Writes to the active TelemetryContext if one is attached, else to the
-    process-level counter so import-time signals survive into the next compile.
-
-    Args:
-        name: The feature counter to bump.
-        by: How much to add. Defaults to 1.
-    """
-    target = TelemetryContext.get()
-    target_dict = target.features_used if target is not None else _recorded_features
-    target_dict[name] = target_dict.get(name, 0) + by
-
 
 @dataclasses.dataclass(frozen=True, kw_only=True, slots=True, eq=False)
 class TelemetryContext(BaseContext):
     """Per-compile telemetry handle attached to the current contextvar."""
 
     start_perf_counter: float = dataclasses.field(default_factory=time.perf_counter)
-    features_used: dict[FeatureName, int] = dataclasses.field(default_factory=dict)
     trigger: CompileTrigger | None = None
     exception: BaseException | None = dataclasses.field(default=None, repr=False)
 

packages/reflex-components-core/src/reflex_components_core/core/upload.py

@@ -38,7 +38,6 @@
 from reflex_base.vars.sequence import ArrayVar, LiteralStringVar
 from reflex_components_sonner.toast import toast
 
-from reflex.utils.telemetry_context import increment_feature
 from reflex_components_core.base.fragment import Fragment
 from reflex_components_core.core._upload import UploadChunkIterator, UploadFile
 from reflex_components_core.core.cond import cond
@@ -289,7 +288,6 @@ def create(cls, *children, **props) -> Component:
         """
         # Mark the Upload component as used in the app.
         cls.is_used = True
-        increment_feature("upload_count")
 
         props.setdefault("multiple", True)
 

pyi_hashes.json

@@ -20,7 +20,7 @@
   "packages/reflex-components-core/src/reflex_components_core/core/helmet.pyi": "7fd81a99bde5b0ff94bb52523597fd5c",
   "packages/reflex-components-core/src/reflex_components_core/core/html.pyi": "753d6ae315369530dad450ed643f5be6",
   "packages/reflex-components-core/src/reflex_components_core/core/sticky.pyi": "ba60a7d9cba75b27a1133bd63a9fbd59",
-  "packages/reflex-components-core/src/reflex_components_core/core/upload.pyi": "91d66d21b80fad4c0ebaa9c88274d2e2",
+  "packages/reflex-components-core/src/reflex_components_core/core/upload.pyi": "2dd6ba6e3a4d61fc1d79eb582a7cc548",
   "packages/reflex-components-core/src/reflex_components_core/core/window_events.pyi": "5e1dcb1130bc8af282783fae329ae6a6",
   "packages/reflex-components-core/src/reflex_components_core/datadisplay/__init__.pyi": "c96fed4da42a13576d64f84e3c7cb25c",
   "packages/reflex-components-core/src/reflex_components_core/el/__init__.pyi": "f09129ddefb57ab4c7769c86dc9a3153",

reflex/app.py

@@ -42,6 +42,7 @@
 from reflex_base.event.context import EventContext
 from reflex_base.event.processor import BaseStateEventProcessor, EventProcessor
 from reflex_base.registry import RegistrationContext
+from reflex_base.telemetry_context import CompileTrigger, TelemetryContext
 from reflex_base.utils import console
 from reflex_base.utils.imports import ImportVar
 from reflex_base.utils.types import ASGIApp, Message, Receive, Scope, Send
@@ -96,11 +97,6 @@
     should_prerender_routes,
 )
 from reflex.utils.misc import run_in_thread
-from reflex.utils.telemetry_context import (
-    CompileTrigger,
-    TelemetryContext,
-    increment_feature,
-)
 from reflex.utils.token_manager import RedisTokenManager, TokenManager
 
 if sys.version_info < (3, 13):
@@ -913,8 +909,6 @@ def add_page(
         state = self._state or State
         route_args = get_route_args(route)
         state.setup_dynamic_args(route_args)
-        if route_args:
-            increment_feature("dynamic_routes_count")
 
         self._load_events[route] = (
             (on_load if isinstance(on_load, list) else [on_load])

reflex/app_mixins/lifespan.py

@@ -15,8 +15,6 @@
 from reflex_base.utils.exceptions import InvalidLifespanTaskTypeError
 from starlette.applications import Starlette
 
-from reflex.utils.telemetry_context import increment_feature
-
 from .mixin import AppMixin
 
 if TYPE_CHECKING:
@@ -200,7 +198,4 @@ def register_lifespan_task(
             functools.update_wrapper(registered_task, task)
         self._lifespan_tasks[registered_task] = None
         console.debug(f"Registered lifespan task: {task_name}")
-        module = getattr(registered_task, "__module__", None) or ""
-        if module != "reflex" and not module.startswith("reflex."):
-            increment_feature("lifespan_tasks_count")
         return task

reflex/istate/shared.py

@@ -13,7 +13,6 @@
 
 from reflex.istate.manager.token import BaseStateToken
 from reflex.state import BaseState, State, _override_base_method
-from reflex.utils.telemetry_context import increment_feature
 
 UPDATE_OTHER_CLIENT_TASKS: set[asyncio.Task] = set()
 LINKED_STATE = TypeVar("LINKED_STATE", bound="SharedStateBaseInternal")
@@ -520,4 +519,3 @@ def __init_subclass__(cls, **kwargs):
             # pulls in all linked states and substates which may not actually be
             # accessed for this event.
             root_state._always_dirty_substates.add(SharedStateBaseInternal.get_name())
-        increment_feature("shared_state_count")

reflex/istate/storage.py

@@ -6,8 +6,6 @@
 
 from reflex_base.utils import format
 
-from reflex.utils.telemetry_context import increment_feature
-
 
 class ClientStorageBase:
     """Base class for client-side storage."""
@@ -75,7 +73,6 @@ def __new__(
         inst.domain = domain
         inst.secure = secure
         inst.same_site = same_site
-        increment_feature("cookie_count")
         return inst
 
 
@@ -112,7 +109,6 @@ def __new__(
             inst = super().__new__(cls, object)
         inst.name = name
         inst.sync = sync
-        increment_feature("local_storage_count")
         return inst
 
 
@@ -145,5 +141,4 @@ def __new__(
         else:
             inst = super().__new__(cls, object)
         inst.name = name
-        increment_feature("session_storage_count")
         return inst

reflex/model.py

@@ -13,8 +13,6 @@
 from reflex_base.utils import console
 from reflex_base.utils.serializers import serializer
 
-from reflex.utils.telemetry_context import increment_feature
-
 if TYPE_CHECKING:
     from typing import TypeVar
 
@@ -202,7 +200,6 @@ def register(cls, model: SQLModelOrSqlAlchemyT) -> SQLModelOrSqlAlchemyT:
                 The model passed in as an argument (Allows decorator usage)
             """
             cls.models.add(model)
-            increment_feature("db_model_count")
             return model
 
         @classmethod

reflex/utils/exec.py

@@ -20,13 +20,13 @@
 from reflex_base.config import get_config
 from reflex_base.constants.base import LogLevel
 from reflex_base.environment import environment
+from reflex_base.telemetry_context import CompileTrigger
 from reflex_base.utils import console
 from reflex_base.utils.decorator import once
 
 from reflex.utils import path_ops
 from reflex.utils.misc import get_module_path
 from reflex.utils.prerequisites import get_web_dir
-from reflex.utils.telemetry_context import CompileTrigger
 
 # For uvicorn windows bug fix (#2335)
 frontend_process = None

reflex/utils/prerequisites.py

@@ -31,9 +31,9 @@
 if typing.TYPE_CHECKING:
     from redis import Redis as RedisSync
     from redis.asyncio import Redis
+    from reflex_base.telemetry_context import CompileTrigger
 
     from reflex.app import App
-    from reflex.utils.telemetry_context import CompileTrigger
 
 
 class AppInfo(NamedTuple):