SQLModel unknown type as prop on BaseModel

[EvanSchalton]

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 SQLModel
from pydantic import BaseModel

class Table(SQLModel, table=True):
    field1: str
    field2: str

class Model(BaseModel):
    data: Table

test = Model(data=Table(field1="test", field2="test"))

a = test.data.field1
# Type of "data" is unknown Pylance(reportUnknownMemberType)
# Type of "field1" is unknown Pylance(reportUnknownMemberType)
# Link: https://github.com/microsoft/pyright/blob/main/docs/configuration.md#reportUnknownMemberType
# (variable) test: Model

Description

1. Create a SQLModel
2. Create a BaseModel that references the SQLModel
3. MyPy can't resolve the SQLModel type on an instance of the BaseModel

Operating System

Linux

Operating System Details

No response

SQLModel Version

0.0.16

Python Version

Python 3.11.8

Additional Context

[aistellar]

I'm having the same issue. It breaks all type hints related to SQLModel classes in VS Code.

[simonmoesorensen]

This might also be relevant here. I experience the same issues when dealing with relationships and methods in a SQLModel.

class Team(SQLModel):
    id: Optional[int] = Field(default=None, primary_key=True)
    name: str
    heroes: List["Hero"] = Relationship(back_populates="team")

    def team_power(self) -> str:
        if self.heroes:
            powers: List[str] = []
            for hero in self.heroes:
                powers.append(hero.power)  # This works?
        else:
            powers = ["No powers"]

        return f"Team has powers: {powers}"


class Hero(SQLModel):
    id: Optional[int] = Field(default=None, primary_key=True)
    team_id: Optional[int] = Field(default=None, foreign_key="team.id")
    team: Team = Relationship(back_populates="heroes")
    power: str

    def hero_power(self) -> str:
        if self.team is not None:  # Type of "team" is unknown
            team = self.team.name  # Type of "team" is unknown
        else:
            team = "No team"
        return f"Hero is in {team} and has {self.power}!"

The strange thing is that it works for hero.power in the Team class but not the self.team.name in the Hero class