refactor: create per-memo component subclasses with tag set at class level

Replace instance-level self.tag assignment with cached dynamically
created ExperimentalMemoComponent subclasses via
_get_experimental_memo_component_class, so the tag is a class-level
attribute rather than set in _post_init.

Author: FarhanAliRaza

pyi_hashes.json

@@ -119,5 +119,5 @@
   "reflex/components/recharts/polar.pyi": "ea4743e8903365ba95bc4b653c47cc4a",
   "reflex/components/recharts/recharts.pyi": "b3d93d085d51053bbb8f65326f34a299",
   "reflex/components/sonner/toast.pyi": "636050fcc919f8ab0903c30dceaa18f1",
-  "reflex/experimental/memo.pyi": "50be0b7fe796412f60f3bd1ead3f829a"
+  "reflex/experimental/memo.pyi": "78b1968972194785f72eab32476bc61d"
 }

reflex/experimental/memo.py

@@ -5,7 +5,7 @@
 import dataclasses
 import inspect
 from collections.abc import Callable
-from functools import update_wrapper
+from functools import cache, update_wrapper
 from typing import Any, get_args, get_origin, get_type_hints
 
 from reflex import constants
@@ -103,8 +103,6 @@ def _post_init(self, **kwargs):
 
         super()._post_init(**kwargs)
 
-        self.tag = definition.export_name
-
         props: dict[str, Any] = {}
         for key, value in {**declared_props, **rest_props}.items():
             camel_cased_key = format.to_camel_case(key)
@@ -116,6 +114,28 @@ def _post_init(self, **kwargs):
         object.__setattr__(self, "get_props", lambda: prop_names)
 
 
+@cache
+def _get_experimental_memo_component_class(
+    export_name: str,
+) -> type[ExperimentalMemoComponent]:
+    """Get the component subclass for an experimental memo export.
+
+    Args:
+        export_name: The exported React component name.
+
+    Returns:
+        A cached component subclass with the tag set at class definition time.
+    """
+    return type(
+        f"ExperimentalMemoComponent_{export_name}",
+        (ExperimentalMemoComponent,),
+        {
+            "__module__": __name__,
+            "tag": export_name,
+        },
+    )
+
+
 EXPERIMENTAL_MEMOS: dict[str, ExperimentalMemoDefinition] = {}
 
 
@@ -877,7 +897,7 @@ def __call__(self, *children: Any, **props: Any) -> ExperimentalMemoComponent:
             raise TypeError(msg)
 
         # Build the component props passed into the memo wrapper.
-        return ExperimentalMemoComponent._create(
+        return _get_experimental_memo_component_class(definition.export_name)._create(
             children=list(children),
             memo_definition=definition,
             **explicit_values,

tests/units/experimental/test_memo.py

@@ -82,10 +82,14 @@ def my_card(
         foo="extra",
         class_name="extra",
     )
+    component_again = my_card(title="World")
 
     assert isinstance(component, ExperimentalMemoComponent)
     assert len(component.children) == 2
     assert component.get_props() == ("title", "foo")
+    assert type(component) is type(component_again)
+    assert type(component).tag == "MyCard"
+    assert type(component).get_fields()["tag"].default == "MyCard"
 
     rendered = component.render()
     assert rendered["name"] == "MyCard"