arize-phoenix 5.0.0__tar.gz → 5.1.1__tar.gz

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/.gitignore

@@ -28,5 +28,6 @@ pyrightconfig.json
 
 # Demo data
 tutorials/internal/demo_llama_index/*.json
+examples/agent_framework_comparison/utils/saved_traces/*.parquet
 .env
 .conda

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 5.0.0
+Version: 5.1.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -27,10 +27,11 @@ Requires-Dist: fastapi
 Requires-Dist: fastapi-mail
 Requires-Dist: grpc-interceptor
 Requires-Dist: grpcio
-Requires-Dist: hdbscan>=0.8.33
+Requires-Dist: hdbscan>=0.8.38
 Requires-Dist: httpx
 Requires-Dist: jinja2
-Requires-Dist: numpy<2
+Requires-Dist: numba>=0.60.0
+Requires-Dist: numpy!=2.0.0
 Requires-Dist: openinference-instrumentation>=0.1.12
 Requires-Dist: openinference-semantic-conventions>=0.1.9
 Requires-Dist: opentelemetry-exporter-otlp

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/integration_tests/pytest.ini

@@ -1,6 +1,9 @@
 [pytest]
 addopts =
-  -rfE -l
+  -ra
+  -n auto
+  --showlocals
+  --exitfirst
   --color=yes
   --code-highlight=yes
   --import-mode=importlib

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/pyproject.toml

@@ -22,11 +22,12 @@ classifiers = [
 ]
 dependencies = [
   "scikit-learn",
-  "numpy<2",  # https://github.com/scikit-learn-contrib/hdbscan/issues/642
+  "numpy!=2.0.0",  # https://github.com/lmcinnes/umap/issues/1138
   "pandas>=1.0",
   "jinja2",
   "umap-learn",
-  "hdbscan>=0.8.33",
+  "hdbscan>=0.8.38",
+  "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
   "starlette",
   "uvicorn",
   "psutil",

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/src/phoenix/pointcloud/clustering.py

@@ -3,7 +3,6 @@ from typing import List, Set
 
 import numpy as np
 import numpy.typing as npt
-from hdbscan import HDBSCAN
 from typing_extensions import TypeAlias
 
 RowIndex: TypeAlias = int
@@ -18,6 +17,8 @@ class Hdbscan:
     cluster_selection_epsilon: float = 0.0
 
     def find_clusters(self, mat: Matrix) -> List[RawCluster]:
+        from hdbscan import HDBSCAN
+
         cluster_ids: npt.NDArray[np.int_] = HDBSCAN(**asdict(self)).fit_predict(mat)
         ans: List[RawCluster] = [set() for _ in range(np.max(cluster_ids) + 1)]
         for row_idx, cluster_id in enumerate(cluster_ids):

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/src/phoenix/pointcloud/projectors.py

@@ -6,12 +6,6 @@ import numpy as np
 import numpy.typing as npt
 from typing_extensions import TypeAlias
 
-with warnings.catch_warnings():
-    from numba.core.errors import NumbaWarning
-
-    warnings.simplefilter("ignore", category=NumbaWarning)
-    from umap import UMAP
-
 Matrix: TypeAlias = npt.NDArray[np.float64]
 
 
@@ -25,6 +19,11 @@ class Umap:
     min_dist: float = 0.1
 
     def project(self, mat: Matrix, n_components: int) -> Matrix:
+        with warnings.catch_warnings():
+            from numba.core.errors import NumbaWarning
+
+            warnings.simplefilter("ignore", category=NumbaWarning)
+            from umap import UMAP
         config = asdict(self)
         config["n_components"] = n_components
         if len(mat) <= n_components:

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/src/phoenix/server/api/types/EmbeddingDimension.py

@@ -422,17 +422,17 @@ class EmbeddingDimension(Node):
             if isinstance(inferences_role, InferencesRole):
                 dataset = model[inferences_role.value]
                 embedding_metadata = EmbeddingMetadata(
-                    prediction_id=dataset[PREDICTION_ID][row_id],
-                    link_to_data=dataset[self.dimension.link_to_data][row_id],
-                    raw_data=dataset[self.dimension.raw_data][row_id],
+                    prediction_id=dataset[PREDICTION_ID].iloc[row_id],
+                    link_to_data=dataset[self.dimension.link_to_data].iloc[row_id],
+                    raw_data=dataset[self.dimension.raw_data].iloc[row_id],
                 )
             elif (corpus := info.context.corpus) is not None:
                 dataset = corpus[PRIMARY]
                 dimension = cast(ms.EmbeddingDimension, corpus[PROMPT])
                 embedding_metadata = EmbeddingMetadata(
-                    prediction_id=dataset[PREDICTION_ID][row_id],
-                    link_to_data=dataset[dimension.link_to_data][row_id],
-                    raw_data=dataset[dimension.raw_data][row_id],
+                    prediction_id=dataset[PREDICTION_ID].iloc[row_id],
+                    link_to_data=dataset[dimension.link_to_data].iloc[row_id],
+                    raw_data=dataset[dimension.raw_data].iloc[row_id],
                 )
             else:
                 continue
@@ -445,10 +445,10 @@ class EmbeddingDimension(Node):
                     event_id=event_id,
                     coordinates=to_gql_coordinates(vector),
                     event_metadata=EventMetadata(
-                        prediction_label=dataset[PREDICTION_LABEL][row_id],
-                        prediction_score=dataset[PREDICTION_SCORE][row_id],
-                        actual_label=dataset[ACTUAL_LABEL][row_id],
-                        actual_score=dataset[ACTUAL_SCORE][row_id],
+                        prediction_label=dataset[PREDICTION_LABEL].iloc[row_id],
+                        prediction_score=dataset[PREDICTION_SCORE].iloc[row_id],
+                        actual_label=dataset[ACTUAL_LABEL].iloc[row_id],
+                        actual_score=dataset[ACTUAL_SCORE].iloc[row_id],
                     ),
                     embedding_metadata=embedding_metadata,
                 )

{arize_phoenix-5.0.0 → arize_phoenix-5.1.1}/src/phoenix/trace/fixtures.py

@@ -145,6 +145,27 @@ demo_llama_index_rag_fixture = TracesFixture(
     ),
 )
 
+demo_code_based_agent_fixture = TracesFixture(
+    name="demo_code_based_agent",
+    project_name="demo_code_based_agent",
+    description="Shows traces captured from a code-based agent.",
+    file_name="agent-demo-traces.parquet",
+)
+
+demo_langgraph_agent_fixture = TracesFixture(
+    name="demo_langgraph_agent",
+    project_name="demo_langgraph_agent",
+    description="Shows traces captured from a Langgraph agent.",
+    file_name="langgraph-demo-traces.parquet",
+)
+
+demo_llamaindex_workflows_agent_fixture = TracesFixture(
+    name="demo_llamaindex_workflows_agent",
+    project_name="demo_llamaindex_workflows_agent",
+    description="Shows traces captured from a LlamaIndex Workflows agent.",
+    file_name="llamaindex-workflow-demo-traces.parquet",
+)
+
 demo_o1_preview_timeseries_testing_fixture = TracesFixture(
     name="demo_o1_preview_timeseries_evals",
     project_name="demo_o1_preview_timeseries",
@@ -257,6 +278,9 @@ random_fixture = TracesFixture(
 TRACES_FIXTURES: List[TracesFixture] = [
     demo_llama_index_rag_fixture,
     demo_llama_index_rag_llm_fixture,
+    demo_langgraph_agent_fixture,
+    demo_code_based_agent_fixture,
+    demo_llamaindex_workflows_agent_fixture,
     demo_o1_preview_timeseries_testing_fixture,
     llama_index_rag_fixture,
     llama_index_rag_fixture_with_davinci,

arize_phoenix-5.1.1/src/phoenix/version.py

@@ -0,0 +1 @@
+__version__ = "5.1.1"

arize_phoenix-5.0.0/src/phoenix/version.py

@@ -1 +0,0 @@
-__version__ = "5.0.0"