diff --git a/alembic/versions/001_initial_schema.py b/alembic/versions/001_initial_schema.py index c85d1b4..2b88480 100644 --- a/alembic/versions/001_initial_schema.py +++ b/alembic/versions/001_initial_schema.py @@ -73,7 +73,7 @@ def upgrade() -> None: sa.Column("id", sa.BigInteger, primary_key=True), sa.Column("source_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), sa.Column("target_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), - sa.Column("lastfm_match", sa.Real), + sa.Column("lastfm_match", sa.REAL), sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()), sa.UniqueConstraint("source_track_id", "target_track_id", name="uq_similarity_link"), ) @@ -105,7 +105,7 @@ def upgrade() -> None: sa.Column("track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), sa.Column("position", sa.Integer, nullable=False), sa.Column("is_known", sa.Boolean, nullable=False), - sa.Column("similarity_score", sa.Real), + sa.Column("similarity_score", sa.REAL), ) diff --git a/src/haunt_fm/models/track.py b/src/haunt_fm/models/track.py index bcdedda..5789cec 100644 --- a/src/haunt_fm/models/track.py +++ b/src/haunt_fm/models/track.py @@ -1,7 +1,7 @@ from datetime import datetime from pgvector.sqlalchemy import Vector -from sqlalchemy import BigInteger, DateTime, Index, Integer, Real, Text, func +from sqlalchemy import REAL, BigInteger, DateTime, Index, Integer, Text, func from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.orm import Mapped, mapped_column, relationship @@ -69,7 +69,7 @@ class SimilarityLink(Base): id: Mapped[int] = mapped_column(BigInteger, primary_key=True) source_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) target_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) - lastfm_match: Mapped[float | None] = mapped_column(Real) + lastfm_match: Mapped[float | None] = mapped_column(REAL) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) __table_args__ = ( @@ -107,7 +107,7 @@ class PlaylistTrack(Base): track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) position: Mapped[int] = mapped_column(Integer, nullable=False) is_known: Mapped[bool] = mapped_column(nullable=False) - similarity_score: Mapped[float | None] = mapped_column(Real) + similarity_score: Mapped[float | None] = mapped_column(REAL) playlist: Mapped[Playlist] = relationship(back_populates="tracks") track: Mapped[Track] = relationship()