prism-models 0.6.5__tar.gz → 0.7.3__tar.gz

{prism_models-0.6.5 → prism_models-0.7.3}/PKG-INFO

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

prism_models-0.7.3/README.md

@@ -0,0 +1,154 @@
+# Installing prism-models from AWS CodeArtifact
+
+This package is published to AWS CodeArtifact. Projects that depend on it need to configure their environment to authenticate with the private registry.
+
+## Configuration
+
+### AWS CodeArtifact Details
+
+- **Domain**: `globalrescue`
+- **Domain Owner**: `687144405948`
+- **Region**: `us-east-2`
+- **Repository URL**: `https://globalrescue-687144405948.d.codeartifact.us-east-2.amazonaws.com/pypi/pypi/simple/`
+
+---
+
+## For Projects Using `uv`
+
+### 1. Add the index to `pyproject.toml`
+
+```toml
+[[tool.uv.index]]
+name = "private"
+url = "https://globalrescue-687144405948.d.codeartifact.us-east-2.amazonaws.com/pypi/pypi/simple/"
+default = true
+
+[[tool.uv.index]]
+name = "pypi"
+url = "https://pypi.org/simple"
+```
+
+### 2. Authenticate before installing
+
+**Local development (SSO):**
+```bash
+aws sso login --profile <your-profile>
+export AWS_PROFILE=<your-profile>
+
+# Get token and set it
+export UV_INDEX_PRIVATE_PASSWORD=$(aws codeartifact get-authorization-token \
+    --domain globalrescue \
+    --domain-owner 687144405948 \
+    --region us-east-2 \
+    --query authorizationToken \
+    --output text)
+
+# Now install
+uv sync
+```
+
+**CI/CD (with IAM role):**
+```bash
+export UV_INDEX_PRIVATE_PASSWORD=$(aws codeartifact get-authorization-token \
+    --domain globalrescue \
+    --domain-owner 687144405948 \
+    --region us-east-2 \
+    --query authorizationToken \
+    --output text)
+
+uv sync
+```
+
+---
+
+## For Projects Using `pip`
+
+### 1. Get the repository URL with credentials
+
+```bash
+aws codeartifact login --tool pip \
+    --domain globalrescue \
+    --domain-owner 687144405948 \
+    --region us-east-2 \
+    --repository pypi
+```
+
+This automatically configures pip to use CodeArtifact for the session.
+
+### 2. Or manually configure pip
+
+```bash
+# Get the token
+TOKEN=$(aws codeartifact get-authorization-token \
+    --domain globalrescue \
+    --domain-owner 687144405948 \
+    --region us-east-2 \
+    --query authorizationToken \
+    --output text)
+
+# Install with extra index
+pip install prism-models \
+    --index-url https://aws:${TOKEN}@globalrescue-687144405948.d.codeartifact.us-east-2.amazonaws.com/pypi/pypi/simple/ \
+    --extra-index-url https://pypi.org/simple/
+```
+
+---
+
+## GitLab CI/CD Example
+
+```yaml
+install-dependencies:
+  script:
+    - |
+      export UV_INDEX_PRIVATE_PASSWORD=$(aws codeartifact get-authorization-token \
+        --domain globalrescue \
+        --domain-owner 687144405948 \
+        --region us-east-2 \
+        --query authorizationToken \
+        --output text)
+    - uv sync
+```
+
+---
+
+## Required AWS Permissions
+
+The IAM user/role needs these permissions:
+
+```json
+{
+  "Version": "2012-10-17",
+  "Statement": [
+    {
+      "Effect": "Allow",
+      "Action": [
+        "codeartifact:GetAuthorizationToken",
+        "codeartifact:GetRepositoryEndpoint",
+        "codeartifact:ReadFromRepository"
+      ],
+      "Resource": "*"
+    },
+    {
+      "Effect": "Allow",
+      "Action": "sts:GetServiceBearerToken",
+      "Resource": "*",
+      "Condition": {
+        "StringEquals": {
+          "sts:AWSServiceName": "codeartifact.amazonaws.com"
+        }
+      }
+    }
+  ]
+}
+```
+
+---
+
+## Troubleshooting
+
+| Error | Solution |
+|-------|----------|
+| `401 Unauthorized` | Token expired - run `aws codeartifact get-authorization-token` again |
+| `ExpiredTokenException` | Run `aws sso login` again (tokens last 12 hours by default) |
+| `Could not find a version` | Check the package name and that it's published to CodeArtifact |
+| `Access Denied` | Check IAM permissions for CodeArtifact access |

{prism_models-0.6.5 → prism_models-0.7.3}/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-0.6.5 → prism_models-0.7.3}/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-0.6.5 → prism_models-0.7.3}/prism_models/chat.py

@@ -32,15 +32,14 @@ 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: To manage user roles,  Docs, Collections and profiles
+    - MANAGER: To provide feedback to GR-ACE
+    - USER: Ability to chat with GR-ACE
     """
     ADMIN = "ADMIN"
-    ACCOUNT_ADMIN = "ACCOUNT_ADMIN"
     MANAGER = "MANAGER"
-    VIEWER = "VIEWER"
+    USER = "USER"
+
 
 
 class ConversationType(PyEnum):
@@ -82,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)
@@ -110,9 +110,9 @@ class Contact(ChatSchemaMixin, BaseModel):
     # Hierarchical permission system
     contact_role: Mapped[ContactRole] = mapped_column(
         Enum(ContactRole),
-        default=ContactRole.VIEWER,
+        default=ContactRole.USER,
         nullable=False,
-        server_default="VIEWER"
+        server_default="USER"
     )
     manager_contact_id = Column(
         Integer,
@@ -138,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-0.6.5 → prism_models-0.7.3}/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-0.7.3/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-0.7.3/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.5 → prism_models-0.7.3}/prism_models.egg-info/PKG-INFO

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

@@ -70,4 +70,6 @@ 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
 prism_models/migration/versions/2025_12_07_2236_88f01df468f2_added_sparse_embeddings_field_in_vector_.py
 prism_models/migration/versions/2025_12_09_1146_6abdd3f001ab_adding_orchestrator_model_to_profile_.py
-prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py
+prism_models/migration/versions/2025_12_17_1220_4d9291a3acb1_removed_qa_pair_chunk.py
+prism_models/migration/versions/2026_01_14_1742_147509a82af8_changed_user_role_enums.py
+prism_models/migration/versions/2026_01_15_1010_3b7d0b2e9f5a_add_contact_profile_table.py

{prism_models-0.6.5 → prism_models-0.7.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "prism-models"
-version = "0.6.5"
+version = "0.7.3"
 requires-python = ">=3.12"
 dependencies = [
   "sqlalchemy[asyncio]>=2.0.0",
@@ -20,3 +20,14 @@ packages = ["prism_models"]
 
 [tool.setuptools.package-data]
 prism_models = ["migration/*", "migration/versions/*"]
+
+[[tool.uv.index]]
+name = "private"
+url = "https://globalrescue-687144405948.d.codeartifact.us-east-2.amazonaws.com/pypi/pypi/simple/"
+publish-url = "https://globalrescue-687144405948.d.codeartifact.us-east-2.amazonaws.com/pypi/pypi/"
+default = true
+
+[[tool.uv.index]]
+name = "pypi"
+url = "https://pypi.org/simple"
+