prism-models 0.1.7__tar.gz → 0.2.1__tar.gz

{prism_models-0.1.7 → prism_models-0.2.1}/PKG-INFO

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

{prism_models-0.1.7 → prism_models-0.2.1}/prism_models/config.py

@@ -26,7 +26,7 @@ class Settings(BaseSettings):
     @property
     def PG_DATABASE_URL_PRISM(self) -> str:
         """Get the Prism database URL."""
-        return self.PG_DATABASE_URL + "/prism2"
+        return self.PG_DATABASE_URL + "/prism"
 
     @property
     def is_development(self) -> bool:

{prism_models-0.1.7 → prism_models-0.2.1}/prism_models/content.py

@@ -97,15 +97,20 @@ class Source(BaseModel):
     s3_directory: Mapped[str | None] = mapped_column(String(2048), nullable=True)
 
     sharepoint_directory: Mapped[str | None] = mapped_column(String(2048), nullable=True)
-    # One-to-one relationship to Collection via unique FK - NOTE NOT UNIQUE
-    collection_id: Mapped[int | None] = mapped_column(
-        Integer, ForeignKey("collection.id"), nullable=False, unique=True, index=True
-    )
-    collection: Mapped[Optional["Collection"]] = relationship(
-        "Collection", back_populates="source", uselist=False
+
+    target_collection_ids: Mapped[list[int] | None] = mapped_column(ARRAY(Integer), nullable=True)
+
+    sync_enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=True)
+
+    default_publish_status: Mapped[DocumentPublishStatus] = mapped_column(
+        Enum(DocumentPublishStatus, native_enum=False),
+        nullable=True,
+        default=DocumentPublishStatus.PREVIEW,
     )
 
 
+
+
     def __repr__(self):
         return f"<Source(id={getattr(self, 'id', None)}, name='{self.name}')>"
 
@@ -121,10 +126,6 @@ class Collection(BaseModel):
     # Relationships
     owner: Mapped[Optional["Contact"]] = relationship("Contact")
     documents: Mapped[list["CollectionDocument"]] = relationship(back_populates="collection")
-    # One-to-one backref to Source
-    source: Mapped[Optional["Source"]] = relationship(
-        "Source", back_populates="collection", uselist=False
-    )
 
     __table_args__ = (UniqueConstraint("name", name="uq_collection_name"),)
 
@@ -248,4 +249,4 @@ class Vector(BaseModel):
     
 
     chunk: Mapped["Chunk"] = relationship(back_populates="vector")
-    qa_pair: Mapped["QAPair"] = relationship(back_populates="vector")
+    qa_pair: Mapped["QAPair"] = relationship(back_populates="vector")

{prism_models-0.1.7 → prism_models-0.2.1}/prism_models/feedback.py

@@ -94,7 +94,6 @@ class Feedback(BaseModel):
     # Relationships
     manager: Mapped["Contact"] = relationship(foreign_keys=[manager_id])
     chunk: Mapped[Optional["Chunk"]] = relationship()
-    qa_pair: Mapped[Optional["QAPair"]] = relationship()
     message: Mapped[Optional["ConversationMessage"]] = relationship()
     assigned_reviewer: Mapped[Optional["Contact"]] = relationship(foreign_keys=[assigned_reviewer_id])
 
@@ -111,7 +110,6 @@ class FeedbackAnalysis(BaseModel):
 
     analysis_type: Mapped[AnalysisType] = mapped_column(String(50), nullable=False)
     llm_response: Mapped[dict[str, Any] | None] = mapped_column(JSON)
-    confidence_score: Mapped[float | None] = mapped_column(Float)
     model_used: Mapped[str | None] = mapped_column(String(255))
     analyzed_at: Mapped[datetime] = mapped_column(TIMESTAMP(timezone=True), server_default=func.now(), nullable=False)
     approved: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
@@ -127,9 +125,8 @@ class FeedbackAnalysis(BaseModel):
 class Augmentation(BaseModel):
     feedback_id: Mapped[int] = mapped_column(Integer, ForeignKey("feedback.id"), nullable=False, index=True)
     original_chunk_id: Mapped[int | None] = mapped_column(Integer, ForeignKey("chunk.id"), nullable=True)
-    original_qa_pair_id: Mapped[int | None] = mapped_column(Integer, ForeignKey("qa_pair.id"), nullable=True)
     generated_chunk_id: Mapped[int | None] = mapped_column(Integer, ForeignKey("chunk.id"), nullable=True)
-    generated_qa_pair_id: Mapped[int | None] = mapped_column(Integer, ForeignKey("qa_pair.id"), nullable=True)
+    
 
     action_type: Mapped[AugmentationAction] = mapped_column(String(50), nullable=False)
     change_summary: Mapped[dict[str, Any] | None] = mapped_column(JSON)
@@ -137,9 +134,7 @@ class Augmentation(BaseModel):
     # Relationships
     feedback: Mapped["Feedback"] = relationship(back_populates="augmentations")
     original_chunk: Mapped[Optional["Chunk"]] = relationship(foreign_keys=[original_chunk_id])
-    original_qa_pair: Mapped[Optional["QAPair"]] = relationship(foreign_keys=[original_qa_pair_id])
     generated_chunk: Mapped[Optional["Chunk"]] = relationship(foreign_keys=[generated_chunk_id])
-    generated_qa_pair: Mapped[Optional["QAPair"]] = relationship(foreign_keys=[generated_qa_pair_id])
 
     def __repr__(self):
         return f"<Augmentation(id={self.id}, action='{self.action_type}', status='{self.status}')>"

prism_models-0.2.1/prism_models/migration/versions/2025_11_03_1738_b733b48d78b5_remove_source.py

@@ -0,0 +1,32 @@
+"""remove source
+
+Revision ID: b733b48d78b5
+Revises: bf5c2ce928e3
+Create Date: 2025-11-03 17:38:08.937565
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'b733b48d78b5'
+down_revision: Union[str, Sequence[str], None] = 'bf5c2ce928e3'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###

prism_models-0.2.1/prism_models/migration/versions/2025_11_06_1450_8a7e56260eba_added_collection_ids_sync_enabled_and_.py

@@ -0,0 +1,36 @@
+"""added collection ids, sync enabled and default publish status to source table
+
+Revision ID: 8a7e56260eba
+Revises: b733b48d78b5
+Create Date: 2025-11-06 14:50:51.787562
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = '8a7e56260eba'
+down_revision: Union[str, Sequence[str], None] = 'b733b48d78b5'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('source', sa.Column('target_collection_ids', postgresql.ARRAY(sa.Integer()), nullable=True))
+    op.add_column('source', sa.Column('sync_enabled', sa.Boolean(), nullable=True))
+    op.add_column('source', sa.Column('default_publish_status', sa.Enum('PREVIEW', 'ACTIVE', 'INACTIVE', name='documentpublishstatus', native_enum=False), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('source', 'default_publish_status')
+    op.drop_column('source', 'sync_enabled')
+    op.drop_column('source', 'target_collection_ids')
+    # ### end Alembic commands ###

prism_models-0.2.1/prism_models/migration/versions/2025_11_06_1626_4c4148ac2d21_removed_qna_related_columns_from_.py

@@ -0,0 +1,40 @@
+"""removed qna related columns from Augmentation and removed confidence score from feedback analysis
+
+Revision ID: 4c4148ac2d21
+Revises: 8a7e56260eba
+Create Date: 2025-11-06 16:26:20.979418
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '4c4148ac2d21'
+down_revision: Union[str, Sequence[str], None] = '8a7e56260eba'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(op.f('augmentation_original_qa_pair_id_fkey'), 'augmentation', type_='foreignkey')
+    op.drop_constraint(op.f('augmentation_generated_qa_pair_id_fkey'), 'augmentation', type_='foreignkey')
+    op.drop_column('augmentation', 'original_qa_pair_id')
+    op.drop_column('augmentation', 'generated_qa_pair_id')
+    op.drop_column('feedback_analysis', 'confidence_score')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('feedback_analysis', sa.Column('confidence_score', sa.DOUBLE_PRECISION(precision=53), autoincrement=False, nullable=True))
+    op.add_column('augmentation', sa.Column('generated_qa_pair_id', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.add_column('augmentation', sa.Column('original_qa_pair_id', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.create_foreign_key(op.f('augmentation_generated_qa_pair_id_fkey'), 'augmentation', 'qa_pair', ['generated_qa_pair_id'], ['id'])
+    op.create_foreign_key(op.f('augmentation_original_qa_pair_id_fkey'), 'augmentation', 'qa_pair', ['original_qa_pair_id'], ['id'])
+    # ### end Alembic commands ###

prism_models-0.2.1/prism_models/migration/versions/2025_11_06_1648_b228888a01ee_added_chunk_config_to_source.py

@@ -0,0 +1,32 @@
+"""added chunk config to source
+
+Revision ID: b228888a01ee
+Revises: 4c4148ac2d21
+Create Date: 2025-11-06 16:48:20.647497
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'b228888a01ee'
+down_revision: Union[str, Sequence[str], None] = '4c4148ac2d21'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###

{prism_models-0.1.7 → prism_models-0.2.1}/prism_models.egg-info/PKG-INFO

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

{prism_models-0.1.7 → prism_models-0.2.1}/prism_models.egg-info/SOURCES.txt

@@ -43,4 +43,8 @@ prism_models/migration/versions/2025_10_23_1228_919d07a93f83_added_sharepoint_di
 prism_models/migration/versions/2025_10_27_1331_ff3be2c4311f_added_agentcollectionaccess_table.py
 prism_models/migration/versions/2025_10_28_1333_b789e61df26a_ad_enum_and_field_added.py
 prism_models/migration/versions/2025_11_03_1633_cc27b20a106e_removed_collection_from_source_.py
-prism_models/migration/versions/2025_11_03_1653_bf5c2ce928e3_removed_collection_from_source_.py
+prism_models/migration/versions/2025_11_03_1653_bf5c2ce928e3_removed_collection_from_source_.py
+prism_models/migration/versions/2025_11_03_1738_b733b48d78b5_remove_source.py
+prism_models/migration/versions/2025_11_06_1450_8a7e56260eba_added_collection_ids_sync_enabled_and_.py
+prism_models/migration/versions/2025_11_06_1626_4c4148ac2d21_removed_qna_related_columns_from_.py
+prism_models/migration/versions/2025_11_06_1648_b228888a01ee_added_chunk_config_to_source.py

{prism_models-0.1.7 → prism_models-0.2.1}/pyproject.toml

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