AttributeError thrown when validate_assignment=True

[Afstkla]

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 pydantic import ConfigDict
from sqlmodel import Field, SQLModel


class Hero(SQLModel, table=True):
    model_config = ConfigDict(validate_assignment=True)

    id: int = Field(primary_key=True)


h1 = Hero(id=1)  # error happens here

Description

I've just upgraded my code to SQLModel 0.0.14 (and Pydantic 2.5, SQLAlchemy 2.0.23). However, when I run the minimal example above, I immediately get the following error:

'Hero' object has no attribute '__pydantic_extra__'
  File "scratch.py", line 11, in <module>
    h1 = Hero(id=1)
         ^^^^^^^^^^
AttributeError: 'Hero' object has no attribute '__pydantic_extra__'

When I apply the patch here, it solves the issue in the example. However, in a longer example like below, specifically when getting models from the database, I still get the same __pydantic_extra__ error:

from pydantic import ConfigDict
from sqlmodel import Field, Session, SQLModel, create_engine, select


class Hero(SQLModel, table=True):
    model_config = ConfigDict(validate_assignment=True)
    id: int = Field(primary_key=True)
    name: str


engine = create_engine("sqlite+pysqlite:///:memory:")

SQLModel.metadata.create_all(engine)

h1 = Hero(name="a")
with Session(engine) as s:
    s.add(h1)
    s.commit()

with Session(engine) as s:
    hero = s.exec(select(Hero).where(Hero.name == "a")).first()

hero.name = "b" # error happens here

Does anyone have any ideas of what I could try, what I'm doing wrong, or where I should look to find the source and submit a PR?

Operating System

macOS

Operating System Details

No response

SQLModel Version

0.0.14

Python Version

3.11.6

Additional Context

Pydantic 2.5.3
SQLAlchemy 2.0.23

[YuriiMotov]

Works well in current version (0.0.24)