fix(db): add composite indices to traces and project_sessions (#9045)

Author: RogerHYang

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

@@ -0,0 +1,33 @@
+"""drop single indices from traces and project_sessions
+
+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")
+    op.drop_index("ix_traces_start_time", table_name="traces")
+
+    op.drop_index("ix_project_sessions_project_id", table_name="project_sessions")
+    op.drop_index("ix_project_sessions_start_time", table_name="project_sessions")
+
+
+def downgrade() -> None:
+    op.create_index("ix_traces_project_rowid", "traces", ["project_rowid"])
+    op.create_index("ix_traces_start_time", "traces", ["start_time"])
+
+    op.create_index("ix_project_sessions_project_id", "project_sessions", ["project_id"])
+    op.create_index("ix_project_sessions_start_time", "project_sessions", ["start_time"])

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

@@ -0,0 +1,38 @@
+"""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
+
+import sqlalchemy as sa
+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(
+        sa.text(
+            "CREATE INDEX ix_traces_project_rowid_start_time "
+            "ON traces (project_rowid, start_time DESC)"
+        )
+    )
+    op.execute(
+        sa.text(
+            "CREATE INDEX 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")
+    op.drop_index("ix_project_sessions_project_id_start_time", table_name="project_sessions")