一对一关联对象DB自增ID存储遇NotNullViolationError问题求助

存在Playlist和PlaylistEmbedding两个SQLModel实体类，二者为一对一关联关系，主键ID均由数据库自动生成，代码定义如下：

class Playlist(SQLModel, table=True):
    playlist_id: int | None = Field(default=None, primary_key=True)
    embedding_id: int = Field(foreign_key="playlistembedding.embedding_id")

class PlaylistEmbedding(SQLModel, table=True):
    embedding_id: int | None = Field(default=None, primary_key=True)
    playlist_id: int = Field(foreign_key="playlist.playlist_id")
    embedding: list[float] = Field(sa_column=Column(Vector()))

尝试按以下逻辑存储对象：

async with get_async_session() as session:
    session.add(playlist)
    await session.flush()
    assert playlist and playlist.playlist_id

    playlist.embedding_id = await get_embedding_for_playlist(playlist)
    await session.merge(playlist)

    await session.commit()

但持续触发报错：

sqlalchemy.exc.IntegrityError: (sqlalchemy.dialects.postgresql.asyncpg.IntegrityError) <class 'asyncpg.exceptions.NotNullViolationError'>: null value in column "embedding_id" of relation "playlist" violates not-null constraint

1. 如何正确存储这两个采用数据库自增ID的一对一关联对象？
2. 如何实现「先保存Playlist并获取其DB生成ID→保存关联该ID的PlaylistEmbedding→更新Playlist关联嵌入ID」的完整流程？

已尝试将flush替换为commit，问题未解决。

问题1：修正实体类，解除循环依赖 #

当前代码的核心问题是两个表互相设置非空外键，形成循环依赖：插入Playlist需要已存在的embedding_id，插入PlaylistEmbedding又需要已存在的playlist_id，导致无法完成任何一方的初始插入。

修改方式：将Playlist的embedding_id设为可空类型，适配先创建Playlist的流程：

class Playlist(SQLModel, table=True):
    playlist_id: int | None = Field(default=None, primary_key=True)
    # 设为可选类型，允许初始插入时为null
    embedding_id: int | None = Field(default=None, foreign_key="playlistembedding.embedding_id")

class PlaylistEmbedding(SQLModel, table=True):
    embedding_id: int | None = Field(default=None, primary_key=True)
    playlist_id: int = Field(foreign_key="playlist.playlist_id")
    embedding: list[float] = Field(sa_column=Column(Vector()))

问题2：实现完整存储流程 #

调整后的实体类支持按需求顺序存储，代码如下：

async with get_async_session() as session:
    # 1. 插入Playlist并获取自增ID
    session.add(playlist)
    await session.flush()  # flush后数据库会为playlist_id赋值
    assert playlist.playlist_id is not None

    # 2. 生成嵌入向量并创建关联的PlaylistEmbedding
    embedding_vector = await get_embedding_for_playlist(playlist)
    playlist_embedding = PlaylistEmbedding(
        playlist_id=playlist.playlist_id,
        embedding=embedding_vector
    )
    session.add(playlist_embedding)
    await session.flush()  # 获取生成的embedding_id

    # 3. 更新Playlist的embedding_id并提交事务
    playlist.embedding_id = playlist_embedding.embedding_id
    await session.commit()  # 一次性提交所有变更

关键说明： #

• 用flush获取自增ID：flush同步变更到数据库但不提交事务，既能拿到生成的ID，又能在后续步骤出错时回滚所有操作。
• 无需调用merge：playlist已被当前session跟踪，直接修改属性后，commit时会自动同步到数据库。
• 外键可空是前提：解除循环依赖后，才能按顺序完成两个对象的插入与关联。

内容的提问来源于stack exchange，提问作者user22638408