diff --git a/projects/pgai/pgai/sqlalchemy/__init__.py b/projects/pgai/pgai/sqlalchemy/__init__.py index 8eae6f1a4..573a5013a 100644 --- a/projects/pgai/pgai/sqlalchemy/__init__.py +++ b/projects/pgai/pgai/sqlalchemy/__init__.py @@ -180,6 +180,7 @@ def __get__( foreign_keys=[ getattr(self._embedding_class, col.name) for col in pk_cols ], + back_populates="parent", **self.relationship_args, ) setattr(owner, f"{self.name}_model", self._embedding_class)