arize-phoenix 8.4.1__tar.gz → 8.6.0__tar.gz

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.4.1
+Version: 8.6.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/api/context.py

@@ -30,6 +30,7 @@ from phoenix.server.api.dataloaders import (
     LatencyMsQuantileDataLoader,
     MinStartOrMaxEndTimeDataLoader,
     NumChildSpansDataLoader,
+    NumSpansPerTraceDataLoader,
     ProjectByNameDataLoader,
     PromptVersionSequenceNumberDataLoader,
     RecordCountDataLoader,
@@ -78,6 +79,7 @@ class DataLoaders:
     latency_ms_quantile: LatencyMsQuantileDataLoader
     min_start_or_max_end_times: MinStartOrMaxEndTimeDataLoader
     num_child_spans: NumChildSpansDataLoader
+    num_spans_per_trace: NumSpansPerTraceDataLoader
     project_fields: TableFieldsDataLoader
     prompt_version_sequence_number: PromptVersionSequenceNumberDataLoader
     record_counts: RecordCountDataLoader

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/api/dataloaders/__init__.py

@@ -18,6 +18,7 @@ from .experiment_sequence_number import ExperimentSequenceNumberDataLoader
 from .latency_ms_quantile import LatencyMsQuantileCache, LatencyMsQuantileDataLoader
 from .min_start_or_max_end_times import MinStartOrMaxEndTimeCache, MinStartOrMaxEndTimeDataLoader
 from .num_child_spans import NumChildSpansDataLoader
+from .num_spans_per_trace import NumSpansPerTraceDataLoader
 from .project_by_name import ProjectByNameDataLoader
 from .prompt_version_sequence_number import PromptVersionSequenceNumberDataLoader
 from .record_counts import RecordCountCache, RecordCountDataLoader
@@ -55,6 +56,7 @@ __all__ = [
     "LatencyMsQuantileDataLoader",
     "MinStartOrMaxEndTimeDataLoader",
     "NumChildSpansDataLoader",
+    "NumSpansPerTraceDataLoader",
     "PromptVersionSequenceNumberDataLoader",
     "RecordCountDataLoader",
     "SessionIODataLoader",

arize_phoenix-8.6.0/src/phoenix/server/api/dataloaders/num_spans_per_trace.py

@@ -0,0 +1,28 @@
+from typing import Iterable
+
+from sqlalchemy import func, select
+from strawberry.dataloader import DataLoader
+from typing_extensions import TypeAlias
+
+from phoenix.db.models import Span, Trace
+from phoenix.server.types import DbSessionFactory
+
+TraceRowId: TypeAlias = int
+
+Key: TypeAlias = TraceRowId
+Result: TypeAlias = int
+
+
+class NumSpansPerTraceDataLoader(DataLoader[Key, Result]):
+    def __init__(self, db: DbSessionFactory) -> None:
+        super().__init__(load_fn=self._load_fn)
+        self._db = db
+
+    async def _load_fn(self, keys: Iterable[Key]) -> list[Result]:
+        stmt = (
+            select(Trace.id, func.count()).join(Span).where(Trace.id.in_(keys)).group_by(Trace.id)
+        )
+        async with self._db() as session:
+            data = await session.stream(stmt)
+            result: dict[Key, Result] = {id_: cnt async for id_, cnt in data}
+        return [result.get(id_, 0) for id_ in keys]

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/api/queries.py

@@ -86,7 +86,7 @@ class ModelsInput:
 @strawberry.type
 class DbTableStats:
     table_name: str
-    num_bytes: int
+    num_bytes: float
 
 
 @strawberry.type
@@ -796,9 +796,13 @@ class Query:
     ) -> list[DbTableStats]:
         if info.context.db.dialect is SupportedSQLDialect.SQLITE:
             stmt = text("SELECT name, sum(pgsize) FROM dbstat group by name;")
-            async with info.context.db() as session:
-                stats = cast(Iterable[tuple[str, int]], await session.execute(stmt))
-            stats = _consolidate_sqlite_db_table_stats(stats)
+            try:
+                async with info.context.db() as session:
+                    stats = cast(Iterable[tuple[str, int]], await session.execute(stmt))
+                stats = _consolidate_sqlite_db_table_stats(stats)
+            except Exception:
+                # TODO: temporary workaround until we can reproduce the error
+                return []
         elif info.context.db.dialect is SupportedSQLDialect.POSTGRESQL:
             stmt = text(f"""\
                 SELECT c.relname, pg_total_relation_size(c.oid)
@@ -807,8 +811,12 @@ class Query:
                 WHERE c.relkind = 'r'
                 AND n.nspname = '{getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA) or "public"}';
             """)
-            async with info.context.db() as session:
-                stats = cast(Iterable[tuple[str, int]], await session.execute(stmt))
+            try:
+                async with info.context.db() as session:
+                    stats = cast(Iterable[tuple[str, int]], await session.execute(stmt))
+            except Exception:
+                # TODO: temporary workaround until we can reproduce the error
+                return []
         else:
             assert_never(info.context.db.dialect)
         return [

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/api/types/Trace.py

@@ -171,6 +171,13 @@ class Trace(Node):
             return None
         return Span(span_rowid=span_rowid)
 
+    @strawberry.field
+    async def num_spans(
+        self,
+        info: Info[Context, None],
+    ) -> int:
+        return await info.context.data_loaders.num_spans_per_trace.load(self.trace_rowid)
+
     @strawberry.field
     async def spans(
         self,

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/app.py

@@ -86,6 +86,7 @@ from phoenix.server.api.dataloaders import (
     LatencyMsQuantileDataLoader,
     MinStartOrMaxEndTimeDataLoader,
     NumChildSpansDataLoader,
+    NumSpansPerTraceDataLoader,
     ProjectByNameDataLoader,
     PromptVersionSequenceNumberDataLoader,
     RecordCountDataLoader,
@@ -616,6 +617,7 @@ def create_graphql_router(
                     ),
                 ),
                 num_child_spans=NumChildSpansDataLoader(db),
+                num_spans_per_trace=NumSpansPerTraceDataLoader(db),
                 project_fields=TableFieldsDataLoader(db, models.Project),
                 prompt_version_sequence_number=PromptVersionSequenceNumberDataLoader(db),
                 record_counts=RecordCountDataLoader(

{arize_phoenix-8.4.1 → arize_phoenix-8.6.0}/src/phoenix/server/static/.vite/manifest.json

@@ -1,22 +1,22 @@
 {
-  "_components-CRlSj2mp.js": {
-    "file": "assets/components-CRlSj2mp.js",
+  "_components-CKK--O4E.js": {
+    "file": "assets/components-CKK--O4E.js",
     "name": "components",
     "imports": [
       "_vendor-CRRq3WgM.js",
-      "_pages-o_YaLL0c.js",
+      "_pages-BsW87AAs.js",
       "_vendor-arizeai-Dq64X_0o.js",
       "_vendor-codemirror-C1oevlym.js",
       "_vendor-three-C-AGeJYv.js"
     ]
   },
-  "_pages-o_YaLL0c.js": {
-    "file": "assets/pages-o_YaLL0c.js",
+  "_pages-BsW87AAs.js": {
+    "file": "assets/pages-BsW87AAs.js",
     "name": "pages",
     "imports": [
       "_vendor-CRRq3WgM.js",
       "_vendor-arizeai-Dq64X_0o.js",
-      "_components-CRlSj2mp.js",
+      "_components-CKK--O4E.js",
       "_vendor-codemirror-C1oevlym.js",
       "_vendor-recharts-CPj01S89.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-YUNmSSFA.js",
+    "file": "assets/index-DqyEzUrX.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-CRRq3WgM.js",
       "_vendor-arizeai-Dq64X_0o.js",
-      "_pages-o_YaLL0c.js",
-      "_components-CRlSj2mp.js",
+      "_pages-BsW87AAs.js",
+      "_components-CKK--O4E.js",
       "_vendor-three-C-AGeJYv.js",
       "_vendor-codemirror-C1oevlym.js",
       "_vendor-shiki-aY7rz1pm.js",