Type "int | None" is not assignable to type "_ColumnExpressionArgument[Any] | _StarOrOne | None"

[IAseven]

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, Relationship, Session, SQLModel, func, select


class TeamPersonLink(SQLModel, table=True):
    team_id: int | None = Field(default=None, foreign_key="team.id", primary_key=True)
    person_id: int | None = Field(default=None, foreign_key="person.id", primary_key=True)

class Team(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    name: str
    people: list["Person"] = Relationship(back_populates="teams", link_model=TeamPersonLink)

class Person(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    name: str
    teams: list[Team] = Relationship(back_populates="people", link_model=TeamPersonLink)


def get_people_with_team_count(session: Session, page: int = 1, page_size: int = 10):
    stmt = (
        select(
            Person.id,
            Person.name,
            func.count(TeamPersonLink.team_id).label("team_count")
        )
        .select_from(Person)
        .outerjoin(TeamPersonLink, Person.id == TeamPersonLink.person_id)
        .group_by(Person.id)
        .limit(page_size)
        .offset((page - 1) * page_size)
    )

    results = session.exec(stmt).all()
    return results

Description

Argument of type "int | None" cannot be assigned to parameter "expression" of type "_ColumnExpressionArgument[Any] | _StarOrOne | None" in function "__init__"
  Type "int | None" is not assignable to type "_ColumnExpressionArgument[Any] | _StarOrOne | None"
    Type "int" is not assignable to type "_ColumnExpressionArgument[Any] | _StarOrOne | None"
      "int" is not assignable to "ColumnElement[Any]"
      "int" is incompatible with protocol "_HasClauseElement[Any]"
        "__clause_element__" is not present
      "int" is not assignable to "SQLCoreOperations[Any]"
      "int" is not assignable to "ExpressionElementRole[Any]"
      "int" is not assignable to "TypedColumnsClauseRole[Any]"
  ...Pylance[reportArgumentType](https://github.com/microsoft/pylance-release/blob/main/docs/diagnostics/reportArgumentType.md)

Argument of type "bool" cannot be assigned to parameter "onclause" of type "_OnClauseArgument | None" in function "outerjoin"
  Type "bool" is not assignable to type "_OnClauseArgument | None"
    "bool" is not assignable to "ColumnElement[Any]"
    "bool" is incompatible with protocol "_HasClauseElement[Any]"
      "__clause_element__" is not present
    "bool" is not assignable to "SQLCoreOperations[Any]"
    "bool" is not assignable to "ExpressionElementRole[Any]"
    "bool" is not assignable to "TypedColumnsClauseRole[Any]"
    Type "bool" is not assignable to type "() -> ColumnElement[Any]"
  ...Pylance[reportArgumentType](https://github.com/microsoft/pylance-release/blob/main/docs/diagnostics/reportArgumentType.md)
(class) TeamPersonLink

Argument of type "int | None" cannot be assigned to parameter "__first" of type "_ColumnExpressionOrStrLabelArgument[Any] | Literal[_NoArg.NO_ARG] | None" in function "group_by"
  Type "int | None" is not assignable to type "_ColumnExpressionOrStrLabelArgument[Any] | Literal[_NoArg.NO_ARG] | None"
    Type "int" is not assignable to type "_ColumnExpressionOrStrLabelArgument[Any] | Literal[_NoArg.NO_ARG] | None"
      "int" is not assignable to "None"
      "int" is not assignable to "str"
      "int" is not assignable to "ColumnElement[Any]"
      "int" is incompatible with protocol "_HasClauseElement[Any]"
        "__clause_element__" is not present
      "int" is not assignable to "SQLCoreOperations[Any]"
  ...Pylance[reportArgumentType](https://github.com/microsoft/pylance-release/blob/main/docs/diagnostics/reportArgumentType.md)

Operating System

Linux

Operating System Details

No response

SQLModel Version

0.0.24

Python Version

Python 3.12.10

Additional Context

[YuriiMotov]

You can solve this using col():

    stmt = (
        select(
            Person.id,
            Person.name,
            func.count(col(TeamPersonLink.team_id)).label("team_count")
        )
        .select_from(Person)
        .outerjoin(TeamPersonLink, col(Person.id) == TeamPersonLink.person_id)
        .group_by(col(Person.id))
        .limit(page_size)
        .offset((page - 1) * page_size)
    )