arize-phoenix 3.4.0__py3-none-any.whl → 3.5.0__py3-none-any.whl

phoenix/server/templates/index.html

@@ -31,7 +31,8 @@
             // injected into the client before React runs
             value: Object.freeze({
                 basename: "{{basename}}",
-                hasCorpus: Boolean("{{has_corpus}}"),
+                hasInferences: Boolean("{{has_inferences}}" == "True"),
+                hasCorpus: Boolean("{{has_corpus}}" == "True"),
                 UMAP: {
                   minDist: parseFloat("{{min_dist}}"),
                   nNeighbors: parseInt("{{n_neighbors}}"),

phoenix/session/data_extractor.py

@@ -22,8 +22,7 @@ class TraceDataExtractor(ABC):
         start_time: Optional[datetime] = None,
         stop_time: Optional[datetime] = None,
         root_spans_only: Optional[bool] = None,
-    ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
-        ...
+    ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]: ...
 
     def get_spans_dataframe(
         self,
@@ -44,8 +43,7 @@ class TraceDataExtractor(ABC):
         )
 
     @abstractmethod
-    def get_evaluations(self) -> List[Evaluations]:
-        ...
+    def get_evaluations(self) -> List[Evaluations]: ...
 
     def get_trace_dataset(self) -> Optional[TraceDataset]:
         if (dataframe := self.get_spans_dataframe()) is None:

phoenix/session/evaluation.py

@@ -5,6 +5,7 @@ A set of **highly experimental** helper functions to
       indexed by `context.span_id` and `document_position`
   - ingest evaluation results into Phoenix via HttpExporter
 """
+
 import logging
 import math
 from typing import (

phoenix/trace/dsl/filter.py

@@ -29,8 +29,7 @@ _VALID_EVAL_ATTRIBUTES: Tuple[str, ...] = tuple(
 
 
 class SupportsGetSpanEvaluation(Protocol):
-    def get_span_evaluation(self, span_id: SpanID, name: str) -> Optional[pb.Evaluation]:
-        ...
+    def get_span_evaluation(self, span_id: SpanID, name: str) -> Optional[pb.Evaluation]: ...
 
 
 @dataclass(frozen=True)

phoenix/trace/dsl/helpers.py

@@ -21,8 +21,9 @@ IS_RETRIEVER = "span_kind == 'RETRIEVER'"
 
 
 class CanQuerySpans(Protocol):
-    def query_spans(self, *query: SpanQuery) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
-        ...
+    def query_spans(
+        self, *query: SpanQuery
+    ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]: ...
 
 
 def get_retrieved_documents(obj: CanQuerySpans) -> pd.DataFrame:

phoenix/trace/dsl/query.py

@@ -38,13 +38,9 @@ _ALIASES = {
     "trace_id": "context.trace_id",
 }
 
-# Because UUIDs is not convertible to Parquet,
-# they need to be converted to string.
-_CONVERT_TO_STRING = (
-    "context.span_id",
-    "context.trace_id",
-    "parent_id",
-)
+# Because span_kind is an enum, it needs to be converted to string,
+# so it's serializable by pyarrow.
+_CONVERT_TO_STRING = ("span_kind",)
 
 
 def _unalias(key: str) -> str:

phoenix/trace/langchain/tracer.py

@@ -2,6 +2,7 @@
 This module is defunct and will be removed in the future. It's currently
 maintaining a dummy class to avoid breaking any import code.
 """
+
 import logging
 import sys
 from typing import Any, Iterator

phoenix/trace/span_evaluations.py

@@ -323,8 +323,7 @@ class DocumentEvaluations(
 class TraceEvaluations(
     Evaluations,
     index_names=MappingProxyType({("context.trace_id", "trace_id"): is_string_dtype}),
-):
-    ...
+): ...
 
 
 def _parse_schema_metadata(schema: Schema) -> Tuple[UUID, str, Type[Evaluations]]:

phoenix/trace/span_json_encoder.py

@@ -2,10 +2,16 @@ import json
 from dataclasses import asdict
 from datetime import datetime
 from enum import Enum
-from typing import Any, List
+from typing import Any, Dict, List, Mapping
 from uuid import UUID
 
-from phoenix.trace.schemas import Span, SpanContext, SpanConversationAttributes, SpanEvent
+from phoenix.trace.schemas import (
+    COMPUTED_PREFIX,
+    Span,
+    SpanContext,
+    SpanConversationAttributes,
+    SpanEvent,
+)
 
 
 class SpanJSONEncoder(json.JSONEncoder):
@@ -34,7 +40,7 @@ class SpanJSONEncoder(json.JSONEncoder):
                 "end_time": obj.end_time,
                 "status_code": obj.status_code,
                 "status_message": obj.status_message,
-                "attributes": obj.attributes,
+                "attributes": _remove_computed_attributes(obj.attributes),
                 "events": [self.default(event) for event in obj.events],
                 "conversation": obj.conversation,
             }
@@ -49,3 +55,7 @@ def span_to_json(span: Span) -> str:
 
 def spans_to_jsonl(spans: List[Span]) -> str:
     return "\n".join(span_to_json(span) for span in spans)
+
+
+def _remove_computed_attributes(attributes: Mapping[str, Any]) -> Dict[str, Any]:
+    return {key: value for key, value in attributes.items() if not key.startswith(COMPUTED_PREFIX)}

phoenix/trace/tracer.py

@@ -2,6 +2,7 @@
 This module is defunct and will be removed in the future. It's currently
 maintaining a dummy class to avoid breaking any import code.
 """
+
 import logging
 import sys
 from typing import Any, Iterator, Protocol
@@ -12,8 +13,7 @@ logger = logging.getLogger(__name__)
 
 
 class SpanExporter(Protocol):
-    def export(self, _: Any) -> None:
-        ...
+    def export(self, _: Any) -> None: ...
 
 
 _DEFUNCT_MSG = (

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.4.0"
+__version__ = "3.5.0"