leocrm/app/plugins/builtins/ai_proactive/models.py

"""SQLAlchemy models for the AI Proactive plugin."""

from __future__ import annotations

import uuid
from datetime import datetime
from typing import Any

from sqlalchemy import (
    Boolean,
    DateTime,
    Float,
    ForeignKey,
    Index,
    Integer,
    String,
    Text,
)
from sqlalchemy.dialects.postgresql import JSONB, UUID as PGUUID
from sqlalchemy.orm import Mapped, mapped_column

from app.core.db import Base, TenantMixin


class ProactiveSuggestion(Base, TenantMixin):
    """A proactive AI suggestion generated from user context."""

    __tablename__ = "ai_proactive_suggestions"
    __table_args__ = (
        Index(
            "ix_suggestions_user",
            "tenant_id",
            "user_id",
            "is_dismissed",
            "created_at",
        ),
        Index("ix_suggestions_entity", "tenant_id", "entity_type", "entity_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    user_id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), nullable=False
    )
    entity_type: Mapped[str] = mapped_column(String(50), nullable=False)
    entity_id: Mapped[uuid.UUID | None] = mapped_column(PGUUID(as_uuid=True), nullable=True)
    suggestion_type: Mapped[str] = mapped_column(
        String(50), nullable=False, default="info"
    )
    title: Mapped[str] = mapped_column(String(200), nullable=False)
    content: Mapped[str] = mapped_column(Text, nullable=False)
    confidence: Mapped[float] = mapped_column(Float, nullable=False, default=0.5)
    actions: Mapped[list[Any]] = mapped_column(JSONB, nullable=False, default=list)
    is_dismissed: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
    is_acted_upon: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
    context_snapshot: Mapped[dict[str, Any]] = mapped_column(
        JSONB, nullable=False, default=dict
    )
    expires_at: Mapped[datetime | None] = mapped_column(
        DateTime(timezone=True), nullable=True
    )


class ContextLog(Base, TenantMixin):
    """Log of user context changes (page views, entity selections)."""

    __tablename__ = "ai_proactive_context_log"
    __table_args__ = (
        Index("ix_context_log_user_time", "tenant_id", "user_id", "created_at"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    user_id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), nullable=False
    )
    page: Mapped[str] = mapped_column(String(200), nullable=False)
    entity_type: Mapped[str | None] = mapped_column(String(50), nullable=True)
    entity_id: Mapped[uuid.UUID | None] = mapped_column(PGUUID(as_uuid=True), nullable=True)
    entity_data: Mapped[dict[str, Any]] = mapped_column(
        JSONB, nullable=False, default=dict
    )


class ProactiveSettings(Base, TenantMixin):
    """Per-user settings for proactive AI."""

    __tablename__ = "ai_proactive_settings"
    __table_args__ = (
        Index("ix_proactive_settings_user", "tenant_id", "user_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    user_id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), nullable=False
    )
    enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
    suggestion_categories: Mapped[list[Any]] = mapped_column(
        JSONB, nullable=False, default=lambda: ["mail", "tasks", "contacts", "companies", "insights"]
    )
    confidence_threshold: Mapped[float] = mapped_column(
        Float, nullable=False, default=0.5
    )
    rate_limit_seconds: Mapped[int] = mapped_column(
        Integer, nullable=False, default=10
    )
    model: Mapped[str] = mapped_column(String(100), nullable=False, default="ollama/deepseek-v4-flash")