Partitioned tables with composite primary keys and composite foreign keys compatible with Relationship()?

[perearnantgx]

First Check

• I added a very descriptive title here.
• I used the GitHub search to find a similar question and didn't find it.
• I searched the SQLModel documentation, with the integrated search.
• I already searched in Google "How to X in SQLModel" and didn't find any information.
• I already read and followed all the tutorial in the docs and didn't find an answer.
• I already checked if it is not related to SQLModel but to Pydantic.
• I already checked if it is not related to SQLModel but to SQLAlchemy.

Commit to Help

• I commit to help with one of those options 👆

Example Code

from typing import Optional, List
from sqlmodel import Field, Relationship, SQLModel,
from sqlalchemy import PrimaryKeyConstraint, ForeignKeyConstraint


class Parent(SQLModel, table=True):
    __table_args__ =(
        PrimaryKeyConstraint("id", "version_id"),       
     {
         "postgresql_partition_by": 'LIST (version_id)'
    }
    )
    id: str = Field(index=True)
    version_id: int = Field(index=True)
    children: Optional[List["Child"]]= Relationship(back_populates="parent")

class Child(SQLModel, table=True):
    __table_args__ = (
        PrimaryKeyConstraint("id", "version_id"),
        ForeignKeyConstraint(
            ['parent_id', 'version_id'],
            ['parent.id', 'parent.version_id'],
        ),
        {
         "postgresql_partition_by": 'LIST (version_id)',
        }     
    )
    id: Optional[str] = Field(index=True)
    version_id: Optional[int] = Field(index=True)
    parent_id: Optional[str] = Field(index=True, default=None)
    parent: Optional[Parent] = Relationship(back_populates="children")

Description

I have two tables in a postgreSQL database: Parent and Child. Both of them are partitioned with the column version_id(int).
The description of the PKs and FKs can be read from the Example Code.

When I try to set a Relationship in SQLModel between the two tables I got this error message:
sqlalchemy.exc.NoForeignKeysError: Could not determine join condition between parent/child tables on relationship Master_Partition.hotels - there are no foreign keys linking these tables. Ensure that referencing columns are associated with a ForeignKey or ForeignKeyConstraint, or specify a 'primaryjoin' expression.

I tried using the arguments primaryjoin and foreign_keys in Relationship() but they seem not to be compatible.
I tried using the arguments primaryjoin and foreign_keys in sqlalchemy relationship(), but then I have problems with the Class declaration.

Is there a way in SQLModel to do this using Relationship?

Operating System

Windows

Operating System Details

No response

SQLModel Version

0.0.16

Python Version

3.12.2

Additional Context

This issue is very similar #222
but it does not seem to work for me