arize-phoenix 3.14.2__py3-none-any.whl → 3.15.1__py3-none-any.whl

{arize_phoenix-3.14.2.dist-info → arize_phoenix-3.15.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 3.14.2
+Version: 3.15.1
 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-3.14.2.dist-info → arize_phoenix-3.15.1.dist-info}/RECORD

@@ -1,17 +1,17 @@
 phoenix/__init__.py,sha256=mrgR7rvpc7LnBeyLzCX9wmRP4kJSJdfKDbj2zDiDjG8,2311
-phoenix/config.py,sha256=eJ7V0eQMc1vT_0LRF1HDXMWbnGLYa4c36WHYfUHrNAE,3872
+phoenix/config.py,sha256=iGZeRwpvQG-JOzlfv642f98rCAHQ1SpGiEvgJsxg-1g,4682
 phoenix/datetime_utils.py,sha256=D955QLrkgrrSdUM6NyqbCeAu2SMsjhR5rHVQEsVUdng,2773
 phoenix/exceptions.py,sha256=X5k9ipUDfwSCwZB-H5zFJLas86Gf9tAx0W4l5TZxp5k,108
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=f6AeyKTuOpy9RCcTCjVH3gx5nYZhbTMFOuv1WSUOB5o,4992
-phoenix/version.py,sha256=f8-0K-K9GknTfMcWvP5jj3XhIXV87K09GjV-81wN-ec,23
+phoenix/version.py,sha256=ZkLw4mpKHbvgufoRYVg6ykxZYyu0FTe3IZCKsCUV2x4,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=C-kDATyJEgP-oqYVKOiQM76Ljs66F6VZdT93_b8kTGk,4725
 phoenix/core/model_schema.py,sha256=lQaTvKS34yurHOJ53YD020uURLfgG3dqKC1NLQftOjA,50222
 phoenix/core/model_schema_adapter.py,sha256=3GkyzqUST4fYi-Bgs8qAam5hwMCdQRZTDLjZ9Bnzdm4,8268
-phoenix/core/project.py,sha256=z-Yrwyg50TqP48mZ8nQC2ZSxrebrJL_OoMp5EIZi_NE,24635
-phoenix/core/traces.py,sha256=_AiOt1XRNMU_XTRXZBsjdYuF_zHn87Vm80mW0Y7PA9Q,2935
+phoenix/core/project.py,sha256=KBntQewZ02XkU9GIHFlncTHjPFokTeCTMY5pQpg8HkE,24817
+phoenix/core/traces.py,sha256=54Xsp-rBMMdeT8KDVrK1M7g4QGXHFlKgVfVEs5KnQzg,3549
 phoenix/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/datasets/dataset.py,sha256=scKVZ7zc6Dpc_ntt-pWhzY-KWqOJEwKePuyNnKSVTGE,30515
 phoenix/datasets/errors.py,sha256=cGp9vxnw4SewFoWBV3ZGMkhE0Kh73lPIv3Ppz_H_RoA,8261
@@ -55,14 +55,14 @@ phoenix/pointcloud/pointcloud.py,sha256=4zAIkKs2xOUbchpj4XDAV-iPMXrfAJ15TG6rlIYG
 phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcBYPU,1057
 phoenix/pointcloud/umap_parameters.py,sha256=lJsEOrbSuSiqI7g4Yt6xj7kgYxEqoep4ZHWLr6VWBqw,1760
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=cNEskfuSyG7XGdt802VZin-keJk78f8Fm0jaFE94Py8,6887
-phoenix/server/main.py,sha256=nprpPH6eBAKwJ7OjY1bZxQtcukfJpv7b_IuTVwchZWM,10351
+phoenix/server/app.py,sha256=7yGzmItFLoTFa5CBZwz4Qb2VZbVbczh_3qRqTTtkaKw,6888
+phoenix/server/main.py,sha256=wrKegfYm-5APU0CqPqpEE_3MUlSgU2LPfpEkYSd870E,9457
 phoenix/server/thread_server.py,sha256=dP6cm6Cf08jNhDA1TRlVZpziu1YgtPDmaeIJMm725eI,2154
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/context.py,sha256=wjCzq4QlszKG1iN-xgu5rRLYPqdvTFqX02aFYPipNoQ,512
 phoenix/server/api/helpers.py,sha256=_V1eVkchZmTkhOfRC4QqR1sUB2xtIxdsMJkDouZq_IE,251
 phoenix/server/api/interceptor.py,sha256=ykDnoC_apUd-llVli3m1CW18kNSIgjz2qZ6m5JmPDu8,1294
-phoenix/server/api/schema.py,sha256=wYwmxM9tqZVd4OK8KN3KusVibThjaC4YSC-GnBlO-TY,7969
+phoenix/server/api/schema.py,sha256=UpYm8wFwHeQfSX25WwDkhBPilIjiXQ4PSoQUclv3IHc,8839
 phoenix/server/api/input_types/ClusterInput.py,sha256=EL4ftvZxQ8mVdruUPcdhMhByORmSmM8S-X6RPqU6GX0,179
 phoenix/server/api/input_types/Coordinates.py,sha256=meTwbIjwTfqx5DGD2DBlH9wQzdQVNM5a8x9dp1FfIgA,173
 phoenix/server/api/input_types/DataQualityMetricInput.py,sha256=LazvmQCCM5m9SDZTpyxQXO1rYF4cmsc3lsR2S9S65X4,1292
@@ -74,9 +74,9 @@ phoenix/server/api/input_types/SpanSort.py,sha256=3ken7KaDUwoZfhaSiP9QRhkNPgGrCY
 phoenix/server/api/input_types/TimeRange.py,sha256=yzx-gxj8mDeGLft1FzU_x1MVEgIG5Pt6-f8PUVDgipQ,522
 phoenix/server/api/input_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/api/routers/evaluation_handler.py,sha256=viXigPYCOzN8bgmu_OcfEwl8jHkwarAw65bnGpvlR98,3852
-phoenix/server/api/routers/span_handler.py,sha256=CdW8pYQW2y2eSW5Y-d6ce2pgmo4v16vGC6ohVo1bkwg,2398
-phoenix/server/api/routers/trace_handler.py,sha256=GVEl5y3QvnElmUVJCPUtQnVAbC6kOvL6JjOfx53JCLI,2412
+phoenix/server/api/routers/evaluation_handler.py,sha256=IkVKe2CMEaHNb_k_eGq1PWBURfbhcg0KjI7m0Eh-9wc,3949
+phoenix/server/api/routers/span_handler.py,sha256=IM4eiQ7GMU7JVyf0oMbrzEcItVudLy_CeYaWcwHOby4,2387
+phoenix/server/api/routers/trace_handler.py,sha256=0Y9QlePySJZyhKJLWM-m1e837HMZm5QcKR-iXshQdG4,2413
 phoenix/server/api/routers/utils.py,sha256=M41BoH-fl37izhRuN2aX7lWm7jOC20A_3uClv9TVUUY,583
 phoenix/server/api/types/Cluster.py,sha256=R08ZKrLl1KK8colxHU57N5XIOTMUwg5ZI50ofPoxxSM,5618
 phoenix/server/api/types/DataQualityMetric.py,sha256=zRKsNvHBu-NdcsunuLhqFpZhi6ks-HMqA1PJD27jTak,590
@@ -109,7 +109,7 @@ phoenix/server/api/types/Retrieval.py,sha256=OhMK2ncjoyp5h1yjKhjlKpoTbQrMHuxmgSF
 phoenix/server/api/types/ScalarDriftMetricEnum.py,sha256=IUAcRPpgL41WdoIgK6cNk2Te38SspXGyEs-S1fY23_A,232
 phoenix/server/api/types/Segments.py,sha256=B6UUWjalZONjWjl_l61A6USPSu15ICXRgzZ4m3vA1yw,2921
 phoenix/server/api/types/SortDir.py,sha256=OUpXhlCzCxPoXSDkJJygEs9Rw9pMymfaZUG5zPTrw4Y,152
-phoenix/server/api/types/Span.py,sha256=obTJMMeSUD8qIen_XIKW0PDdadlFu2pj4dPpxjb7uI8,12328
+phoenix/server/api/types/Span.py,sha256=qyJgHpj9MkYl_xsLfVm5XvoDCRkFYEt-xVNQJ4MUVow,12678
 phoenix/server/api/types/TimeSeries.py,sha256=QbLfxHnwYsMsirpq4tx9us6ha7YtAVzK4m8mAL3fMt0,5200
 phoenix/server/api/types/UMAPPoints.py,sha256=8l9RJXi308qty4MdHb2pBbiU6ZuLbrRRxXNbPhXoxKI,1639
 phoenix/server/api/types/ValidationResult.py,sha256=pHwdYk4J7SJ5xhlWWHg_6qWkfk4rjOx-bSkGHvkDE3Q,142
@@ -127,25 +127,25 @@ 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/index.css,sha256=KKGpx4iwF91VGRm0YN-4cn8oC-oIqC6HecoPf0x3ZM8,1885
-phoenix/server/static/index.js,sha256=-z9E12Tquch7keYjnWNreqYiDvwwCa7f3Q95hoRtlqM,3175772
+phoenix/server/static/index.js,sha256=yw3BlmllF9--GmxXZYWUqIkMA-hElFhd9kAVNphSoCE,3176874
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=lO2wGA5XsftPg03rw_VcyaYf_4vegtlWbIT5ms4fA_c,1982
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/session/client.py,sha256=9qlybwGGdPHql7n6Mye8feuh5_5AMq34xSajvcPn2yg,5853
+phoenix/session/client.py,sha256=EMhzmsqquZbuuH0NX137npYLVJWEKI5OieeCBsVGh3A,7933
 phoenix/session/data_extractor.py,sha256=0Kf-2mKY_YbYoD2fZkAYpKdFgsXrC3OKQ5d2iZsGgAI,1947
 phoenix/session/evaluation.py,sha256=YCv1XkWHi7vM_W5V7rorrrAxadv78wuMPeCVJvf5-oE,5444
-phoenix/session/session.py,sha256=1RV6FBNqNDDrPFYU2qLffmI4RCRKy22KCLZFcH9oqJA,24556
+phoenix/session/session.py,sha256=9NsZCvrozDfFNc2b4dhadSuYHAO7rJLCvrgA0qangqg,24919
 phoenix/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/storage/spanstore/__init__.py,sha256=Q1tfrRcYCTVbxQGWGCkfb2i1YTweiYixQZ-kNG1ViZQ,228
-phoenix/storage/spanstore/text_file.py,sha256=Fzpl_UDVN6_Qa8kpjhDkjdO7xb_I3g-UuTseenkV42c,2853
-phoenix/trace/__init__.py,sha256=DOuudYVyWlhUnIy-E_aqvJ6ulmk_q7ghuctuXQCGCJA,763
+phoenix/storage/span_store/__init__.py,sha256=ib-1BWPX4zMfIxS_qC-intyZciGMBJkAOcpb06d3jgQ,586
+phoenix/storage/span_store/text_file.py,sha256=Fzpl_UDVN6_Qa8kpjhDkjdO7xb_I3g-UuTseenkV42c,2853
+phoenix/trace/__init__.py,sha256=hXo7FzA-utf7CD9PiZQYfQ-0irqHVzfSA4j_RLSmKK4,891
 phoenix/trace/errors.py,sha256=wB1z8qdPckngdfU-TORToekvg3344oNFAA83_hC2yFY,180
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
 phoenix/trace/exporter.py,sha256=vh2RO1CpP143HxIX94KV0qks8p1x66RE3Tgf8kcBCCg,4519
 phoenix/trace/fixtures.py,sha256=HOjuYOB_xtR7JUeLz7WpEroiGj4E5_SxVLSjBYUy8RQ,7055
 phoenix/trace/otel.py,sha256=mpuj_eOdBUPAcslpxk8XZflb9kkzpMJ8X9acJS8ACPA,15322
-phoenix/trace/projects.py,sha256=uIicJWWBAVwiRY-8EH2YlQmCUzsdspTxsz-ceK35m8E,1826
+phoenix/trace/projects.py,sha256=2BwlNjFE-uwpqYtCu5YyBiYZk9wRPpM13vh3-Cv7GkA,2157
 phoenix/trace/schemas.py,sha256=6PVPnQIIDsWijwCwU19TGKG3xhFCYFY9K_5IqR5dWF4,5904
 phoenix/trace/span_evaluations.py,sha256=T67grfU71iANMM0g6dp2OQ_ZjtkON2viZz95_U9FLQA,12954
 phoenix/trace/span_json_decoder.py,sha256=IAFakPRqSMYxTPKYFMiXYxm7U-FipdN8_xbvapDS0Qc,3131
@@ -154,7 +154,7 @@ phoenix/trace/trace_dataset.py,sha256=RpHIfZLbMmULOIb-fKXJkQLhIdC0sJlAOTjlyJppMY
 phoenix/trace/utils.py,sha256=7LurVGXn245cjj4MJsc7v6jq4DSJkpK6YGBfIaSywuw,1307
 phoenix/trace/dsl/__init__.py,sha256=WIQIjJg362XD3s50OsPJJ0xbDsGp41bSv7vDllLrPuA,144
 phoenix/trace/dsl/filter.py,sha256=paLpcSMnHdgCfcvcroaqOoCe2retAZ5ocp_5cNTnv9s,14167
-phoenix/trace/dsl/helpers.py,sha256=CP6WaJpP7_WdI1Yoip7tDWcYDXZRg6xgNBRS2gojjMc,1728
+phoenix/trace/dsl/helpers.py,sha256=TG8EFZAjvRwjXpxitEGAc4QpF3vn4jpqhI_Tcwp5mE4,2134
 phoenix/trace/dsl/missing.py,sha256=BWPOHr2_tBkPDgVeq8GVXXVbNbJiBelu4NtwHBg6mTE,1435
 phoenix/trace/dsl/query.py,sha256=k0guhWBEo6L7ZJH5FJs2-iGSnWXdUUqu09gd-8M4CGg,14783
 phoenix/trace/langchain/__init__.py,sha256=F37GfD1pd5Kuw7R7iRUM1zXXpO8xEcycNZh5dwqBXNk,109
@@ -169,9 +169,10 @@ phoenix/trace/v1/evaluation_pb2.pyi,sha256=cCbbx06gwQmaH14s3J1X25TtaARh-k1abbxQd
 phoenix/utilities/__init__.py,sha256=8w1Ivw0KO9YKWrhcdnO73cSVqP9VHAp0pSfsi_oDiuQ,672
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
-phoenix/utilities/project.py,sha256=AzyuKxh3IByoKapA-AIRzZNFw_c0zh9RNjhh3BT_YWI,416
-arize_phoenix-3.14.2.dist-info/METADATA,sha256=cW_o29Zdw_iIxdLHoi3SD3eW76e-gKTy3Svhhy9PsR8,29204
-arize_phoenix-3.14.2.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-arize_phoenix-3.14.2.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-3.14.2.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-3.14.2.dist-info/RECORD,,
+phoenix/utilities/project.py,sha256=qWsvKnG1oKhOFUowXf9qiOL2ia7jaFe_ijFFHEt8GJo,431
+phoenix/utilities/span_store.py,sha256=13UK0rE4wQd70yl___WsDRnH0ru-xErng9_Ml7zfEwE,978
+arize_phoenix-3.15.1.dist-info/METADATA,sha256=Gz78s9n05nQlE0BgDupaBvSpJs8ldCrwtM_rWV7DnpQ,29204
+arize_phoenix-3.15.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+arize_phoenix-3.15.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-3.15.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-3.15.1.dist-info/RECORD,,

phoenix/config.py

@@ -1,5 +1,6 @@
 import os
 import tempfile
+from enum import Enum
 from pathlib import Path
 from typing import List, Optional
 
@@ -21,6 +22,11 @@ ENV_PHOENIX_PROJECT_NAME = "PHOENIX_PROJECT_NAME"
 """
 The project name to use when logging traces and evals. defaults to 'default'.
 """
+ENV_SPAN_STORAGE_TYPE = "__DANGEROUS__PHOENIX_SPAN_STORAGE_TYPE"
+"""
+**EXPERIMENTAL**
+The type of span storage to use.
+"""
 
 
 def _get_temp_path() -> Path:
@@ -127,4 +133,27 @@ def get_env_collector_endpoint() -> Optional[str]:
 
 
 def get_env_project_name() -> str:
-    return os.getenv(ENV_PHOENIX_PROJECT_NAME) or "default"
+    return os.getenv(ENV_PHOENIX_PROJECT_NAME) or DEFAULT_PROJECT_NAME
+
+
+def get_env_span_storage_type() -> Optional["SpanStorageType"]:
+    """
+    Get the type of span storage to use.
+    """
+    if not (env_type_str := os.getenv(ENV_SPAN_STORAGE_TYPE)):
+        return None
+    try:
+        return SpanStorageType(env_type_str.lower())
+    except ValueError:
+        raise ValueError(
+            f"⚠️ Invalid span storage type value `{env_type_str}` defined by the "
+            f"environment variable `{ENV_SPAN_STORAGE_TYPE}`. Valid values are: "
+            f"{', '.join(t.value for t in SpanStorageType)}."
+        )
+
+
+class SpanStorageType(Enum):
+    TEXT_FILES = "text-files"
+
+
+DEFAULT_PROJECT_NAME = "default"

phoenix/core/project.py

@@ -91,6 +91,7 @@ class Project:
     def __init__(self) -> None:
         self._spans = _Spans()
         self._evals = _Evals()
+        self._is_archived = False
 
     @property
     def last_updated_at(self) -> Optional[datetime]:
@@ -192,6 +193,13 @@ class Project:
     def export_evaluations(self) -> List[Evaluations]:
         return self._evals.export_evaluations()
 
+    def archive(self) -> None:
+        self._is_archived = True
+
+    @property
+    def is_archived(self) -> bool:
+        return self._is_archived
+
 
 class _Spans:
     def __init__(self) -> None:

phoenix/core/traces.py

@@ -8,13 +8,13 @@ from typing import DefaultDict, Iterator, Optional, Tuple, Union
 from typing_extensions import assert_never
 
 import phoenix.trace.v1 as pb
+from phoenix.config import DEFAULT_PROJECT_NAME
 from phoenix.core.project import (
     END_OF_QUEUE,
     Project,
     _ProjectName,
 )
 from phoenix.trace.schemas import Span
-from phoenix.utilities.project import DEFAULT_PROJECT_NAME
 
 _SpanItem = Tuple[Span, _ProjectName]
 _EvalItem = Tuple[pb.Evaluation, _ProjectName]
@@ -38,10 +38,26 @@ class Traces:
         with self._lock:
             return self._projects.get(project_name)
 
-    def get_projects(self) -> Iterator[Tuple[str, "Project"]]:
+    def get_projects(self) -> Iterator[Tuple[int, str, "Project"]]:
         with self._lock:
-            projects = tuple(self._projects.items())
-        yield from projects
+            for project_id, (project_name, project) in enumerate(self._projects.items()):
+                if project.is_archived:
+                    continue
+                yield project_id, project_name, project
+
+    def archive_project(self, id: int) -> Optional["Project"]:
+        with self._lock:
+            active_projects = {
+                project_id: project
+                for project_id, _, project in self.get_projects()
+                if not project.is_archived
+            }
+            if len(active_projects) <= 1:
+                return None
+            if project := active_projects.get(id):
+                project.archive()
+                return project
+        return None
 
     def put(
         self,

phoenix/server/api/routers/evaluation_handler.py

@@ -15,11 +15,11 @@ from starlette.status import (
 )
 
 import phoenix.trace.v1 as pb
+from phoenix.config import DEFAULT_PROJECT_NAME
 from phoenix.core.traces import Traces
 from phoenix.server.api.routers.utils import table_to_bytes
 from phoenix.session.evaluation import encode_evaluations
 from phoenix.trace.span_evaluations import Evaluations
-from phoenix.utilities.project import DEFAULT_PROJECT_NAME
 
 
 class EvaluationHandler(HTTPEndpoint):
@@ -27,6 +27,7 @@ class EvaluationHandler(HTTPEndpoint):
 
     async def post(self, request: Request) -> Response:
         content_type = request.headers.get("content-type")
+        project_name = request.headers.get("project-name", DEFAULT_PROJECT_NAME)
         if content_type == "application/x-pandas-arrow":
             return await self._process_pyarrow(request)
         if content_type != "application/x-protobuf":
@@ -51,7 +52,7 @@ class EvaluationHandler(HTTPEndpoint):
                 content="Request body is invalid",
                 status_code=HTTP_422_UNPROCESSABLE_ENTITY,
             )
-        self.traces.put(evaluation)
+        self.traces.put(evaluation, project_name=project_name)
         return Response()
 
     async def get(self, request: Request) -> Response:

phoenix/server/api/routers/span_handler.py

@@ -7,11 +7,11 @@ from starlette.requests import Request
 from starlette.responses import Response, StreamingResponse
 from starlette.status import HTTP_404_NOT_FOUND, HTTP_422_UNPROCESSABLE_ENTITY
 
+from phoenix.config import DEFAULT_PROJECT_NAME
 from phoenix.core.traces import Traces
 from phoenix.server.api.routers.utils import df_to_bytes, from_iso_format
 from phoenix.trace.dsl import SpanQuery
 from phoenix.utilities import query_spans
-from phoenix.utilities.project import DEFAULT_PROJECT_NAME
 
 
 class SpanHandler(HTTPEndpoint):

phoenix/server/api/routers/trace_handler.py

@@ -14,7 +14,7 @@ from starlette.responses import Response
 from starlette.status import HTTP_415_UNSUPPORTED_MEDIA_TYPE, HTTP_422_UNPROCESSABLE_ENTITY
 
 from phoenix.core.traces import Traces
-from phoenix.storage.spanstore import SpanStore
+from phoenix.storage.span_store import SpanStore
 from phoenix.trace.otel import decode
 from phoenix.utilities.project import get_project_name
 

phoenix/server/api/schema.py

@@ -56,8 +56,8 @@ class Query:
             []
             if (traces := info.context.traces) is None
             else [
-                Project(id_attr=i, name=name, project=project)
-                for i, (name, project) in enumerate(traces.get_projects())
+                Project(id_attr=project_id, name=project_name, project=project)
+                for project_id, project_name, project in traces.get_projects()
             ]
         )
         return connection_from_list(data=data, args=args)
@@ -86,8 +86,11 @@ class Query:
             return to_gql_embedding_dimension(node_id, embedding_dimension)
         elif type_name == "Project":
             if (traces := info.context.traces) is not None:
-                projects = list(traces.get_projects())
-                if node_id < len(projects):
+                projects = {
+                    project_id: (project_name, project)
+                    for project_id, project_name, project in traces.get_projects()
+                }
+                if node_id in projects:
                     name, project = projects[node_id]
                     return Project(id_attr=node_id, name=name, project=project)
             raise Exception(f"Unknown project: {id}")
@@ -224,7 +227,26 @@ class Query:
 
 
 @strawberry.type
-class Mutation(ExportEventsMutation): ...
+class Mutation(ExportEventsMutation):
+    @strawberry.mutation
+    def delete_project(self, info: Info[Context, None], id: GlobalID) -> Query:
+        if (traces := info.context.traces) is None:
+            return Query()
+        type_name, node_id = from_global_id(str(id))
+        if type_name != "Project":
+            return Query()
+        traces.archive_project(node_id)
+        return Query()
+
+    @strawberry.mutation
+    def archive_project(self, info: Info[Context, None], id: GlobalID) -> Query:
+        if (traces := info.context.traces) is None:
+            return Query()
+        type_name, node_id = from_global_id(str(id))
+        if type_name != "Project":
+            return Query()
+        traces.archive_project(node_id)
+        return Query()
 
 
 schema = strawberry.Schema(query=Query, mutation=Mutation)

phoenix/server/api/types/Span.py

@@ -4,12 +4,14 @@ from datetime import datetime
 from enum import Enum
 from typing import Any, DefaultDict, Dict, List, Mapping, Optional, Sized, cast
 
+import numpy as np
 import strawberry
 from openinference.semconv.trace import EmbeddingAttributes, SpanAttributes
 from strawberry import ID, UNSET
 from strawberry.types import Info
 
 import phoenix.trace.schemas as trace_schema
+from phoenix.config import DEFAULT_PROJECT_NAME
 from phoenix.core.project import Project, WrappedSpan
 from phoenix.metrics.retrieval_metrics import RetrievalMetrics
 from phoenix.server.api.context import Context
@@ -17,7 +19,6 @@ from phoenix.server.api.types.DocumentRetrievalMetrics import DocumentRetrievalM
 from phoenix.server.api.types.Evaluation import DocumentEvaluation, SpanEvaluation
 from phoenix.server.api.types.MimeType import MimeType
 from phoenix.trace.schemas import ComputedAttributes, SpanID
-from phoenix.utilities.project import DEFAULT_PROJECT_NAME
 
 EMBEDDING_EMBEDDINGS = SpanAttributes.EMBEDDING_EMBEDDINGS
 EMBEDDING_VECTOR = EmbeddingAttributes.EMBEDDING_VECTOR
@@ -249,7 +250,7 @@ def to_gql_span(span: WrappedSpan, project: Project) -> "Span":
         ),
         attributes=json.dumps(
             _nested_attributes(_hide_embedding_vectors(span.attributes)),
-            default=_json_encode,
+            cls=_JSONEncoder,
         ),
         metadata=_convert_metadata_to_string(span.attributes.get(METADATA)),
         num_documents=num_documents,
@@ -302,10 +303,19 @@ def to_gql_span(span: WrappedSpan, project: Project) -> "Span":
     )
 
 
-def _json_encode(v: Any) -> str:
-    if isinstance(v, datetime):
-        return v.isoformat()
-    return str(v)
+class _JSONEncoder(json.JSONEncoder):
+    def default(self, obj: Any) -> Any:
+        if isinstance(obj, datetime):
+            return obj.isoformat()
+        if isinstance(obj, Enum):
+            return obj.value
+        if isinstance(obj, np.ndarray):
+            return list(obj)
+        if isinstance(obj, np.integer):
+            return int(obj)
+        if isinstance(obj, np.floating):
+            return float(obj)
+        return super().default(obj)
 
 
 def _trie() -> DefaultDict[str, Any]:

phoenix/server/app.py

@@ -28,7 +28,7 @@ from phoenix.server.api.routers.evaluation_handler import EvaluationHandler
 from phoenix.server.api.routers.span_handler import SpanHandler
 from phoenix.server.api.routers.trace_handler import TraceHandler
 from phoenix.server.api.schema import schema
-from phoenix.storage.spanstore import SpanStore
+from phoenix.storage.span_store import SpanStore
 
 logger = logging.getLogger(__name__)
 

phoenix/server/main.py

@@ -16,7 +16,6 @@ from phoenix.config import (
     get_env_host,
     get_env_port,
     get_pids_path,
-    get_storage_dir,
 )
 from phoenix.core.model_schema_adapter import create_model_from_datasets
 from phoenix.core.traces import Traces
@@ -29,8 +28,7 @@ from phoenix.pointcloud.umap_parameters import (
     UMAPParameters,
 )
 from phoenix.server.app import create_app
-from phoenix.storage.spanstore import SpanStore
-from phoenix.storage.spanstore.text_file import TextFileSpanStoreImpl
+from phoenix.storage.span_store import SpanStore
 from phoenix.trace.fixtures import (
     TRACES_FIXTURES,
     _download_traces_fixture,
@@ -39,7 +37,7 @@ from phoenix.trace.fixtures import (
 )
 from phoenix.trace.otel import decode, encode
 from phoenix.trace.span_json_decoder import json_string_to_span
-from phoenix.utilities.project import get_project_name
+from phoenix.utilities.span_store import get_span_store, load_traces_data_from_store
 
 logger = logging.getLogger(__name__)
 
@@ -108,15 +106,6 @@ def _load_items(
         queue.put(item)
 
 
-def _load_from_store(traces: Traces, span_store: SpanStore) -> None:
-    for traces_data in span_store.load():
-        for resource_spans in traces_data.resource_spans:
-            project_name = get_project_name(resource_spans.resource.attributes)
-            for scope_span in resource_spans.scope_spans:
-                for span in scope_span.spans:
-                    traces.put(decode(span), project_name=project_name)
-
-
 DEFAULT_UMAP_PARAMS_STR = f"{DEFAULT_MIN_DIST},{DEFAULT_N_NEIGHBORS},{DEFAULT_N_SAMPLES}"
 
 if __name__ == "__main__":
@@ -142,8 +131,6 @@ if __name__ == "__main__":
     parser.add_argument("--debug", action="store_false")
     subparsers = parser.add_subparsers(dest="command", required=True)
     serve_parser = subparsers.add_parser("serve")
-    experimental_parser = subparsers.add_parser("extremely-dangerous-experimental-span-storage")
-    experimental_parser.add_argument("--storage-path", type=str, required=False)
     datasets_parser = subparsers.add_parser("datasets")
     datasets_parser.add_argument("--primary", type=str, required=True)
     datasets_parser.add_argument("--reference", type=str, required=False)
@@ -200,19 +187,14 @@ if __name__ == "__main__":
         )
         trace_dataset_name = args.trace_fixture
         simulate_streaming = args.simulate_streaming
-    elif args.command == "extremely-dangerous-experimental-span-storage":
-        span_store_path = (
-            get_storage_dir() if args.storage_path is None else Path(args.storage_path)
-        )
-        span_store = TextFileSpanStoreImpl(span_store_path)
 
     model = create_model_from_datasets(
         primary_dataset,
         reference_dataset,
     )
     traces = Traces()
-    if span_store:
-        Thread(target=_load_from_store, args=(traces, span_store), daemon=True).start()
+    if span_store := get_span_store():
+        Thread(target=load_traces_data_from_store, args=(traces, span_store), daemon=True).start()
     if trace_dataset_name is not None:
         fixture_spans = list(
             # Apply `encode` here because legacy jsonl files contains UUIDs as strings.