quick fix

This commit is contained in:
2026-01-12 20:30:59 +01:00
parent bf01724b3e
commit 743bed67f3
16 changed files with 146 additions and 102 deletions

View File

@@ -3,10 +3,19 @@
from datetime import datetime
from typing import TYPE_CHECKING
from sqlalchemy import BigInteger, Boolean, Float, ForeignKey, String, Text, UniqueConstraint
from sqlalchemy import (
BigInteger,
Boolean,
DateTime,
Float,
ForeignKey,
String,
Text,
UniqueConstraint,
)
from sqlalchemy.orm import Mapped, mapped_column, relationship
from .base import Base
from .base import Base, utc_now
if TYPE_CHECKING:
from .conversation import Conversation, Message
@@ -24,8 +33,8 @@ class User(Base):
discord_username: Mapped[str] = mapped_column(String(255))
discord_display_name: Mapped[str | None] = mapped_column(String(255))
custom_name: Mapped[str | None] = mapped_column(String(255))
first_seen_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
last_seen_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
first_seen_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
last_seen_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
# Relationships
@@ -88,8 +97,10 @@ class UserFact(Base):
confidence: Mapped[float] = mapped_column(Float, default=1.0)
source: Mapped[str] = mapped_column(String(50), default="conversation")
is_active: Mapped[bool] = mapped_column(Boolean, default=True, index=True)
learned_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
last_referenced_at: Mapped[datetime | None] = mapped_column(default=None)
learned_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
last_referenced_at: Mapped[datetime | None] = mapped_column(
DateTime(timezone=True), default=None
)
# Relationships
user: Mapped["User"] = relationship(back_populates="facts")