Spaces:
Runtime error
Runtime error
File size: 2,173 Bytes
b74c6ae 95412b7 b74c6ae 95412b7 b74c6ae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | """message_feedback: thumbs up/down on chat answers, plus messages.recall_json (how the answer was produced)
Revision ID: 0015_message_feedback
Revises: 0014_chat_citations
Create Date: 2026-10-02
"""
import sqlalchemy as sa
from alembic import op
revision = "0015_message_feedback"
down_revision = "0014_chat_citations"
branch_labels = None
depends_on = None
def upgrade() -> None:
# 0001 builds from the live models, so a fresh database may already have these.
insp = sa.inspect(op.get_bind())
if "recall_json" not in {c["name"] for c in insp.get_columns("messages")}:
with op.batch_alter_table("messages") as b:
b.add_column(sa.Column("recall_json", sa.JSON(), nullable=True))
if "message_feedback" in insp.get_table_names():
return
op.create_table(
"message_feedback",
sa.Column("id", sa.Uuid(), primary_key=True),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
sa.Column("workspace_id", sa.Uuid(), sa.ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False),
sa.Column("notebook_id", sa.Uuid(), sa.ForeignKey("notebooks.id", ondelete="SET NULL")),
sa.Column("chat_id", sa.Uuid(), sa.ForeignKey("chats.id", ondelete="SET NULL")),
sa.Column("message_id", sa.Uuid(), sa.ForeignKey("messages.id", ondelete="SET NULL")),
sa.Column("rating", sa.Integer(), nullable=False),
sa.Column("reasons", sa.JSON(), nullable=False),
sa.Column("comment", sa.Text()),
sa.Column("question", sa.Text(), nullable=False),
sa.Column("answer", sa.Text(), nullable=False),
sa.Column("sources", sa.JSON(), nullable=False),
sa.Column("recall", sa.JSON(), nullable=False),
)
op.create_index("ix_message_feedback_workspace_id", "message_feedback", ["workspace_id"])
op.create_index("ix_message_feedback_message_id", "message_feedback", ["message_id"], unique=True)
def downgrade() -> None:
op.drop_table("message_feedback")
with op.batch_alter_table("messages") as b:
b.drop_column("recall_json")
|