Skip to content

Commit 281df9d

Browse files
committed
Add data collection campaign management table and update chat_message schema
1 parent 9142481 commit 281df9d

2 files changed

Lines changed: 109 additions & 0 deletions

File tree

.env.example

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
PG_USER=welearn_datastack
2+
PG_PASSWORD=...
3+
PG_HOST=...
4+
PG_PORT=5432
5+
PG_DB=welearn_datastack
6+
PG_DRIVER=postgresql+psycopg2
7+
PG_SCHEMA=document_related,corpus_related,user_related,agent_related
8+
LOG_LEVEL=INFO
9+
LOG_FORMAT=[%(asctime)s][%(name)s][%(levelname)s] - %(message)s
Lines changed: 100 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,100 @@
1+
"""data collection
2+
3+
Revision ID: 2ad4895b2674
4+
Revises: 068312e7800c
5+
Create Date: 2026-01-16 15:55:41.447852
6+
7+
"""
8+
9+
from typing import Sequence, Union
10+
11+
import sqlalchemy as sa
12+
from alembic import op
13+
from sqlalchemy.dialects import postgresql
14+
15+
# revision identifiers, used by Alembic.
16+
revision: str = "2ad4895b2674"
17+
down_revision: Union[str, None] = "068312e7800c"
18+
branch_labels: Union[str, Sequence[str], None] = None
19+
depends_on: Union[str, Sequence[str], None] = None
20+
21+
22+
def upgrade() -> None:
23+
op.create_table(
24+
"data_collection_campaign_management",
25+
sa.Column(
26+
"id", sa.Uuid(), server_default=sa.func.gen_random_uuid(), nullable=False
27+
),
28+
sa.Column("is_active", sa.Boolean(), nullable=False),
29+
sa.Column("end_at", postgresql.TIMESTAMP(), nullable=False),
30+
sa.Column(
31+
"created_at", postgresql.TIMESTAMP(), server_default="NOW()", nullable=False
32+
),
33+
sa.PrimaryKeyConstraint("id"),
34+
schema="user_related",
35+
)
36+
37+
op.add_column(
38+
"chat_message",
39+
sa.Column("role", sa.String(), nullable=False),
40+
schema="user_related",
41+
)
42+
op.add_column(
43+
"chat_message",
44+
sa.Column("inferred_user_id", sa.Uuid(), nullable=False),
45+
schema="user_related",
46+
)
47+
op.add_column(
48+
"chat_message",
49+
sa.Column("conversation_id", sa.Uuid(), nullable=False),
50+
schema="user_related",
51+
)
52+
op.drop_constraint(
53+
op.f("message_user_id_fkey"),
54+
"chat_message",
55+
schema="user_related",
56+
type_="foreignkey",
57+
)
58+
op.create_foreign_key(
59+
"message_inferred_user_id_fkey",
60+
"chat_message",
61+
"inferred_user",
62+
["inferred_user_id"],
63+
["id"],
64+
source_schema="user_related",
65+
referent_schema="user_related",
66+
)
67+
op.drop_column("chat_message", "user_id", schema="user_related")
68+
op.add_column(
69+
"returned_document",
70+
sa.Column("is_clicked", sa.Boolean(), nullable=False),
71+
schema="user_related",
72+
)
73+
74+
75+
def downgrade() -> None:
76+
op.drop_column("returned_document", "is_clicked", schema="user_related")
77+
op.add_column(
78+
"chat_message",
79+
sa.Column("user_id", sa.Uuid(), nullable=False),
80+
schema="user_related",
81+
)
82+
op.drop_constraint(
83+
"message_inferred_user_id_fkey",
84+
"chat_message",
85+
schema="user_related",
86+
type_="foreignkey",
87+
)
88+
op.create_foreign_key(
89+
op.f("message_user_id_fkey"),
90+
"chat_message",
91+
"user_profile",
92+
["user_id"],
93+
["id"],
94+
source_schema="user_related",
95+
referent_schema="user_related",
96+
)
97+
op.drop_column("chat_message", "conversation_id", schema="user_related")
98+
op.drop_column("chat_message", "inferred_user_id", schema="user_related")
99+
op.drop_column("chat_message", "role", schema="user_related")
100+
op.drop_table("data_collection_campaign_management", schema="user_related")

0 commit comments

Comments
 (0)