add sql revisions

aana_chat_with_video/alembic/versions/b9860676dd49_set_server_default_for_task_completed_.py

@@ -0,0 +1,47 @@
+"""Set server default for task.completed_at and task.assigned_at to none and add num_retries.
+Revision ID: b9860676dd49
+Revises: 5ad873484aa3
+Create Date: 2024-08-22 07:54:55.921710
+"""
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "b9860676dd49"
+down_revision: str | None = "5ad873484aa3"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Upgrade database to this revision from previous."""
+    with op.batch_alter_table("tasks", schema=None) as batch_op:
+        batch_op.alter_column(
+            "completed_at",
+            server_default=None,
+        )
+        batch_op.alter_column(
+            "assigned_at",
+            server_default=None,
+        )
+        batch_op.add_column(
+            sa.Column(
+                "num_retries",
+                sa.Integer(),
+                nullable=False,
+                comment="Number of retries",
+                server_default=sa.text("0"),
+            )
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database from this revision to previous."""
+    with op.batch_alter_table("tasks", schema=None) as batch_op:
+        batch_op.drop_column("num_retries")
+
+    # ### end Alembic commands ###

aana_chat_with_video/alembic/versions/d93a90261ee5_added_extended_video.py

@@ -1,7 +1,7 @@
 """added extended video.
 
 Revision ID: d93a90261ee5
-Revises: 5ad873484aa3
+Revises: b9860676dd49
 Create Date: 2024-07-29 12:41:04.976640
 
 """
@@ -12,7 +12,7 @@
 
 # revision identifiers, used by Alembic.
 revision: str = 'd93a90261ee5'
-down_revision: str | None = '5ad873484aa3'
+down_revision: str | None = 'b9860676dd49'
 branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None