arize-phoenix 11.38.0__py3-none-any.whl → 12.2.0__py3-none-any.whl

phoenix/db/migrations/versions/0df286449799_add_session_annotations_table.py

@@ -0,0 +1,105 @@
+"""add session annotations table
+
+Revision ID: 0df286449799
+Revises: 735d3d93c33e
+Create Date: 2025-08-06 11:27:01.479664
+"""
+
+from typing import Any, Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy import JSON
+from sqlalchemy.dialects import postgresql
+from sqlalchemy.ext.compiler import compiles
+
+# revision identifiers, used by Alembic.
+revision: str = "0df286449799"
+down_revision: Union[str, None] = "735d3d93c33e"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+class JSONB(JSON):
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    __visit_name__ = "JSONB"
+
+
+@compiles(JSONB, "sqlite")
+def _(*args: Any, **kwargs: Any) -> str:
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    return "JSONB"
+
+
+JSON_ = (
+    JSON()
+    .with_variant(
+        postgresql.JSONB(),
+        "postgresql",
+    )
+    .with_variant(
+        JSONB(),
+        "sqlite",
+    )
+)
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+def upgrade() -> None:
+    op.create_table(
+        "project_session_annotations",
+        sa.Column("id", _Integer, primary_key=True),
+        sa.Column(
+            "project_session_id",
+            _Integer,
+            sa.ForeignKey("project_sessions.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.Column("name", sa.String, nullable=False),
+        sa.Column("label", sa.String),
+        sa.Column("score", sa.Float),
+        sa.Column("explanation", sa.String),
+        sa.Column("metadata", JSON_, nullable=False),
+        sa.Column(
+            "annotator_kind",
+            sa.String,
+            sa.CheckConstraint(
+                "annotator_kind IN ('LLM', 'CODE', 'HUMAN')",
+                name="valid_annotator_kind",
+            ),
+            nullable=False,
+        ),
+        sa.Column(
+            "user_id",
+            _Integer,
+            sa.ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True,
+        ),
+        sa.Column("identifier", sa.String, server_default="", nullable=False),
+        sa.Column(
+            "source",
+            sa.String,
+            sa.CheckConstraint("source IN ('API', 'APP')", name="valid_source"),
+            nullable=False,
+        ),
+        sa.Column(
+            "created_at", sa.TIMESTAMP(timezone=True), server_default=sa.func.now(), nullable=False
+        ),
+        sa.Column(
+            "updated_at",
+            sa.TIMESTAMP(timezone=True),
+            server_default=sa.func.now(),
+            onupdate=sa.func.now(),
+            nullable=False,
+        ),
+        sa.UniqueConstraint("name", "project_session_id", "identifier"),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("project_session_annotations")

phoenix/db/migrations/versions/272b66ff50f8_drop_single_indices.py

@@ -0,0 +1,119 @@
+"""drop single indices from traces, project_sessions, and experiment_runs
+
+Revision ID: 272b66ff50f8
+Revises: a20694b15f82
+Create Date: 2025-08-11 20:37:46.941940
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "272b66ff50f8"
+down_revision: Union[str, None] = "a20694b15f82"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.drop_index(
+        "ix_traces_project_rowid",
+        table_name="traces",
+        if_exists=True,
+    )
+    op.drop_index(
+        "ix_traces_start_time",
+        table_name="traces",
+        if_exists=True,
+    )
+
+    op.drop_index(
+        "ix_project_sessions_project_id",
+        table_name="project_sessions",
+        if_exists=True,
+    )
+    op.drop_index(
+        "ix_project_sessions_start_time",
+        table_name="project_sessions",
+        if_exists=True,
+    )
+
+    op.drop_index(
+        "ix_experiment_runs_experiment_id",
+        table_name="experiment_runs",
+        if_exists=True,
+    )
+    op.drop_index(
+        "ix_experiment_run_annotations_experiment_run_id",
+        table_name="experiment_run_annotations",
+        if_exists=True,
+    )
+
+    op.drop_index(
+        "ix_dataset_example_revisions_dataset_example_id",
+        table_name="dataset_example_revisions",
+        if_exists=True,
+    )
+
+    op.drop_index(
+        "ix_span_cost_details_span_cost_id",
+        table_name="span_cost_details",
+        if_exists=True,
+    )
+
+
+def downgrade() -> None:
+    op.create_index(
+        "ix_traces_project_rowid",
+        "traces",
+        ["project_rowid"],
+        if_not_exists=True,
+    )
+    op.create_index(
+        "ix_traces_start_time",
+        "traces",
+        ["start_time"],
+        if_not_exists=True,
+    )
+
+    op.create_index(
+        "ix_project_sessions_project_id",
+        "project_sessions",
+        ["project_id"],
+        if_not_exists=True,
+    )
+    op.create_index(
+        "ix_project_sessions_start_time",
+        "project_sessions",
+        ["start_time"],
+        if_not_exists=True,
+    )
+
+    op.create_index(
+        "ix_experiment_runs_experiment_id",
+        "experiment_runs",
+        ["experiment_id"],
+        if_not_exists=True,
+    )
+    op.create_index(
+        "ix_experiment_run_annotations_experiment_run_id",
+        "experiment_run_annotations",
+        ["experiment_run_id"],
+        if_not_exists=True,
+    )
+
+    op.create_index(
+        "ix_dataset_example_revisions_dataset_example_id",
+        "dataset_example_revisions",
+        ["dataset_example_id"],
+        if_not_exists=True,
+    )
+
+    op.create_index(
+        "ix_span_cost_details_span_cost_id",
+        "span_cost_details",
+        ["span_cost_id"],
+        if_not_exists=True,
+    )

phoenix/db/migrations/versions/58228d933c91_dataset_labels.py

@@ -0,0 +1,67 @@
+"""dataset_labels
+
+Revision ID: 58228d933c91
+Revises: 699f655af132
+Create Date: 2025-09-05 17:47:34.637329
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "58228d933c91"
+down_revision: Union[str, None] = "699f655af132"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+def upgrade() -> None:
+    op.create_table(
+        "dataset_labels",
+        sa.Column("id", _Integer, primary_key=True),
+        sa.Column("name", sa.String, nullable=False, unique=True),
+        sa.Column("description", sa.String, nullable=True),
+        sa.Column("color", sa.String, nullable=False),
+        sa.Column(
+            "user_id",
+            _Integer,
+            sa.ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True,
+            index=True,
+        ),
+    )
+
+    op.create_table(
+        "datasets_dataset_labels",
+        sa.Column(
+            "dataset_id",
+            _Integer,
+            sa.ForeignKey("datasets.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column(
+            "dataset_label_id",
+            _Integer,
+            sa.ForeignKey("dataset_labels.id", ondelete="CASCADE"),
+            nullable=False,
+            # index on the second element of the composite primary key
+            index=True,
+        ),
+        sa.PrimaryKeyConstraint(
+            "dataset_id",
+            "dataset_label_id",
+        ),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("datasets_dataset_labels")
+    op.drop_table("dataset_labels")

phoenix/db/migrations/versions/699f655af132_experiment_tags.py

@@ -0,0 +1,57 @@
+"""experiment_tags
+
+Revision ID: 699f655af132
+Revises: d0690a79ea51
+Create Date: 2025-09-05 13:14:22.676233
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "699f655af132"
+down_revision: Union[str, None] = "d0690a79ea51"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+def upgrade() -> None:
+    op.create_table(
+        "experiment_tags",
+        sa.Column("id", _Integer, primary_key=True),
+        sa.Column(
+            "experiment_id",
+            _Integer,
+            sa.ForeignKey("experiments.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.Column(
+            "dataset_id",
+            _Integer,
+            sa.ForeignKey("datasets.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column(
+            "user_id",
+            _Integer,
+            sa.ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True,
+            index=True,
+        ),
+        sa.Column("name", sa.String, nullable=False),
+        sa.Column("description", sa.String, nullable=True),
+        sa.UniqueConstraint("dataset_id", "name"),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("experiment_tags")

phoenix/db/migrations/versions/735d3d93c33e_add_composite_indices.py

@@ -0,0 +1,41 @@
+"""add composite indices to traces and project_sessions
+
+Revision ID: 735d3d93c33e
+Revises: 272b66ff50f8
+Create Date: 2025-08-11 20:52:47.477712
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "735d3d93c33e"
+down_revision: Union[str, None] = "272b66ff50f8"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.execute(
+        "CREATE INDEX IF NOT EXISTS ix_traces_project_rowid_start_time "
+        "ON traces (project_rowid, start_time DESC)"
+    )
+    op.execute(
+        "CREATE INDEX IF NOT EXISTS ix_project_sessions_project_id_start_time "
+        "ON project_sessions (project_id, start_time DESC)"
+    )
+
+
+def downgrade() -> None:
+    op.drop_index(
+        "ix_traces_project_rowid_start_time",
+        table_name="traces",
+        if_exists=True,
+    )
+    op.drop_index(
+        "ix_project_sessions_project_id_start_time",
+        table_name="project_sessions",
+        if_exists=True,
+    )

phoenix/db/migrations/versions/ab513d89518b_add_user_id_on_dataset_versions.py

@@ -0,0 +1,40 @@
+"""add user_id on dataset_versions
+
+Revision ID: ab513d89518b
+Revises: 01a8342c9cdf
+Create Date: 2025-09-26 11:00:06.961920
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "ab513d89518b"
+down_revision: Union[str, None] = "01a8342c9cdf"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+def upgrade() -> None:
+    with op.batch_alter_table("dataset_versions") as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "user_id",
+                _Integer,
+                sa.ForeignKey("users.id", ondelete="SET NULL"),
+                nullable=True,
+            ),
+        )
+
+
+def downgrade() -> None:
+    with op.batch_alter_table("dataset_versions") as batch_op:
+        batch_op.drop_column("user_id")

phoenix/db/migrations/versions/d0690a79ea51_users_on_experiments.py

@@ -0,0 +1,40 @@
+"""add user_id on experiments
+
+Revision ID: d0690a79ea51
+Revises: ab513d89518b
+Create Date: 2025-08-26 19:12:47.849806
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "d0690a79ea51"
+down_revision: Union[str, None] = "ab513d89518b"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+def upgrade() -> None:
+    with op.batch_alter_table("experiments") as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "user_id",
+                _Integer,
+                sa.ForeignKey("users.id", ondelete="SET NULL"),
+                nullable=True,
+            ),
+        )
+
+
+def downgrade() -> None:
+    with op.batch_alter_table("experiments") as batch_op:
+        batch_op.drop_column("user_id")

phoenix/db/migrations/versions/deb2c81c0bb2_dataset_splits.py

@@ -0,0 +1,139 @@
+"""dataset_splits
+
+Revision ID: deb2c81c0bb2
+Revises: 58228d933c91
+Create Date: 2025-09-08 15:50:12.066217
+
+"""
+
+from typing import Any, Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy import JSON
+from sqlalchemy.dialects import postgresql
+from sqlalchemy.ext.compiler import compiles
+
+
+class JSONB(JSON):
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    __visit_name__ = "JSONB"
+
+
+@compiles(JSONB, "sqlite")
+def _(*args: Any, **kwargs: Any) -> str:
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    return "JSONB"
+
+
+JSON_ = (
+    JSON()
+    .with_variant(
+        postgresql.JSONB(),
+        "postgresql",
+    )
+    .with_variant(
+        JSONB(),
+        "sqlite",
+    )
+)
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+
+# revision identifiers, used by Alembic.
+revision: str = "deb2c81c0bb2"
+down_revision: Union[str, None] = "e76cbd66ffc3"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # Create dataset_splits table
+    op.create_table(
+        "dataset_splits",
+        sa.Column("id", _Integer, primary_key=True),
+        sa.Column(
+            "user_id",
+            _Integer,
+            sa.ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True,
+            index=True,
+        ),
+        sa.Column("name", sa.String, nullable=False, unique=True),
+        sa.Column("description", sa.String, nullable=True),
+        sa.Column("color", sa.String, nullable=False),
+        sa.Column("metadata", JSON_, nullable=False),
+        sa.Column(
+            "created_at",
+            sa.TIMESTAMP(timezone=True),
+            nullable=False,
+            server_default=sa.func.now(),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.TIMESTAMP(timezone=True),
+            nullable=False,
+            server_default=sa.func.now(),
+            onupdate=sa.func.now(),
+        ),
+    )
+
+    # Create crosswalk table: dataset_splits_dataset_examples
+    op.create_table(
+        "dataset_splits_dataset_examples",
+        sa.Column(
+            "dataset_split_id",
+            _Integer,
+            sa.ForeignKey("dataset_splits.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column(
+            "dataset_example_id",
+            _Integer,
+            sa.ForeignKey("dataset_examples.id", ondelete="CASCADE"),
+            nullable=False,
+            # index on the second element of the composite primary key
+            index=True,
+        ),
+        sa.PrimaryKeyConstraint(
+            "dataset_split_id",
+            "dataset_example_id",
+        ),
+    )
+
+    # Create experiments_dataset_splits table The rational of this table is to
+    # gather examples for a specific dataset split for a specific experiment.
+    # Select all dataset examples where examples belong to a dataset split and
+    # examples belong to a experiment.
+
+    op.create_table(
+        "experiments_dataset_splits",
+        sa.Column(
+            "experiment_id",
+            _Integer,
+            sa.ForeignKey("experiments.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column(
+            "dataset_split_id",
+            _Integer,
+            sa.ForeignKey("dataset_splits.id", ondelete="CASCADE"),
+            nullable=False,
+            # index on the second element of the composite primary key
+            index=True,
+        ),
+        sa.PrimaryKeyConstraint(
+            "experiment_id",
+            "dataset_split_id",
+        ),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("experiments_dataset_splits")
+    op.drop_table("dataset_splits_dataset_examples")
+    op.drop_table("dataset_splits")

phoenix/db/migrations/versions/e76cbd66ffc3_add_experiments_dataset_examples.py

@@ -0,0 +1,87 @@
+"""add experiments_dataset_examples junction table
+
+Revision ID: e76cbd66ffc3
+Revises: deb2c81c0bb2
+Create Date: 2025-09-23 12:33:13.554164
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+_Integer = sa.Integer().with_variant(
+    sa.BigInteger(),
+    "postgresql",
+)
+
+# revision identifiers, used by Alembic.
+revision: str = "e76cbd66ffc3"
+down_revision: Union[str, None] = "58228d933c91"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+BACKFILL = """\
+INSERT INTO experiments_dataset_examples (
+    experiment_id,
+    dataset_example_id,
+    dataset_example_revision_id
+)
+SELECT
+    ranked.experiment_id,
+    ranked.dataset_example_id,
+    ranked.dataset_example_revision_id
+FROM (
+    SELECT
+        e.id as experiment_id,
+        der.dataset_example_id,
+        der.id as dataset_example_revision_id,
+        der.revision_kind,
+        ROW_NUMBER() OVER (
+            PARTITION BY e.id, der.dataset_example_id
+            ORDER BY der.dataset_version_id DESC
+        ) as rn
+    FROM experiments e
+        JOIN dataset_examples de ON de.dataset_id = e.dataset_id
+        JOIN dataset_example_revisions der ON der.dataset_example_id = de.id
+    WHERE der.dataset_version_id <= e.dataset_version_id
+) ranked
+WHERE ranked.rn = 1
+    AND ranked.revision_kind != 'DELETE'
+"""
+
+
+def upgrade() -> None:
+    op.create_table(
+        "experiments_dataset_examples",
+        sa.Column(
+            "experiment_id",
+            _Integer,
+            sa.ForeignKey("experiments.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column(
+            "dataset_example_id",
+            _Integer,
+            sa.ForeignKey("dataset_examples.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.Column(
+            "dataset_example_revision_id",
+            _Integer,
+            sa.ForeignKey("dataset_example_revisions.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.PrimaryKeyConstraint(
+            "experiment_id",
+            "dataset_example_id",
+        ),
+    )
+    op.execute(BACKFILL)
+
+
+def downgrade() -> None:
+    op.drop_table("experiments_dataset_examples")