arize-phoenix 8.2.2__py3-none-any.whl → 8.4.0__py3-none-any.whl

phoenix/server/api/types/Trace.py

@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Annotated, Optional, Union
 import strawberry
 from openinference.semconv.trace import SpanAttributes
 from sqlalchemy import desc, select
-from strawberry import UNSET, Private, lazy
+from strawberry import ID, UNSET, Private, lazy
 from strawberry.relay import Connection, GlobalID, Node, NodeID
 from strawberry.types import Info
 from typing_extensions import TypeAlias
@@ -24,6 +24,7 @@ from phoenix.server.api.types.Span import Span
 from phoenix.server.api.types.TraceAnnotation import TraceAnnotation, to_gql_trace_annotation
 
 if TYPE_CHECKING:
+    from phoenix.server.api.types.Project import Project
     from phoenix.server.api.types.ProjectSession import ProjectSession
 
 ProjectRowId: TypeAlias = int
@@ -33,49 +34,127 @@ TraceRowId: TypeAlias = int
 @strawberry.type
 class Trace(Node):
     trace_rowid: NodeID[TraceRowId]
-    project_rowid: Private[ProjectRowId]
-    project_session_rowid: Private[Optional[int]]
-    trace_id: str
-    start_time: datetime
-    end_time: datetime
+    db_trace: Private[models.Trace] = UNSET
+
+    def __post_init__(self) -> None:
+        if self.db_trace and self.trace_rowid != self.db_trace.id:
+            raise ValueError("Trace ID mismatch")
+
+    @strawberry.field
+    async def trace_id(
+        self,
+        info: Info[Context, None],
+    ) -> ID:
+        if self.db_trace:
+            trace_id = self.db_trace.trace_id
+        else:
+            trace_id = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.trace_id),
+            )
+        return ID(trace_id)
+
+    @strawberry.field
+    async def start_time(
+        self,
+        info: Info[Context, None],
+    ) -> datetime:
+        if self.db_trace:
+            start_time = self.db_trace.start_time
+        else:
+            start_time = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.start_time),
+            )
+        return start_time
+
+    @strawberry.field
+    async def end_time(
+        self,
+        info: Info[Context, None],
+    ) -> datetime:
+        if self.db_trace:
+            end_time = self.db_trace.end_time
+        else:
+            end_time = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.end_time),
+            )
+        return end_time
 
     @strawberry.field
     async def latency_ms(
         self,
         info: Info[Context, None],
     ) -> Optional[float]:
-        async with info.context.db() as session:
-            latency = await session.scalar(
-                select(
-                    models.Trace.latency_ms,
-                ).where(models.Trace.id == self.trace_rowid)
+        if self.db_trace:
+            latency_ms = self.db_trace.latency_ms
+        else:
+            latency_ms = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.latency_ms),
             )
-        return latency
+        return latency_ms
 
     @strawberry.field
-    async def project_id(self) -> GlobalID:
+    async def project(
+        self,
+        info: Info[Context, None],
+    ) -> Annotated["Project", strawberry.lazy(".Project")]:
+        if self.db_trace:
+            project_rowid = self.db_trace.project_rowid
+        else:
+            project_rowid = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.project_rowid),
+            )
+        from phoenix.server.api.types.Project import Project
+
+        return Project(project_rowid=project_rowid)
+
+    @strawberry.field
+    async def project_id(
+        self,
+        info: Info[Context, None],
+    ) -> GlobalID:
+        if self.db_trace:
+            project_rowid = self.db_trace.project_rowid
+        else:
+            project_rowid = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.project_rowid),
+            )
         from phoenix.server.api.types.Project import Project
 
-        return GlobalID(type_name=Project.__name__, node_id=str(self.project_rowid))
+        return GlobalID(type_name=Project.__name__, node_id=str(project_rowid))
 
     @strawberry.field
-    async def project_session_id(self) -> Optional[GlobalID]:
-        if self.project_session_rowid is None:
+    async def project_session_id(
+        self,
+        info: Info[Context, None],
+    ) -> Optional[GlobalID]:
+        if self.db_trace:
+            project_session_rowid = self.db_trace.project_session_rowid
+        else:
+            project_session_rowid = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.project_session_rowid),
+            )
+        if project_session_rowid is None:
             return None
         from phoenix.server.api.types.ProjectSession import ProjectSession
 
-        return GlobalID(type_name=ProjectSession.__name__, node_id=str(self.project_session_rowid))
+        return GlobalID(type_name=ProjectSession.__name__, node_id=str(project_session_rowid))
 
     @strawberry.field
     async def session(
         self,
         info: Info[Context, None],
     ) -> Union[Annotated["ProjectSession", lazy(".ProjectSession")], None]:
-        if self.project_session_rowid is None:
+        if self.db_trace:
+            project_session_rowid = self.db_trace.project_session_rowid
+        else:
+            project_session_rowid = await info.context.data_loaders.trace_fields.load(
+                (self.trace_rowid, models.Trace.project_session_rowid),
+            )
+        if project_session_rowid is None:
             return None
         from phoenix.server.api.types.ProjectSession import to_gql_project_session
 
-        stmt = select(models.ProjectSession).filter_by(id=self.project_session_rowid)
+        stmt = select(models.ProjectSession).filter_by(id=project_session_rowid)
         async with info.context.db() as session:
             project_session = await session.scalar(stmt)
         if project_session is None:
@@ -141,16 +220,5 @@ class Trace(Node):
         return [to_gql_trace_annotation(annotation) for annotation in annotations]
 
 
-def to_gql_trace(trace: models.Trace) -> Trace:
-    return Trace(
-        trace_rowid=trace.id,
-        project_rowid=trace.project_rowid,
-        project_session_rowid=trace.project_session_rowid,
-        trace_id=trace.trace_id,
-        start_time=trace.start_time,
-        end_time=trace.end_time,
-    )
-
-
 INPUT_VALUE = SpanAttributes.INPUT_VALUE.split(".")
 OUTPUT_VALUE = SpanAttributes.OUTPUT_VALUE.split(".")

phoenix/server/app.py

@@ -85,6 +85,7 @@ from phoenix.server.api.dataloaders import (
     ExperimentSequenceNumberDataLoader,
     LatencyMsQuantileDataLoader,
     MinStartOrMaxEndTimeDataLoader,
+    NumChildSpansDataLoader,
     ProjectByNameDataLoader,
     PromptVersionSequenceNumberDataLoader,
     RecordCountDataLoader,
@@ -97,8 +98,8 @@ from phoenix.server.api.dataloaders import (
     SpanByIdDataLoader,
     SpanDatasetExamplesDataLoader,
     SpanDescendantsDataLoader,
-    SpanFieldsDataLoader,
     SpanProjectsDataLoader,
+    TableFieldsDataLoader,
     TokenCountDataLoader,
     TraceByTraceIdsDataLoader,
     TraceRootSpansDataLoader,
@@ -614,6 +615,8 @@ def create_graphql_router(
                         else None
                     ),
                 ),
+                num_child_spans=NumChildSpansDataLoader(db),
+                project_fields=TableFieldsDataLoader(db, models.Project),
                 prompt_version_sequence_number=PromptVersionSequenceNumberDataLoader(db),
                 record_counts=RecordCountDataLoader(
                     db,
@@ -626,7 +629,7 @@ def create_graphql_router(
                 session_token_usages=SessionTokenUsagesDataLoader(db),
                 session_trace_latency_ms_quantile=SessionTraceLatencyMsQuantileDataLoader(db),
                 span_annotations=SpanAnnotationsDataLoader(db),
-                span_fields=SpanFieldsDataLoader(db),
+                span_fields=TableFieldsDataLoader(db, models.Span),
                 span_by_id=SpanByIdDataLoader(db),
                 span_dataset_examples=SpanDatasetExamplesDataLoader(db),
                 span_descendants=SpanDescendantsDataLoader(db),
@@ -636,6 +639,7 @@ def create_graphql_router(
                     cache_map=cache_for_dataloaders.token_count if cache_for_dataloaders else None,
                 ),
                 trace_by_trace_ids=TraceByTraceIdsDataLoader(db),
+                trace_fields=TableFieldsDataLoader(db, models.Trace),
                 trace_root_spans=TraceRootSpansDataLoader(db),
                 project_by_name=ProjectByNameDataLoader(db),
                 users=UsersDataLoader(db),

phoenix/server/static/.vite/manifest.json

@@ -1,32 +1,28 @@
 {
-  "_components-MeFAEc1z.js": {
-    "file": "assets/components-MeFAEc1z.js",
+  "_components-BgFPI6sn.js": {
+    "file": "assets/components-BgFPI6sn.js",
     "name": "components",
     "imports": [
-      "_vendor-Cqfydjep.js",
-      "_pages-NrL4hb9q.js",
-      "_vendor-arizeai-WnerlUPN.js",
-      "_vendor-codemirror-D-ZZKLFq.js",
+      "_vendor-CRRq3WgM.js",
+      "_pages-CmDiPH1A.js",
+      "_vendor-arizeai-Dq64X_0o.js",
+      "_vendor-codemirror-C1oevlym.js",
       "_vendor-three-C-AGeJYv.js"
     ]
   },
-  "_pages-NrL4hb9q.js": {
-    "file": "assets/pages-NrL4hb9q.js",
+  "_pages-CmDiPH1A.js": {
+    "file": "assets/pages-CmDiPH1A.js",
     "name": "pages",
     "imports": [
-      "_vendor-Cqfydjep.js",
-      "_vendor-arizeai-WnerlUPN.js",
-      "_components-MeFAEc1z.js",
-      "_vendor-codemirror-D-ZZKLFq.js",
-      "_vendor-recharts-KY97ZPfK.js"
+      "_vendor-CRRq3WgM.js",
+      "_vendor-arizeai-Dq64X_0o.js",
+      "_components-BgFPI6sn.js",
+      "_vendor-codemirror-C1oevlym.js",
+      "_vendor-recharts-CPj01S89.js"
     ]
   },
-  "_vendor-Cg6lcjUC.css": {
-    "file": "assets/vendor-Cg6lcjUC.css",
-    "src": "_vendor-Cg6lcjUC.css"
-  },
-  "_vendor-Cqfydjep.js": {
-    "file": "assets/vendor-Cqfydjep.js",
+  "_vendor-CRRq3WgM.js": {
+    "file": "assets/vendor-CRRq3WgM.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-C-AGeJYv.js"
@@ -35,33 +31,37 @@
       "assets/vendor-Cg6lcjUC.css"
     ]
   },
-  "_vendor-arizeai-WnerlUPN.js": {
-    "file": "assets/vendor-arizeai-WnerlUPN.js",
+  "_vendor-Cg6lcjUC.css": {
+    "file": "assets/vendor-Cg6lcjUC.css",
+    "src": "_vendor-Cg6lcjUC.css"
+  },
+  "_vendor-arizeai-Dq64X_0o.js": {
+    "file": "assets/vendor-arizeai-Dq64X_0o.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-Cqfydjep.js"
+      "_vendor-CRRq3WgM.js"
     ]
   },
-  "_vendor-codemirror-D-ZZKLFq.js": {
-    "file": "assets/vendor-codemirror-D-ZZKLFq.js",
+  "_vendor-codemirror-C1oevlym.js": {
+    "file": "assets/vendor-codemirror-C1oevlym.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-Cqfydjep.js",
-      "_vendor-shiki-D5K9GnFn.js"
+      "_vendor-CRRq3WgM.js",
+      "_vendor-shiki-aY7rz1pm.js"
     ]
   },
-  "_vendor-recharts-KY97ZPfK.js": {
-    "file": "assets/vendor-recharts-KY97ZPfK.js",
+  "_vendor-recharts-CPj01S89.js": {
+    "file": "assets/vendor-recharts-CPj01S89.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-Cqfydjep.js"
+      "_vendor-CRRq3WgM.js"
     ]
   },
-  "_vendor-shiki-D5K9GnFn.js": {
-    "file": "assets/vendor-shiki-D5K9GnFn.js",
+  "_vendor-shiki-aY7rz1pm.js": {
+    "file": "assets/vendor-shiki-aY7rz1pm.js",
     "name": "vendor-shiki",
     "imports": [
-      "_vendor-Cqfydjep.js"
+      "_vendor-CRRq3WgM.js"
     ]
   },
   "_vendor-three-C-AGeJYv.js": {
@@ -69,19 +69,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-BSRuZ-_J.js",
+    "file": "assets/index-CIkk8uHr.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-Cqfydjep.js",
-      "_vendor-arizeai-WnerlUPN.js",
-      "_pages-NrL4hb9q.js",
-      "_components-MeFAEc1z.js",
+      "_vendor-CRRq3WgM.js",
+      "_vendor-arizeai-Dq64X_0o.js",
+      "_pages-CmDiPH1A.js",
+      "_components-BgFPI6sn.js",
       "_vendor-three-C-AGeJYv.js",
-      "_vendor-codemirror-D-ZZKLFq.js",
-      "_vendor-shiki-D5K9GnFn.js",
-      "_vendor-recharts-KY97ZPfK.js"
+      "_vendor-codemirror-C1oevlym.js",
+      "_vendor-shiki-aY7rz1pm.js",
+      "_vendor-recharts-CPj01S89.js"
     ]
   }
 }