refactor: remove obsolete migration files and update initialization migration

migrations/versions/3292a4761b48_files_and_vector_records.py

@@ -1,48 +0,0 @@
-"""files_and_vector_records
-
-Revision ID: 3292a4761b48
-Revises: 8652e8501339
-Create Date: 2024-10-28 13:03:53.442309
-
-"""
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-
-
-# revision identifiers, used by Alembic.
-revision: str = '3292a4761b48'
-down_revision: Union[str, None] = '8652e8501339'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('files',
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('file_name', sa.String(), nullable=False),
-    sa.Column('uploaded_at', sa.DateTime(), nullable=False),
-    sa.Column('processed_at', sa.DateTime(), nullable=True),
-    sa.Column('slack_file_id', sa.String(), nullable=True),
-    sa.Column('user_id', sa.Integer(), nullable=False),
-    sa.Column('mime_type', sa.String(), nullable=False),
-    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
-    sa.PrimaryKeyConstraint('id')
-    )
-    op.create_table('vector_records',
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('file_id', sa.Integer(), nullable=False),
-    sa.Column('record_id', sa.Integer(), nullable=False),
-    sa.ForeignKeyConstraint(['file_id'], ['files.id'], ),
-    sa.PrimaryKeyConstraint('id')
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table('vector_records')
-    op.drop_table('files')
-    # ### end Alembic commands ###

migrations/versions/653e10175e07_record_id_column.py

@@ -1,36 +0,0 @@
-"""record id column
-
-Revision ID: 653e10175e07
-Revises: 96747434d72b
-Create Date: 2024-11-04 23:26:52.258704
-
-"""
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-
-
-# revision identifiers, used by Alembic.
-revision: str = '653e10175e07'
-down_revision: Union[str, None] = '96747434d72b'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.alter_column('vector_records', 'record_id',
-               existing_type=sa.INTEGER(),
-               type_=sa.String(),
-               existing_nullable=False)
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.alter_column('vector_records', 'record_id',
-               existing_type=sa.String(),
-               type_=sa.INTEGER(),
-               existing_nullable=False)
-    # ### end Alembic commands ###

migrations/versions/{8652e8501339_add_user_and_slack_team.py → 6f82114b956f_initialisation.py}

@@ -1,8 +1,8 @@
-"""add user and slack team
+"""Initialisation
 
-Revision ID: 8652e8501339
+Revision ID: 6f82114b956f
 Revises: 
-Create Date: 2024-10-24 16:52:59.936776
+Create Date: 2024-11-07 14:04:41.141506
 
 """
 from typing import Sequence, Union
@@ -12,7 +12,7 @@ import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision: str = '8652e8501339'
+revision: str = '6f82114b956f'
 down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
@@ -36,15 +36,45 @@ def upgrade() -> None:
     sa.Column('slack_team_id', sa.Integer(), nullable=True),
     sa.Column('email_id', sa.String(), nullable=False),
     sa.Column('last_message_sent_at', sa.DateTime(), nullable=True),
+    sa.Column('next_content_delivery_date', sa.DateTime(), nullable=True),
     sa.Column('created_at', sa.DateTime(), nullable=False),
     sa.ForeignKeyConstraint(['slack_team_id'], ['slack_teams.id'], ),
     sa.PrimaryKeyConstraint('id')
     )
+    op.create_table('files',
+    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
+    sa.Column('file_name', sa.String(), nullable=False),
+    sa.Column('uploaded_at', sa.DateTime(), nullable=False),
+    sa.Column('processed_at', sa.DateTime(), nullable=True),
+    sa.Column('slack_file_id', sa.String(), nullable=True),
+    sa.Column('user_id', sa.Integer(), nullable=False),
+    sa.Column('mime_type', sa.String(), nullable=False),
+    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('personalized_content',
+    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
+    sa.Column('content', sa.Text(), nullable=False),
+    sa.Column('user_id', sa.Integer(), nullable=False),
+    sa.Column('delivered_at', sa.DateTime(), nullable=True),
+    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('vector_records',
+    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
+    sa.Column('file_id', sa.Integer(), nullable=False),
+    sa.Column('record_id', sa.String(), nullable=False),
+    sa.ForeignKeyConstraint(['file_id'], ['files.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('vector_records')
+    op.drop_table('personalized_content')
+    op.drop_table('files')
     op.drop_table('users')
     op.drop_table('slack_teams')
     # ### end Alembic commands ###

migrations/versions/96747434d72b_next_delivery_date_and_personalised_.py

@@ -1,40 +0,0 @@
-"""next_delivery_date and personalised column
-
-Revision ID: 96747434d72b
-Revises: 3292a4761b48
-Create Date: 2024-10-30 10:58:35.523977
-
-"""
-
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-
-
-# revision identifiers, used by Alembic.
-revision: str = '96747434d72b'
-down_revision: Union[str, None] = "3292a4761b48"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('personalized_content',
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('content', sa.Text(), nullable=False),
-    sa.Column('user_id', sa.Integer(), nullable=False),
-    sa.Column('delivered_at', sa.DateTime(), nullable=True),
-    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
-    sa.PrimaryKeyConstraint('id')
-    )
-    op.add_column('users', sa.Column('next_content_delivery_date', sa.DateTime(), nullable=True))
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_column('users', 'next_content_delivery_date')
-    op.drop_table('personalized_content')
-    # ### end Alembic commands ###

poetry.lock

@@ -120,6 +120,24 @@ files = [
 [package.dependencies]
 frozenlist = ">=1.1.0"
 
+[[package]]
+name = "aiosqlite"
+version = "0.20.0"
+description = "asyncio bridge to the standard sqlite3 module"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "aiosqlite-0.20.0-py3-none-any.whl", hash = "sha256:36a1deaca0cac40ebe32aac9977a6e2bbc7f5189f23f4a54d5908986729e5bd6"},
+    {file = "aiosqlite-0.20.0.tar.gz", hash = "sha256:6d35c8c256637f4672f843c31021464090805bf925385ac39473fb16eaaca3d7"},
+]
+
+[package.dependencies]
+typing_extensions = ">=4.0"
+
+[package.extras]
+dev = ["attribution (==1.7.0)", "black (==24.2.0)", "coverage[toml] (==7.4.1)", "flake8 (==7.0.0)", "flake8-bugbear (==24.2.6)", "flit (==3.9.0)", "mypy (==1.8.0)", "ufmt (==2.3.0)", "usort (==1.0.8.post1)"]
+docs = ["sphinx (==7.2.6)", "sphinx-mdinclude (==0.5.3)"]
+
 [[package]]
 name = "alembic"
 version = "1.14.0"
@@ -3332,4 +3350,4 @@ propcache = ">=0.2.0"
 [metadata]
 lock-version = "2.0"
 python-versions = "3.12.*"
-content-hash = "5930615910191cc57c6919c9904b481049afd96fe5daf163ce6ed3569cc60b25"
+content-hash = "628d29d4f3ed77432eff709cd821db42cc00c75c03ee84a30e120ea6014e2252"

pyproject.toml

@@ -28,6 +28,7 @@ pinecone = "^5.3.1"
 langchain-pinecone = "^0.2.0"
 langchain-core = "^0.3.12"
 sqlalchemy = "^2.0.36"
+aiosqlite = "^0.20.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest-env = "^1.1.5"