arize-phoenix 3.19.4__py3-none-any.whl → 3.20.0__py3-none-any.whl

phoenix/server/main.py

@@ -19,8 +19,8 @@ from phoenix.config import (
 )
 from phoenix.core.model_schema_adapter import create_model_from_datasets
 from phoenix.core.traces import Traces
-from phoenix.datasets.dataset import EMPTY_DATASET, Dataset
-from phoenix.datasets.fixtures import FIXTURES, get_datasets
+from phoenix.inferences.fixtures import FIXTURES, get_datasets
+from phoenix.inferences.inferences import EMPTY_DATASET, Inferences
 from phoenix.pointcloud.umap_parameters import (
     DEFAULT_MIN_DIST,
     DEFAULT_N_NEIGHBORS,
@@ -114,9 +114,9 @@ if __name__ == "__main__":
     trace_dataset_name: Optional[str] = None
     simulate_streaming: Optional[bool] = None
 
-    primary_dataset: Dataset = EMPTY_DATASET
-    reference_dataset: Optional[Dataset] = None
-    corpus_dataset: Optional[Dataset] = None
+    primary_dataset: Inferences = EMPTY_DATASET
+    reference_dataset: Optional[Inferences] = None
+    corpus_dataset: Optional[Inferences] = None
 
     # automatically remove the pid file when the process is being gracefully terminated
     atexit.register(_remove_pid_file)
@@ -158,14 +158,14 @@ if __name__ == "__main__":
         primary_dataset_name = args.primary
         reference_dataset_name = args.reference
         corpus_dataset_name = args.corpus
-        primary_dataset = Dataset.from_name(primary_dataset_name)
+        primary_dataset = Inferences.from_name(primary_dataset_name)
         reference_dataset = (
-            Dataset.from_name(reference_dataset_name)
+            Inferences.from_name(reference_dataset_name)
             if reference_dataset_name is not None
             else None
         )
         corpus_dataset = (
-            None if corpus_dataset_name is None else Dataset.from_name(corpus_dataset_name)
+            None if corpus_dataset_name is None else Inferences.from_name(corpus_dataset_name)
         )
     elif args.command == "fixture":
         fixture_name = args.fixture

phoenix/session/evaluation.py

@@ -1,5 +1,5 @@
 """
-A set of **highly experimental** helper functions to
+A set of helper functions to
   - extract spans from Phoenix for evaluation
     - explode retrieved documents from (horizontal) lists to a (vertical) series
       indexed by `context.span_id` and `document_position`
@@ -73,7 +73,6 @@ def _extract_subject_id_from_index(
     value: Union[str, Sequence[Any]],
 ) -> pb.Evaluation.SubjectId:
     """
-    (**Highly Experimental**)
     Returns `SubjectId` given the format of `index_names`. Allowed formats are:
         - DocumentRetrievalId
             - index_names=["context.span_id", "document_position"]

phoenix/session/session.py

@@ -36,7 +36,7 @@ from phoenix.config import (
 )
 from phoenix.core.model_schema_adapter import create_model_from_datasets
 from phoenix.core.traces import Traces
-from phoenix.datasets.dataset import EMPTY_DATASET, Dataset
+from phoenix.inferences.inferences import EMPTY_DATASET, Inferences
 from phoenix.pointcloud.umap_parameters import get_umap_parameters
 from phoenix.server.app import create_app
 from phoenix.server.thread_server import ThreadServer
@@ -104,9 +104,9 @@ class Session(TraceDataExtractor, ABC):
 
     def __init__(
         self,
-        primary_dataset: Dataset,
-        reference_dataset: Optional[Dataset] = None,
-        corpus_dataset: Optional[Dataset] = None,
+        primary_dataset: Inferences,
+        reference_dataset: Optional[Inferences] = None,
+        corpus_dataset: Optional[Inferences] = None,
         trace_dataset: Optional[TraceDataset] = None,
         default_umap_parameters: Optional[Mapping[str, Any]] = None,
         host: Optional[str] = None,
@@ -194,9 +194,9 @@ _session: Optional[Session] = None
 class ProcessSession(Session):
     def __init__(
         self,
-        primary_dataset: Dataset,
-        reference_dataset: Optional[Dataset] = None,
-        corpus_dataset: Optional[Dataset] = None,
+        primary_dataset: Inferences,
+        reference_dataset: Optional[Inferences] = None,
+        corpus_dataset: Optional[Inferences] = None,
         trace_dataset: Optional[TraceDataset] = None,
         default_umap_parameters: Optional[Mapping[str, Any]] = None,
         host: Optional[str] = None,
@@ -215,9 +215,9 @@ class ProcessSession(Session):
             notebook_env=notebook_env,
         )
         primary_dataset.to_disc()
-        if isinstance(reference_dataset, Dataset):
+        if isinstance(reference_dataset, Inferences):
             reference_dataset.to_disc()
-        if isinstance(corpus_dataset, Dataset):
+        if isinstance(corpus_dataset, Inferences):
             corpus_dataset.to_disc()
         if isinstance(trace_dataset, TraceDataset):
             trace_dataset.to_disc()
@@ -284,9 +284,9 @@ class ProcessSession(Session):
 class ThreadSession(Session):
     def __init__(
         self,
-        primary_dataset: Dataset,
-        reference_dataset: Optional[Dataset] = None,
-        corpus_dataset: Optional[Dataset] = None,
+        primary_dataset: Inferences,
+        reference_dataset: Optional[Inferences] = None,
+        corpus_dataset: Optional[Inferences] = None,
         trace_dataset: Optional[TraceDataset] = None,
         default_umap_parameters: Optional[Mapping[str, Any]] = None,
         host: Optional[str] = None,
@@ -424,9 +424,9 @@ class ThreadSession(Session):
 
 
 def launch_app(
-    primary: Optional[Dataset] = None,
-    reference: Optional[Dataset] = None,
-    corpus: Optional[Dataset] = None,
+    primary: Optional[Inferences] = None,
+    reference: Optional[Inferences] = None,
+    corpus: Optional[Inferences] = None,
     trace: Optional[TraceDataset] = None,
     default_umap_parameters: Optional[Mapping[str, Any]] = None,
     host: Optional[str] = None,
@@ -447,7 +447,7 @@ def launch_app(
     corpus : Dataset, optional
         The dataset containing corpus for LLM context retrieval.
     trace: TraceDataset, optional
-        **Experimental** The trace dataset containing the trace data.
+        The trace dataset containing the trace data.
     host: str, optional
         The host on which the server runs. It can also be set using environment
         variable `PHOENIX_HOST`, otherwise it defaults to `127.0.0.1`.

phoenix/utilities/deprecation.py

@@ -0,0 +1,30 @@
+import functools
+import warnings
+from typing import Any, Callable, Type, TypeVar
+
+GenericClass = TypeVar("GenericClass", bound=Type[Any])
+CallableType = TypeVar("CallableType", bound=Callable[..., Any])
+
+
+def deprecated_class(message: str) -> Callable[[GenericClass], GenericClass]:
+    def decorator(original_class: GenericClass) -> GenericClass:
+        @functools.wraps(original_class)
+        def new_class(*args: Any, **kwargs: Any) -> Any:
+            warnings.warn(message, DeprecationWarning, stacklevel=2)
+            return original_class(*args, **kwargs)
+
+        return new_class  # type: ignore
+
+    return decorator
+
+
+def deprecated(message: str) -> Callable[[CallableType], CallableType]:
+    def decorator(original_func: CallableType) -> CallableType:
+        @functools.wraps(original_func)
+        def new_func(*args: Any, **kwargs: Any) -> Any:
+            warnings.warn(message, DeprecationWarning, stacklevel=2)
+            return original_func(*args, **kwargs)
+
+        return new_func  # type: ignore
+
+    return decorator

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.19.4"
+__version__ = "3.20.0"