add fields on end user

api/migrations/versions/2025_03_08_1621-0dab3b2ce369_add_memory_fields_to_end_user.py

@@ -0,0 +1,35 @@
+"""add memory fields to end user
+
+Revision ID: 0dab3b2ce369
+Revises: c11c071cf21e
+Create Date: 2025-03-08 16:21:08.903178
+
+"""
+from alembic import op
+import models as models
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '0dab3b2ce369'
+down_revision = 'c11c071cf21e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('end_users', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('memory', sa.Text(), nullable=True))
+        batch_op.add_column(sa.Column('memory_updated_at', sa.DateTime(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('end_users', schema=None) as batch_op:
+        batch_op.drop_column('memory_updated_at')
+        batch_op.drop_column('memory')
+
+    # ### end Alembic commands ###

api/models/model.py

@@ -1324,6 +1324,8 @@ class EndUser(UserMixin, db.Model):  # type: ignore[name-defined]
     session_id: Mapped[str] = mapped_column()
     gender = db.Column(db.Integer, nullable=False, server_default=db.text("0"))  # 0: unknown, 1: male, 2: female
     extra_profile = db.Column(db.JSON, nullable=True)  # JSON format, e.g. { "major":"engineer" }
+    memory = db.Column(db.Text, nullable=True)  # Long text to store user memory
+    memory_updated_at = db.Column(db.DateTime, nullable=True)  # To record when memory was last updated
     created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
     updated_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())