fix: add auto-migration for remote_id and summary columns in main.py lifespan

alembic/versions/dc6a7b8c9d0e_add_remote_id_to_research_paper.py

@@ -0,0 +1,29 @@
+"""add remote_id to research_paper
+
+Revision ID: dc6a7b8c9d0e
+Revises: 8a2d3e4f5b6c
+Create Date: 2026-04-26 13:00:00.000000
+
+"""
+from typing import Sequence, Union
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision: str = 'dc6a7b8c9d0e'
+down_revision: Union[str, None] = '8a2d3e4f5b6c'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # Add remote_id and summary columns to research_papers table
+    op.add_column('research_papers', sa.Column('remote_id', sa.Text(), nullable=True))
+    op.add_column('research_papers', sa.Column('summary', sa.Text(), nullable=True))
+    op.create_index(op.f('ix_research_papers_remote_id'), 'research_papers', ['remote_id'], unique=False)
+
+
+def downgrade() -> None:
+    op.drop_index(op.f('ix_research_papers_remote_id'), table_name='research_papers')
+    op.drop_column('research_papers', 'summary')
+    op.drop_column('research_papers', 'remote_id')

app/main.py

@@ -65,6 +65,20 @@ async def lifespan(app: FastAPI):
                 with engine.connect().execution_options(isolation_level="AUTOCOMMIT") as ddl_conn:
                     ddl_conn.execute(text("ALTER TABLE users ADD COLUMN research_profile TEXT"))
                 logger.info("Added 'research_profile' column to users table.")
+
+            # Check if 'remote_id' column exists in research_papers
+            res_remote = conn.execute(text("SELECT column_name FROM information_schema.columns WHERE table_name='research_papers' AND column_name='remote_id'")).fetchone()
+            if not res_remote:
+                with engine.connect().execution_options(isolation_level="AUTOCOMMIT") as ddl_conn:
+                    ddl_conn.execute(text("ALTER TABLE research_papers ADD COLUMN remote_id TEXT"))
+                logger.info("Added 'remote_id' column to research_papers table.")
+
+            # Check if 'summary' column exists in research_papers
+            res_sum = conn.execute(text("SELECT column_name FROM information_schema.columns WHERE table_name='research_papers' AND column_name='summary'")).fetchone()
+            if not res_sum:
+                with engine.connect().execution_options(isolation_level="AUTOCOMMIT") as ddl_conn:
+                    ddl_conn.execute(text("ALTER TABLE research_papers ADD COLUMN summary TEXT"))
+                logger.info("Added 'summary' column to research_papers table.")
     except Exception as e:
         logger.warning(f"Auto-migration warning (non-fatal): {str(e)}")