arize-phoenix 3.9.0__tar.gz → 3.10.0__tar.gz

{arize_phoenix-3.9.0 → arize_phoenix-3.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 3.9.0
+Version: 3.10.0
 Summary: ML Observability in your notebook
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-3.9.0 → arize_phoenix-3.10.0}/src/phoenix/config.py

@@ -17,6 +17,10 @@ ENV_PHOENIX_WORKING_DIR = "PHOENIX_WORKING_DIR"
 The directory in which to save, load, and export datasets. This directory must
 be accessible by both the Phoenix server and the notebook environment.
 """
+ENV_PHOENIX_PROJECT_NAME = "PHOENIX_PROJECT_NAME"
+"""
+The project name to use when logging traces and evals. defaults to 'default'.
+"""
 
 
 def _get_temp_path() -> Path:
@@ -113,3 +117,7 @@ def get_env_host() -> str:
 
 def get_env_collector_endpoint() -> Optional[str]:
     return os.getenv(ENV_PHOENIX_COLLECTOR_ENDPOINT)
+
+
+def get_env_project_name() -> str:
+    return os.getenv(ENV_PHOENIX_PROJECT_NAME) or "default"

{arize_phoenix-3.9.0 → arize_phoenix-3.10.0}/src/phoenix/server/api/schema.py

@@ -1,7 +1,5 @@
 from collections import defaultdict
-from datetime import datetime
-from itertools import chain
-from typing import Dict, List, Optional, Set, Tuple, Union, cast
+from typing import Dict, List, Optional, Set, Union
 
 import numpy as np
 import numpy.typing as npt
@@ -10,7 +8,6 @@ from strawberry import ID, UNSET
 from strawberry.types import Info
 from typing_extensions import Annotated
 
-from phoenix.core.project import DEFAULT_PROJECT_NAME
 from phoenix.pointcloud.clustering import Hdbscan
 from phoenix.server.api.helpers import ensure_list
 from phoenix.server.api.input_types.ClusterInput import ClusterInput
@@ -18,15 +15,10 @@ from phoenix.server.api.input_types.Coordinates import (
     InputCoordinate2D,
     InputCoordinate3D,
 )
-from phoenix.server.api.input_types.SpanSort import SpanSort
 from phoenix.server.api.types.Cluster import Cluster, to_gql_clusters
 from phoenix.server.api.types.Project import Project
-from phoenix.trace.dsl import SpanFilter
-from phoenix.trace.schemas import SpanID, TraceID
 
 from .context import Context
-from .input_types.TimeRange import TimeRange
-from .types.DatasetInfo import TraceDatasetInfo
 from .types.DatasetRole import AncillaryDatasetRole, DatasetRole
 from .types.Dimension import to_gql_dimension
 from .types.EmbeddingDimension import (
@@ -35,15 +27,12 @@ from .types.EmbeddingDimension import (
     DEFAULT_MIN_SAMPLES,
     to_gql_embedding_dimension,
 )
-from .types.EvaluationSummary import EvaluationSummary
 from .types.Event import create_event_id, unpack_event_id
 from .types.ExportEventsMutation import ExportEventsMutation
 from .types.Functionality import Functionality
 from .types.Model import Model
 from .types.node import GlobalID, Node, from_global_id
 from .types.pagination import Connection, ConnectionArgs, Cursor, connection_from_list
-from .types.Span import Span, to_gql_span
-from .types.ValidationResult import ValidationResult
 
 
 @strawberry.type
@@ -233,190 +222,6 @@ class Query:
             clustered_events=clustered_events,
         )
 
-    @strawberry.field
-    def streaming_last_updated_at(
-        self,
-        info: Info[Context, None],
-    ) -> Optional[datetime]:
-        last_updated_at: Optional[datetime] = None
-        if (traces := info.context.traces) is not None and (
-            traces_last_updated_at := traces.last_updated_at
-        ) is not None:
-            last_updated_at = (
-                traces_last_updated_at
-                if last_updated_at is None
-                else max(last_updated_at, traces_last_updated_at)
-            )
-        return last_updated_at
-
-    @strawberry.field
-    def spans(
-        self,
-        info: Info[Context, None],
-        time_range: Optional[TimeRange] = UNSET,
-        trace_ids: Optional[List[ID]] = UNSET,
-        first: Optional[int] = 50,
-        last: Optional[int] = UNSET,
-        after: Optional[Cursor] = UNSET,
-        before: Optional[Cursor] = UNSET,
-        sort: Optional[SpanSort] = UNSET,
-        root_spans_only: Optional[bool] = UNSET,
-        filter_condition: Optional[str] = UNSET,
-    ) -> Connection[Span]:
-        args = ConnectionArgs(
-            first=first,
-            after=after if isinstance(after, Cursor) else None,
-            last=last,
-            before=before if isinstance(before, Cursor) else None,
-        )
-        if not (traces := info.context.traces) or not (
-            project := traces.get_project(DEFAULT_PROJECT_NAME)
-        ):
-            return connection_from_list(data=[], args=args)
-        predicate = (
-            SpanFilter(
-                condition=filter_condition,
-                evals=project,
-            )
-            if filter_condition
-            else None
-        )
-        if not trace_ids:
-            spans = traces.get_spans(
-                start_time=time_range.start if time_range else None,
-                stop_time=time_range.end if time_range else None,
-                root_spans_only=root_spans_only,
-            )
-        else:
-            spans = chain.from_iterable(
-                traces.get_trace(trace_id) for trace_id in map(TraceID, trace_ids)
-            )
-        if predicate:
-            spans = filter(predicate, spans)
-        if sort:
-            spans = sort(spans, evals=project)
-        data = list(map(to_gql_span, spans))
-        return connection_from_list(data=data, args=args)
-
-    @strawberry.field(
-        description="Names of all available evaluations for spans. "
-        "(The list contains no duplicates.)"
-    )  # type: ignore
-    def span_evaluation_names(
-        self,
-        info: Info[Context, None],
-    ) -> List[str]:
-        if not (traces := info.context.traces) or not (
-            project := traces.get_project(DEFAULT_PROJECT_NAME)
-        ):
-            return []
-        return project.get_span_evaluation_names()
-
-    @strawberry.field(
-        description="Names of available document evaluations.",
-    )  # type: ignore
-    def document_evaluation_names(
-        self,
-        info: Info[Context, None],
-        span_id: Optional[ID] = UNSET,
-    ) -> List[str]:
-        if not (traces := info.context.traces) or not (
-            project := traces.get_project(DEFAULT_PROJECT_NAME)
-        ):
-            return []
-        return project.get_document_evaluation_names(
-            None if span_id is UNSET else SpanID(span_id),
-        )
-
-    @strawberry.field
-    def span_evaluation_summary(
-        self,
-        info: Info[Context, None],
-        evaluation_name: str,
-        time_range: Optional[TimeRange] = UNSET,
-        filter_condition: Optional[str] = UNSET,
-    ) -> Optional[EvaluationSummary]:
-        if not (traces := info.context.traces) or not (
-            project := traces.get_project(DEFAULT_PROJECT_NAME)
-        ):
-            return None
-        predicate = (
-            SpanFilter(
-                condition=filter_condition,
-                evals=project,
-            )
-            if filter_condition
-            else None
-        )
-        span_ids = project.get_span_evaluation_span_ids(evaluation_name)
-        if not span_ids:
-            return None
-        spans = traces.get_spans(
-            start_time=time_range.start if time_range else None,
-            stop_time=time_range.end if time_range else None,
-            span_ids=span_ids,
-        )
-        if predicate:
-            spans = filter(predicate, spans)
-        evaluations = tuple(
-            evaluation
-            for span in spans
-            if (
-                evaluation := project.get_span_evaluation(
-                    span.context.span_id,
-                    evaluation_name,
-                )
-            )
-            is not None
-        )
-        if not evaluations:
-            return None
-        labels = project.get_span_evaluation_labels(evaluation_name)
-        return EvaluationSummary(evaluations, labels)
-
-    @strawberry.field
-    def trace_dataset_info(
-        self,
-        info: Info[Context, None],
-    ) -> Optional[TraceDatasetInfo]:
-        if (traces := info.context.traces) is None:
-            return None
-        if not (span_count := traces.span_count):
-            return None
-        start_time, stop_time = cast(
-            Tuple[datetime, datetime],
-            traces.right_open_time_range,
-        )
-        latency_ms_p50, latency_ms_p99 = traces.root_span_latency_ms_quantiles(0.50, 0.99)
-        return TraceDatasetInfo(
-            start_time=start_time,
-            end_time=stop_time,
-            record_count=span_count,
-            token_count_total=traces.token_count_total,
-            latency_ms_p50=latency_ms_p50,
-            latency_ms_p99=latency_ms_p99,
-        )
-
-    @strawberry.field
-    def validate_span_filter_condition(
-        self, info: Info[Context, None], condition: str
-    ) -> ValidationResult:
-        traces = info.context.traces
-        project = traces.get_project(DEFAULT_PROJECT_NAME) if traces else None
-        valid_eval_names = project.get_span_evaluation_names() if project else ()
-        try:
-            SpanFilter(
-                condition=condition,
-                evals=project,
-                valid_eval_names=valid_eval_names,
-            )
-            return ValidationResult(is_valid=True, error_message=None)
-        except SyntaxError as e:
-            return ValidationResult(
-                is_valid=False,
-                error_message=e.msg,
-            )
-
 
 @strawberry.type
 class Mutation(ExportEventsMutation): ...

{arize_phoenix-3.9.0 → arize_phoenix-3.10.0}/src/phoenix/server/api/types/Dataset.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import Iterable, List, Optional, Set, Union
 
 import strawberry
@@ -8,14 +9,16 @@ import phoenix.core.model_schema as ms
 from phoenix.core.model_schema import FEATURE, TAG, ScalarDimension
 
 from ..input_types.DimensionInput import DimensionInput
-from .DatasetInfo import DatasetInfo
 from .DatasetRole import AncillaryDatasetRole, DatasetRole
 from .Dimension import Dimension, to_gql_dimension
 from .Event import Event, create_event, create_event_id, parse_event_ids_by_dataset_role
 
 
 @strawberry.type
-class Dataset(DatasetInfo):
+class Dataset:
+    start_time: datetime = strawberry.field(description="The start bookend of the data")
+    end_time: datetime = strawberry.field(description="The end bookend of the data")
+    record_count: int = strawberry.field(description="The record count of the data")
     dataset: strawberry.Private[ms.Dataset]
     dataset_role: strawberry.Private[Union[DatasetRole, AncillaryDatasetRole]]
     model: strawberry.Private[ms.Model]

{arize_phoenix-3.9.0 → arize_phoenix-3.10.0}/src/phoenix/server/api/types/Project.py

@@ -19,6 +19,7 @@ from phoenix.server.api.types.pagination import (
     connection_from_list,
 )
 from phoenix.server.api.types.Span import Span, to_gql_span
+from phoenix.server.api.types.ValidationResult import ValidationResult
 from phoenix.trace.dsl import SpanFilter
 from phoenix.trace.schemas import SpanID, TraceID
 
@@ -219,3 +220,19 @@ class Project(Node):
         self,
     ) -> Optional[datetime]:
         return self.project.last_updated_at
+
+    @strawberry.field
+    def validate_span_filter_condition(self, condition: str) -> ValidationResult:
+        valid_eval_names = self.project.get_span_evaluation_names()
+        try:
+            SpanFilter(
+                condition=condition,
+                evals=self.project,
+                valid_eval_names=valid_eval_names,
+            )
+            return ValidationResult(is_valid=True, error_message=None)
+        except SyntaxError as e:
+            return ValidationResult(
+                is_valid=False,
+                error_message=e.msg,
+            )