"""ORM model for user feedback on runs.""" from __future__ import annotations from datetime import UTC, datetime from sqlalchemy import JSON, DateTime, String, Text, UniqueConstraint from sqlalchemy.orm import Mapped, mapped_column from deerflow.persistence.base import Base class FeedbackRow(Base): __tablename__ = "feedback" __table_args__ = (UniqueConstraint("thread_id", "run_id", "user_id", name="uq_feedback_thread_run_user"),) feedback_id: Mapped[str] = mapped_column(String(64), primary_key=True) run_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True) thread_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True) user_id: Mapped[str | None] = mapped_column(String(64), index=True) rating: Mapped[int] = mapped_column(nullable=False) # +1 (thumbs-up) or -1 (thumbs-down) comment: Mapped[str | None] = mapped_column(Text) # Optional text feedback from the user tags: Mapped[list[str] | None] = mapped_column(JSON, nullable=True) # Optional thumbs-down reason slugs from the feedback dialog # (language-neutral, e.g. ["incorrect", "slow"]) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))