prism-models 0.6.6__py3-none-any.whl → 0.7.3__py3-none-any.whl

prism_models/__init__.py

@@ -3,7 +3,15 @@
 __version__ = "0.1.0"
 
 
-from prism_models.agent_profile import Agent, AgentCollectionAccess, AgentProfile, AgentProfileStatus, Profile, ProfileCollectionAccess
+from prism_models.agent_profile import (
+    Agent,
+    AgentCollectionAccess,
+    AgentProfile,
+    AgentProfileStatus,
+    ContactProfile,
+    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, Source, Vector
@@ -24,6 +32,7 @@ __all__ = [
     "Collection",
     "CollectionDocument",
     "Contact",
+    "ContactProfile",
     "Conversation",
     "ConversationMessage",
     "ConversationMessageMetadata",

prism_models/agent_profile.py

@@ -42,6 +42,37 @@ class Profile(BaseModel):
     def __repr__(self) -> str:
         return f"<Profile(id={self.id}, name='{self.name}', active={self.is_active})>"
 
+
+class ContactProfile(BaseModel):
+    """Association between Contact and Profile."""
+
+    contact_id: Mapped[int] = mapped_column(
+        Integer,
+        ForeignKey("contact.id", ondelete="CASCADE"),
+        nullable=False,
+        index=True,
+    )
+    profile_id: Mapped[int] = mapped_column(
+        Integer,
+        ForeignKey("profile.id", ondelete="CASCADE"),
+        nullable=False,
+        index=True,
+    )
+    assigned_by_contact_id: Mapped[int | None] = mapped_column(
+        Integer,
+        ForeignKey("contact.id", ondelete="SET NULL"),
+        nullable=True,
+        index=True,
+    )
+
+    contact: Mapped["Contact"] = relationship(foreign_keys=[contact_id], back_populates="contact_profiles")
+    profile: Mapped["Profile"] = relationship()
+    assigned_by_contact: Mapped["Contact"] = relationship(foreign_keys=[assigned_by_contact_id])
+
+    __table_args__ = (
+        UniqueConstraint("contact_id", "profile_id", name="uq_contact_profile"),
+    )
+
 class AgentType(str, Enum):
     ORCHESTRATOR_AGENT = "ORCHESTRATOR"
     DOMAIN_AGENT = "DOMAIN"

prism_models/chat.py

@@ -39,7 +39,7 @@ class ContactRole(PyEnum):
     ADMIN = "ADMIN"
     MANAGER = "MANAGER"
     USER = "USER"
-    
+
 
 
 class ConversationType(PyEnum):
@@ -81,6 +81,7 @@ class Contact(ChatSchemaMixin, BaseModel):
     Relationships:
         - conversations: One-to-many relationship with Conversation model
         - user_preferences: One-to-one relationship with UserPreferences model
+        - contact_profiles: One-to-many relationship with ContactProfile model
     """
 
     email = Column(String(300), nullable=True)
@@ -111,7 +112,7 @@ class Contact(ChatSchemaMixin, BaseModel):
         Enum(ContactRole),
         default=ContactRole.USER,
         nullable=False,
-        server_default="VIEWER"
+        server_default="USER"
     )
     manager_contact_id = Column(
         Integer,
@@ -137,6 +138,12 @@ class Contact(ChatSchemaMixin, BaseModel):
         cascade="all, delete-orphan",
         uselist=False  # One-to-one relationship
     )
+    contact_profiles: Mapped[list["ContactProfile"]] = relationship(
+        "ContactProfile",
+        back_populates="contact",
+        cascade="all, delete-orphan",
+        foreign_keys="ContactProfile.contact_id",
+    )
 
 
 class UserPreferences(BaseModel):

prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py

@@ -37,7 +37,7 @@ def upgrade() -> None:
     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_index(op.f('idx_document_title_trgm'), table_name='document', postgresql_ops={'title': 'gin_trgm_ops'}, postgresql_using='gin', if_exists=True)
     # ### end Alembic commands ###
 
 

prism_models/migration/versions/2026_01_14_1742_147509a82af8_changed_user_role_enums.py

@@ -0,0 +1,54 @@
+"""Changed user role enums
+
+Revision ID: 147509a82af8
+Revises: 4d9291a3acb1
+Create Date: 2026-01-14 17:42:10.143289
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '147509a82af8'
+down_revision: Union[str, Sequence[str], None] = '4d9291a3acb1'
+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.execute("CREATE TYPE contactrole_new AS ENUM ('ADMIN', 'MANAGER', 'USER')")
+    op.execute("ALTER TABLE contact ALTER COLUMN contact_role DROP DEFAULT")
+    op.execute(
+        "ALTER TABLE contact ALTER COLUMN contact_role TYPE contactrole_new "
+        "USING (CASE "
+        "WHEN contact_role::text IN ('ADMIN', 'MANAGER') THEN contact_role::text "
+        "ELSE 'USER' "
+        "END)::contactrole_new"
+    )
+    op.execute("ALTER TABLE contact ALTER COLUMN contact_role SET DEFAULT 'USER'")
+    op.execute("DROP TYPE contactrole")
+    op.execute("ALTER TYPE contactrole_new RENAME TO contactrole")
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.execute("CREATE TYPE contactrole_old AS ENUM ('ADMIN', 'ACCOUNT_ADMIN', 'MANAGER', 'VIEWER')")
+    op.execute("ALTER TABLE contact ALTER COLUMN contact_role DROP DEFAULT")
+    op.execute(
+        "ALTER TABLE contact ALTER COLUMN contact_role TYPE contactrole_old "
+        "USING (CASE "
+        "WHEN contact_role::text IN ('ADMIN', 'MANAGER') THEN contact_role::text "
+        "ELSE 'VIEWER' "
+        "END)::contactrole_old"
+    )
+    op.execute("ALTER TABLE contact ALTER COLUMN contact_role SET DEFAULT 'VIEWER'")
+    op.execute("DROP TYPE contactrole")
+    op.execute("ALTER TYPE contactrole_old RENAME TO contactrole")
+    # ### end Alembic commands ###

prism_models/migration/versions/2026_01_15_1010_3b7d0b2e9f5a_add_contact_profile_table.py

@@ -0,0 +1,71 @@
+"""Add contact_profile table
+
+Revision ID: 3b7d0b2e9f5a
+Revises: 147509a82af8
+Create Date: 2026-01-15 10:10:00.000000
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+
+# revision identifiers, used by Alembic.
+revision: str = "3b7d0b2e9f5a"
+down_revision: Union[str, Sequence[str], None] = "147509a82af8"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    op.create_table(
+        "contact_profile",
+        sa.Column("contact_id", sa.Integer(), nullable=False),
+        sa.Column("profile_id", sa.Integer(), nullable=False),
+        sa.Column("assigned_by_contact_id", sa.Integer(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["assigned_by_contact_id"],
+            ["contact.id"],
+            name=op.f("contact_profile_assigned_by_contact_id_fkey"),
+            ondelete="SET NULL",
+        ),
+        sa.ForeignKeyConstraint(
+            ["contact_id"],
+            ["contact.id"],
+            name=op.f("contact_profile_contact_id_fkey"),
+            ondelete="CASCADE",
+        ),
+        sa.ForeignKeyConstraint(
+            ["profile_id"],
+            ["profile.id"],
+            name=op.f("contact_profile_profile_id_fkey"),
+            ondelete="CASCADE",
+        ),
+        sa.PrimaryKeyConstraint("id", name=op.f("contact_profile_pkey")),
+        sa.UniqueConstraint("contact_id", "profile_id", name="uq_contact_profile"),
+    )
+    op.create_index(
+        op.f("contact_profile_assigned_by_contact_id_idx"),
+        "contact_profile",
+        ["assigned_by_contact_id"],
+        unique=False,
+    )
+    op.create_index(op.f("contact_profile_contact_id_idx"), "contact_profile", ["contact_id"], unique=False)
+    op.create_index(op.f("contact_profile_id_idx"), "contact_profile", ["id"], unique=False)
+    op.create_index(op.f("contact_profile_profile_id_idx"), "contact_profile", ["profile_id"], unique=False)
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    op.drop_index(op.f("contact_profile_profile_id_idx"), table_name="contact_profile")
+    op.drop_index(op.f("contact_profile_id_idx"), table_name="contact_profile")
+    op.drop_index(op.f("contact_profile_contact_id_idx"), table_name="contact_profile")
+    op.drop_index(op.f("contact_profile_assigned_by_contact_id_idx"), table_name="contact_profile")
+    op.drop_table("contact_profile")

{prism_models-0.6.6.dist-info → prism_models-0.7.3.dist-info}/METADATA

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

{prism_models-0.6.6.dist-info → prism_models-0.7.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
-prism_models/__init__.py,sha256=Z4IOhsD992zLMKmo9a2eWdcxThSrl2N9igGn_mRKM_4,1420
-prism_models/agent_profile.py,sha256=SSZhrMSGYXr2Skc3niw1SiCZGK5JhwUyyH7Au-UXuPQ,7190
+prism_models/__init__.py,sha256=jD-ASYrK4-FW1txMuUnUCwvZCHN7d9ko8b_lCLuZS9s,1491
+prism_models/agent_profile.py,sha256=sNO9NfOpOFXgpkvxMry-5rcHNfFdWXtMV6b45JA0Wr8,8159
 prism_models/base.py,sha256=Ka8zNToTiTG0jNgzqqj2goGmPXt8zxuuSMYvotnb8wY,2281
-prism_models/chat.py,sha256=IktJGUCKqrSWLBSD6XpuZBSbuMUYVNApu_BEZtjf6A8,14909
+prism_models/chat.py,sha256=0nsp1u-zzibIYHEAvuz_lu8qzulzzVD7vOjSrwcHMjI,15205
 prism_models/config.py,sha256=gy_6HknnG17Clv8X_EBhaXEBFBLGR1PLjvN5-SFTo5E,962
 prism_models/content.py,sha256=mzVZw54UiLIdWB9-A_HdjigJNNWBevmTMbGW6copXBM,10361
 prism_models/feedback.py,sha256=Z2-Fj51xGhoC2j0i6NtkbqnWzPBzJV7F7g9nB7l18I0,6702
@@ -63,8 +63,10 @@ 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=QnbVia8YU2_Rp0LWFvbA2lUy7kmdrGLbDmTHw-YQAoM,4752
-prism_models-0.6.6.dist-info/METADATA,sha256=HDaddbPooYJE6IPRyngjbuo3o5QnEwFqcrDlxhaA_Zk,283
-prism_models-0.6.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prism_models-0.6.6.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
-prism_models-0.6.6.dist-info/RECORD,,
+prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py,sha256=p0WnO1SYNqtFum15k5BwPwIWEUuGZBpEbLEtbDx-z8w,4768
+prism_models/migration/versions/2026_01_14_1742_147509a82af8_changed_user_role_enums.py,sha256=S4ZSqgbCe0Xrp3Z1E10J1vg3Arh_bmCxk7YP1g67eDQ,2000
+prism_models/migration/versions/2026_01_15_1010_3b7d0b2e9f5a_add_contact_profile_table.py,sha256=5G-ZmYZGXJ4TNdZms3WDwhHKbUMVp2M94f5uBSVW-j4,2790
+prism_models-0.7.3.dist-info/METADATA,sha256=Xw0NmGjf3n6i9J7GCW0H2Y7LoYMGklO5Agn2Brsb9nc,283
+prism_models-0.7.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prism_models-0.7.3.dist-info/top_level.txt,sha256=UNzqwpLgFYU0EoyB9LiB1jTtc89A1sQ24fSEyNVvgJI,13
+prism_models-0.7.3.dist-info/RECORD,,