arize-phoenix 5.5.2__py3-none-any.whl → 5.7.0__py3-none-any.whl

phoenix/trace/fixtures.py

@@ -2,6 +2,7 @@ import logging
 import shutil
 from binascii import hexlify
 from collections import defaultdict
+from collections.abc import Iterable, Iterator, Sequence
 from dataclasses import dataclass, field, replace
 from datetime import datetime, timezone
 from io import StringIO
@@ -9,15 +10,8 @@ from random import getrandbits
 from tempfile import NamedTemporaryFile
 from time import sleep, time
 from typing import (
-    DefaultDict,
-    Dict,
-    Iterable,
-    Iterator,
-    List,
     NamedTuple,
     Optional,
-    Sequence,
-    Tuple,
     cast,
 )
 from urllib.parse import urljoin
@@ -280,7 +274,7 @@ random_fixture = TracesFixture(
     file_name="random.jsonl",
 )
 
-TRACES_FIXTURES: List[TracesFixture] = [
+TRACES_FIXTURES: list[TracesFixture] = [
     demo_llama_index_rag_fixture,
     demo_llama_index_rag_llm_fixture,
     demo_langgraph_agent_fixture,
@@ -298,10 +292,10 @@ TRACES_FIXTURES: List[TracesFixture] = [
     anthropic_tools_fixture,
 ]
 
-NAME_TO_TRACES_FIXTURE: Dict[str, TracesFixture] = {
+NAME_TO_TRACES_FIXTURE: dict[str, TracesFixture] = {
     fixture.name: fixture for fixture in TRACES_FIXTURES
 }
-PROJ_NAME_TO_TRACES_FIXTURE: DefaultDict[str, List[TracesFixture]] = defaultdict(list)
+PROJ_NAME_TO_TRACES_FIXTURE: defaultdict[str, list[TracesFixture]] = defaultdict(list)
 for fixture in TRACES_FIXTURES:
     if fixture.project_name:
         PROJ_NAME_TO_TRACES_FIXTURE[fixture.project_name].append(fixture)
@@ -322,7 +316,7 @@ def get_trace_fixture_by_name(fixture_name: str) -> TracesFixture:
     return NAME_TO_TRACES_FIXTURE[fixture_name]
 
 
-def get_trace_fixtures_by_project_name(proj_name: str) -> List[TracesFixture]:
+def get_trace_fixtures_by_project_name(proj_name: str) -> list[TracesFixture]:
     """
     Returns a dictionary of project name (key) and set of TracesFixtures (value)
     whose project name matches the input name.
@@ -437,7 +431,7 @@ def _read_eval_fixture(eval_fixture: EvaluationFixture) -> Iterator[pb.Evaluatio
             explanation=StringValue(value=cast(str, explanation)) if explanation else None,
         )
         if isinstance(eval_fixture, DocumentEvaluationFixture):
-            span_id, document_position = cast(Tuple[str, int], index)
+            span_id, document_position = cast(tuple[str, int], index)
             # Legacy fixture files contain UUID strings for span_ids. The hyphens in these
             # strings need to be removed because we are also removing the hyphens from the
             # span_ids of their corresponding traces. In general, hyphen is not an allowed
@@ -476,10 +470,10 @@ def _url(
 def reset_fixture_span_ids_and_timestamps(
     spans: Iterable[Span],
     evals: Iterable[pb.Evaluation] = (),
-) -> Tuple[List[Span], List[pb.Evaluation]]:
+) -> tuple[list[Span], list[pb.Evaluation]]:
     old_spans, old_evals = list(spans), list(evals)
-    new_trace_ids: Dict[str, str] = {}
-    new_span_ids: Dict[str, str] = {}
+    new_trace_ids: dict[str, str] = {}
+    new_span_ids: dict[str, str] = {}
     for old_span in old_spans:
         new_trace_ids[old_span.context.trace_id] = _new_trace_id()
         new_span_ids[old_span.context.span_id] = _new_span_id()
@@ -495,8 +489,8 @@ def reset_fixture_span_ids_and_timestamps(
             new_span_ids[span_id] = _new_span_id()
     max_end_time = max(old_span.end_time for old_span in old_spans)
     time_diff = datetime.now(timezone.utc) - max_end_time
-    new_spans: List[Span] = []
-    new_evals: List[pb.Evaluation] = []
+    new_spans: list[Span] = []
+    new_evals: list[pb.Evaluation] = []
     for old_span in old_spans:
         new_trace_id = new_trace_ids[old_span.context.trace_id]
         new_span_id = new_span_ids[old_span.context.span_id]

phoenix/trace/otel.py

@@ -1,19 +1,9 @@
 import json
 from binascii import hexlify, unhexlify
+from collections.abc import Iterable, Iterator, Mapping, Sequence
 from datetime import datetime, timezone
 from types import MappingProxyType
-from typing import (
-    Any,
-    Dict,
-    Iterable,
-    Iterator,
-    Mapping,
-    Optional,
-    Sequence,
-    SupportsFloat,
-    Tuple,
-    cast,
-)
+from typing import Any, Optional, SupportsFloat, cast
 
 import numpy as np
 import opentelemetry.proto.trace.v1.trace_pb2 as otlp
@@ -133,7 +123,7 @@ def _decode_unix_nano(time_unix_nano: int) -> datetime:
 
 def _decode_key_values(
     key_values: Iterable[KeyValue],
-) -> Iterator[Tuple[str, Any]]:
+) -> Iterator[tuple[str, Any]]:
     return ((kv.key, _decode_value(kv.value)) for kv in key_values)
 
 
@@ -169,7 +159,7 @@ _STATUS_DECODING = MappingProxyType(
 )
 
 
-def _decode_status(otlp_status: otlp.Status) -> Tuple[SpanStatusCode, StatusMessage]:
+def _decode_status(otlp_status: otlp.Status) -> tuple[SpanStatusCode, StatusMessage]:
     status_code = _STATUS_DECODING.get(otlp_status.code, SpanStatusCode.UNSET)
     return status_code, otlp_status.message
 
@@ -186,7 +176,7 @@ def encode_span_to_otlp(span: Span) -> otlp.Span:
     start_time_unix_nano: int = int(span.start_time.timestamp() * _BILLION)
     end_time_unix_nano: int = int(span.end_time.timestamp() * _BILLION) if span.end_time else 0
 
-    attributes: Dict[str, Any] = dict(span.attributes)
+    attributes: dict[str, Any] = dict(span.attributes)
 
     for key, value in span.attributes.items():
         if value is None:

phoenix/trace/projects.py

@@ -1,5 +1,6 @@
 import types
-from typing import Any, Callable, Optional, Type
+from collections.abc import Callable
+from typing import Any, Optional
 
 from openinference.semconv.resource import ResourceAttributes
 from opentelemetry.sdk import trace
@@ -58,7 +59,7 @@ class using_project:
 
     def __exit__(
         self,
-        exc_type: Optional[Type[BaseException]],
+        exc_type: Optional[type[BaseException]],
         exc_value: Optional[BaseException],
         traceback: Optional[types.TracebackType],
     ) -> None:

phoenix/trace/schemas.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 from enum import Enum
-from typing import Any, List, Mapping, NamedTuple, Optional
+from typing import Any, Mapping, NamedTuple, Optional
 from uuid import UUID
 
 EXCEPTION_TYPE = "exception.type"
@@ -169,7 +169,7 @@ class Span:
     OpenTelemetry Inspiration:
     https://opentelemetry.io/docs/concepts/signals/traces/#span-events
     """
-    events: List[SpanEvent]
+    events: list[SpanEvent]
 
     """
     An extension of the OpenTelemetry Span interface to include the

phoenix/trace/span_evaluations.py

@@ -1,10 +1,11 @@
 import json
 from abc import ABC
+from collections.abc import Callable, Mapping, Sequence
 from dataclasses import dataclass, field
 from itertools import product
 from pathlib import Path
 from types import MappingProxyType
-from typing import Any, Callable, List, Mapping, Optional, Sequence, Set, Tuple, Type, Union
+from typing import Any, Optional, Union
 from uuid import UUID, uuid4
 
 import pandas as pd
@@ -20,11 +21,11 @@ EVAL_PARQUET_FILE_NAME = "evaluations-{id}.parquet"
 
 
 class NeedsNamedIndex(ABC):
-    index_names: Mapping[Tuple[str, ...], Callable[[Any], bool]]
-    all_valid_index_name_sorted_combos: Set[Tuple[str, ...]]
+    index_names: Mapping[tuple[str, ...], Callable[[Any], bool]]
+    all_valid_index_name_sorted_combos: set[tuple[str, ...]]
 
     @classmethod
-    def preferred_names(cls) -> List[str]:
+    def preferred_names(cls) -> list[str]:
         return [choices[0] for choices in cls.index_names.keys()]
 
     @classmethod
@@ -43,7 +44,7 @@ class NeedsNamedIndex(ABC):
         )
 
     @classmethod
-    def find_valid_index_names(cls, dtypes: "pd.Series[Any]") -> Optional[List[str]]:
+    def find_valid_index_names(cls, dtypes: "pd.Series[Any]") -> Optional[list[str]]:
         valid_names = []
         for names, check_type in cls.index_names.items():
             for name in names:
@@ -92,7 +93,7 @@ class Evaluations(NeedsNamedIndex, NeedsResultColumns, ABC):
             f"dataframe=<rows: {len(self.dataframe)!r}>)"
         )
 
-    def __dir__(self) -> List[str]:
+    def __dir__(self) -> list[str]:
         return ["get_dataframe"]
 
     def get_dataframe(self, prefix_columns_with_name: bool = True) -> pd.DataFrame:
@@ -153,7 +154,7 @@ class Evaluations(NeedsNamedIndex, NeedsResultColumns, ABC):
 
     def __init_subclass__(
         cls,
-        index_names: Mapping[Tuple[str, ...], Callable[[Any], bool]],
+        index_names: Mapping[tuple[str, ...], Callable[[Any], bool]],
         **kwargs: Any,
     ) -> None:
         super().__init_subclass__(**kwargs)
@@ -327,7 +328,7 @@ class TraceEvaluations(
 ): ...
 
 
-def _parse_schema_metadata(schema: Schema) -> Tuple[UUID, str, Type[Evaluations]]:
+def _parse_schema_metadata(schema: Schema) -> tuple[UUID, str, type[Evaluations]]:
     """
     Validates and parses the pyarrow schema metadata.
     """

phoenix/trace/span_json_decoder.py

@@ -1,6 +1,6 @@
 import json
 from datetime import datetime
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from openinference.semconv.trace import SpanAttributes
 
@@ -22,7 +22,7 @@ INPUT_MIME_TYPE = SpanAttributes.INPUT_MIME_TYPE
 OUTPUT_MIME_TYPE = SpanAttributes.OUTPUT_MIME_TYPE
 
 
-def json_to_attributes(obj: Optional[Dict[str, Any]]) -> Dict[str, Any]:
+def json_to_attributes(obj: Optional[dict[str, Any]]) -> dict[str, Any]:
     if obj is None:
         return {}
     if not isinstance(obj, dict):
@@ -34,7 +34,7 @@ def json_to_attributes(obj: Optional[Dict[str, Any]]) -> Dict[str, Any]:
     return obj
 
 
-def json_to_span(data: Dict[str, Any]) -> Any:
+def json_to_span(data: dict[str, Any]) -> Any:
     """
     A hook for json.loads to convert a dict to a Span object.
     """

phoenix/trace/span_json_encoder.py

@@ -2,7 +2,7 @@ import json
 from dataclasses import asdict
 from datetime import datetime
 from enum import Enum
-from typing import Any, List
+from typing import Any
 from uuid import UUID
 
 import numpy as np
@@ -60,5 +60,5 @@ def span_to_json(span: Span) -> str:
     return json.dumps(span, cls=SpanJSONEncoder)
 
 
-def spans_to_jsonl(spans: List[Span]) -> str:
+def spans_to_jsonl(spans: list[Span]) -> str:
     return "\n".join(span_to_json(span) for span in spans)

phoenix/trace/trace_dataset.py

@@ -1,7 +1,8 @@
 import json
+from collections.abc import Iterable, Iterator
 from datetime import datetime
 from pathlib import Path
-from typing import Any, Iterable, Iterator, List, Optional, Tuple, Union, cast
+from typing import Any, Optional, Union, cast
 from uuid import UUID, uuid4
 from warnings import warn
 
@@ -116,7 +117,7 @@ class TraceDataset:
     A human readable name for the dataset.
     """
     dataframe: pd.DataFrame
-    evaluations: List[Evaluations] = []
+    evaluations: list[Evaluations] = []
     _id: UUID
     _data_file_name: str = "data.parquet"
 
@@ -152,11 +153,11 @@ class TraceDataset:
         self.evaluations = list(evaluations)
 
     @classmethod
-    def from_spans(cls, spans: List[Span]) -> "TraceDataset":
+    def from_spans(cls, spans: list[Span]) -> "TraceDataset":
         """Creates a TraceDataset from a list of spans.
 
         Args:
-            spans (List[Span]): A list of spans.
+            spans (list[Span]): A list of spans.
 
         Returns:
             TraceDataset: A TraceDataset containing the spans.
@@ -349,7 +350,7 @@ class TraceDataset:
         return pd.concat([df, evals_df], axis=1)
 
 
-def _parse_schema_metadata(schema: Schema) -> Tuple[UUID, str, List[UUID]]:
+def _parse_schema_metadata(schema: Schema) -> tuple[UUID, str, list[UUID]]:
     """
     Returns parsed metadata from a parquet schema or raises an exception if the
     metadata is invalid.

phoenix/trace/utils.py

@@ -2,7 +2,7 @@ import json
 import os
 import re
 from traceback import format_exception
-from typing import List, Optional, Tuple, cast
+from typing import Optional, cast
 from urllib import request
 
 import pandas as pd
@@ -14,16 +14,16 @@ def parse_file_extension(file_path: str) -> str:
 
 def download_json_traces_fixture(
     url: str,
-) -> List[str]:
+) -> list[str]:
     """
     Stores the traces fixture as list of jsons from the jsonl files in the phoenix bucket.
     """
 
     with request.urlopen(url) as f:
-        return cast(List[str], f.readlines())
+        return cast(list[str], f.readlines())
 
 
-def json_lines_to_df(lines: List[str]) -> pd.DataFrame:
+def json_lines_to_df(lines: list[str]) -> pd.DataFrame:
     """
     Convert a list of JSON line strings to a Pandas DataFrame.
     """
@@ -56,9 +56,9 @@ def get_stacktrace(exception: BaseException) -> str:
 _VERSION_TRIPLET_REGEX = re.compile(r"(\d+)\.(\d+)\.(\d+)")
 
 
-def extract_version_triplet(version: str) -> Optional[Tuple[int, int, int]]:
+def extract_version_triplet(version: str) -> Optional[tuple[int, int, int]]:
     return (
-        cast(Tuple[int, int, int], tuple(map(int, match.groups())))
+        cast(tuple[int, int, int], tuple(map(int, match.groups())))
         if (match := _VERSION_TRIPLET_REGEX.search(version))
         else None
     )

phoenix/utilities/deprecation.py

@@ -1,8 +1,9 @@
 import functools
 import warnings
-from typing import Any, Callable, Type, TypeVar
+from collections.abc import Callable
+from typing import Any, TypeVar
 
-GenericClass = TypeVar("GenericClass", bound=Type[Any])
+GenericClass = TypeVar("GenericClass", bound=type[Any])
 CallableType = TypeVar("CallableType", bound=Callable[..., Any])
 
 

phoenix/utilities/error_handling.py

@@ -1,12 +1,13 @@
 import logging
 import traceback
-from typing import Any, Callable, Iterable, Optional, Type, TypeVar, cast
+from collections.abc import Callable, Iterable
+from typing import Any, Optional, TypeVar, cast
 
 F = TypeVar("F", bound=Callable[..., Any])
 
 
 def graceful_fallback(
-    fallback_method: Callable[..., Any], exceptions: Optional[Iterable[Type[BaseException]]] = None
+    fallback_method: Callable[..., Any], exceptions: Optional[Iterable[type[BaseException]]] = None
 ) -> Callable[[F], F]:
     """
     Decorator that reroutes failing functions to a specified fallback method.

phoenix/utilities/json.py

@@ -1,8 +1,9 @@
 import dataclasses
 import datetime
+from collections.abc import Mapping, Sequence
 from enum import Enum
 from pathlib import Path
-from typing import Any, Mapping, Sequence, Union, get_args, get_origin
+from typing import Any, Union, get_args, get_origin
 
 import numpy as np
 from strawberry import UNSET

phoenix/utilities/logging.py

@@ -1,6 +1,6 @@
 # A collection of printing and logging utilities
 
-from typing import Any, List
+from typing import Any
 
 from tqdm.auto import tqdm
 
@@ -10,7 +10,7 @@ def printif(condition: bool, *args: Any, **kwargs: Any) -> None:
         tqdm.write(*args, **kwargs)
 
 
-def log_a_list(list_of_str: List[str], join_word: str) -> str:
+def log_a_list(list_of_str: list[str], join_word: str) -> str:
     if list_of_str is None or len(list_of_str) == 0:
         return ""
     if len(list_of_str) == 1:

phoenix/utilities/project.py

@@ -1,4 +1,4 @@
-from typing import Iterable
+from collections.abc import Iterable
 
 from openinference.semconv.resource import ResourceAttributes
 from opentelemetry.proto.common.v1.common_pb2 import KeyValue

phoenix/utilities/re.py

@@ -1,6 +1,5 @@
 import logging
 from re import compile, split
-from typing import Dict, List
 from urllib.parse import unquote
 
 logger = logging.getLogger(__name__)
@@ -19,7 +18,7 @@ _HEADER_PATTERN = compile(_KEY_VALUE_FORMAT)
 _DELIMITER_PATTERN = compile(r"[ \t]*,[ \t]*")
 
 
-def parse_env_headers(s: str) -> Dict[str, str]:
+def parse_env_headers(s: str) -> dict[str, str]:
     """
     Parse ``s``, which is a ``str`` instance containing HTTP headers encoded
     for use in ENV variables per the W3C Baggage HTTP header format at
@@ -28,8 +27,8 @@ def parse_env_headers(s: str) -> Dict[str, str]:
 
     src: https://github.com/open-telemetry/opentelemetry-python/blob/2d5cd58f33bd8a16f45f30be620a96699bc14297/opentelemetry-api/src/opentelemetry/util/re.py#L52
     """
-    headers: Dict[str, str] = {}
-    headers_list: List[str] = split(_DELIMITER_PATTERN, s)
+    headers: dict[str, str] = {}
+    headers_list: list[str] = split(_DELIMITER_PATTERN, s)
     for header in headers_list:
         if not header:  # empty string
             continue

phoenix/utilities/template_formatters.py

@@ -1,12 +1,13 @@
 import re
 from abc import ABC, abstractmethod
+from collections.abc import Iterable
 from string import Formatter
-from typing import Any, Iterable, Set
+from typing import Any
 
 
 class TemplateFormatter(ABC):
     @abstractmethod
-    def parse(self, template: str) -> Set[str]:
+    def parse(self, template: str) -> set[str]:
         """
         Parse the template and return a set of variable names.
         """
@@ -18,7 +19,9 @@ class TemplateFormatter(ABC):
         """
         template_variable_names = self.parse(template)
         if missing_template_variables := template_variable_names - set(variables.keys()):
-            raise ValueError(f"Missing template variables: {', '.join(missing_template_variables)}")
+            raise TemplateFormatterError(
+                f"Missing template variable(s): {', '.join(missing_template_variables)}"
+            )
         return self._format(template, template_variable_names, **variables)
 
     @abstractmethod
@@ -37,7 +40,7 @@ class FStringTemplateFormatter(TemplateFormatter):
     'world'
     """
 
-    def parse(self, template: str) -> Set[str]:
+    def parse(self, template: str) -> set[str]:
         return set(field_name for _, field_name, _, _ in Formatter().parse(template) if field_name)
 
     def _format(self, template: str, variable_names: Iterable[str], **variables: Any) -> str:
@@ -57,7 +60,7 @@ class MustacheTemplateFormatter(TemplateFormatter):
 
     PATTERN = re.compile(r"(?<!\\){{\s*(\w+)\s*}}")
 
-    def parse(self, template: str) -> Set[str]:
+    def parse(self, template: str) -> set[str]:
         return set(match for match in re.findall(self.PATTERN, template))
 
     def _format(self, template: str, variable_names: Iterable[str], **variables: Any) -> str:
@@ -68,3 +71,11 @@ class MustacheTemplateFormatter(TemplateFormatter):
                 string=template,
             )
         return template
+
+
+class TemplateFormatterError(Exception):
+    """
+    An error raised when template formatting fails.
+    """
+
+    pass

phoenix/version.py

@@ -1 +1 @@
-__version__ = "5.5.2"
+__version__ = "5.7.0"