arize-phoenix 3.3.0__tar.gz → 3.4.0__tar.gz

{arize_phoenix-3.3.0 → arize_phoenix-3.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 3.3.0
+Version: 3.4.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
@@ -61,6 +61,8 @@ Requires-Dist: pytest-lazy-fixture; extra == 'dev'
 Requires-Dist: pytest==7.4.4; extra == 'dev'
 Requires-Dist: ruff==0.1.5; extra == 'dev'
 Requires-Dist: strawberry-graphql[debug-server]==0.208.2; extra == 'dev'
+Provides-Extra: evals
+Requires-Dist: arize-phoenix-evals>=0.0.3; extra == 'evals'
 Provides-Extra: experimental
 Requires-Dist: tenacity; extra == 'experimental'
 Provides-Extra: llama-index
@@ -303,12 +305,12 @@ df = download_benchmark_dataset(
 df = df.sample(100)
 df = df.rename(
     columns={
-        "query_text": "query",
+        "query_text": "input",
         "document_text": "reference",
     },
 )
 model = OpenAIModel(
-    model_name="gpt-4",
+    model="gpt-4",
     temperature=0.0,
 )
 rails =list(RAG_RELEVANCY_PROMPT_RAILS_MAP.values())

{arize_phoenix-3.3.0 → arize_phoenix-3.4.0}/README.md

@@ -232,12 +232,12 @@ df = download_benchmark_dataset(
 df = df.sample(100)
 df = df.rename(
     columns={
-        "query_text": "query",
+        "query_text": "input",
         "document_text": "reference",
     },
 )
 model = OpenAIModel(
-    model_name="gpt-4",
+    model="gpt-4",
     temperature=0.0,
 )
 rails =list(RAG_RELEVANCY_PROMPT_RAILS_MAP.values())

{arize_phoenix-3.3.0/src → arize_phoenix-3.4.0}/phoenix/__init__.py

@@ -1,3 +1,9 @@
+import sys
+from importlib.abc import Loader, MetaPathFinder
+from importlib.machinery import ModuleSpec
+from types import ModuleType
+from typing import Any, Optional
+
 from .datasets.dataset import Dataset
 from .datasets.fixtures import ExampleDatasets, load_example
 from .datasets.schema import EmbeddingColumnNames, RetrievalEmbeddingColumnNames, Schema
@@ -41,4 +47,26 @@ __all__ = [
     "NotebookEnvironment",
     "log_evaluations",
     "Client",
+    "evals",
 ]
+
+
+class PhoenixEvalsFinder(MetaPathFinder):
+    def find_spec(self, fullname: Any, path: Any, target: Any = None) -> Optional[ModuleSpec]:
+        if fullname == "phoenix.evals":
+            return ModuleSpec(fullname, PhoenixEvalsLoader())
+        return None
+
+
+class PhoenixEvalsLoader(Loader):
+    def create_module(self, spec: ModuleSpec) -> None:
+        return None
+
+    def exec_module(self, module: ModuleType) -> None:
+        raise ImportError(
+            "The optional `phoenix.evals` package is not installed. "
+            "Please install `phoenix` with the `evals` extra: `pip install 'arize-phoenix[evals]'`."
+        )
+
+
+sys.meta_path.append(PhoenixEvalsFinder())

{arize_phoenix-3.3.0/src → arize_phoenix-3.4.0}/phoenix/core/traces.py

@@ -10,7 +10,6 @@ from typing import (
     Dict,
     Iterable,
     Iterator,
-    List,
     Optional,
     Set,
     SupportsFloat,
@@ -120,7 +119,6 @@ class Traces:
         self._parent_span_ids: Dict[SpanID, ParentSpanID] = {}
         self._traces: DefaultDict[TraceID, Set[ReadableSpan]] = defaultdict(set)
         self._child_spans: DefaultDict[SpanID, Set[ReadableSpan]] = defaultdict(set)
-        self._orphan_spans: DefaultDict[ParentSpanID, List[otlp.Span]] = defaultdict(list)
         self._num_documents: DefaultDict[SpanID, int] = defaultdict(int)
         self._start_time_sorted_spans: SortedKeyList[ReadableSpan] = SortedKeyList(
             key=lambda span: span.start_time,
@@ -239,7 +237,7 @@ class Traces:
 
     @property
     def span_count(self) -> int:
-        """Total number of spans (excluding orphan spans if any)"""
+        """Total number of spans"""
         return len(self._spans)
 
     @property