arize-phoenix 6.1.0__py3-none-any.whl → 7.0.0__py3-none-any.whl

phoenix/server/api/types/TokenUsage.py

@@ -0,0 +1,11 @@
+import strawberry
+
+
+@strawberry.type
+class TokenUsage:
+    prompt: int = 0
+    completion: int = 0
+
+    @strawberry.field
+    async def total(self) -> int:
+        return self.prompt + self.completion

phoenix/server/api/types/Trace.py

@@ -1,11 +1,13 @@
 from __future__ import annotations
 
-from typing import Optional
+from datetime import datetime
+from typing import TYPE_CHECKING, Annotated, Optional, Union
 
 import strawberry
+from openinference.semconv.trace import SpanAttributes
 from sqlalchemy import desc, select
 from sqlalchemy.orm import contains_eager
-from strawberry import UNSET, Private
+from strawberry import UNSET, Private, lazy
 from strawberry.relay import Connection, GlobalID, Node, NodeID
 from strawberry.types import Info
 
@@ -21,12 +23,18 @@ from phoenix.server.api.types.SortDir import SortDir
 from phoenix.server.api.types.Span import Span, to_gql_span
 from phoenix.server.api.types.TraceAnnotation import TraceAnnotation, to_gql_trace_annotation
 
+if TYPE_CHECKING:
+    from phoenix.server.api.types.ProjectSession import ProjectSession
+
 
 @strawberry.type
 class Trace(Node):
     id_attr: NodeID[int]
     project_rowid: Private[int]
+    project_session_rowid: Private[Optional[int]]
     trace_id: str
+    start_time: datetime
+    end_time: datetime
 
     @strawberry.field
     async def project_id(self) -> GlobalID:
@@ -34,6 +42,40 @@ class Trace(Node):
 
         return GlobalID(type_name=Project.__name__, node_id=str(self.project_rowid))
 
+    @strawberry.field
+    async def project_session_id(self) -> Optional[GlobalID]:
+        if self.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))
+
+    @strawberry.field
+    async def session(
+        self,
+        info: Info[Context, None],
+    ) -> Union[Annotated["ProjectSession", lazy(".ProjectSession")], None]:
+        if self.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)
+        async with info.context.db() as session:
+            project_session = await session.scalar(stmt)
+        if project_session is None:
+            return None
+        return to_gql_project_session(project_session)
+
+    @strawberry.field
+    async def root_span(
+        self,
+        info: Info[Context, None],
+    ) -> Optional[Span]:
+        span = await info.context.data_loaders.trace_root_spans.load(self.id_attr)
+        if span is None:
+            return None
+        return to_gql_span(span)
+
     @strawberry.field
     async def spans(
         self,
@@ -82,3 +124,18 @@ class Trace(Node):
                 stmt = stmt.order_by(models.TraceAnnotation.created_at.desc())
             annotations = await session.scalars(stmt)
         return [to_gql_trace_annotation(annotation) for annotation in annotations]
+
+
+def to_gql_trace(trace: models.Trace) -> Trace:
+    return Trace(
+        id_attr=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

@@ -87,12 +87,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,
 )
@@ -609,6 +615,12 @@ def create_graphql_router(
                     db,
                     cache_map=cache_for_dataloaders.record_count if cache_for_dataloaders else None,
                 ),
+                session_first_inputs=SessionIODataLoader(db, "first_input"),
+                session_last_outputs=SessionIODataLoader(db, "last_output"),
+                session_num_traces=SessionNumTracesDataLoader(db),
+                session_num_traces_with_error=SessionNumTracesWithErrorDataLoader(db),
+                session_token_usages=SessionTokenUsagesDataLoader(db),
+                session_trace_latency_ms_quantile=SessionTraceLatencyMsQuantileDataLoader(db),
                 span_annotations=SpanAnnotationsDataLoader(db),
                 span_dataset_examples=SpanDatasetExamplesDataLoader(db),
                 span_descendants=SpanDescendantsDataLoader(db),
@@ -617,7 +629,8 @@ def create_graphql_router(
                     db,
                     cache_map=cache_for_dataloaders.token_count if cache_for_dataloaders else None,
                 ),
-                trace_row_ids=TraceRowIdsDataLoader(db),
+                trace_by_trace_ids=TraceByTraceIdsDataLoader(db),
+                trace_root_spans=TraceRootSpansDataLoader(db),
                 project_by_name=ProjectByNameDataLoader(db),
                 users=UsersDataLoader(db),
                 user_roles=UserRolesDataLoader(db),

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

@@ -1,32 +1,32 @@
 {
-  "_components-CdiZ1Osh.js": {
-    "file": "assets/components-CdiZ1Osh.js",
+  "_components-DKH6AzJw.js": {
+    "file": "assets/components-DKH6AzJw.js",
     "name": "components",
     "imports": [
-      "_vendor-Bnv1dNRQ.js",
-      "_pages-FArMEfgg.js",
-      "_vendor-arizeai-BG6iwyLC.js",
-      "_vendor-codemirror-BotnVFFX.js",
+      "_vendor-Cb3zlNNd.js",
+      "_pages-CVY3Nv4Z.js",
+      "_vendor-arizeai-Buo4e1A6.js",
+      "_vendor-codemirror-BuAQiUVf.js",
       "_vendor-three-DwGkEfCM.js"
     ]
   },
-  "_pages-FArMEfgg.js": {
-    "file": "assets/pages-FArMEfgg.js",
+  "_pages-CVY3Nv4Z.js": {
+    "file": "assets/pages-CVY3Nv4Z.js",
     "name": "pages",
     "imports": [
-      "_vendor-Bnv1dNRQ.js",
-      "_vendor-arizeai-BG6iwyLC.js",
-      "_components-CdiZ1Osh.js",
-      "_vendor-recharts-Dy5gEFzQ.js",
-      "_vendor-codemirror-BotnVFFX.js"
+      "_vendor-Cb3zlNNd.js",
+      "_vendor-arizeai-Buo4e1A6.js",
+      "_components-DKH6AzJw.js",
+      "_vendor-recharts-Cl9dK5tC.js",
+      "_vendor-codemirror-BuAQiUVf.js"
     ]
   },
   "_vendor-!~{003}~.js": {
     "file": "assets/vendor-DxkFTwjz.css",
     "src": "_vendor-!~{003}~.js"
   },
-  "_vendor-Bnv1dNRQ.js": {
-    "file": "assets/vendor-Bnv1dNRQ.js",
+  "_vendor-Cb3zlNNd.js": {
+    "file": "assets/vendor-Cb3zlNNd.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-DwGkEfCM.js"
@@ -35,25 +35,33 @@
       "assets/vendor-DxkFTwjz.css"
     ]
   },
-  "_vendor-arizeai-BG6iwyLC.js": {
-    "file": "assets/vendor-arizeai-BG6iwyLC.js",
+  "_vendor-arizeai-Buo4e1A6.js": {
+    "file": "assets/vendor-arizeai-Buo4e1A6.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-Bnv1dNRQ.js"
+      "_vendor-Cb3zlNNd.js"
     ]
   },
-  "_vendor-codemirror-BotnVFFX.js": {
-    "file": "assets/vendor-codemirror-BotnVFFX.js",
+  "_vendor-codemirror-BuAQiUVf.js": {
+    "file": "assets/vendor-codemirror-BuAQiUVf.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-Bnv1dNRQ.js"
+      "_vendor-Cb3zlNNd.js",
+      "_vendor-shiki-CazYUixL.js"
     ]
   },
-  "_vendor-recharts-Dy5gEFzQ.js": {
-    "file": "assets/vendor-recharts-Dy5gEFzQ.js",
+  "_vendor-recharts-Cl9dK5tC.js": {
+    "file": "assets/vendor-recharts-Cl9dK5tC.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-Bnv1dNRQ.js"
+      "_vendor-Cb3zlNNd.js"
+    ]
+  },
+  "_vendor-shiki-CazYUixL.js": {
+    "file": "assets/vendor-shiki-CazYUixL.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-Cb3zlNNd.js"
     ]
   },
   "_vendor-three-DwGkEfCM.js": {
@@ -61,18 +69,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-D_sCOjlG.js",
+    "file": "assets/index-DLV87qiO.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-Bnv1dNRQ.js",
-      "_vendor-arizeai-BG6iwyLC.js",
-      "_pages-FArMEfgg.js",
-      "_components-CdiZ1Osh.js",
+      "_vendor-Cb3zlNNd.js",
+      "_vendor-arizeai-Buo4e1A6.js",
+      "_pages-CVY3Nv4Z.js",
+      "_components-DKH6AzJw.js",
       "_vendor-three-DwGkEfCM.js",
-      "_vendor-recharts-Dy5gEFzQ.js",
-      "_vendor-codemirror-BotnVFFX.js"
+      "_vendor-recharts-Cl9dK5tC.js",
+      "_vendor-codemirror-BuAQiUVf.js",
+      "_vendor-shiki-CazYUixL.js"
     ]
   }
 }