arize-phoenix 6.2.0__tar.gz → 7.0.1__tar.gz

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 6.2.0
+Version: 7.0.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -44,7 +44,7 @@ Requires-Dist: scipy
 Requires-Dist: sqlalchemy[asyncio]<3,>=2.0.4
 Requires-Dist: sqlean-py>=3.45.1
 Requires-Dist: starlette
-Requires-Dist: strawberry-graphql==0.243.1
+Requires-Dist: strawberry-graphql==0.247.0
 Requires-Dist: tqdm
 Requires-Dist: typing-extensions>=4.6
 Requires-Dist: uvicorn
@@ -66,7 +66,7 @@ Requires-Dist: opentelemetry-sdk; extra == 'container'
 Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
-Requires-Dist: strawberry-graphql[opentelemetry]==0.243.1; extra == 'container'
+Requires-Dist: strawberry-graphql[opentelemetry]==0.247.0; extra == 'container'
 Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
@@ -97,7 +97,7 @@ Requires-Dist: pytest-postgresql; extra == 'dev'
 Requires-Dist: pytest-xdist; extra == 'dev'
 Requires-Dist: pytest==8.3.3; extra == 'dev'
 Requires-Dist: ruff==0.6.9; extra == 'dev'
-Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.243.1; extra == 'dev'
+Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.247.0; extra == 'dev'
 Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
   "starlette",
   "uvicorn",
   "psutil",
-  "strawberry-graphql==0.243.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql==0.247.0",  # need to pin version because we're monkey-patching
   "pyarrow",
   "typing-extensions>=4.6",
   "scipy",
@@ -87,7 +87,7 @@ dev = [
   "pytest-postgresql",
   "asyncpg",
   "psycopg[binary,pool]",
-  "strawberry-graphql[debug-server,opentelemetry]==0.243.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[debug-server,opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
   "pre-commit",
   "arize[AutoEmbeddings, LLM_Evaluation]",
   "llama-index>=0.10.3",
@@ -133,7 +133,7 @@ container = [
   "opentelemetry-instrumentation-sqlalchemy",
   "opentelemetry-instrumentation-grpc",
   "py-grpc-prometheus",
-  "strawberry-graphql[opentelemetry]==0.243.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
   "uvloop; platform_system != 'Windows'",
   "fast-hdbscan>=0.2.0",
   "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
@@ -229,7 +229,7 @@ dependencies = [
   "py-grpc-prometheus",
   "pypistats",  # this is needed to type-check third-party packages
   "requests",  # this is needed to type-check third-party packages
-  "strawberry-graphql[opentelemetry]==0.243.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
   "tenacity",
   "types-cachetools",
   "types-protobuf",

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/src/phoenix/db/engines.py

@@ -163,11 +163,23 @@ def aio_postgresql_engine(
     log_to_stdout: bool = False,
     log_migrations_to_stdout: bool = True,
 ) -> AsyncEngine:
-    engine = create_async_engine(url=url, echo=log_to_stdout, json_serializer=_dumps)
+    url_query = dict(url.query)
+    sslmode = url_query.pop("sslmode", None) or url_query.pop("ssl", None)
+    engine = create_async_engine(
+        url=url.set(
+            # https://github.com/MagicStack/asyncpg/issues/737
+            query={**url_query, "ssl": sslmode} if sslmode else url_query,
+        ),
+        echo=log_to_stdout,
+        json_serializer=_dumps,
+    )
     if not migrate:
         return engine
     sync_engine = sqlalchemy.create_engine(
-        url=url.set(drivername="postgresql+psycopg"),
+        url=url.set(
+            drivername="postgresql+psycopg",
+            query={**url_query, "sslmode": sslmode} if sslmode else url_query,
+        ),
         echo=log_migrations_to_stdout,
         json_serializer=_dumps,
     )

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/src/phoenix/db/insertion/span.py

@@ -28,42 +28,77 @@ async def insert_span(
     dialect = SupportedSQLDialect(session.bind.dialect.name)
     if (
         project_rowid := await session.scalar(
-            select(models.Project.id).where(models.Project.name == project_name)
+            select(models.Project.id).filter_by(name=project_name)
         )
     ) is None:
         project_rowid = await session.scalar(
-            insert(models.Project).values(dict(name=project_name)).returning(models.Project.id)
+            insert(models.Project).values(name=project_name).returning(models.Project.id)
         )
     assert project_rowid is not None
-    if trace := await session.scalar(
-        select(models.Trace).where(models.Trace.trace_id == span.context.trace_id)
-    ):
-        trace_rowid = trace.id
-        if span.start_time < trace.start_time or trace.end_time < span.end_time:
-            trace_start_time = min(trace.start_time, span.start_time)
-            trace_end_time = max(trace.end_time, span.end_time)
-            await session.execute(
-                update(models.Trace)
-                .where(models.Trace.id == trace_rowid)
-                .values(
-                    start_time=trace_start_time,
-                    end_time=trace_end_time,
-                )
-            )
+
+    trace_id = span.context.trace_id
+    trace: models.Trace = await session.scalar(
+        select(models.Trace).filter_by(trace_id=trace_id)
+    ) or models.Trace(trace_id=trace_id)
+
+    if trace.id is not None:
+        # Trace record may need to be updated.
+        if trace.end_time < span.end_time:
+            trace.end_time = span.end_time
+            trace.project_rowid = project_rowid
+        if span.start_time < trace.start_time:
+            trace.start_time = span.start_time
     else:
-        trace_rowid = cast(
-            int,
-            await session.scalar(
-                insert(models.Trace)
-                .values(
-                    project_rowid=project_rowid,
-                    trace_id=span.context.trace_id,
-                    start_time=span.start_time,
-                    end_time=span.end_time,
-                )
-                .returning(models.Trace.id)
-            ),
+        # Trace record needs to be persisted for the first time.
+        trace.start_time = span.start_time
+        trace.end_time = span.end_time
+        trace.project_rowid = project_rowid
+        session.add(trace)
+
+    session_id = get_attribute_value(span.attributes, SpanAttributes.SESSION_ID)
+    session_id = str(session_id).strip() if session_id is not None else ""
+    assert isinstance(session_id, str)
+
+    project_session: Optional[models.ProjectSession] = None
+    if trace.project_session_rowid is not None:
+        # ProjectSession record already exists in database for this Trace record, so we fetch
+        # it because it may need to be updated. However, the session_id on the span, if exists,
+        # will be ignored at this point. Otherwise, if session_id is different, we will need
+        # to create a new ProjectSession record, as well as to determine whether the old record
+        # needs to be deleted if this is the last Trace associated with it.
+        project_session = await session.scalar(
+            select(models.ProjectSession).filter_by(id=trace.project_session_rowid)
         )
+    elif session_id:
+        project_session = await session.scalar(
+            select(models.ProjectSession).filter_by(session_id=session_id)
+        ) or models.ProjectSession(session_id=session_id)
+
+    if project_session is not None:
+        if project_session.id is None:
+            # ProjectSession record needs to be persisted for the first time.
+            project_session.start_time = trace.start_time
+            project_session.end_time = trace.end_time
+            project_session.project_id = project_rowid
+            session.add(project_session)
+            await session.flush()
+            assert project_session.id is not None
+            trace.project_session_rowid = project_session.id
+        else:
+            # ProjectSession record may need to be updated.
+            if trace.project_session_rowid is None:
+                trace.project_session_rowid = project_session.id
+            if trace.start_time < project_session.start_time:
+                project_session.start_time = trace.start_time
+            if project_session.end_time < trace.end_time:
+                project_session.end_time = trace.end_time
+
+    await session.flush()
+    assert trace.id is not None
+    assert project_session is None or (
+        project_session.id is not None and project_session.id == trace.project_session_rowid
+    )
+
     cumulative_error_count = int(span.status_code is SpanStatusCode.ERROR)
     cumulative_llm_token_count_prompt = cast(
         int, get_attribute_value(span.attributes, SpanAttributes.LLM_TOKEN_COUNT_PROMPT) or 0
@@ -94,7 +129,7 @@ async def insert_span(
         insert_on_conflict(
             dict(
                 span_id=span.context.span_id,
-                trace_rowid=trace_rowid,
+                trace_rowid=trace.id,
                 parent_id=span.parent_id,
                 span_kind=span.span_kind.value,
                 name=span.name,

arize_phoenix-7.0.1/src/phoenix/db/migrations/data_migration_scripts/populate_project_sessions.py

@@ -0,0 +1,199 @@
+# /// script
+# dependencies = [
+#   "arize-phoenix[pg]",
+# ]
+# ///
+"""
+Populate the `project_sessions` table with data from the traces and spans tables.
+
+Environment variables.
+
+- `PHOENIX_SQL_DATABASE_URL` must be set to the database connection string.
+- (optional) Postgresql schema can be set via `PHOENIX_SQL_DATABASE_SCHEMA`.
+"""
+
+import os
+from datetime import datetime
+from time import perf_counter
+from typing import Any, Optional, Union
+
+import sqlean
+from openinference.semconv.trace import SpanAttributes
+from sqlalchemy import (
+    JSON,
+    Engine,
+    NullPool,
+    create_engine,
+    event,
+    func,
+    insert,
+    make_url,
+    select,
+    update,
+)
+from sqlalchemy.dialects import postgresql
+from sqlalchemy.ext.compiler import compiles
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
+
+from phoenix.config import ENV_PHOENIX_SQL_DATABASE_SCHEMA, get_env_database_connection_str
+from phoenix.db.engines import set_postgresql_search_path
+
+
+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(),  # type: ignore
+        "postgresql",
+    )
+    .with_variant(
+        JSONB(),
+        "sqlite",
+    )
+)
+
+
+class Base(DeclarativeBase): ...
+
+
+class ProjectSession(Base):
+    __tablename__ = "project_sessions"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    session_id: Mapped[str]
+    project_id: Mapped[int]
+    start_time: Mapped[datetime]
+    end_time: Mapped[datetime]
+
+
+class Trace(Base):
+    __tablename__ = "traces"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    project_session_rowid: Mapped[Union[int, None]]
+    project_rowid: Mapped[int]
+    start_time: Mapped[datetime]
+    end_time: Mapped[datetime]
+
+
+class Span(Base):
+    __tablename__ = "spans"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    trace_rowid: Mapped[int]
+    parent_id: Mapped[Optional[str]]
+    attributes: Mapped[dict[str, Any]] = mapped_column(JSON_, nullable=False)
+
+
+SESSION_ID = SpanAttributes.SESSION_ID.split(".")
+USER_ID = SpanAttributes.USER_ID.split(".")
+
+
+def populate_project_sessions(
+    engine: Engine,
+) -> None:
+    sessions_from_span = (
+        select(
+            Span.attributes[SESSION_ID].as_string().label("session_id"),
+            Trace.project_rowid.label("project_id"),
+            Trace.start_time.label("start_time"),
+            func.row_number()
+            .over(
+                partition_by=Span.attributes[SESSION_ID],
+                order_by=[Trace.start_time, Trace.id, Span.id],
+            )
+            .label("rank"),
+            func.max(Trace.end_time)
+            .over(partition_by=Span.attributes[SESSION_ID])
+            .label("end_time"),
+        )
+        .join_from(Span, Trace, Span.trace_rowid == Trace.id)
+        .where(Span.parent_id.is_(None))
+        .where(Span.attributes[SESSION_ID].as_string() != "")
+        .subquery()
+    )
+    sessions_for_trace_id = (
+        select(
+            Span.trace_rowid,
+            ProjectSession.id.label("project_session_rowid"),
+        )
+        .join_from(
+            Span,
+            ProjectSession,
+            Span.attributes[SESSION_ID].as_string() == ProjectSession.session_id,
+        )
+        .where(Span.parent_id.is_(None))
+        .where(Span.attributes[SESSION_ID].as_string() != "")
+        .subquery()
+    )
+    start_time = perf_counter()
+    with sessionmaker(engine).begin() as session:
+        session.execute(
+            insert(ProjectSession).from_select(
+                [
+                    "session_id",
+                    "project_id",
+                    "start_time",
+                    "end_time",
+                ],
+                select(
+                    sessions_from_span.c.session_id,
+                    sessions_from_span.c.project_id,
+                    sessions_from_span.c.start_time,
+                    sessions_from_span.c.end_time,
+                ).where(sessions_from_span.c.rank == 1),
+            )
+        )
+        session.execute(
+            (
+                update(Trace)
+                .values(project_session_rowid=sessions_for_trace_id.c.project_session_rowid)
+                .where(Trace.id == sessions_for_trace_id.c.trace_rowid)
+            )
+        )
+    elapsed_time = perf_counter() - start_time
+    print(f"✅ Populated project_sessions in {elapsed_time:.3f} seconds.")
+
+
+if __name__ == "__main__":
+    sql_database_url = make_url(get_env_database_connection_str())
+    print(f"Using database URL: {sql_database_url}")
+    ans = input("Is that correct? [y]/n: ")
+    if ans.lower().startswith("n"):
+        url = input("Please enter the correct database URL: ")
+        sql_database_url = make_url(url)
+    backend = sql_database_url.get_backend_name()
+    if backend == "sqlite":
+        file = sql_database_url.database
+        engine = create_engine(
+            url=sql_database_url.set(drivername="sqlite"),
+            creator=lambda: sqlean.connect(f"file:///{file}", uri=True),
+            poolclass=NullPool,
+            echo=True,
+        )
+    elif backend == "postgresql":
+        schema = os.getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA)
+        if schema:
+            print(f"Using schema: {schema}")
+        else:
+            print("No PostgreSQL schema set. (This is the default.)")
+        ans = input("Is that correct? [y]/n: ")
+        if ans.lower().startswith("n"):
+            schema = input("Please enter the correct schema: ")
+        engine = create_engine(
+            url=sql_database_url.set(drivername="postgresql+psycopg"),
+            poolclass=NullPool,
+            echo=True,
+        )
+        if schema:
+            event.listen(engine, "connect", set_postgresql_search_path(schema))
+    else:
+        raise ValueError(f"Unknown database backend: {backend}")
+    populate_project_sessions(engine)

arize_phoenix-7.0.1/src/phoenix/db/migrations/versions/4ded9e43755f_create_project_sessions_table.py

@@ -0,0 +1,66 @@
+"""create project_session table
+
+Revision ID: 4ded9e43755f
+Revises: cd164e83824f
+Create Date: 2024-10-08 22:53:24.539786
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "4ded9e43755f"
+down_revision: Union[str, None] = "cd164e83824f"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "project_sessions",
+        sa.Column("id", sa.Integer, primary_key=True),
+        sa.Column("session_id", sa.String, unique=True, nullable=False),
+        sa.Column(
+            "project_id",
+            sa.Integer,
+            sa.ForeignKey("projects.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.Column(
+            "start_time",
+            sa.TIMESTAMP(timezone=True),
+            index=True,
+            nullable=False,
+        ),
+        sa.Column(
+            "end_time",
+            sa.TIMESTAMP(timezone=True),
+            index=True,
+            nullable=False,
+        ),
+    )
+    with op.batch_alter_table("traces") as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "project_session_rowid",
+                sa.Integer,
+                sa.ForeignKey("project_sessions.id", ondelete="CASCADE"),
+                nullable=True,
+            ),
+        )
+    op.create_index(
+        "ix_traces_project_session_rowid",
+        "traces",
+        ["project_session_rowid"],
+    )
+
+
+def downgrade() -> None:
+    op.drop_index("ix_traces_project_session_rowid")
+    with op.batch_alter_table("traces") as batch_op:
+        batch_op.drop_column("project_session_rowid")
+    op.drop_table("project_sessions")

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/src/phoenix/db/models.py

@@ -156,14 +156,37 @@ class Project(Base):
     )
 
 
+class ProjectSession(Base):
+    __tablename__ = "project_sessions"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    session_id: Mapped[str] = mapped_column(String, nullable=False, unique=True)
+    project_id: Mapped[int] = mapped_column(
+        ForeignKey("projects.id", ondelete="CASCADE"),
+        nullable=False,
+        index=True,
+    )
+    start_time: Mapped[datetime] = mapped_column(UtcTimeStamp, index=True, nullable=False)
+    end_time: Mapped[datetime] = mapped_column(UtcTimeStamp, index=True, nullable=False)
+    traces: Mapped[list["Trace"]] = relationship(
+        "Trace",
+        back_populates="project_session",
+        uselist=True,
+    )
+
+
 class Trace(Base):
     __tablename__ = "traces"
     id: Mapped[int] = mapped_column(primary_key=True)
     project_rowid: Mapped[int] = mapped_column(
         ForeignKey("projects.id", ondelete="CASCADE"),
+        nullable=False,
         index=True,
     )
     trace_id: Mapped[str]
+    project_session_rowid: Mapped[Optional[int]] = mapped_column(
+        ForeignKey("project_sessions.id", ondelete="CASCADE"),
+        index=True,
+    )
     start_time: Mapped[datetime] = mapped_column(UtcTimeStamp, index=True)
     end_time: Mapped[datetime] = mapped_column(UtcTimeStamp)
 
@@ -188,6 +211,10 @@ class Trace(Base):
         cascade="all, delete-orphan",
         uselist=True,
     )
+    project_session: Mapped[ProjectSession] = relationship(
+        "ProjectSession",
+        back_populates="traces",
+    )
     experiment_runs: Mapped[list["ExperimentRun"]] = relationship(
         primaryjoin="foreign(ExperimentRun.trace_id) == Trace.trace_id",
         back_populates="trace",

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/src/phoenix/server/api/context.py

@@ -31,12 +31,18 @@ from phoenix.server.api.dataloaders import (
     MinStartOrMaxEndTimeDataLoader,
     ProjectByNameDataLoader,
     RecordCountDataLoader,
+    SessionIODataLoader,
+    SessionNumTracesDataLoader,
+    SessionNumTracesWithErrorDataLoader,
+    SessionTokenUsagesDataLoader,
+    SessionTraceLatencyMsQuantileDataLoader,
     SpanAnnotationsDataLoader,
     SpanDatasetExamplesDataLoader,
     SpanDescendantsDataLoader,
     SpanProjectsDataLoader,
     TokenCountDataLoader,
-    TraceRowIdsDataLoader,
+    TraceByTraceIdsDataLoader,
+    TraceRootSpansDataLoader,
     UserRolesDataLoader,
     UsersDataLoader,
 )
@@ -68,12 +74,19 @@ class DataLoaders:
     latency_ms_quantile: LatencyMsQuantileDataLoader
     min_start_or_max_end_times: MinStartOrMaxEndTimeDataLoader
     record_counts: RecordCountDataLoader
+    session_first_inputs: SessionIODataLoader
+    session_last_outputs: SessionIODataLoader
+    session_num_traces: SessionNumTracesDataLoader
+    session_num_traces_with_error: SessionNumTracesWithErrorDataLoader
+    session_token_usages: SessionTokenUsagesDataLoader
+    session_trace_latency_ms_quantile: SessionTraceLatencyMsQuantileDataLoader
     span_annotations: SpanAnnotationsDataLoader
     span_dataset_examples: SpanDatasetExamplesDataLoader
     span_descendants: SpanDescendantsDataLoader
     span_projects: SpanProjectsDataLoader
     token_counts: TokenCountDataLoader
-    trace_row_ids: TraceRowIdsDataLoader
+    trace_by_trace_ids: TraceByTraceIdsDataLoader
+    trace_root_spans: TraceRootSpansDataLoader
     project_by_name: ProjectByNameDataLoader
     users: UsersDataLoader
     user_roles: UserRolesDataLoader

{arize_phoenix-6.2.0 → arize_phoenix-7.0.1}/src/phoenix/server/api/dataloaders/__init__.py

@@ -19,12 +19,18 @@ from .latency_ms_quantile import LatencyMsQuantileCache, LatencyMsQuantileDataLo
 from .min_start_or_max_end_times import MinStartOrMaxEndTimeCache, MinStartOrMaxEndTimeDataLoader
 from .project_by_name import ProjectByNameDataLoader
 from .record_counts import RecordCountCache, RecordCountDataLoader
+from .session_io import SessionIODataLoader
+from .session_num_traces import SessionNumTracesDataLoader
+from .session_num_traces_with_error import SessionNumTracesWithErrorDataLoader
+from .session_token_usages import SessionTokenUsagesDataLoader
+from .session_trace_latency_ms_quantile import SessionTraceLatencyMsQuantileDataLoader
 from .span_annotations import SpanAnnotationsDataLoader
 from .span_dataset_examples import SpanDatasetExamplesDataLoader
 from .span_descendants import SpanDescendantsDataLoader
 from .span_projects import SpanProjectsDataLoader
 from .token_counts import TokenCountCache, TokenCountDataLoader
-from .trace_row_ids import TraceRowIdsDataLoader
+from .trace_by_trace_ids import TraceByTraceIdsDataLoader
+from .trace_root_spans import TraceRootSpansDataLoader
 from .user_roles import UserRolesDataLoader
 from .users import UsersDataLoader
 
@@ -45,11 +51,17 @@ __all__ = [
     "LatencyMsQuantileDataLoader",
     "MinStartOrMaxEndTimeDataLoader",
     "RecordCountDataLoader",
+    "SessionIODataLoader",
+    "SessionNumTracesDataLoader",
+    "SessionNumTracesWithErrorDataLoader",
+    "SessionTokenUsagesDataLoader",
+    "SessionTraceLatencyMsQuantileDataLoader",
     "SpanDatasetExamplesDataLoader",
     "SpanDescendantsDataLoader",
     "SpanProjectsDataLoader",
     "TokenCountDataLoader",
-    "TraceRowIdsDataLoader",
+    "TraceByTraceIdsDataLoader",
+    "TraceRootSpansDataLoader",
     "ProjectByNameDataLoader",
     "SpanAnnotationsDataLoader",
     "UsersDataLoader",