arize-phoenix 5.1.6__tar.gz → 5.2.0__tar.gz

{arize_phoenix-5.1.6 → arize_phoenix-5.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 5.1.6
+Version: 5.2.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -55,6 +55,8 @@ Requires-Dist: umap-learn
 Requires-Dist: uvicorn
 Requires-Dist: wrapt
 Provides-Extra: container
+Requires-Dist: fast-hdbscan>=0.2.0; extra == 'container'
+Requires-Dist: numba>=0.60.0; extra == 'container'
 Requires-Dist: opentelemetry-exporter-otlp; extra == 'container'
 Requires-Dist: opentelemetry-instrumentation-fastapi; extra == 'container'
 Requires-Dist: opentelemetry-instrumentation-grpc; extra == 'container'
@@ -65,6 +67,7 @@ Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
 Requires-Dist: strawberry-graphql[opentelemetry]==0.243.1; extra == 'container'
+Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
@@ -101,6 +104,10 @@ Requires-Dist: tox==4.18.1; extra == 'dev'
 Requires-Dist: types-tabulate; extra == 'dev'
 Requires-Dist: uv==0.4.8; extra == 'dev'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'dev'
+Provides-Extra: embeddings
+Requires-Dist: fast-hdbscan>=0.2.0; extra == 'embeddings'
+Requires-Dist: numba>=0.60.0; extra == 'embeddings'
+Requires-Dist: umap-learn; extra == 'embeddings'
 Provides-Extra: evals
 Provides-Extra: experimental
 Provides-Extra: llama-index

{arize_phoenix-5.1.6 → arize_phoenix-5.2.0}/pyproject.toml

@@ -62,6 +62,10 @@ dependencies = [
 ]
 dynamic = ["version"]
 
+[project.scripts]
+arize-phoenix = "phoenix.server.main:main"
+phoenix = "phoenix.server.main:main"
+
 [project.optional-dependencies]
 dev = [
   "gcsfs",
@@ -99,6 +103,11 @@ dev = [
   "uvloop; platform_system != 'Windows'",
   "grpc-interceptor[testing]",
 ]
+embeddings = [
+  "fast-hdbscan>=0.2.0",
+  "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
+  "umap-learn",
+]
 evals = []
 experimental = []
 llama-index = [
@@ -124,6 +133,9 @@ container = [
   "py-grpc-prometheus",
   "strawberry-graphql[opentelemetry]==0.243.1",  # need to pin version because we're monkey-patching
   "uvloop; platform_system != 'Windows'",
+  "fast-hdbscan>=0.2.0",
+  "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
+  "umap-learn",
 ]
 test = [
 ]

{arize_phoenix-5.1.6 → arize_phoenix-5.2.0}/src/phoenix/server/app.py

@@ -632,6 +632,15 @@ def create_app(
     oauth2_client_configs: Optional[List[OAuth2ClientConfig]] = None,
     bulk_inserter_factory: Optional[Callable[..., BulkInserter]] = None,
 ) -> FastAPI:
+    if model.embedding_dimensions:
+        try:
+            import fast_hdbscan  # noqa: F401
+            import umap  # noqa: F401
+        except ImportError as exc:
+            raise ImportError(
+                "To visualize embeddings, please install `umap-learn` and `fast-hdbscan` "
+                "via `pip install arize-phoenix[embeddings]`"
+            ) from exc
     logger.info(f"Server umap params: {umap_params}")
     bulk_inserter_factory = bulk_inserter_factory or BulkInserter
     startup_callbacks_list: List[_Callback] = list(startup_callbacks)

{arize_phoenix-5.1.6 → arize_phoenix-5.2.0}/src/phoenix/server/main.py

@@ -2,7 +2,7 @@ import atexit
 import codecs
 import os
 import sys
-from argparse import ArgumentParser
+from argparse import SUPPRESS, ArgumentParser
 from importlib.metadata import version
 from pathlib import Path
 from threading import Thread
@@ -130,6 +130,9 @@ DEFAULT_UMAP_PARAMS_STR = f"{DEFAULT_MIN_DIST},{DEFAULT_N_NEIGHBORS},{DEFAULT_N_
 
 
 def main() -> None:
+    initialize_settings()
+    setup_logging()
+
     primary_inferences_name: str
     reference_inferences_name: Optional[str]
     trace_dataset_name: Optional[str] = None
@@ -138,23 +141,29 @@ def main() -> None:
     reference_inferences: Optional[Inferences] = None
     corpus_inferences: Optional[Inferences] = None
 
-    # automatically remove the pid file when the process is being gracefully terminated
     atexit.register(_remove_pid_file)
 
-    parser = ArgumentParser()
-    parser.add_argument("--database-url", required=False)
-    parser.add_argument("--export_path")
-    parser.add_argument("--host", type=str, required=False)
-    parser.add_argument("--port", type=int, required=False)
-    parser.add_argument("--read-only", action="store_true", required=False)  # Default is False
-    parser.add_argument("--no-internet", action="store_true")
-    parser.add_argument("--umap_params", type=str, required=False, default=DEFAULT_UMAP_PARAMS_STR)
-    parser.add_argument("--debug", action="store_true")
-    # Whether the app is running in a development environment
-    parser.add_argument("--dev", action="store_true")
-    parser.add_argument("--no-ui", action="store_true")
-
-    subparsers = parser.add_subparsers(dest="command", required=True)
+    parser = ArgumentParser(usage="phoenix serve", add_help=False)
+    parser.add_argument(
+        "-h",
+        "--help",
+        action="help",
+        help=SUPPRESS,
+    )
+    parser.add_argument("--database-url", required=False, help=SUPPRESS)
+    parser.add_argument("--export_path", help=SUPPRESS)
+    parser.add_argument("--host", type=str, required=False, help=SUPPRESS)
+    parser.add_argument("--port", type=int, required=False, help=SUPPRESS)
+    parser.add_argument("--read-only", action="store_true", required=False, help=SUPPRESS)
+    parser.add_argument("--no-internet", action="store_true", help=SUPPRESS)
+    parser.add_argument(
+        "--umap_params", type=str, required=False, default=DEFAULT_UMAP_PARAMS_STR, help=SUPPRESS
+    )
+    parser.add_argument("--debug", action="store_true", help=SUPPRESS)
+    parser.add_argument("--dev", action="store_true", help=SUPPRESS)
+    parser.add_argument("--no-ui", action="store_true", help=SUPPRESS)
+
+    subparsers = parser.add_subparsers(dest="command", required=True, help=SUPPRESS)
 
     serve_parser = subparsers.add_parser("serve")
     serve_parser.add_argument(
@@ -186,7 +195,7 @@ def main() -> None:
     )
     serve_parser.add_argument(
         "--force-fixture-ingestion",
-        action="store_true",  # default is False
+        action="store_true",
         required=False,
         help=(
             "Whether or not to check the database age before adding the fixtures. "
@@ -196,7 +205,7 @@ def main() -> None:
     )
     serve_parser.add_argument(
         "--scaffold-datasets",
-        action="store_true",  # default is False
+        action="store_true",
         required=False,
         help=(
             "Whether or not to add any datasets defined in "
@@ -213,15 +222,13 @@ def main() -> None:
 
     fixture_parser = subparsers.add_parser("fixture")
     fixture_parser.add_argument("fixture", type=str, choices=[fixture.name for fixture in FIXTURES])
-    fixture_parser.add_argument("--primary-only", action="store_true")  # Default is False
+    fixture_parser.add_argument("--primary-only", action="store_true")
 
     trace_fixture_parser = subparsers.add_parser("trace-fixture")
     trace_fixture_parser.add_argument(
         "fixture", type=str, choices=[fixture.name for fixture in TRACES_FIXTURES]
     )
-    trace_fixture_parser.add_argument(
-        "--simulate-streaming", action="store_true"
-    )  # Default is False
+    trace_fixture_parser.add_argument("--simulate-streaming", action="store_true")
 
     demo_parser = subparsers.add_parser("demo")
     demo_parser.add_argument("fixture", type=str, choices=[fixture.name for fixture in FIXTURES])
@@ -272,7 +279,6 @@ def main() -> None:
         )
         trace_dataset_name = args.trace_fixture
     elif args.command == "serve":
-        # We use sets to avoid duplicates
         if args.with_fixture:
             primary_inferences, reference_inferences, corpus_inferences = get_inferences(
                 str(args.with_fixture),
@@ -293,7 +299,6 @@ def main() -> None:
         scaffold_datasets = args.scaffold_datasets
     host: Optional[str] = args.host or get_env_host()
     if host == "::":
-        # TODO(dustin): why is this necessary? it's not type compliant
         host = None
 
     port = args.port or get_env_port()
@@ -416,6 +421,4 @@ def initialize_settings() -> None:
 
 
 if __name__ == "__main__":
-    initialize_settings()
-    setup_logging()
     main()

arize_phoenix-5.2.0/src/phoenix/version.py

@@ -0,0 +1 @@
+__version__ = "5.2.0"

arize_phoenix-5.1.6/src/phoenix/version.py

@@ -1 +0,0 @@
-__version__ = "5.1.6"