prism-models 0.6.3__tar.gz → 0.6.5__tar.gz

{prism_models-0.6.3 → prism_models-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prism-models
-Version: 0.6.3
+Version: 0.6.5
 Requires-Python: >=3.12
 Requires-Dist: sqlalchemy[asyncio]>=2.0.0
 Requires-Dist: alembic>=1.16.0

{prism_models-0.6.3 → prism_models-0.6.5}/prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py

@@ -11,6 +11,8 @@ from alembic import op
 import sqlalchemy as sa
 from sqlalchemy.dialects import postgresql
 
+contact_role_enum = sa.Enum('ADMIN', 'ACCOUNT_ADMIN', 'MANAGER', 'VIEWER', name='contactrole')
+
 # revision identifiers, used by Alembic.
 revision: str = '4d9291a3acb1'
 down_revision: Union[str, Sequence[str], None] = '6abdd3f001ab'
@@ -21,32 +23,33 @@ depends_on: Union[str, Sequence[str], None] = None
 def upgrade() -> None:
     """Upgrade schema."""
     # ### commands auto generated by Alembic - please adjust! ###
+    # Drop references to qa_pair before removing the table to satisfy FK constraints.
+    op.drop_constraint(op.f('vector_qa_pair_id_key'), 'vector', type_='unique')
+    op.drop_constraint(op.f('vector_qa_pair_id_fkey'), 'vector', type_='foreignkey')
+    op.drop_column('vector', 'qa_pair_id')
+
     op.drop_table('qa_pair_chunk_association')
     op.drop_index(op.f('qa_pair_document_id_idx'), table_name='qa_pair')
     op.drop_index(op.f('qa_pair_id_idx'), table_name='qa_pair')
     op.drop_table('qa_pair')
-    op.add_column('contact', sa.Column('contact_role', sa.Enum('ADMIN', 'ACCOUNT_ADMIN', 'MANAGER', 'VIEWER', name='contactrole'), server_default='VIEWER', nullable=False))
+    contact_role_enum.create(op.get_bind(), checkfirst=True)
+    op.add_column('contact', sa.Column('contact_role', contact_role_enum, server_default='VIEWER', nullable=False))
     op.add_column('contact', sa.Column('manager_contact_id', sa.Integer(), nullable=True))
     op.create_index(op.f('contact_manager_contact_id_idx'), 'contact', ['manager_contact_id'], unique=False)
     op.create_foreign_key(op.f('contact_manager_contact_id_fkey'), 'contact', 'contact', ['manager_contact_id'], ['id'], ondelete='SET NULL')
     op.drop_index(op.f('idx_document_title_trgm'), table_name='document', postgresql_ops={'title': 'gin_trgm_ops'}, postgresql_using='gin')
-    op.drop_constraint(op.f('vector_qa_pair_id_key'), 'vector', type_='unique')
-    op.drop_constraint(op.f('vector_qa_pair_id_fkey'), 'vector', type_='foreignkey')
-    op.drop_column('vector', 'qa_pair_id')
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     """Downgrade schema."""
     # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('vector', sa.Column('qa_pair_id', sa.INTEGER(), autoincrement=False, nullable=True))
-    op.create_foreign_key(op.f('vector_qa_pair_id_fkey'), 'vector', 'qa_pair', ['qa_pair_id'], ['id'])
-    op.create_unique_constraint(op.f('vector_qa_pair_id_key'), 'vector', ['qa_pair_id'], postgresql_nulls_not_distinct=False)
     op.create_index(op.f('idx_document_title_trgm'), 'document', ['title'], unique=False, postgresql_ops={'title': 'gin_trgm_ops'}, postgresql_using='gin')
     op.drop_constraint(op.f('contact_manager_contact_id_fkey'), 'contact', type_='foreignkey')
     op.drop_index(op.f('contact_manager_contact_id_idx'), table_name='contact')
     op.drop_column('contact', 'manager_contact_id')
     op.drop_column('contact', 'contact_role')
+    contact_role_enum.drop(op.get_bind(), checkfirst=True)
     op.create_table('qa_pair',
     sa.Column('document_id', sa.INTEGER(), autoincrement=False, nullable=False),
     sa.Column('question', sa.TEXT(), autoincrement=False, nullable=False),
@@ -70,4 +73,7 @@ def downgrade() -> None:
     sa.ForeignKeyConstraint(['qa_pair_id'], ['qa_pair.id'], name=op.f('qa_pair_chunk_association_qa_pair_id_fkey')),
     sa.PrimaryKeyConstraint('qa_pair_id', 'chunk_id', name=op.f('qa_pair_chunk_association_pkey'))
     )
+    op.add_column('vector', sa.Column('qa_pair_id', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.create_foreign_key(op.f('vector_qa_pair_id_fkey'), 'vector', 'qa_pair', ['qa_pair_id'], ['id'])
+    op.create_unique_constraint(op.f('vector_qa_pair_id_key'), 'vector', ['qa_pair_id'], postgresql_nulls_not_distinct=False)
     # ### end Alembic commands ###

{prism_models-0.6.3 → prism_models-0.6.5}/prism_models.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prism-models
-Version: 0.6.3
+Version: 0.6.5
 Requires-Python: >=3.12
 Requires-Dist: sqlalchemy[asyncio]>=2.0.0
 Requires-Dist: alembic>=1.16.0

{prism_models-0.6.3 → prism_models-0.6.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "prism-models"
-version = "0.6.3"
+version = "0.6.5"
 requires-python = ">=3.12"
 dependencies = [
   "sqlalchemy[asyncio]>=2.0.0",