|
@@ -10,6 +10,7 @@ from typing import (
|
|
|
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)
|