arize-phoenix 8.4.1__py3-none-any.whl → 8.6.0__py3-none-any.whl

{arize_phoenix-8.4.1.dist-info → arize_phoenix-8.6.0.dist-info}/METADATA

@@ -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.dist-info → arize_phoenix-8.6.0.dist-info}/RECORD

@@ -6,7 +6,7 @@ phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=kpW1WL0kiB8XJsO6XycvZVJ-lBkNoenhQ7atCvBoSe8,5365
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=6yCd3vhnQem22chqsom8x3AUg4DeTLyaMeAcm2HpMtk,22
+phoenix/version.py,sha256=L5qrX1zzvIkOeVl7vP_Kx44NHIZT14fKxoLPGspE9Yg,22
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=qBFraOtmwCCnWJltKNP18DDG0mULXigytlFsa6YOz6k,4837
@@ -80,7 +80,7 @@ phoenix/pointcloud/pointcloud.py,sha256=SN_1wXZcwKrtSnHGZLDZGx71orqE1WyVF7E-D58d
 phoenix/pointcloud/projectors.py,sha256=TQgwc9cJDjJkin1WZyZzgl3HsYrLLiyWD7Czy4jNW3U,1088
 phoenix/pointcloud/umap_parameters.py,sha256=db_WEPoamuWtopZx7tQfAXPnoE0MS8FkAV0_ThjEx_Q,1735
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=MRy_pDDS0R63Wp8roLCxr_sDe_ZFZJLgwq7-n1mOYTo,39290
+phoenix/server/app.py,sha256=3jSAFGkwr4Fqf0k3hKOkmhw1W1j9JI_7c2IOYZ2ewBk,39390
 phoenix/server/bearer_auth.py,sha256=0UudvkAS_dxna5JEJJhGUYwB6Ny-e22ssX5Mm79QwCk,5907
 phoenix/server/dml_event.py,sha256=MjJmVEKytq75chBOSyvYDusUnEbg1pHpIjR3pZkUaJA,2838
 phoenix/server/dml_event_handler.py,sha256=EZLXmCvx4pJrCkz29gxwKwmvmUkTtPCHw6klR-XM8qE,8258
@@ -96,14 +96,14 @@ phoenix/server/types.py,sha256=gJJPBcDRkQ9VHZIt_aLqG_OBbGt1oWp4e3W3Jp61oKs,7409
 phoenix/server/api/README.md,sha256=Pyq1PLPgTzXAswrfIhGXrjI3Skq8it2jTVnanT6Ba4Q,1162
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/auth.py,sha256=nywpmfMI1trZTbZRD3oBj4kFjzg_vnxDljcM431T1eY,1246
-phoenix/server/api/context.py,sha256=zeIANzIfUPfs3odkDGvFs6RNRXYmpsz1DJoYlXNR6nM,6375
+phoenix/server/api/context.py,sha256=OopBkMnY48TzulTtfuay3MXmhbFIWPtPoAsbhCW-Pl0,6459
 phoenix/server/api/exceptions.py,sha256=TA0JuY2YRnj35qGuMSQ8d0ToHum9gWm9W--3fSKHrX0,1171
 phoenix/server/api/interceptor.py,sha256=ykDnoC_apUd-llVli3m1CW18kNSIgjz2qZ6m5JmPDu8,1294
-phoenix/server/api/queries.py,sha256=Zygu1nR2Ktkw0-Bc3vnpG5C9LSI_DHRxfujxlOJ7_BU,34542
+phoenix/server/api/queries.py,sha256=QKZkLGzqDRa7EDfUADnk3jI0Rzsc_g_fe4y4oOBGbKw,34866
 phoenix/server/api/schema.py,sha256=fcs36xQwFF_Qe41_5cWR8wYpDvOrnbcyTeo5WNMbDsA,1702
 phoenix/server/api/subscriptions.py,sha256=DSIgQF6lQqkbc7D0AaI5R4g3hIHbU04H5Y2UIpwmpy0,22989
 phoenix/server/api/utils.py,sha256=quCBRcusc6PUq9tJq7M8PgwFZp7nXgVAxtbw8feribY,833
-phoenix/server/api/dataloaders/__init__.py,sha256=ZgJVbqIH-Y4M863fMIOsf4Ka9iVFmWtzqr0a4mQ92BU,4266
+phoenix/server/api/dataloaders/__init__.py,sha256=LiOF9d95TPOD3pYuE906trLq-_q1VP8JQ9B6SHicbbM,4360
 phoenix/server/api/dataloaders/annotation_summaries.py,sha256=2sHmIDX7n8tuPeBTs9bMKtlMKWn_Ph9awTZqmwn2Owc,5505
 phoenix/server/api/dataloaders/average_experiment_run_latency.py,sha256=GLFoFAbztOH-0FVzzZ77mATIO63UcjB50j3qXiNi-tE,1811
 phoenix/server/api/dataloaders/dataset_example_revisions.py,sha256=xF7M2dg3UmjhdCrscnztCIBBI0cg3RF48IIqvilpc18,4623
@@ -119,6 +119,7 @@ phoenix/server/api/dataloaders/experiment_sequence_number.py,sha256=zM_f78fnqhpp
 phoenix/server/api/dataloaders/latency_ms_quantile.py,sha256=CHWV9nvEEM3xqpXWubywtLtFLzbwyLU85EaP_kQajJU,7363
 phoenix/server/api/dataloaders/min_start_or_max_end_times.py,sha256=1jYglTXv4S8k7OWSGdgeJ2OxVCmZtX8a6sjJp9GPZ04,2783
 phoenix/server/api/dataloaders/num_child_spans.py,sha256=655g9SP_3gfT4qqukfgKzLy5uBLQmnVQBP-mZL3in0s,1200
+phoenix/server/api/dataloaders/num_spans_per_trace.py,sha256=nNAxj6QYcTc8oykfJZxH0vtS9tE83oXoArVwQTJHzTw,922
 phoenix/server/api/dataloaders/project_by_name.py,sha256=O24Rjs0ZQaYtSnPfA6YBQfbHfUgNsHKUcc_gIKyK9vo,1086
 phoenix/server/api/dataloaders/prompt_version_sequence_number.py,sha256=tBaCbpI9KgjCACvpd-5hZy5TDSHH7PJPn4bd1xJuN-g,1314
 phoenix/server/api/dataloaders/record_counts.py,sha256=NLlMEaJ-HxiJavqbSkMbzOrIRzqzYxO8-t2mfp1vXqM,4160
@@ -285,7 +286,7 @@ phoenix/server/api/types/SystemApiKey.py,sha256=2ym8EgsTBIvxx1l9xZ-2YMovz58ZwYb_
 phoenix/server/api/types/TimeSeries.py,sha256=nuuZtfHmOhTjeB8_SvZ5PUQexAkTcPScwYeFC5RUlRU,5491
 phoenix/server/api/types/TokenUsage.py,sha256=g-PjAGVigpchQgkXAuC5sc53fn2YwAgfeXkGmFPi_TE,201
 phoenix/server/api/types/ToolDefinition.py,sha256=T6UH2vcbuPBDy7jKYOqMth2NdqxMPgDBf11Tpbt5Yb8,187
-phoenix/server/api/types/Trace.py,sha256=kH0Tv3m8chVbxMiQ27_W5v8gz6RsvGvl2qS9eoELXlw,7992
+phoenix/server/api/types/Trace.py,sha256=Yhy3olalZ03k3kJkg2DHwYhBM2sUe_HIRtDY258gvFc,8193
 phoenix/server/api/types/TraceAnnotation.py,sha256=OW6A2zr1gomOuG0XQe55dk15XXX2DSM0DzatRbHWH5A,1256
 phoenix/server/api/types/UMAPPoints.py,sha256=49sWnxjcAJKRzqUY71Fa0tOPti5XjIIFT5cSg6oNu_U,1650
 phoenix/server/api/types/User.py,sha256=iTVUrI8U6-asOhBt1bOZNtoRAaRNC4WHgR1Uv2rHnWQ,1975
@@ -312,10 +313,10 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
-phoenix/server/static/.vite/manifest.json,sha256=sQ_lHc6TkDsvc2g9rCZb9s_G0YvJMBpzB4MslJGv7Qo,2165
-phoenix/server/static/assets/components-CRlSj2mp.js,sha256=ZN4Xq4Fbtdwg81XezQGRogB0upxqx9MZA4uvPsIo8wc,420595
-phoenix/server/static/assets/index-YUNmSSFA.js,sha256=QQKgYv2KwukZh449x-BfcWEd1DDydJRIZt1bqIhWhNw,59131
-phoenix/server/static/assets/pages-o_YaLL0c.js,sha256=7GvBpyRWdkzw31J2Rx29pz3gGFE9Fha45R---hF2GyM,829587
+phoenix/server/static/.vite/manifest.json,sha256=luCFNUeHPVshRd78xnNZreAR3XvhJkhyGdgSS7yyyP0,2165
+phoenix/server/static/assets/components-CKK--O4E.js,sha256=eo2djbxR-LTaZAf4CESyOJIxVEg3PLAYWtAweI9R9oY,420930
+phoenix/server/static/assets/index-DqyEzUrX.js,sha256=TcU9eL7uVGb2eQ-urd34X6dxtmEJO9QGDom4QcSuN3E,59131
+phoenix/server/static/assets/pages-BsW87AAs.js,sha256=str6WZPgXHwot8bIrUs_iiZcq5-O1nT6J_gstH1KiNU,831608
 phoenix/server/static/assets/vendor-CRRq3WgM.js,sha256=J5mIL57AMCQb5lg2CVj-4e5dGb4NOhDeZNBN61fIROM,2235517
 phoenix/server/static/assets/vendor-Cg6lcjUC.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
 phoenix/server/static/assets/vendor-arizeai-Dq64X_0o.js,sha256=jAFsbq-7Q6v08obSM4NW5oMHsouV6Nd9DDV5A9YqsRE,202331
@@ -362,9 +363,9 @@ phoenix/utilities/project.py,sha256=auVpARXkDb-JgeX5f2aStyFIkeKvGwN9l7qrFeJMVxI,
 phoenix/utilities/re.py,sha256=6YyUWIkv0zc2SigsxfOWIHzdpjKA_TZo2iqKq7zJKvw,2081
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/template_formatters.py,sha256=gh9PJD6WEGw7TEYXfSst1UR4pWWwmjxMLrDVQ_CkpkQ,2779
-arize_phoenix-8.4.1.dist-info/METADATA,sha256=XLNuzNF17bXHp3CKKOqfoAG5DwB_-pP6fRj_AicKVbs,23646
-arize_phoenix-8.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-8.4.1.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-8.4.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-8.4.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-8.4.1.dist-info/RECORD,,
+arize_phoenix-8.6.0.dist-info/METADATA,sha256=wCzR7lODUmq8aN0V5thNYNJpYFjd6Mvxb6tkLuHMbtk,23646
+arize_phoenix-8.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-8.6.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-8.6.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-8.6.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-8.6.0.dist-info/RECORD,,

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

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",

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]

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 [

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,

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(

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",