arize-phoenix 11.6.1__tar.gz → 11.7.0__tar.gz

{arize_phoenix-11.6.1 → arize_phoenix-11.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 11.6.1
+Version: 11.7.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://arize.com/docs/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -40,7 +40,7 @@ Requires-Dist: opentelemetry-sdk
 Requires-Dist: opentelemetry-semantic-conventions
 Requires-Dist: pandas>=1.0
 Requires-Dist: prometheus-client
-Requires-Dist: protobuf<6.0,>=4.25.8
+Requires-Dist: protobuf>=4.25.8
 Requires-Dist: psutil
 Requires-Dist: pyarrow
 Requires-Dist: pydantic>=2.1.0
@@ -65,13 +65,13 @@ Requires-Dist: fast-hdbscan>=0.2.0; extra == 'container'
 Requires-Dist: google-generativeai; extra == 'container'
 Requires-Dist: numba>=0.60.0; extra == 'container'
 Requires-Dist: openai>=1.0.0; extra == 'container'
-Requires-Dist: opentelemetry-exporter-otlp; extra == 'container'
-Requires-Dist: opentelemetry-instrumentation-fastapi; extra == 'container'
-Requires-Dist: opentelemetry-instrumentation-grpc; extra == 'container'
-Requires-Dist: opentelemetry-instrumentation-sqlalchemy; extra == 'container'
-Requires-Dist: opentelemetry-proto>=1.12.0; extra == 'container'
-Requires-Dist: opentelemetry-sdk; extra == 'container'
-Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
+Requires-Dist: opentelemetry-exporter-otlp==1.33.1; extra == 'container'
+Requires-Dist: opentelemetry-instrumentation-fastapi==0.54b1; extra == 'container'
+Requires-Dist: opentelemetry-instrumentation-grpc==0.54b1; extra == 'container'
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy==0.54b1; extra == 'container'
+Requires-Dist: opentelemetry-proto==1.33.1; extra == 'container'
+Requires-Dist: opentelemetry-sdk==1.33.1; extra == 'container'
+Requires-Dist: opentelemetry-semantic-conventions==0.54b1; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
 Requires-Dist: strawberry-graphql[opentelemetry]==0.270.1; extra == 'container'

{arize_phoenix-11.6.1 → arize_phoenix-11.7.0}/pyproject.toml

@@ -34,7 +34,7 @@ dependencies = [
   "typing-extensions>=4.6",
   "scipy",
   "wrapt>=1.17.2",
-  "protobuf>=4.25.8, <6.0",
+  "protobuf>=4.25.8",
   "grpcio",
   "grpc-interceptor",
   "tqdm",
@@ -131,13 +131,15 @@ container = [
   "openai>=1.0.0",
   "azure-identity",
   "aiohttp",  # used by azure-identity via azure.core.pipeline.transport
-  "opentelemetry-sdk",
-  "opentelemetry-proto>=1.12.0",
-  "opentelemetry-exporter-otlp",
-  "opentelemetry-semantic-conventions",
-  "opentelemetry-instrumentation-fastapi",
-  "opentelemetry-instrumentation-sqlalchemy",
-  "opentelemetry-instrumentation-grpc",
+  # Note: when updating opentelemetry dependencies, all packages must use versions
+  # released on the same date (e.g., 1.33.1/0.54b1) to ensure compatibility.
+  "opentelemetry-sdk==1.33.1",
+  "opentelemetry-proto==1.33.1",
+  "opentelemetry-exporter-otlp==1.33.1",
+  "opentelemetry-semantic-conventions==0.54b1",
+  "opentelemetry-instrumentation-fastapi==0.54b1",  # https://github.com/open-telemetry/opentelemetry-python-contrib/issues/3637
+  "opentelemetry-instrumentation-sqlalchemy==0.54b1",
+  "opentelemetry-instrumentation-grpc==0.54b1",
   "py-grpc-prometheus",
   "strawberry-graphql[opentelemetry]==0.270.1",
   "uvloop; platform_system != 'Windows'",

{arize_phoenix-11.6.1 → arize_phoenix-11.7.0}/src/phoenix/experiments/functions.py

@@ -10,7 +10,7 @@ from copy import deepcopy
 from dataclasses import replace
 from datetime import datetime, timezone
 from itertools import product
-from typing import Any, Literal, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union, cast
 from urllib.parse import urljoin
 
 import httpx
@@ -65,6 +65,41 @@ from phoenix.trace.attributes import flatten
 from phoenix.utilities.client import VersionedAsyncClient, VersionedClient
 from phoenix.utilities.json import jsonify
 
+if TYPE_CHECKING:
+    from phoenix.client.resources.datasets import Dataset as ClientDataset
+
+
+def _convert_client_dataset(new_dataset: "ClientDataset") -> Dataset:
+    """
+    Converts Dataset objects from `phoenix.client` to Dataset objects compatible with experiments.
+    """
+    examples_dict: dict[str, Example] = {}
+    for example_data in new_dataset.examples:
+        legacy_example = Example(
+            id=example_data["id"],
+            input=example_data["input"],
+            output=example_data["output"],
+            metadata=example_data["metadata"],
+            updated_at=datetime.fromisoformat(example_data["updated_at"]),
+        )
+        examples_dict[legacy_example.id] = legacy_example
+
+    return Dataset(
+        id=new_dataset.id,
+        version_id=new_dataset.version_id,
+        examples=examples_dict,
+    )
+
+
+def _is_new_client_dataset(dataset: Any) -> bool:
+    """Check if dataset is from new client (has list examples)."""
+    try:
+        from phoenix.client.resources.datasets import Dataset as _ClientDataset
+
+        return isinstance(dataset, _ClientDataset)
+    except ImportError:
+        return False
+
 
 def _phoenix_clients() -> tuple[httpx.Client, httpx.AsyncClient]:
     return VersionedClient(
@@ -85,7 +120,7 @@ RateLimitErrors: TypeAlias = Union[type[BaseException], Sequence[type[BaseExcept
 
 
 def run_experiment(
-    dataset: Dataset,
+    dataset: Union[Dataset, Any],  # Accept both legacy and new client datasets
     task: ExperimentTask,
     evaluators: Optional[Evaluators] = None,
     *,
@@ -166,11 +201,20 @@ def run_experiment(
         RanExperiment: The results of the experiment and evaluation. Additional evaluations can be
             added to the experiment using the `evaluate_experiment` function.
     """
+    # Auto-convert client Dataset objects to legacy format
+    normalized_dataset: Dataset
+    if _is_new_client_dataset(dataset):
+        normalized_dataset = _convert_client_dataset(cast("ClientDataset", dataset))
+    else:
+        normalized_dataset = dataset
+
     task_signature = inspect.signature(task)
     _validate_task_signature(task_signature)
 
-    if not dataset.examples:
-        raise ValueError(f"Dataset has no examples: {dataset.id=}, {dataset.version_id=}")
+    if not normalized_dataset.examples:
+        raise ValueError(
+            f"Dataset has no examples: {normalized_dataset.id=}, {normalized_dataset.version_id=}"
+        )
     # Add this to the params once supported in the UI
     repetitions = 1
     assert repetitions > 0, "Must run the experiment at least once."
@@ -179,7 +223,7 @@ def run_experiment(
     sync_client, async_client = _phoenix_clients()
 
     payload = {
-        "version_id": dataset.version_id,
+        "version_id": normalized_dataset.version_id,
         "name": experiment_name,
         "description": experiment_description,
         "metadata": experiment_metadata,
@@ -187,23 +231,23 @@ def run_experiment(
     }
     if not dry_run:
         experiment_response = sync_client.post(
-            f"/v1/datasets/{dataset.id}/experiments",
+            f"/v1/datasets/{normalized_dataset.id}/experiments",
             json=payload,
         )
         experiment_response.raise_for_status()
         exp_json = experiment_response.json()["data"]
         project_name = exp_json["project_name"]
         experiment = Experiment(
-            dataset_id=dataset.id,
-            dataset_version_id=dataset.version_id,
+            dataset_id=normalized_dataset.id,
+            dataset_version_id=normalized_dataset.version_id,
             repetitions=repetitions,
             id=exp_json["id"],
             project_name=project_name,
         )
     else:
         experiment = Experiment(
-            dataset_id=dataset.id,
-            dataset_version_id=dataset.version_id,
+            dataset_id=normalized_dataset.id,
+            dataset_version_id=normalized_dataset.version_id,
             repetitions=repetitions,
             id=DRY_RUN,
             project_name="",
@@ -216,18 +260,18 @@ def run_experiment(
     print("🧪 Experiment started.")
     if dry_run:
         examples = {
-            (ex := dataset[i]).id: ex
-            for i in pd.Series(range(len(dataset)))
-            .sample(min(len(dataset), int(dry_run)), random_state=42)
+            (ex := normalized_dataset[i]).id: ex
+            for i in pd.Series(range(len(normalized_dataset)))
+            .sample(min(len(normalized_dataset), int(dry_run)), random_state=42)
             .sort_values()
         }
         id_selection = "\n".join(examples)
         print(f"🌵️ This is a dry-run for these example IDs:\n{id_selection}")
-        dataset = replace(dataset, examples=examples)
+        normalized_dataset = replace(normalized_dataset, examples=examples)
     else:
-        dataset_experiments_url = get_dataset_experiments_url(dataset_id=dataset.id)
+        dataset_experiments_url = get_dataset_experiments_url(dataset_id=normalized_dataset.id)
         experiment_compare_url = get_experiment_url(
-            dataset_id=dataset.id,
+            dataset_id=normalized_dataset.id,
             experiment_id=experiment.id,
         )
         print(f"📺 View dataset experiments: {dataset_experiments_url}")
@@ -497,7 +541,7 @@ def run_experiment(
 
     test_cases = [
         TestCase(example=deepcopy(ex), repetition_number=rep)
-        for ex, rep in product(dataset.examples.values(), range(1, repetitions + 1))
+        for ex, rep in product(normalized_dataset.examples.values(), range(1, repetitions + 1))
     ]
     task_runs, _execution_details = executor.run(test_cases)
     print("✅ Task runs completed.")
@@ -513,7 +557,7 @@ def run_experiment(
             task_runs.append(ExperimentRun.from_dict(run))
 
         # Check if we got all expected runs
-        expected_runs = len(dataset.examples) * repetitions
+        expected_runs = len(normalized_dataset.examples) * repetitions
         actual_runs = len(task_runs)
         if actual_runs < expected_runs:
             print(
@@ -521,12 +565,14 @@ def run_experiment(
                 "completed successfully."
             )
 
-    params = ExperimentParameters(n_examples=len(dataset.examples), n_repetitions=repetitions)
+    params = ExperimentParameters(
+        n_examples=len(normalized_dataset.examples), n_repetitions=repetitions
+    )
     task_summary = TaskSummary.from_task_runs(params, task_runs)
     ran_experiment: RanExperiment = object.__new__(RanExperiment)
     ran_experiment.__init__(  # type: ignore[misc]
         params=params,
-        dataset=dataset,
+        dataset=normalized_dataset,
         runs={r.id: r for r in task_runs if r is not None},
         task_summary=task_summary,
         **_asdict(experiment),

{arize_phoenix-11.6.1 → arize_phoenix-11.7.0}/src/phoenix/experiments/tracing.py

@@ -8,7 +8,7 @@ from typing import Any, Optional
 
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import ReadableSpan
-from opentelemetry.trace import INVALID_TRACE_ID
+from opentelemetry.trace import INVALID_SPAN_ID
 from wrapt import apply_patch, resolve_path, wrap_function_wrapper
 
 
@@ -29,7 +29,7 @@ class SpanModifier:
         Args:
           span: ReadableSpan: the span to modify
         """
-        if (ctx := span._context) is None or ctx.span_id == INVALID_TRACE_ID:
+        if (ctx := span._context) is None or ctx.span_id == INVALID_SPAN_ID:
             return
         span._resource = span._resource.merge(self._resource)
 

{arize_phoenix-11.6.1 → arize_phoenix-11.7.0}/src/phoenix/server/api/auth.py

@@ -29,7 +29,7 @@ class IsLocked(Authorization):
     message = "Operations that write or modify data are locked"
 
     def has_permission(self, source: Any, info: Info, **kwargs: Any) -> bool:
-        return not info.context.db.should_not_insert_or_update
+        return not (info.context.db.should_not_insert_or_update or info.context.locked)
 
 
 MSG_ADMIN_ONLY = "Only admin can perform this action"

arize_phoenix-11.7.0/src/phoenix/server/static/.vite/manifest.json

@@ -0,0 +1,87 @@
+{
+  "_components-J3qjrjBf.js": {
+    "file": "assets/components-J3qjrjBf.js",
+    "name": "components",
+    "imports": [
+      "_vendor-BnPh9i9e.js",
+      "_pages-CW1UdBht.js",
+      "_vendor-arizeai-Cr9o_Iu_.js",
+      "_vendor-codemirror-k3zCIjlN.js",
+      "_vendor-three-C5WAXd5r.js"
+    ]
+  },
+  "_pages-CW1UdBht.js": {
+    "file": "assets/pages-CW1UdBht.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-BnPh9i9e.js",
+      "_vendor-arizeai-Cr9o_Iu_.js",
+      "_components-J3qjrjBf.js",
+      "_vendor-codemirror-k3zCIjlN.js",
+      "_vendor-recharts-BdblEuGB.js"
+    ]
+  },
+  "_vendor-BnPh9i9e.js": {
+    "file": "assets/vendor-BnPh9i9e.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-C5WAXd5r.js"
+    ],
+    "css": [
+      "assets/vendor-WIZid84E.css"
+    ]
+  },
+  "_vendor-WIZid84E.css": {
+    "file": "assets/vendor-WIZid84E.css",
+    "src": "_vendor-WIZid84E.css"
+  },
+  "_vendor-arizeai-Cr9o_Iu_.js": {
+    "file": "assets/vendor-arizeai-Cr9o_Iu_.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-BnPh9i9e.js"
+    ]
+  },
+  "_vendor-codemirror-k3zCIjlN.js": {
+    "file": "assets/vendor-codemirror-k3zCIjlN.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-BnPh9i9e.js",
+      "_vendor-shiki-DPtuv2M4.js"
+    ]
+  },
+  "_vendor-recharts-BdblEuGB.js": {
+    "file": "assets/vendor-recharts-BdblEuGB.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-BnPh9i9e.js"
+    ]
+  },
+  "_vendor-shiki-DPtuv2M4.js": {
+    "file": "assets/vendor-shiki-DPtuv2M4.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-BnPh9i9e.js"
+    ]
+  },
+  "_vendor-three-C5WAXd5r.js": {
+    "file": "assets/vendor-three-C5WAXd5r.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-CEObsQf_.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-BnPh9i9e.js",
+      "_vendor-arizeai-Cr9o_Iu_.js",
+      "_pages-CW1UdBht.js",
+      "_components-J3qjrjBf.js",
+      "_vendor-three-C5WAXd5r.js",
+      "_vendor-codemirror-k3zCIjlN.js",
+      "_vendor-shiki-DPtuv2M4.js",
+      "_vendor-recharts-BdblEuGB.js"
+    ]
+  }
+}