prism-models 0.6.2__py3-none-any.whl → 0.6.4__py3-none-any.whl

prism_models/__init__.py

@@ -6,7 +6,7 @@ __version__ = "0.1.0"
 from prism_models.agent_profile import Agent, AgentCollectionAccess, AgentProfile, AgentProfileStatus, Profile, ProfileCollectionAccess
 from prism_models.base import POSTGRES_NAMING_CONVENTION, Base, BaseModel, TimestampMixin
 from prism_models.chat import Contact, Conversation, ConversationMessage, ConversationMessageMetadata, UserPreferences
-from prism_models.content import Chunk, ChunkConfig, Collection, CollectionDocument, Document, IntegrationConfig, QAPair, Source, Vector
+from prism_models.content import Chunk, ChunkConfig, Collection, CollectionDocument, Document, IntegrationConfig, Source, Vector
 from prism_models.feedback import Augmentation, Feedback, FeedbackAnalysis, FeedbackStatus, FeedbackType
 from prism_models.qdrant import QdrantVectorPayload, DestinationVectorPayload, PydanticType
 
@@ -38,7 +38,6 @@ __all__ = [
     "Profile",
     "ProfileCollectionAccess",
     "PydanticType",
-    "QAPair",
     "QdrantVectorPayload",
     "Source",
     "TimestampMixin",

prism_models/chat.py

@@ -27,6 +27,22 @@ class ContactSource(PyEnum):
     MICROSOFT_AD = "MICROSOFT_AD"
 
 
+class ContactRole(PyEnum):
+    """
+    Hierarchical permission roles for contacts.
+
+    Permission hierarchy (highest to lowest):
+    - ADMIN: Full access to all contacts in the system
+    - ACCOUNT_ADMIN: Full access to contacts within their account
+    - MANAGER: Can view/edit contacts in their team (direct reports)
+    - VIEWER: Read-only access to own data and limited team visibility
+    """
+    ADMIN = "ADMIN"
+    ACCOUNT_ADMIN = "ACCOUNT_ADMIN"
+    MANAGER = "MANAGER"
+    VIEWER = "VIEWER"
+
+
 class ConversationType(PyEnum):
     TRAVEL_GUIDE = "TRAVEL_GUIDE"
     MSA_CHAT = "MSA_CHAT"
@@ -91,6 +107,28 @@ class Contact(ChatSchemaMixin, BaseModel):
     azure_ad_object_id = Column(String(36), nullable=True, unique=True, index=True)
     azure_ad_tenant_id = Column(String(36), nullable=True, index=True)
 
+    # Hierarchical permission system
+    contact_role: Mapped[ContactRole] = mapped_column(
+        Enum(ContactRole),
+        default=ContactRole.VIEWER,
+        nullable=False,
+        server_default="VIEWER"
+    )
+    manager_contact_id = Column(
+        Integer,
+        ForeignKey("contact.id", ondelete="SET NULL"),
+        nullable=True,
+        index=True
+    )
+
+    # Self-referential relationship for hierarchy
+    manager = relationship(
+        "Contact",
+        remote_side="Contact.id",
+        foreign_keys=[manager_contact_id],
+        backref="direct_reports"
+    )
+
     conversations = relationship("Conversation", back_populates="contact", cascade="all, delete-orphan")
 
     # preferences relationship

prism_models/content.py

@@ -4,21 +4,18 @@ from typing import Any, Optional
 from sqlalchemy import (
     JSON,
     Boolean,
-    Column,
     Enum,
     Float,
     ForeignKey,
-    Index,
     Integer,
     String,
-    Table,
     Text,
     UniqueConstraint,
 )
 from sqlalchemy.dialects.postgresql import ARRAY
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
-from prism_models.base import Base, BaseModel
+from prism_models.base import BaseModel
 from prism_models.chat import Contact
 from prism_models.qdrant import QdrantVectorPayload, PydanticType
 from prism_models.qdrant import SparseEmbedding
@@ -174,7 +171,6 @@ class Document(BaseModel):
     uploaded_by: Mapped[Optional["Contact"]] = relationship("Contact", foreign_keys=[uploaded_by_id])
     collections: Mapped[list["CollectionDocument"]] = relationship(back_populates="document")
     chunks: Mapped[list["Chunk"]] = relationship(back_populates="document", cascade="all, delete-orphan")
-    qa_pairs: Mapped[list["QAPair"]] = relationship(back_populates="document", cascade="all, delete-orphan")
 
 
     def __repr__(self):
@@ -220,14 +216,6 @@ class ChunkConfig(BaseModel):
     documents: Mapped[list["Document"]] = relationship("Document", back_populates="chunk_config")
 
 
-qa_pair_chunk_association_table = Table(
-    "qa_pair_chunk_association",
-    Base.metadata,
-    Column("qa_pair_id", ForeignKey("qa_pair.id"), primary_key=True),
-    Column("chunk_id", ForeignKey("chunk.id"), primary_key=True),
-)
-
-
 class Chunk(BaseModel):
     document_id: Mapped[int] = mapped_column(ForeignKey("document.id"), nullable=False, index=True)
     chunk_config_id: Mapped[int | None] = mapped_column(ForeignKey("chunk_config.id"))
@@ -238,23 +226,10 @@ class Chunk(BaseModel):
 
     document: Mapped["Document"] = relationship(back_populates="chunks")
     chunk_config: Mapped[Optional["ChunkConfig"]] = relationship()
-    qa_pairs: Mapped[list["QAPair"]] = relationship(secondary=qa_pair_chunk_association_table, back_populates="source_chunks")
     vector: Mapped[Optional["Vector"]] = relationship(back_populates="chunk", uselist=False, cascade="all, delete-orphan")
 
-
-class QAPair(BaseModel):
-    document_id: Mapped[int] = mapped_column(ForeignKey("document.id"), nullable=False, index=True)
-    question: Mapped[str] = mapped_column(Text, nullable=False)
-    answer: Mapped[str] = mapped_column(Text, nullable=False)
-    provenance: Mapped[ProvenanceType] = mapped_column(String(50), default=ProvenanceType.GENERATED)
-    confidence_score: Mapped[float | None] = mapped_column(Float, nullable=True)
-    document: Mapped["Document"] = relationship(back_populates="qa_pairs")
-    source_chunks: Mapped[list["Chunk"]] = relationship(secondary=qa_pair_chunk_association_table, back_populates="qa_pairs")
-    vector: Mapped[Optional["Vector"]] = relationship(back_populates="qa_pair", uselist=False, cascade="all, delete-orphan")
-
 class Vector(BaseModel):
     chunk_id: Mapped[int] = mapped_column(ForeignKey("chunk.id"), nullable=True, unique=True)
-    qa_pair_id: Mapped[int] = mapped_column(ForeignKey("qa_pair.id"), nullable=True, unique=True)
     model: Mapped[str] = mapped_column(String(100))
     qdrant_point_id: Mapped[str | None] = mapped_column(String(100), unique=True)
     
@@ -273,4 +248,3 @@ class Vector(BaseModel):
     )
 
     chunk: Mapped["Chunk"] = relationship(back_populates="vector")
-    qa_pair: Mapped["QAPair"] = relationship(back_populates="vector")

prism_models/feedback.py

@@ -19,7 +19,7 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from prism_models.base import BaseModel
 from prism_models.chat import Contact, ConversationMessage
-from prism_models.content import Chunk, Document, QAPair
+from prism_models.content import Chunk, Document
 
 # MessageFeedback model removed - using new Feedback model below
 

prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py

@@ -0,0 +1,79 @@
+"""REMOVED: QA PAIR CHUNK
+
+Revision ID: 4d9291a3acb1
+Revises: 6abdd3f001ab
+Create Date: 2025-12-17 12:20:30.848462
+
+"""
+from typing import Sequence, Union
+
+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'
+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! ###
+    # 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')
+    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')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    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),
+    sa.Column('answer', sa.TEXT(), autoincrement=False, nullable=False),
+    sa.Column('provenance', sa.VARCHAR(length=50), autoincrement=False, nullable=False),
+    sa.Column('confidence_score', sa.DOUBLE_PRECISION(precision=53), autoincrement=False, nullable=True),
+    sa.Column('id', sa.INTEGER(), server_default=sa.text("nextval('qa_pair_id_seq'::regclass)"), autoincrement=True, nullable=False),
+    sa.Column('created_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), autoincrement=False, nullable=False),
+    sa.Column('updated_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), autoincrement=False, nullable=False),
+    sa.Column('deleted_at', postgresql.TIMESTAMP(timezone=True), autoincrement=False, nullable=True),
+    sa.ForeignKeyConstraint(['document_id'], ['document.id'], name='qa_pair_document_id_fkey'),
+    sa.PrimaryKeyConstraint('id', name='qa_pair_pkey'),
+    postgresql_ignore_search_path=False
+    )
+    op.create_index(op.f('qa_pair_id_idx'), 'qa_pair', ['id'], unique=False)
+    op.create_index(op.f('qa_pair_document_id_idx'), 'qa_pair', ['document_id'], unique=False)
+    op.create_table('qa_pair_chunk_association',
+    sa.Column('qa_pair_id', sa.INTEGER(), autoincrement=False, nullable=False),
+    sa.Column('chunk_id', sa.INTEGER(), autoincrement=False, nullable=False),
+    sa.ForeignKeyConstraint(['chunk_id'], ['chunk.id'], name=op.f('qa_pair_chunk_association_chunk_id_fkey')),
+    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.2.dist-info → prism_models-0.6.4.dist-info}/METADATA

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

{prism_models-0.6.2.dist-info → prism_models-0.6.4.dist-info}/RECORD

@@ -1,10 +1,10 @@
-prism_models/__init__.py,sha256=MmKTV3lCKovI9BZVMtPvBdRDoJwttIbe0NoQdzDcj7k,1442
+prism_models/__init__.py,sha256=Z4IOhsD992zLMKmo9a2eWdcxThSrl2N9igGn_mRKM_4,1420
 prism_models/agent_profile.py,sha256=SSZhrMSGYXr2Skc3niw1SiCZGK5JhwUyyH7Au-UXuPQ,7190
 prism_models/base.py,sha256=Ka8zNToTiTG0jNgzqqj2goGmPXt8zxuuSMYvotnb8wY,2281
-prism_models/chat.py,sha256=giEOHfG-22zmgH9Ca6ydce7pfzuOWSrORZPFEGNrz5g,13967
+prism_models/chat.py,sha256=EwxENfZ2dkAwgrUzqcQZRRzbwiW7J_tJ53qBT2UvsDg,15055
 prism_models/config.py,sha256=gy_6HknnG17Clv8X_EBhaXEBFBLGR1PLjvN5-SFTo5E,962
-prism_models/content.py,sha256=wCBpqj3sxMMeDPmudHXmMcC1ekH7ynK2KO0rhr8QE00,11799
-prism_models/feedback.py,sha256=wOgF_hx0-b0Yn8CrVyC0hWHbcs3jIGVNe11dCv-BsW8,6710
+prism_models/content.py,sha256=mzVZw54UiLIdWB9-A_HdjigJNNWBevmTMbGW6copXBM,10361
+prism_models/feedback.py,sha256=Z2-Fj51xGhoC2j0i6NtkbqnWzPBzJV7F7g9nB7l18I0,6702
 prism_models/qdrant.py,sha256=Ta0fsLsxbT3Evdm8KEo-zw6IzoNckuWDmDZNfBtqjfA,3950
 prism_models/migration/README,sha256=MVlc9TYmr57RbhXET6QxgyCcwWP7w-vLkEsirENqiIQ,38
 prism_models/migration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -63,7 +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-0.6.2.dist-info/METADATA,sha256=-8QMpG3bPU8ThGlOSaL5wVWtN0Ine3c9VnZtCWKhKkE,283
-prism_models-0.6.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prism_models-0.6.2.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
-prism_models-0.6.2.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.4.dist-info/METADATA,sha256=W3MMiA5CcdMhLTsxcRPLD5HYZjBLro5OXoov5VaIofg,283
+prism_models-0.6.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prism_models-0.6.4.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
+prism_models-0.6.4.dist-info/RECORD,,