avoid recursive calls for PASSTHROUGH memo components

Author: masenf

packages/reflex-base/src/reflex_base/components/component.py

@@ -1381,7 +1381,7 @@ def render(self) -> dict:
         self._cached_render_result = rendered_dict
         return rendered_dict
 
-    def _get_component_hash(self) -> str:
+    def _get_component_hash(self, shallow: bool = False) -> str:
         """Get a stable content hash for this component.
 
         The hash incorporates the rendered JSX dict plus the component's
@@ -1392,16 +1392,32 @@ def _get_component_hash(self) -> str:
         components differing only in ``on_mount``, which is excluded
         from ``_render`` props but lives in the lifecycle hook).
 
+        Args:
+            shallow: If True, only hash the component's own render output and
+                directly defined hooks, imports, custom code, and app-wrap
+                components, excluding any of those from child components.
+
         Returns:
             The hex digest content hash.
         """
         hasher = md5(usedforsecurity=False)
         _update_deterministic_hash(hasher, self.render())
-        _update_deterministic_hash(hasher, dict(self._get_all_imports()))
-        _update_deterministic_hash(hasher, dict(self._get_all_hooks_internal()))
-        _update_deterministic_hash(hasher, dict(self._get_all_hooks()))
-        _update_deterministic_hash(hasher, dict(self._get_all_custom_code()))
-        _update_deterministic_hash(hasher, dict(self._get_all_app_wrap_components()))
+        if shallow:
+            # For non-snapshot strategies, we only hash the component's own hooks, imports, custom code, and app-wrap components
+            _update_deterministic_hash(hasher, dict(self._get_imports()))
+            _update_deterministic_hash(hasher, dict(self._get_hooks_internal()))
+            _update_deterministic_hash(hasher, dict(self._get_added_hooks()))
+            _update_deterministic_hash(hasher, self._get_hooks())
+            _update_deterministic_hash(hasher, self._get_custom_code())
+            _update_deterministic_hash(hasher, dict(self._get_app_wrap_components()))
+        else:
+            _update_deterministic_hash(hasher, dict(self._get_all_imports()))
+            _update_deterministic_hash(hasher, dict(self._get_all_hooks_internal()))
+            _update_deterministic_hash(hasher, dict(self._get_all_hooks()))
+            _update_deterministic_hash(hasher, dict(self._get_all_custom_code()))
+            _update_deterministic_hash(
+                hasher, dict(self._get_all_app_wrap_components())
+            )
         return hasher.hexdigest()
 
     def _compute_memo_tag(self) -> str:
@@ -1418,8 +1434,16 @@ def _compute_memo_tag(self) -> str:
         Returns:
             The stable tag name.
         """
+        from reflex_base.components.memoize_helpers import (
+            MemoizationStrategy,
+            get_memoization_strategy,
+        )
+
+        comp_hash = self._get_component_hash(
+            shallow=get_memoization_strategy(self) == MemoizationStrategy.PASSTHROUGH
+        )
         return format.format_state_name(
-            f"{type(self).__qualname__}_{self.tag or 'Comp'}_{self._get_component_hash()}"
+            f"{type(self).__qualname__}_{self.tag or 'Comp'}_{comp_hash}"
         ).capitalize()
 
     def _replace_prop_names(self, rendered_dict: dict) -> None: