299 lines
10 KiB
Python
299 lines
10 KiB
Python
|
"""add tenant-scoped expense application memory
|
||
|
|
|
||
|
|
Revision ID: 20260714_0005
|
||
|
|
Revises: 20260714_0004
|
||
|
|
Create Date: 2026-07-14 17:10:00
|
||
|
|
"""
|
||
|
|
|
||
|
|
from collections.abc import Sequence
|
||
|
|
|
||
|
|
import sqlalchemy as sa
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
|
||
|
|
revision: str = "20260714_0005"
|
||
|
|
down_revision: str | None = "20260714_0004"
|
||
|
|
branch_labels: str | Sequence[str] | None = None
|
||
|
|
depends_on: str | Sequence[str] | None = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
op.create_unique_constraint(
|
||
|
|
"uq_ai_decision_feedback_tenant_id",
|
||
|
|
"ai_decision_feedback",
|
||
|
|
["tenant_id", "id"],
|
||
|
|
)
|
||
|
|
op.create_unique_constraint(
|
||
|
|
"uq_workflow_outcomes_tenant_id",
|
||
|
|
"workflow_outcomes",
|
||
|
|
["tenant_id", "id"],
|
||
|
|
)
|
||
|
|
|
||
|
|
op.create_table(
|
||
|
|
"memory_entries",
|
||
|
|
sa.Column("id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("tenant_id", sa.String(length=64), nullable=False),
|
||
|
|
sa.Column(
|
||
|
|
"scope_type",
|
||
|
|
sa.String(length=20),
|
||
|
|
server_default="user",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column("scope_id", sa.String(length=120), nullable=False),
|
||
|
|
sa.Column(
|
||
|
|
"scene",
|
||
|
|
sa.String(length=50),
|
||
|
|
server_default="travel_application",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column(
|
||
|
|
"field_key",
|
||
|
|
sa.String(length=60),
|
||
|
|
server_default="transport_mode",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column("generation", sa.Integer(), server_default="1", nullable=False),
|
||
|
|
sa.Column("value_json", sa.JSON(), nullable=False),
|
||
|
|
sa.Column("value_fingerprint", sa.String(length=80), nullable=False),
|
||
|
|
sa.Column(
|
||
|
|
"status",
|
||
|
|
sa.String(length=20),
|
||
|
|
server_default="candidate",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column("evidence_count", sa.Integer(), server_default="0", nullable=False),
|
||
|
|
sa.Column(
|
||
|
|
"approved_evidence_count",
|
||
|
|
sa.Integer(),
|
||
|
|
server_default="0",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column(
|
||
|
|
"confidence",
|
||
|
|
sa.Numeric(precision=5, scale=4),
|
||
|
|
server_default="0",
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column(
|
||
|
|
"last_evidence_at",
|
||
|
|
sa.DateTime(timezone=True),
|
||
|
|
server_default=sa.func.now(),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column("candidate_expires_at", sa.DateTime(timezone=True), nullable=False),
|
||
|
|
sa.Column("activated_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
sa.Column("active_expires_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
sa.Column("suppressed_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
sa.Column("expired_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
sa.Column("revoked_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
sa.Column("revoked_reason", sa.String(length=255), nullable=True),
|
||
|
|
sa.Column("superseded_by_id", sa.String(length=36), nullable=True),
|
||
|
|
sa.Column(
|
||
|
|
"created_at",
|
||
|
|
sa.DateTime(timezone=True),
|
||
|
|
server_default=sa.func.now(),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.Column(
|
||
|
|
"updated_at",
|
||
|
|
sa.DateTime(timezone=True),
|
||
|
|
server_default=sa.func.now(),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"scope_type = 'user'",
|
||
|
|
name="ck_memory_entries_scope_type",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"scene = 'travel_application'",
|
||
|
|
name="ck_memory_entries_scene",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"field_key = 'transport_mode'",
|
||
|
|
name="ck_memory_entries_field_key",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"status IN ('candidate', 'active', 'suppressed', 'expired', 'revoked')",
|
||
|
|
name="ck_memory_entries_status",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"generation >= 1",
|
||
|
|
name="ck_memory_entries_generation",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"evidence_count >= 0 AND approved_evidence_count >= 0 "
|
||
|
|
"AND approved_evidence_count <= evidence_count",
|
||
|
|
name="ck_memory_entries_evidence_counts",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"confidence >= 0 AND confidence <= 1",
|
||
|
|
name="ck_memory_entries_confidence",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"candidate_expires_at > created_at",
|
||
|
|
name="ck_memory_entries_candidate_expiry",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"active_expires_at IS NULL OR ("
|
||
|
|
"activated_at IS NOT NULL AND active_expires_at > activated_at)",
|
||
|
|
name="ck_memory_entries_active_expiry",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"status != 'active' OR (activated_at IS NOT NULL AND active_expires_at IS NOT NULL)",
|
||
|
|
name="ck_memory_entries_active_fields",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"status != 'suppressed' OR suppressed_at IS NOT NULL",
|
||
|
|
name="ck_memory_entries_suppressed_fields",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"status != 'expired' OR expired_at IS NOT NULL",
|
||
|
|
name="ck_memory_entries_expired_fields",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"status != 'revoked' OR (revoked_at IS NOT NULL AND length(revoked_reason) > 0)",
|
||
|
|
name="ck_memory_entries_revoked_fields",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"superseded_by_id IS NULL OR status IN ('suppressed', 'revoked')",
|
||
|
|
name="ck_memory_entries_superseded_status",
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
"superseded_by_id IS NULL OR superseded_by_id != id",
|
||
|
|
name="ck_memory_entries_not_self_superseded",
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "superseded_by_id"],
|
||
|
|
["memory_entries.tenant_id", "memory_entries.id"],
|
||
|
|
name="fk_memory_entries_tenant_superseded_by",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
sa.UniqueConstraint(
|
||
|
|
"tenant_id",
|
||
|
|
"id",
|
||
|
|
name="uq_memory_entries_tenant_id",
|
||
|
|
),
|
||
|
|
sa.UniqueConstraint(
|
||
|
|
"tenant_id",
|
||
|
|
"scope_type",
|
||
|
|
"scope_id",
|
||
|
|
"scene",
|
||
|
|
"field_key",
|
||
|
|
"generation",
|
||
|
|
name="uq_memory_entries_generation",
|
||
|
|
),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
"ix_memory_entries_scope_lookup",
|
||
|
|
"memory_entries",
|
||
|
|
["tenant_id", "scope_type", "scope_id", "scene", "field_key", "status"],
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
"ix_memory_entries_status_expiry",
|
||
|
|
"memory_entries",
|
||
|
|
["tenant_id", "status", "candidate_expires_at", "active_expires_at"],
|
||
|
|
)
|
||
|
|
|
||
|
|
op.create_table(
|
||
|
|
"memory_evidence_links",
|
||
|
|
sa.Column("id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("tenant_id", sa.String(length=64), nullable=False),
|
||
|
|
sa.Column("memory_entry_id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("expense_case_id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("decision_id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("feedback_id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column("outcome_id", sa.String(length=36), nullable=False),
|
||
|
|
sa.Column(
|
||
|
|
"created_at",
|
||
|
|
sa.DateTime(timezone=True),
|
||
|
|
server_default=sa.func.now(),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "memory_entry_id"],
|
||
|
|
["memory_entries.tenant_id", "memory_entries.id"],
|
||
|
|
name="fk_memory_evidence_links_tenant_entry",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "expense_case_id"],
|
||
|
|
["expense_cases.tenant_id", "expense_cases.id"],
|
||
|
|
name="fk_memory_evidence_links_tenant_case",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "decision_id"],
|
||
|
|
["ai_decisions.tenant_id", "ai_decisions.id"],
|
||
|
|
name="fk_memory_evidence_links_tenant_decision",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "feedback_id"],
|
||
|
|
["ai_decision_feedback.tenant_id", "ai_decision_feedback.id"],
|
||
|
|
name="fk_memory_evidence_links_tenant_feedback",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["tenant_id", "outcome_id"],
|
||
|
|
["workflow_outcomes.tenant_id", "workflow_outcomes.id"],
|
||
|
|
name="fk_memory_evidence_links_tenant_outcome",
|
||
|
|
ondelete="RESTRICT",
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
sa.UniqueConstraint(
|
||
|
|
"tenant_id",
|
||
|
|
"id",
|
||
|
|
name="uq_memory_evidence_links_tenant_id",
|
||
|
|
),
|
||
|
|
sa.UniqueConstraint(
|
||
|
|
"tenant_id",
|
||
|
|
"memory_entry_id",
|
||
|
|
"expense_case_id",
|
||
|
|
name="uq_memory_evidence_links_entry_case",
|
||
|
|
),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
"ix_memory_evidence_links_entry_time",
|
||
|
|
"memory_evidence_links",
|
||
|
|
["tenant_id", "memory_entry_id", "created_at"],
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
"ix_memory_evidence_links_sources",
|
||
|
|
"memory_evidence_links",
|
||
|
|
["tenant_id", "decision_id", "feedback_id", "outcome_id"],
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
op.drop_index(
|
||
|
|
"ix_memory_evidence_links_sources",
|
||
|
|
table_name="memory_evidence_links",
|
||
|
|
)
|
||
|
|
op.drop_index(
|
||
|
|
"ix_memory_evidence_links_entry_time",
|
||
|
|
table_name="memory_evidence_links",
|
||
|
|
)
|
||
|
|
op.drop_table("memory_evidence_links")
|
||
|
|
|
||
|
|
op.drop_index(
|
||
|
|
"ix_memory_entries_status_expiry",
|
||
|
|
table_name="memory_entries",
|
||
|
|
)
|
||
|
|
op.drop_index(
|
||
|
|
"ix_memory_entries_scope_lookup",
|
||
|
|
table_name="memory_entries",
|
||
|
|
)
|
||
|
|
op.drop_table("memory_entries")
|
||
|
|
|
||
|
|
op.drop_constraint(
|
||
|
|
"uq_workflow_outcomes_tenant_id",
|
||
|
|
"workflow_outcomes",
|
||
|
|
type_="unique",
|
||
|
|
)
|
||
|
|
op.drop_constraint(
|
||
|
|
"uq_ai_decision_feedback_tenant_id",
|
||
|
|
"ai_decision_feedback",
|
||
|
|
type_="unique",
|
||
|
|
)
|