prism-models 0.6.3__py3-none-any.whl → 0.6.5__py3-none-any.whl

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.dist-info → prism_models-0.6.5.dist-info}/METADATA

@@ -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.dist-info → prism_models-0.6.5.dist-info}/RECORD

@@ -63,8 +63,8 @@ prism_models/migration/versions/2025_11_24_1124_dc57fb290621_added_conversation_
 prism_models/migration/versions/2025_11_25_1847_c6c36663acd5_added_user_preferences_table.py,sha256=4E51jRKENB-pU2yg_Pf2zxORWFqPfHcQ4aQmblPS4Ew,2438
 prism_models/migration/versions/2025_12_07_2236_88f01df468f2_added_sparse_embeddings_field_in_vector_.py,sha256=Qrg6ouy11sESaPJcfkfh9Vk8CoOW34vrMmF69OStZDg,995
 prism_models/migration/versions/2025_12_09_1146_6abdd3f001ab_adding_orchestrator_model_to_profile_.py,sha256=eiCgTyFW7TbifwXPbZtsYW9_UPYwyCMF4I7eXnMOTwQ,937
-prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py,sha256=t9mWTA6Yu4Ko2zlfkSkEelerXn2NfB79TWziRoRRJYA,4506
-prism_models-0.6.3.dist-info/METADATA,sha256=NP-pdnndSfRfqT3dfjwRX8h5EgQZaPIY0ske8rsypKU,283
-prism_models-0.6.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prism_models-0.6.3.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
-prism_models-0.6.3.dist-info/RECORD,,
+prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py,sha256=QnbVia8YU2_Rp0LWFvbA2lUy7kmdrGLbDmTHw-YQAoM,4752
+prism_models-0.6.5.dist-info/METADATA,sha256=e35cEPTl8s-1RdPFXOxsAo_K-GwrcH_J7FkOB3jQ6Jk,283
+prism_models-0.6.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prism_models-0.6.5.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
+prism_models-0.6.5.dist-info/RECORD,,