arize-phoenix 7.0.1__tar.gz → 7.1.1__tar.gz

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 7.0.1
+Version: 7.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
@@ -13,7 +13,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: <3.13,>=3.9
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: <3.14,>=3.9
 Requires-Dist: aioitertools
 Requires-Dist: aiosqlite
 Requires-Dist: alembic<2,>=1.3.0
@@ -44,13 +45,13 @@ Requires-Dist: scipy
 Requires-Dist: sqlalchemy[asyncio]<3,>=2.0.4
 Requires-Dist: sqlean-py>=3.45.1
 Requires-Dist: starlette
-Requires-Dist: strawberry-graphql==0.247.0
+Requires-Dist: strawberry-graphql==0.253.1
 Requires-Dist: tqdm
 Requires-Dist: typing-extensions>=4.6
 Requires-Dist: uvicorn
 Requires-Dist: websockets
-Requires-Dist: wrapt; sys_platform != 'darwin'
-Requires-Dist: wrapt<1.17; sys_platform == 'darwin'
+Requires-Dist: wrapt<1.17; sys_platform == 'darwin' and platform_machine == 'x86_64'
+Requires-Dist: wrapt>=1.17; sys_platform != 'darwin' or platform_machine != 'x86_64'
 Provides-Extra: container
 Requires-Dist: anthropic; extra == 'container'
 Requires-Dist: fast-hdbscan>=0.2.0; extra == 'container'
@@ -66,7 +67,7 @@ Requires-Dist: opentelemetry-sdk; extra == 'container'
 Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
-Requires-Dist: strawberry-graphql[opentelemetry]==0.247.0; extra == 'container'
+Requires-Dist: strawberry-graphql[opentelemetry]==0.253.1; extra == 'container'
 Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
@@ -97,7 +98,7 @@ Requires-Dist: pytest-postgresql; extra == 'dev'
 Requires-Dist: pytest-xdist; extra == 'dev'
 Requires-Dist: pytest==8.3.3; extra == 'dev'
 Requires-Dist: ruff==0.6.9; extra == 'dev'
-Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.247.0; extra == 'dev'
+Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.253.1; extra == 'dev'
 Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'
@@ -198,7 +199,7 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 | [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅             |
 | Jupyter Notebooks                                                                                                | ✅             |
 | [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅             |
-| [Sessions](https://github.com/Arize-ai/phoenix/issues/2619)                                                      | In Progress 🚧 |
+| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅             |
 | Prompt Management                                                                                                | Coming soon ⏱️ |
 
 ## Tracing Integrations

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/README.md

@@ -74,7 +74,7 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 | [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅             |
 | Jupyter Notebooks                                                                                                | ✅             |
 | [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅             |
-| [Sessions](https://github.com/Arize-ai/phoenix/issues/2619)                                                      | In Progress 🚧 |
+| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅             |
 | Prompt Management                                                                                                | Coming soon ⏱️ |
 
 ## Tracing Integrations

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/pyproject.toml

@@ -2,7 +2,7 @@
 name = "arize-phoenix"
 description = "AI Observability and Evaluation"
 readme = "README.md"
-requires-python = ">=3.9, <3.13"
+requires-python = ">=3.9, <3.14"
 license = {text="Elastic-2.0"}
 license-files = { paths = ["LICENSE", "IP_NOTICE"] }
 keywords = [
@@ -19,6 +19,7 @@ classifiers = [
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
   "scikit-learn",
@@ -28,12 +29,12 @@ dependencies = [
   "starlette",
   "uvicorn",
   "psutil",
-  "strawberry-graphql==0.247.0",  # need to pin version because we're monkey-patching
+  "strawberry-graphql==0.253.1",  # need to pin version because we're monkey-patching
   "pyarrow",
   "typing-extensions>=4.6",
   "scipy",
-  "wrapt<1.17; sys_platform == 'darwin'",
-  "wrapt; sys_platform != 'darwin'",
+  "wrapt<1.17; sys_platform == 'darwin' and platform_machine == 'x86_64'",
+  "wrapt>=1.17; sys_platform != 'darwin' or platform_machine != 'x86_64'",
   "protobuf>=3.20.2, <6.0",
   "grpcio",
   "grpc-interceptor",
@@ -87,7 +88,7 @@ dev = [
   "pytest-postgresql",
   "asyncpg",
   "psycopg[binary,pool]",
-  "strawberry-graphql[debug-server,opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[debug-server,opentelemetry]==0.253.1",  # need to pin version because we're monkey-patching
   "pre-commit",
   "arize[AutoEmbeddings, LLM_Evaluation]",
   "llama-index>=0.10.3",
@@ -133,7 +134,7 @@ container = [
   "opentelemetry-instrumentation-sqlalchemy",
   "opentelemetry-instrumentation-grpc",
   "py-grpc-prometheus",
-  "strawberry-graphql[opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[opentelemetry]==0.253.1",  # need to pin version because we're monkey-patching
   "uvloop; platform_system != 'Windows'",
   "fast-hdbscan>=0.2.0",
   "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
@@ -229,7 +230,7 @@ dependencies = [
   "py-grpc-prometheus",
   "pypistats",  # this is needed to type-check third-party packages
   "requests",  # this is needed to type-check third-party packages
-  "strawberry-graphql[opentelemetry]==0.247.0",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[opentelemetry]==0.253.1",  # need to pin version because we're monkey-patching
   "tenacity",
   "types-cachetools",
   "types-protobuf",

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/auth.py

@@ -20,6 +20,18 @@ class IsNotReadOnly(Authorization):
         return not info.context.read_only
 
 
+class IsLocked(Authorization):
+    """
+    Disables mutations and subscriptions that create or update data but allows
+    queries and delete mutations.
+    """
+
+    message = "Operations that write or modify data are locked"
+
+    def has_permission(self, source: Any, info: Info, **kwargs: Any) -> bool:
+        return not info.context.locked
+
+
 MSG_ADMIN_ONLY = "Only admin can perform this action"
 
 

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/context.py

@@ -108,6 +108,7 @@ class Context(BaseContext):
     event_queue: CanPutItem[DmlEvent] = _NoOp()
     corpus: Optional[Model] = None
     read_only: bool = False
+    locked: bool = False
     auth_enabled: bool = False
     secret: Optional[str] = None
     token_store: Optional[TokenStore] = None

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/helpers/playground_clients.py

@@ -772,6 +772,7 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
     provider_key=GenerativeProviderKey.GEMINI,
     model_names=[
         PROVIDER_DEFAULT,
+        "gemini-2.0-flash-exp",
         "gemini-1.5-flash",
         "gemini-1.5-flash-8b",
         "gemini-1.5-pro",

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/input_types/DataQualityMetricInput.py

@@ -8,7 +8,10 @@ from typing_extensions import Annotated
 from phoenix.core.model_schema import Column
 from phoenix.metrics import Metric
 from phoenix.metrics.mixins import UnaryOperator
-from phoenix.server.api.types.DataQualityMetric import DataQualityMetric
+from phoenix.server.api.types.DataQualityMetric import (
+    DATA_QUALITY_METRIC_FACTORIES,
+    DataQualityMetric,
+)
 
 
 @strawberry.input
@@ -26,7 +29,7 @@ class DataQualityMetricInput:
     metric_instance: strawberry.Private[Metric] = field(init=False)
 
     def __post_init__(self) -> None:
-        metric_instance = self.metric.value()
+        metric_instance = DATA_QUALITY_METRIC_FACTORIES[self.metric]()
         if isinstance(metric_instance, UnaryOperator):
             if not isinstance(self.column_name, str):
                 raise ValueError(f"column name must not be null for {self.metric.name}")

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/input_types/PerformanceMetricInput.py

@@ -3,7 +3,10 @@ import strawberry
 from phoenix.core.model_schema import ACTUAL_LABEL, PREDICTION_LABEL, Column, Model
 from phoenix.metrics import Metric
 from phoenix.metrics.mixins import EvaluationMetric
-from phoenix.server.api.types.PerformanceMetric import PerformanceMetric
+from phoenix.server.api.types.PerformanceMetric import (
+    PERFORMANCE_METRIC_FUNCTIONS,
+    PerformanceMetric,
+)
 
 
 @strawberry.input
@@ -14,5 +17,5 @@ class PerformanceMetricInput:
         return EvaluationMetric(
             actual=Column(model[ACTUAL_LABEL].name),
             predicted=Column(model[PREDICTION_LABEL].name),
-            eval=self.metric.value,
+            eval=PERFORMANCE_METRIC_FUNCTIONS[self.metric],
         )

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/api_key_mutations.py

@@ -8,7 +8,7 @@ from strawberry.relay import GlobalID
 from strawberry.types import Info
 
 from phoenix.db import enums, models
-from phoenix.server.api.auth import IsAdmin, IsNotReadOnly
+from phoenix.server.api.auth import IsAdmin, IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import Unauthorized
 from phoenix.server.api.queries import Query
@@ -60,7 +60,7 @@ class DeleteApiKeyMutationPayload:
 
 @strawberry.type
 class ApiKeyMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin, IsLocked])  # type: ignore
     async def create_system_api_key(
         self, info: Info[Context, None], input: CreateApiKeyInput
     ) -> CreateSystemApiKeyMutationPayload:
@@ -101,7 +101,7 @@ class ApiKeyMutationMixin:
             query=Query(),
         )
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_user_api_key(
         self, info: Info[Context, None], input: CreateUserApiKeyInput
     ) -> CreateUserApiKeyMutationPayload:

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/chat_mutations.py

@@ -25,7 +25,7 @@ from typing_extensions import assert_never
 from phoenix.datetime_utils import local_now, normalize_datetime
 from phoenix.db import models
 from phoenix.db.helpers import get_dataset_example_revisions
-from phoenix.server.api.auth import IsNotReadOnly
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, CustomGraphQLError, NotFound
 from phoenix.server.api.helpers.dataset_helpers import get_dataset_example_output
@@ -120,7 +120,7 @@ class ChatCompletionOverDatasetMutationPayload:
 
 @strawberry.type
 class ChatCompletionMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     @classmethod
     async def chat_completion_over_dataset(
         cls,
@@ -275,7 +275,7 @@ class ChatCompletionMutationMixin:
             payload.examples.append(example_payload)
         return payload
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     @classmethod
     async def chat_completion(
         cls, info: Info[Context, None], input: ChatCompletionInput

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/dataset_mutations.py

@@ -12,7 +12,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.auth import IsNotReadOnly
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, NotFound
 from phoenix.server.api.helpers.dataset_helpers import (
@@ -44,7 +44,7 @@ class DatasetMutationPayload:
 
 @strawberry.type
 class DatasetMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_dataset(
         self,
         info: Info[Context, None],
@@ -67,7 +67,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetInsertEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_dataset(
         self,
         info: Info[Context, None],
@@ -96,7 +96,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetInsertEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def add_spans_to_dataset(
         self,
         info: Info[Context, None],
@@ -205,7 +205,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetInsertEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def add_examples_to_dataset(
         self, info: Info[Context, None], input: AddExamplesToDatasetInput
     ) -> DatasetMutationPayload:
@@ -331,7 +331,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetInsertEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def delete_dataset(
         self,
         info: Info[Context, None],
@@ -362,7 +362,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetDeleteEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_dataset_examples(
         self,
         info: Info[Context, None],
@@ -454,7 +454,7 @@ class DatasetMutationMixin:
         info.context.event_queue.put(DatasetInsertEvent((dataset.id,)))
         return DatasetMutationPayload(dataset=to_gql_dataset(dataset))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def delete_dataset_examples(
         self, info: Info[Context, None], input: DeleteDatasetExamplesInput
     ) -> DatasetMutationPayload:

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/span_annotations_mutations.py

@@ -6,7 +6,7 @@ from strawberry import UNSET
 from strawberry.types import Info
 
 from phoenix.db import models
-from phoenix.server.api.auth import IsNotReadOnly
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.input_types.CreateSpanAnnotationInput import CreateSpanAnnotationInput
 from phoenix.server.api.input_types.DeleteAnnotationsInput import DeleteAnnotationsInput
@@ -25,7 +25,7 @@ class SpanAnnotationMutationPayload:
 
 @strawberry.type
 class SpanAnnotationMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_span_annotations(
         self, info: Info[Context, None], input: list[CreateSpanAnnotationInput]
     ) -> SpanAnnotationMutationPayload:
@@ -59,7 +59,7 @@ class SpanAnnotationMutationMixin:
             query=Query(),
         )
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_span_annotations(
         self, info: Info[Context, None], input: list[PatchAnnotationInput]
     ) -> SpanAnnotationMutationPayload:

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/trace_annotations_mutations.py

@@ -6,7 +6,7 @@ from strawberry import UNSET
 from strawberry.types import Info
 
 from phoenix.db import models
-from phoenix.server.api.auth import IsNotReadOnly
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.input_types.CreateTraceAnnotationInput import CreateTraceAnnotationInput
 from phoenix.server.api.input_types.DeleteAnnotationsInput import DeleteAnnotationsInput
@@ -25,7 +25,7 @@ class TraceAnnotationMutationPayload:
 
 @strawberry.type
 class TraceAnnotationMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_trace_annotations(
         self, info: Info[Context, None], input: list[CreateTraceAnnotationInput]
     ) -> TraceAnnotationMutationPayload:
@@ -59,7 +59,7 @@ class TraceAnnotationMutationMixin:
             query=Query(),
         )
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_trace_annotations(
         self, info: Info[Context, None], input: list[PatchAnnotationInput]
     ) -> TraceAnnotationMutationPayload:

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/mutations/user_mutations.py

@@ -22,7 +22,7 @@ from phoenix.auth import (
     validate_password_format,
 )
 from phoenix.db import enums, models
-from phoenix.server.api.auth import IsAdmin, IsNotReadOnly
+from phoenix.server.api.auth import IsAdmin, IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import Conflict, NotFound, Unauthorized
 from phoenix.server.api.input_types.UserRoleInput import UserRoleInput
@@ -81,7 +81,7 @@ class UserMutationPayload:
 
 @strawberry.type
 class UserMutationMixin:
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin, IsLocked])  # type: ignore
     async def create_user(
         self,
         info: Info[Context, None],
@@ -112,7 +112,7 @@ class UserMutationMixin:
                 raise Conflict(_user_operation_error_message(error))
         return UserMutationPayload(user=to_gql_user(user))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin, IsLocked])  # type: ignore
     async def patch_user(
         self,
         info: Info[Context, None],
@@ -155,7 +155,7 @@ class UserMutationMixin:
             await info.context.log_out(user.id)
         return UserMutationPayload(user=to_gql_user(user))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_viewer(
         self,
         info: Info[Context, None],
@@ -196,7 +196,7 @@ class UserMutationMixin:
             response.delete_cookie(PHOENIX_ACCESS_TOKEN_COOKIE_NAME)
         return UserMutationPayload(user=to_gql_user(user))
 
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin])  # type: ignore
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsAdmin, IsLocked])  # type: ignore
     async def delete_users(
         self,
         info: Info[Context, None],

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/subscriptions.py

@@ -25,7 +25,7 @@ from typing_extensions import TypeAlias, assert_never
 
 from phoenix.datetime_utils import local_now, normalize_datetime
 from phoenix.db import models
-from phoenix.server.api.auth import IsNotReadOnly
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, CustomGraphQLError, NotFound
 from phoenix.server.api.helpers.playground_clients import (
@@ -89,7 +89,7 @@ PLAYGROUND_PROJECT_NAME = "playground"
 
 @strawberry.type
 class Subscription:
-    @strawberry.subscription(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.subscription(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def chat_completion(
         self, info: Info[Context, None], input: ChatCompletionInput
     ) -> AsyncIterator[ChatCompletionSubscriptionPayload]:
@@ -167,7 +167,7 @@ class Subscription:
         info.context.event_queue.put(SpanInsertEvent(ids=(playground_project_id,)))
         yield ChatCompletionSubscriptionResult(span=to_gql_span(db_span))
 
-    @strawberry.subscription(permission_classes=[IsNotReadOnly])  # type: ignore
+    @strawberry.subscription(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def chat_completion_over_dataset(
         self, info: Info[Context, None], input: ChatCompletionOverDatasetInput
     ) -> AsyncIterator[ChatCompletionSubscriptionPayload]:

arize_phoenix-7.1.1/src/phoenix/server/api/types/DataQualityMetric.py

@@ -0,0 +1,40 @@
+from enum import Enum, auto
+from functools import partial
+from typing import Callable, Mapping, cast
+
+import strawberry
+
+from phoenix.metrics import Metric
+from phoenix.metrics.metrics import Cardinality, Count, Max, Mean, Min, PercentEmpty, Quantile, Sum
+
+
+@strawberry.enum
+class DataQualityMetric(Enum):
+    cardinality = auto()
+    percentEmpty = auto()
+    mean = auto()
+    sum = auto()
+    min = auto()
+    max = auto()
+    count = auto()
+    p01 = auto()
+    p25 = auto()
+    p50 = auto()
+    p75 = auto()
+    p99 = auto()
+
+
+DATA_QUALITY_METRIC_FACTORIES: Mapping[DataQualityMetric, Callable[[], Metric]] = {
+    DataQualityMetric.cardinality: cast(Callable[[], Metric], Cardinality),
+    DataQualityMetric.percentEmpty: cast(Callable[[], Metric], PercentEmpty),
+    DataQualityMetric.mean: cast(Callable[[], Metric], Mean),
+    DataQualityMetric.sum: cast(Callable[[], Metric], Sum),
+    DataQualityMetric.min: cast(Callable[[], Metric], Min),
+    DataQualityMetric.max: cast(Callable[[], Metric], Max),
+    DataQualityMetric.count: cast(Callable[[], Metric], Count),
+    DataQualityMetric.p01: cast(Callable[[], Metric], partial(Quantile, probability=0.01)),
+    DataQualityMetric.p25: cast(Callable[[], Metric], partial(Quantile, probability=0.25)),
+    DataQualityMetric.p50: cast(Callable[[], Metric], partial(Quantile, probability=0.50)),
+    DataQualityMetric.p75: cast(Callable[[], Metric], partial(Quantile, probability=0.75)),
+    DataQualityMetric.p99: cast(Callable[[], Metric], partial(Quantile, probability=0.99)),
+}

arize_phoenix-7.1.1/src/phoenix/server/api/types/PerformanceMetric.py

@@ -0,0 +1,16 @@
+from enum import Enum, auto
+from typing import Callable, Mapping, cast
+
+import strawberry
+
+from phoenix.metrics.wrappers import SkEval
+
+
+@strawberry.enum
+class PerformanceMetric(Enum):
+    accuracyScore = auto()
+
+
+PERFORMANCE_METRIC_FUNCTIONS: Mapping[PerformanceMetric, Callable[..., float]] = {
+    PerformanceMetric.accuracyScore: cast(Callable[..., float], SkEval.accuracy_score),
+}

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/types/TimeSeries.py

@@ -15,7 +15,10 @@ from phoenix.metrics.timeseries import timeseries
 from phoenix.server.api.input_types.Granularity import Granularity, to_timestamps
 from phoenix.server.api.input_types.TimeRange import TimeRange
 from phoenix.server.api.interceptor import GqlValueMediator
-from phoenix.server.api.types.DataQualityMetric import DataQualityMetric
+from phoenix.server.api.types.DataQualityMetric import (
+    DATA_QUALITY_METRIC_FACTORIES,
+    DataQualityMetric,
+)
 from phoenix.server.api.types.InferencesRole import InferencesRole
 from phoenix.server.api.types.ScalarDriftMetricEnum import ScalarDriftMetric
 from phoenix.server.api.types.VectorDriftMetricEnum import VectorDriftMetric
@@ -100,7 +103,7 @@ def get_data_quality_timeseries_data(
     granularity: Granularity,
     inferences_role: InferencesRole,
 ) -> list[TimeSeriesDataPoint]:
-    metric_instance = metric.value()
+    metric_instance = DATA_QUALITY_METRIC_FACTORIES[metric]()
     if isinstance(metric_instance, UnaryOperator):
         metric_instance = replace(
             metric_instance,

{arize_phoenix-7.0.1 → arize_phoenix-7.1.1}/src/phoenix/server/api/types/Trace.py

@@ -36,6 +36,19 @@ class Trace(Node):
     start_time: datetime
     end_time: datetime
 
+    @strawberry.field
+    async def latency_ms(
+        self,
+        info: Info[Context, None],
+    ) -> Optional[float]:
+        async with info.context.db() as session:
+            latency = await session.scalar(
+                select(
+                    models.Trace.latency_ms,
+                ).where(models.Trace.id == self.id_attr)
+            )
+        return latency
+
     @strawberry.field
     async def project_id(self) -> GlobalID:
         from phoenix.server.api.types.Project import Project

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

@@ -0,0 +1,87 @@
+{
+  "_components-DaxEPah8.js": {
+    "file": "assets/components-DaxEPah8.js",
+    "name": "components",
+    "imports": [
+      "_vendor-DwOx_a-2.js",
+      "_pages-CY7Swxhi.js",
+      "_vendor-arizeai-BWSsCZNG.js",
+      "_vendor-codemirror-WrxmCiOZ.js",
+      "_vendor-three-DwGkEfCM.js"
+    ]
+  },
+  "_pages-CY7Swxhi.js": {
+    "file": "assets/pages-CY7Swxhi.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-DwOx_a-2.js",
+      "_vendor-arizeai-BWSsCZNG.js",
+      "_components-DaxEPah8.js",
+      "_vendor-codemirror-WrxmCiOZ.js",
+      "_vendor-recharts-Bj7q9ORc.js"
+    ]
+  },
+  "_vendor-!~{003}~.js": {
+    "file": "assets/vendor-DxkFTwjz.css",
+    "src": "_vendor-!~{003}~.js"
+  },
+  "_vendor-DwOx_a-2.js": {
+    "file": "assets/vendor-DwOx_a-2.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-DwGkEfCM.js"
+    ],
+    "css": [
+      "assets/vendor-DxkFTwjz.css"
+    ]
+  },
+  "_vendor-arizeai-BWSsCZNG.js": {
+    "file": "assets/vendor-arizeai-BWSsCZNG.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-DwOx_a-2.js"
+    ]
+  },
+  "_vendor-codemirror-WrxmCiOZ.js": {
+    "file": "assets/vendor-codemirror-WrxmCiOZ.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-DwOx_a-2.js",
+      "_vendor-shiki-RnWcZtzE.js"
+    ]
+  },
+  "_vendor-recharts-Bj7q9ORc.js": {
+    "file": "assets/vendor-recharts-Bj7q9ORc.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-DwOx_a-2.js"
+    ]
+  },
+  "_vendor-shiki-RnWcZtzE.js": {
+    "file": "assets/vendor-shiki-RnWcZtzE.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-DwOx_a-2.js"
+    ]
+  },
+  "_vendor-three-DwGkEfCM.js": {
+    "file": "assets/vendor-three-DwGkEfCM.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-DxfKd3lv.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-DwOx_a-2.js",
+      "_vendor-arizeai-BWSsCZNG.js",
+      "_pages-CY7Swxhi.js",
+      "_components-DaxEPah8.js",
+      "_vendor-three-DwGkEfCM.js",
+      "_vendor-codemirror-WrxmCiOZ.js",
+      "_vendor-shiki-RnWcZtzE.js",
+      "_vendor-recharts-Bj7q9ORc.js"
+    ]
+  }
+}