arize-phoenix 4.28.1__py3-none-any.whl → 4.30.0__py3-none-any.whl

phoenix/server/api/mutations/dataset_mutations.py

@@ -13,6 +13,7 @@ from strawberry.types import Info
 from phoenix.db import models
 from phoenix.db.helpers import get_eval_trace_ids_for_datasets, get_project_names_for_datasets
 from phoenix.server.api.context import Context
+from phoenix.server.api.exceptions import BadRequest, NotFound
 from phoenix.server.api.helpers.dataset_helpers import (
     get_dataset_example_input,
     get_dataset_example_output,
@@ -362,7 +363,7 @@ class DatasetMutationMixin:
                 expected_type_name=Dataset.__name__,
             )
         except ValueError:
-            raise ValueError(f"Unknown dataset: {input.dataset_id}")
+            raise NotFound(f"Unknown dataset: {input.dataset_id}")
         project_names_stmt = get_project_names_for_datasets(dataset_id)
         eval_trace_ids_stmt = get_eval_trace_ids_for_datasets(dataset_id)
         stmt = (
@@ -372,7 +373,7 @@ class DatasetMutationMixin:
             project_names = await session.scalars(project_names_stmt)
             eval_trace_ids = await session.scalars(eval_trace_ids_stmt)
             if not (dataset := await session.scalar(stmt)):
-                raise ValueError(f"Unknown dataset: {input.dataset_id}")
+                raise NotFound(f"Unknown dataset: {input.dataset_id}")
         await asyncio.gather(
             delete_projects(info.context.db, *project_names),
             delete_traces(info.context.db, *eval_trace_ids),
@@ -388,7 +389,7 @@ class DatasetMutationMixin:
         input: PatchDatasetExamplesInput,
     ) -> DatasetMutationPayload:
         if not (patches := input.patches):
-            raise ValueError("Must provide examples to patch.")
+            raise BadRequest("Must provide examples to patch.")
         by_numeric_id = [
             (
                 from_global_id_with_expected_type(patch.example_id, DatasetExample.__name__),
@@ -399,9 +400,9 @@ class DatasetMutationMixin:
         ]
         example_ids, _, patches = map(list, zip(*sorted(by_numeric_id)))
         if len(set(example_ids)) < len(example_ids):
-            raise ValueError("Cannot patch the same example more than once per mutation.")
+            raise BadRequest("Cannot patch the same example more than once per mutation.")
         if any(patch.is_empty() for patch in patches):
-            raise ValueError("Received one or more empty patches that contain no fields to update.")
+            raise BadRequest("Received one or more empty patches that contain no fields to update.")
         version_description = input.version_description or None
         version_metadata = input.version_metadata
         async with info.context.db() as session:
@@ -419,9 +420,9 @@ class DatasetMutationMixin:
                 )
             ).all()
             if not datasets:
-                raise ValueError("No examples found.")
+                raise NotFound("No examples found.")
             if len(set(ds.id for ds in datasets)) > 1:
-                raise ValueError("Examples must come from the same dataset.")
+                raise BadRequest("Examples must come from the same dataset.")
             dataset = datasets[0]
 
             revision_ids = (
@@ -445,7 +446,7 @@ class DatasetMutationMixin:
                 )
             ).all()
             if (num_missing_examples := len(example_ids) - len(revisions)) > 0:
-                raise ValueError(f"{num_missing_examples} example(s) could not be found.")
+                raise NotFound(f"{num_missing_examples} example(s) could not be found.")
 
             version_id = await session.scalar(
                 insert(models.DatasetVersion)

phoenix/server/api/mutations/experiment_mutations.py

@@ -9,6 +9,7 @@ from strawberry.types import Info
 from phoenix.db import models
 from phoenix.db.helpers import get_eval_trace_ids_for_experiments, get_project_names_for_experiments
 from phoenix.server.api.context import Context
+from phoenix.server.api.exceptions import CustomGraphQLError
 from phoenix.server.api.input_types.DeleteExperimentsInput import DeleteExperimentsInput
 from phoenix.server.api.mutations.auth import IsAuthenticated
 from phoenix.server.api.types.Experiment import Experiment, to_gql_experiment
@@ -52,7 +53,7 @@ class ExperimentMutationMixin:
             }
             if unknown_experiment_ids := set(experiment_ids) - set(experiments.keys()):
                 await savepoint.rollback()
-                raise ValueError(
+                raise CustomGraphQLError(
                     "Failed to delete experiment(s), "
                     "probably due to invalid input experiment ID(s): "
                     + str(

phoenix/server/api/queries.py

@@ -33,6 +33,7 @@ from phoenix.db.models import (
 )
 from phoenix.pointcloud.clustering import Hdbscan
 from phoenix.server.api.context import Context
+from phoenix.server.api.exceptions import NotFound
 from phoenix.server.api.helpers import ensure_list
 from phoenix.server.api.input_types.ClusterInput import ClusterInput
 from phoenix.server.api.input_types.Coordinates import (
@@ -391,7 +392,7 @@ class Query:
             async with info.context.db() as session:
                 project = (await session.execute(project_stmt)).first()
             if project is None:
-                raise ValueError(f"Unknown project: {id}")
+                raise NotFound(f"Unknown project: {id}")
             return Project(
                 id_attr=project.id,
                 name=project.name,
@@ -406,7 +407,7 @@ class Query:
             async with info.context.db() as session:
                 trace = (await session.execute(trace_stmt)).first()
             if trace is None:
-                raise ValueError(f"Unknown trace: {id}")
+                raise NotFound(f"Unknown trace: {id}")
             return Trace(
                 id_attr=trace.id, trace_id=trace.trace_id, project_rowid=trace.project_rowid
             )
@@ -421,13 +422,13 @@ class Query:
             async with info.context.db() as session:
                 span = await session.scalar(span_stmt)
             if span is None:
-                raise ValueError(f"Unknown span: {id}")
+                raise NotFound(f"Unknown span: {id}")
             return to_gql_span(span)
         elif type_name == Dataset.__name__:
             dataset_stmt = select(models.Dataset).where(models.Dataset.id == node_id)
             async with info.context.db() as session:
                 if (dataset := await session.scalar(dataset_stmt)) is None:
-                    raise ValueError(f"Unknown dataset: {id}")
+                    raise NotFound(f"Unknown dataset: {id}")
             return to_gql_dataset(dataset)
         elif type_name == DatasetExample.__name__:
             example_id = node_id
@@ -453,7 +454,7 @@ class Query:
                     )
                 )
             if not example:
-                raise ValueError(f"Unknown dataset example: {id}")
+                raise NotFound(f"Unknown dataset example: {id}")
             return DatasetExample(
                 id_attr=example.id,
                 created_at=example.created_at,
@@ -464,7 +465,7 @@ class Query:
                     select(models.Experiment).where(models.Experiment.id == node_id)
                 )
             if not experiment:
-                raise ValueError(f"Unknown experiment: {id}")
+                raise NotFound(f"Unknown experiment: {id}")
             return Experiment(
                 id_attr=experiment.id,
                 name=experiment.name,
@@ -485,9 +486,9 @@ class Query:
                         )
                     )
                 ):
-                    raise ValueError(f"Unknown experiment run: {id}")
+                    raise NotFound(f"Unknown experiment run: {id}")
             return to_gql_experiment_run(run)
-        raise Exception(f"Unknown node type: {type_name}")
+        raise NotFound(f"Unknown node type: {type_name}")
 
     @strawberry.field
     def clusters(

phoenix/server/api/routers/v1/experiments.py

@@ -2,7 +2,7 @@ from datetime import datetime
 from random import getrandbits
 from typing import Any, Dict, List, Optional
 
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Path
 from pydantic import Field
 from sqlalchemy import select
 from starlette.requests import Request
@@ -18,7 +18,7 @@ from phoenix.server.dml_event import ExperimentInsertEvent
 from .pydantic_compat import V1RoutesBaseModel
 from .utils import ResponseBody, add_errors_to_responses
 
-router = APIRouter(tags=["experiments"], include_in_schema=False)
+router = APIRouter(tags=["experiments"], include_in_schema=True)
 
 
 def _short_uuid() -> str:
@@ -90,8 +90,8 @@ class CreateExperimentResponseBody(ResponseBody[Experiment]):
 )
 async def create_experiment(
     request: Request,
-    dataset_id: str,
     request_body: CreateExperimentRequestBody,
+    dataset_id: str = Path(..., title="Dataset ID"),
 ) -> CreateExperimentResponseBody:
     dataset_globalid = GlobalID.from_id(dataset_id)
     try:
@@ -266,7 +266,7 @@ class ListExperimentsResponseBody(ResponseBody[List[Experiment]]):
 )
 async def list_experiments(
     request: Request,
-    dataset_id: str,
+    dataset_id: str = Path(..., title="Dataset ID"),
 ) -> ListExperimentsResponseBody:
     dataset_gid = GlobalID.from_id(dataset_id)
     try:

phoenix/server/api/schema.py

@@ -1,5 +1,6 @@
 import strawberry
 
+from phoenix.server.api.exceptions import get_mask_errors_extension
 from phoenix.server.api.mutations import Mutation
 from phoenix.server.api.queries import Query
 
@@ -10,4 +11,5 @@ from phoenix.server.api.queries import Query
 schema = strawberry.Schema(
     query=Query,
     mutation=Mutation,
+    extensions=[get_mask_errors_extension()],
 )

phoenix/server/static/.vite/manifest.json

@@ -1,32 +1,32 @@
 {
-  "_components-BYH03rjA.js": {
-    "file": "assets/components-BYH03rjA.js",
+  "_components-CkSg5zK4.js": {
+    "file": "assets/components-CkSg5zK4.js",
     "name": "components",
     "imports": [
-      "_vendor-aSQri0vz.js",
-      "_vendor-arizeai-CsdcB1NH.js",
-      "_pages-DnbxgoTK.js",
+      "_vendor-DsnEJuEV.js",
+      "_vendor-arizeai-DtynTLNi.js",
+      "_pages-C6emDFIO.js",
       "_vendor-three-DwGkEfCM.js",
-      "_vendor-codemirror-CYHkhs7D.js"
+      "_vendor-codemirror-C5to5cK4.js"
     ]
   },
-  "_pages-DnbxgoTK.js": {
-    "file": "assets/pages-DnbxgoTK.js",
+  "_pages-C6emDFIO.js": {
+    "file": "assets/pages-C6emDFIO.js",
     "name": "pages",
     "imports": [
-      "_vendor-aSQri0vz.js",
-      "_components-BYH03rjA.js",
-      "_vendor-arizeai-CsdcB1NH.js",
-      "_vendor-recharts-B0sannek.js",
-      "_vendor-codemirror-CYHkhs7D.js"
+      "_vendor-DsnEJuEV.js",
+      "_components-CkSg5zK4.js",
+      "_vendor-arizeai-DtynTLNi.js",
+      "_vendor-recharts-reihe2SJ.js",
+      "_vendor-codemirror-C5to5cK4.js"
     ]
   },
   "_vendor-!~{003}~.js": {
     "file": "assets/vendor-DxkFTwjz.css",
     "src": "_vendor-!~{003}~.js"
   },
-  "_vendor-aSQri0vz.js": {
-    "file": "assets/vendor-aSQri0vz.js",
+  "_vendor-DsnEJuEV.js": {
+    "file": "assets/vendor-DsnEJuEV.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-DwGkEfCM.js"
@@ -35,25 +35,25 @@
       "assets/vendor-DxkFTwjz.css"
     ]
   },
-  "_vendor-arizeai-CsdcB1NH.js": {
-    "file": "assets/vendor-arizeai-CsdcB1NH.js",
+  "_vendor-arizeai-DtynTLNi.js": {
+    "file": "assets/vendor-arizeai-DtynTLNi.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-aSQri0vz.js"
+      "_vendor-DsnEJuEV.js"
     ]
   },
-  "_vendor-codemirror-CYHkhs7D.js": {
-    "file": "assets/vendor-codemirror-CYHkhs7D.js",
+  "_vendor-codemirror-C5to5cK4.js": {
+    "file": "assets/vendor-codemirror-C5to5cK4.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-aSQri0vz.js"
+      "_vendor-DsnEJuEV.js"
     ]
   },
-  "_vendor-recharts-B0sannek.js": {
-    "file": "assets/vendor-recharts-B0sannek.js",
+  "_vendor-recharts-reihe2SJ.js": {
+    "file": "assets/vendor-recharts-reihe2SJ.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-aSQri0vz.js"
+      "_vendor-DsnEJuEV.js"
     ]
   },
   "_vendor-three-DwGkEfCM.js": {
@@ -61,18 +61,18 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-fqdjNpYm.js",
+    "file": "assets/index-DTecsU5w.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-aSQri0vz.js",
-      "_vendor-arizeai-CsdcB1NH.js",
-      "_pages-DnbxgoTK.js",
-      "_components-BYH03rjA.js",
+      "_vendor-DsnEJuEV.js",
+      "_vendor-arizeai-DtynTLNi.js",
+      "_pages-C6emDFIO.js",
+      "_components-CkSg5zK4.js",
       "_vendor-three-DwGkEfCM.js",
-      "_vendor-recharts-B0sannek.js",
-      "_vendor-codemirror-CYHkhs7D.js"
+      "_vendor-recharts-reihe2SJ.js",
+      "_vendor-codemirror-C5to5cK4.js"
     ]
   }
 }