models.py

from sqlalchemy import Column, String, Text, ForeignKey, JSON, Boolean, BLOB, DateTime, TIMESTAMP
from sqlalchemy.ext.declarative import declarative_base
from datetime import datetime, timezone

Base = declarative_base()

# Define the models
class Account(Base):
    __tablename__ = "accounts"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    name = Column(String)
    username = Column(String)
    email = Column(String, nullable=False)
    avatarUrl = Column(String)
    details = Column(JSON, default={}, nullable=False)

class Memory(Base):
    __tablename__ = "memories"

    id = Column(String, primary_key=True)
    type = Column(String, nullable=False)
    createdAt = Column(TIMESTAMP, default=datetime.now(timezone.utc))
    content = Column(Text, nullable=False)
    embedding = Column(BLOB, nullable=False)  # Assuming embedding is stored as a string representation of a blob
    userId = Column(String, ForeignKey("accounts.id"))
    roomId = Column(String)
    agentId = Column(String, ForeignKey("accounts.id"))
    unique = Column(Boolean, default=True, nullable=False)

class Goal(Base):
    __tablename__ = "goals"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    userId = Column(String, ForeignKey("accounts.id"))
    name = Column(String)
    status = Column(String)
    description = Column(String)
    roomId = Column(String)
    objectives = Column(JSON, default=[], nullable=False)

class Log(Base):
    __tablename__ = "logs"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    userId = Column(String, nullable=False)
    body = Column(Text, nullable=False)
    type = Column(String, nullable=False)
    roomId = Column(String, nullable=False)

class Participant(Base):
    __tablename__ = "participants"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    userId = Column(String, ForeignKey("accounts.id"))
    roomId = Column(String)
    userState = Column(String)
    last_message_read = Column(String)

class Relationship(Base):
    __tablename__ = "relationships"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    userA = Column(String, ForeignKey("accounts.id"), nullable=False)
    userB = Column(String, ForeignKey("accounts.id"), nullable=False)
    status = Column(String, nullable=False)
    userId = Column(String, ForeignKey("accounts.id"), nullable=False)

class Room(Base):
    __tablename__ = "rooms"

    id = Column(String, primary_key=True)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))

class Cache(Base):
    __tablename__ = "cache"

    key = Column(String, primary_key=True)
    agentId = Column(String, primary_key=True)
    value = Column(JSON, default={}, nullable=False)
    createdAt = Column(DateTime, default=datetime.now(timezone.utc))
    expiresAt = Column(DateTime)