prism-models 0.1.0__py3-none-any.whl

prism_models/migration/versions/2025_10_03_1452_f9c750ec2a0b_1_to_1_relationship_between_collection_.py

@@ -0,0 +1,52 @@
+"""1 to 1 relationship between collection and source
+
+Revision ID: f9c750ec2a0b
+Revises: 7d1cb343a63f
+Create Date: 2025-10-03 14:52:06.699081
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'f9c750ec2a0b'
+down_revision: Union[str, Sequence[str], None] = '7d1cb343a63f'
+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('type', sa.Enum('S3', 'SHARE_POINT', 'GRID', 'CRM', 'CONFLUENCE', 'CUSTOM', name='sourcename', native_enum=False), nullable=True))
+    op.add_column('source', sa.Column('collection_id', sa.Integer(), nullable=False))
+    op.alter_column('source', 'name',
+               existing_type=sa.VARCHAR(length=23),
+               type_=sa.String(length=255),
+               existing_nullable=False)
+    op.drop_index(op.f('source_name_idx'), table_name='source')
+    op.create_index(op.f('source_name_idx'), 'source', ['name'], unique=True)
+    op.create_index(op.f('source_collection_id_idx'), 'source', ['collection_id'], unique=True)
+    op.create_index(op.f('source_type_idx'), 'source', ['type'], unique=False)
+    op.create_foreign_key(op.f('source_collection_id_fkey'), 'source', 'collection', ['collection_id'], ['id'])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(op.f('source_collection_id_fkey'), 'source', type_='foreignkey')
+    op.drop_index(op.f('source_type_idx'), table_name='source')
+    op.drop_index(op.f('source_collection_id_idx'), table_name='source')
+    op.drop_index(op.f('source_name_idx'), table_name='source')
+    op.create_index(op.f('source_name_idx'), 'source', ['name'], unique=False)
+    op.alter_column('source', 'name',
+               existing_type=sa.String(length=255),
+               type_=sa.VARCHAR(length=23),
+               existing_nullable=False)
+    op.drop_column('source', 'collection_id')
+    op.drop_column('source', 'type')
+    # ### end Alembic commands ###

prism_models/migration/versions/2025_10_07_1722_5cfa0c462948_added_travel_advisory_enum.py

@@ -0,0 +1,38 @@
+"""added travel advisory enum
+
+Revision ID: 5cfa0c462948
+Revises: f9c750ec2a0b
+Create Date: 2025-10-07 17:22:46.670468
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '5cfa0c462948'
+down_revision: Union[str, Sequence[str], None] = 'f9c750ec2a0b'
+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.alter_column('source', 'type',
+               existing_type=sa.VARCHAR(length=11),
+               type_=sa.Enum('S3', 'SHARE_POINT', 'GRID', 'CRM', 'CONFLUENCE', 'CUSTOM', 'TRAVEL_ADVISORY', name='sourcename', native_enum=False),
+               existing_nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('source', 'type',
+               existing_type=sa.Enum('S3', 'SHARE_POINT', 'GRID', 'CRM', 'CONFLUENCE', 'CUSTOM', 'TRAVEL_ADVISORY', name='sourcename', native_enum=False),
+               type_=sa.VARCHAR(length=11),
+               existing_nullable=True)
+    # ### end Alembic commands ###

prism_models/migration/versions/2025_10_08_1304_c91eb8e38cc7_added_destination_report_and_event_.py

@@ -0,0 +1,38 @@
+"""Added Destination Report and event sources
+
+Revision ID: c91eb8e38cc7
+Revises: 5cfa0c462948
+Create Date: 2025-10-08 13:04:14.325976
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'c91eb8e38cc7'
+down_revision: Union[str, Sequence[str], None] = '5cfa0c462948'
+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.alter_column('source', 'type',
+               existing_type=sa.VARCHAR(length=15),
+               type_=sa.Enum('S3', 'SHARE_POINT', 'GRID', 'GRID_DESTINATION_REPORT', 'GRID_EVENT_REPORT', 'CRM', 'CONFLUENCE', 'CUSTOM', 'TRAVEL_ADVISORY', name='sourcename', native_enum=False),
+               existing_nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('source', 'type',
+               existing_type=sa.Enum('S3', 'SHARE_POINT', 'GRID', 'GRID_DESTINATION_REPORT', 'GRID_EVENT_REPORT', 'CRM', 'CONFLUENCE', 'CUSTOM', 'TRAVEL_ADVISORY', name='sourcename', native_enum=False),
+               type_=sa.VARCHAR(length=15),
+               existing_nullable=True)
+    # ### end Alembic commands ###

prism_models/migration/versions/2025_10_09_1308_796b720ea35f_added_qa_id_to_vovetor.py

@@ -0,0 +1,42 @@
+"""added qa id to vovetor
+
+Revision ID: 796b720ea35f
+Revises: c91eb8e38cc7
+Create Date: 2025-10-09 13:08:56.972719
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '796b720ea35f'
+down_revision: Union[str, Sequence[str], None] = 'c91eb8e38cc7'
+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('vector', sa.Column('qa_pair_id', sa.Integer(), nullable=True))
+    op.alter_column('vector', 'chunk_id',
+               existing_type=sa.INTEGER(),
+               nullable=True)
+    op.create_unique_constraint(op.f('vector_qa_pair_id_key'), 'vector', ['qa_pair_id'])
+    op.create_foreign_key(op.f('vector_qa_pair_id_fkey'), 'vector', 'qa_pair', ['qa_pair_id'], ['id'])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(op.f('vector_qa_pair_id_fkey'), 'vector', type_='foreignkey')
+    op.drop_constraint(op.f('vector_qa_pair_id_key'), 'vector', type_='unique')
+    op.alter_column('vector', 'chunk_id',
+               existing_type=sa.INTEGER(),
+               nullable=False)
+    op.drop_column('vector', 'qa_pair_id')
+    # ### end Alembic commands ###

prism_models/migration/versions/2025_10_16_1611_663c66268631_added_sharepoint_drive_item_id_as_an_.py

@@ -0,0 +1,32 @@
+"""added sharepoint_drive_item_id as an optional field to document table
+
+Revision ID: 663c66268631
+Revises: 796b720ea35f
+Create Date: 2025-10-16 16:11:31.728752
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '663c66268631'
+down_revision: Union[str, Sequence[str], None] = '796b720ea35f'
+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('document', sa.Column('sharepoint_drive_item_id', sa.String(length=1024), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('document', 'sharepoint_drive_item_id')
+    # ### end Alembic commands ###

prism_models/qdrant.py

@@ -0,0 +1,97 @@
+from typing import Optional, List, Any
+from datetime import datetime
+from pydantic import BaseModel, Field, UUID4, ValidationError
+from sqlalchemy import JSON, TypeDecorator
+from sqlalchemy.engine.interfaces import Dialect
+from enum import Enum
+
+class ChunkTypeEnum(str, Enum):
+    TEXT = "TEXT"
+    Q_AND_A = "Q/A"
+    AUGMENTED_TEXT = "AUGMENTED_TEXT"
+    AUGMENTED_Q_AND_A = "AUGMENTED_Q/A"
+
+class QdrantVectorPayload(BaseModel):
+    virtual_collection_names: Optional[List[str]] = Field(
+        default=None, description="Names of the virtual collections"
+    )
+    uuid: Optional[UUID4] = Field(
+        default=None, description="Unique identifier for the vector"
+    )
+    chunk_text: Optional[str] = Field(
+        default=None, description="Text content of the chunk"
+    )
+    is_active: Optional[bool] = Field(
+        default=True, description="Whether the vector is active"
+    )
+    deleted_at: Optional[datetime] = Field(
+        default=None, description="Timestamp when the vector was deleted"
+    )
+
+    chunk_type: ChunkTypeEnum = Field(
+        default=ChunkTypeEnum.TEXT, 
+        description="Type of the chunk"
+    )
+
+    class Config:
+        orm_mode = True
+
+
+class DestinationVectorPayload(QdrantVectorPayload):
+    destination_id: Optional[int] = Field(
+        default=None, description="ID of the destination associated with the chunk"
+    )
+    report_title: Optional[str] = Field(
+        default=None, description="Title of the report"
+    )
+    destination_name: Optional[str] = Field(
+        default=None, description="Name of the destination"
+    )
+
+
+class PydanticType(TypeDecorator):
+    """
+    A SQLAlchemy column type that validates JSON data against a Pydantic model.
+    
+    Usage:
+        class MyModel(Base):
+            data: Mapped[dict] = mapped_column(PydanticType(MyPydanticModel))
+    """
+    impl = JSON
+    cache_ok = True
+    
+    def __init__(self, pydantic_model: type[BaseModel], *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.pydantic_model = pydantic_model
+    
+    def process_bind_param(self, value: Any, dialect: Dialect) -> dict | None:
+        """
+        Validate and convert Pydantic model to dict before storing in database.
+        """
+        if value is None:
+            return None
+        
+        if isinstance(value, self.pydantic_model):
+            return value.model_dump(mode='json')
+        
+        if isinstance(value, dict):
+            try:
+                validated = self.pydantic_model(**value)
+                return validated.model_dump(mode='json')
+            except ValidationError as e:
+                raise ValueError(f"Invalid data for {self.pydantic_model.__name__}: {e}")
+        
+        raise ValueError(f"Expected {self.pydantic_model.__name__} or dict, got {type(value)}")
+    
+    def process_result_value(self, value: dict | None, dialect: Dialect) -> BaseModel | None:
+        """
+        Convert dict from database to Pydantic model instance.
+        """
+        if value is None:
+            return None
+        
+        try:
+            return self.pydantic_model(**value)
+        except ValidationError as e:
+            print(f"Warning: Invalid data in database for {self.pydantic_model.__name__}: {e}")
+            return None

prism_models-0.1.0.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: prism-models
+Version: 0.1.0
+Requires-Python: >=3.12
+Requires-Dist: sqlalchemy[asyncio]>=2.0.0
+Requires-Dist: alembic>=1.16.0
+Requires-Dist: asyncpg>=0.29.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pydantic-settings>=2.1.0
+Requires-Dist: uuid7>=0.1.0

prism_models-0.1.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+prism_models/__init__.py,sha256=QNgzPjt8lzBV7j0e70zUoJurYiLMSsVcWEHnQz4hO-c,1396
+prism_models/agent_profile.py,sha256=SlZyd59q5DhdMmbcW9SAQCy44sgF_I7arz-M8_71an4,3789
+prism_models/base.py,sha256=Ka8zNToTiTG0jNgzqqj2goGmPXt8zxuuSMYvotnb8wY,2281
+prism_models/chat.py,sha256=zncfBav1YdB8AdC-9McY9fsKZ3-2plLLhy_NMd8JoyA,11784
+prism_models/config.py,sha256=gy_6HknnG17Clv8X_EBhaXEBFBLGR1PLjvN5-SFTo5E,962
+prism_models/content.py,sha256=7QEfWYAwpj3ltFdQS5e9bBNrtpGttOYpKhvja-5ecD8,10993
+prism_models/feedback.py,sha256=tMZISQ0mk7_1ADsK4bO36_VZ0f-JDdAPcfWzIZBk0_g,6202
+prism_models/qdrant.py,sha256=ORIpLv1fbRJiL9qq2F-WurtY42MGEPku4ojY_-cdsRY,3266
+prism_models/migration/README,sha256=MVlc9TYmr57RbhXET6QxgyCcwWP7w-vLkEsirENqiIQ,38
+prism_models/migration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prism_models/migration/env.py,sha256=Uobqv_peoEUYAZST0D0hO7xHPQe8OPxqZ0yV1xuNE2c,4096
+prism_models/migration/script.py.mako,sha256=04kgeBtNMa4cCnG8CfQcKt6P6rnloIfj8wy0u_DBydM,704
+prism_models/migration/versions/2025_09_11_1516_161f8829d93f_initial_schema.py,sha256=6nkB5DLbTNCk84SYsKz0kUzyhvDPjkkC9HuwN37QUao,35166
+prism_models/migration/versions/2025_09_11_1558_5e011849ea76_changes_for_feedback.py,sha256=KmglWbDxyOSctzgTpyis0TC0-rEqZKrhM9xgFiTWF-s,5434
+prism_models/migration/versions/2025_09_14_2243_059af231c2b2_profile_entities.py,sha256=aKxOOEDPo_wZwh93vJNXXQDdO9dqlPJ5pK3mY_Hef9U,6331
+prism_models/migration/versions/2025_09_15_1646_3219fec0bb10_agent_changes.py,sha256=0b7TyOyEBU-HaINIMGYZjhWHbz5m5LEB1Bjz-IZ72qQ,868
+prism_models/migration/versions/2025_09_16_1627_f2013b08daac_rename_metadata_to_additional_data.py,sha256=ihzMYcVK-37DN1jYsuOMISvBjnGDwViDTaj4FczONRs,1415
+prism_models/migration/versions/2025_09_17_1147_327febbf555f_display_name_added.py,sha256=5DhctzmhumdDt157ytUi0JNQPCt6KXF9_u_ADutR5pw,1021
+prism_models/migration/versions/2025_09_18_1106_b0bcb7ca1dc9_add_support_for_profile_on_create_convo.py,sha256=MWZRiCnVKaojXOGMpr_G41611yIXnAuBzAeeV6Nusw4,1445
+prism_models/migration/versions/2025_09_18_1511_bbc1955191e6_preview_mode.py,sha256=-lokRulsgfjt1LXY2e1pcLFSw40ELICpRM6QLec9pjM,1043
+prism_models/migration/versions/2025_09_26_1115_6eb70e848451_added_publish_status_to_document.py,sha256=i_yNpWCgIG9t8nfPR58xerzcgCqKFMl7lxPvixGkHPU,1463
+prism_models/migration/versions/2025_09_26_1240_f8b0ea2e743c_drop_unique_title_version_on_document.py,sha256=Td2ijXUdUrN68aFizK5SKYC4XwOw3DPKwEGFjycv7Qs,1480
+prism_models/migration/versions/2025_09_26_1505_07dc8c2589e0_added_chunk_id_and_vector_embeddings_to_.py,sha256=6S3qCe0JBx-LNZN3jhGN_9RrKBOKCDVuywzZxveDRaQ,1916
+prism_models/migration/versions/2025_09_29_1220_46ba2693b883_add_markdown_markdown_file_path_s3_.py,sha256=NRZFSr4Zm9XWUujx7-QUBKL1BdICMOKuNVxw3PZsg9Y,934
+prism_models/migration/versions/2025_10_02_1520_bf1472a9b021_removed_doc_id_from_config_table_and_.py,sha256=aB-k8Dqw39YHxgg_hBcPEDyDmFxjn5d1KEQuTCUJ_8E,1155
+prism_models/migration/versions/2025_10_02_1525_6c0e63e0fef8_removed_doc_id_from_config_table_.py,sha256=-4OSytv3-L_Yc3TI2pUB8gpPH3iQzuU3KzeRqZdaD9Q,1228
+prism_models/migration/versions/2025_10_02_1608_1b3eb48f5017_config_id_on_delete_will_be_set_to_null.py,sha256=k9d2CSFCegeeJulUtI3sXmff9ttDi9pftLsgWIwiqm0,1224
+prism_models/migration/versions/2025_10_03_1109_ac85b606d8a4_added_docling_hybrid_to_chunkstrategy.py,sha256=B1z0SvnnkXAAHHQeQQu1aMJGQml66yoTj1sL0BraK6Y,787
+prism_models/migration/versions/2025_10_03_1204_7d1cb343a63f_added_s3_bucket_and_s3_dir_in_source_.py,sha256=qkiMOFumgssJz2dceyN5CNd8hlbY9xruJVozQZrvOe0,1563
+prism_models/migration/versions/2025_10_03_1452_f9c750ec2a0b_1_to_1_relationship_between_collection_.py,sha256=u-zRopfKHEO-2nNBmGXALkPl4mVnIwj2Y2rfcWKTAow,2264
+prism_models/migration/versions/2025_10_07_1722_5cfa0c462948_added_travel_advisory_enum.py,sha256=dxhTunpblAaWN6KNMMYuoZmFNcZ1VMrR5SfOwcDLO4Y,1310
+prism_models/migration/versions/2025_10_08_1304_c91eb8e38cc7_added_destination_report_and_event_.py,sha256=7Re8GVdOC4KMIP14MWpVAD6LlV4ow0uP4w_jYF8iT_g,1422
+prism_models/migration/versions/2025_10_09_1308_796b720ea35f_added_qa_id_to_vovetor.py,sha256=tW58ejf2_NcaK038Igf3zicZQymeVt4I4uzZULNtHV0,1467
+prism_models/migration/versions/2025_10_16_1611_663c66268631_added_sharepoint_drive_item_id_as_an_.py,sha256=IaG9NDyTGENq65GyrhGoScBb-U82x5V7NBKpG9PYy5Q,968
+prism_models-0.1.0.dist-info/METADATA,sha256=4Vmm86piE_0ms7RFvGjVvYo7GO1RGwpjTr-YpSwGzc4,283
+prism_models-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prism_models-0.1.0.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
+prism_models-0.1.0.dist-info/RECORD,,

prism_models-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

prism_models-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+prism_models