quick fix
This commit is contained in:
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user