Typechecking on id field when querying

[travis-cook-sfdc]

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 sqlmodel import Field, Session, SQLModel, create_engine, select

class Hero(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    name: str


hero_1 = Hero(name="Foo")

engine = create_engine("sqlite://database.db")

SQLModel.metadata.create_all(engine)

with Session(engine) as session:
    session.add(hero_1)
    session.commit()

with Session(engine) as session:
    hero = session.exec(select(Hero)).one()
    x = dict[int, Hero] = {hero.id: hero}

Description

• Create a Hero Model
• Save is to the database
• Query it from the database
• Run mypy
• See error:

scratch.py:20: error: Dict entry 0 has incompatible type "int | None": "Hero"; expected "int": "Hero"  [dict-item

Is there anyway for sqlmodel / mypy to resolve the fact that id can't be optional here?

Operating System

macOS

Operating System Details

No response

SQLModel Version

0.0.8

Python Version

3.11

Additional Context

No response

[YuriiMotov]

You can add assert hero.id is not None before that line.

I don't think there are other methods. Only checking the value in runtime