diff --git a/reflex/utils/types.py b/reflex/utils/types.py index abed498eeb9..4187d44345f 100644 --- a/reflex/utils/types.py +++ b/reflex/utils/types.py @@ -10,6 +10,7 @@ Any, Callable, Iterable, + List, Literal, Optional, Type, @@ -164,7 +165,11 @@ def get_attribute_access_type(cls: GenericType, name: str) -> GenericType | None prop = descriptor.property if not isinstance(prop, Relationship): return None - return prop.mapper.class_ + class_ = prop.mapper.class_ + if prop.uselist: + return List[class_] + else: + return class_ elif isinstance(cls, type) and issubclass(cls, Model): # Check in the annotations directly (for sqlmodel.Relationship) hints = get_type_hints(cls)