From 0ab7f5e51d9a6babc39835a011061329883c6775 Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Mon, 8 Sep 2025 16:35:47 +0800
Subject: [PATCH] Add File

---
 .../alembic/versions/033_chat_origin_ddl.py   | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 backend/alembic/versions/033_chat_origin_ddl.py

diff --git a/backend/alembic/versions/033_chat_origin_ddl.py b/backend/alembic/versions/033_chat_origin_ddl.py
new file mode 100644
index 0000000..911835a
--- /dev/null
+++ b/backend/alembic/versions/033_chat_origin_ddl.py
@@ -0,0 +1,30 @@
+"""033_chat_origin_ddl
+
+Revision ID: 3cb5d6a54f2e
+Revises: 6549e47f9adc
+Create Date: 2025-07-22 22:00:48.599729
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = '3cb5d6a54f2e'
+down_revision = '6549e47f9adc'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('chat', sa.Column('origin', sa.Integer(), nullable=True, default=0))  # 0: default, 1: mcp, 2: assistant
+    op.execute('UPDATE chat SET origin = 0 WHERE origin IS NULL')
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('chat', 'origin')
+    # ### end Alembic commands ###