feat: composite index on spans table

Author: RogerHYang

scripts/ddl/postgresql_schema.sql

@@ -529,6 +529,13 @@ create index ix_latency
 create index ix_cumulative_llm_token_count_total
     on public.spans ((cumulative_llm_token_count_prompt + cumulative_llm_token_count_completion));
 
+create index ix_spans_trace_rowid_start_time
+    on public.spans (trace_rowid, start_time);
+
+create index ix_spans_trace_rowid_where_parent_id_is_null
+    on public.spans (trace_rowid, start_time)
+    where (parent_id IS NULL);
+
 create table public.span_annotations
 (
     id             serial

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

@@ -1,4 +1,4 @@
-"""drop single indices from traces and project_sessions
+"""drop single indices from traces, project_sessions, and spans
 
 Revision ID: 272b66ff50f8
 Revises: a20694b15f82
@@ -24,10 +24,14 @@ def upgrade() -> None:
     op.drop_index("ix_project_sessions_project_id", table_name="project_sessions")
     op.drop_index("ix_project_sessions_start_time", table_name="project_sessions")
 
+    op.drop_index("ix_spans_trace_rowid", table_name="spans")
+
 
 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"])
+
+    op.create_index("ix_spans_trace_rowid", "spans", ["trace_rowid"])

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

@@ -1,4 +1,4 @@
-"""add composite indices to traces and project_sessions
+"""add composite indices to traces, project_sessions, and partial index to spans
 
 Revision ID: 735d3d93c33e
 Revises: 272b66ff50f8
@@ -31,8 +31,19 @@ def upgrade() -> None:
             "ON project_sessions (project_id, start_time DESC)"
         )
     )
+    op.execute(
+        sa.text("CREATE INDEX ix_spans_trace_rowid_start_time ON spans (trace_rowid, start_time)")
+    )
+    op.execute(
+        sa.text(
+            "CREATE INDEX ix_spans_trace_rowid_start_time_where_parent_id_is_null "
+            "ON spans (trace_rowid, start_time) WHERE parent_id IS NULL"
+        )
+    )
 
 
 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")
+    op.drop_index("ix_spans_trace_rowid_start_time", table_name="spans")
+    op.drop_index("ix_spans_trace_rowid_where_parent_id_is_null", table_name="spans")