from .base import BaseModel
from . import db


class Workspace(BaseModel):
    __tablename__ = "workspaces"

    user_id = db.Column(
        db.BigInteger, db.ForeignKey("users.id"), nullable=False, comment="用户ID"
    )
    name = db.Column(db.String(100), nullable=False, comment="工作空间名称")
    description = db.Column(db.Text, nullable=True, comment="描述")
    is_default = db.Column(
        db.Boolean, default=False, nullable=False, comment="是否默认"
    )

    agents = db.relationship(
        "Agent", backref="workspace", lazy=True, cascade="all, delete-orphan"
    )
    conversations = db.relationship(
        "Conversation", backref="workspace", lazy=True, cascade="all, delete-orphan"
    )
    memories = db.relationship(
        "Memory", backref="workspace", lazy=True, cascade="all, delete-orphan"
    )