Type inheritance doesn't work in a simple select statement

[muratgozel]

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

import os
import datetime
from typing import Annotated
from fastapi import FastAPI, Depends
from pydantic import BaseModel
from sqlmodel import Field, Session, SQLModel, create_engine, select, JSON


class Config:
    pg_conn_str = os.environ.get("PG_CONN_STR") or "postgresql://localhost:5432/dev"
    debug = True if os.environ.get("DEBUG") else False


config = Config()


class Otp(SQLModel, table=True):
    rid: str = Field(primary_key=True, nullable=False)
    credential: str = Field(nullable=False)
    credential_type: str = Field(nullable=False)
    otp: int = Field()
    additional_data: JSON
    created_at: datetime.datetime = Field(default_factory=lambda: datetime.datetime.now(datetime.timezone.utc))
    updated_at: datetime.datetime = Field()
    removed_at: datetime.datetime = Field()


class OtpLog(SQLModel, table=True):
    rid: str = Field(primary_key=True, nullable=False)
    otp_rid: str = Field(foreign_key="otp.rid", ondelete="CASCADE", nullable=False)
    action: str = Field(nullable=False)
    created_at: datetime.datetime = Field(default_factory=lambda: datetime.datetime.now(datetime.timezone.utc))


engine = create_engine(config.pg_conn_str)


def get_session():
    with Session(engine) as session:
        yield session


SessionDep = Annotated[Session, Depends(get_session)]


app = FastAPI()


@app.get('/')
def home():
    return {'dev': '0.1.0'}


class OtpRequestBody(BaseModel):
    email: str


@app.post('/otp')
async def otp_request(body: OtpRequestBody, session: SessionDep):
    statement = select(Otp).where(Otp.credential == body.email, Otp.removed_at == None)
    otp = session.exec(statement).first()

    if otp is not None:
        return { 'error': { 'code': 'duplicate_request' } }

    return {}

Description

PyCharm shows the following type error for the session.exec(statement). Looks like the statement inherits SelectBase from sqlalchemy and not the Select from sqlmodel.

Operating System

macOS

Operating System Details

No response

SQLModel Version

0.0.24

Python Version

3.13.3

Additional Context

• I haven't changed anything in the editor.
• You can test it by yourself from the example code I provided.

[disonwei]

Encountered the same problem.