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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prism-models
-Version: 0.1.7
+Version: 0.1.9
 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.1.9}/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.1.9}/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.9/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.1.9/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.1.7 → prism_models-0.1.9}/prism_models.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prism-models
-Version: 0.1.7
+Version: 0.1.9
 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.1.9}/prism_models.egg-info/SOURCES.txt

@@ -43,4 +43,6 @@ 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-0.1.7 → prism_models-0.1.9}/pyproject.toml

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