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

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.
         """
@@ -37,7 +38,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 +58,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:

phoenix/version.py

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