arize-phoenix 5.1.6__py3-none-any.whl → 5.2.0__py3-none-any.whl

{arize_phoenix-5.1.6.dist-info → arize_phoenix-5.2.0.dist-info}/METADATA

@@ -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.dist-info → arize_phoenix-5.2.0.dist-info}/RECORD

@@ -6,7 +6,7 @@ phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=aTxhcOA1pZHB6U-B3TEcp6fqDF5oT0xCUvEUNMZVTUQ,5175
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=Q46Pugq4dvkkf_16U7KgXoGZYZmz7sapzjpUKHTufZ0,22
+phoenix/version.py,sha256=VqMNAkt1E9Y42lSKxWJZRHYFFLxlGbjtTti_CRNjAv8,22
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=km_a--PBHOuA337ClRw9xqhOHhrUT6Rl9pz_zV0JYkQ,4843
@@ -73,13 +73,13 @@ phoenix/pointcloud/pointcloud.py,sha256=4zAIkKs2xOUbchpj4XDAV-iPMXrfAJ15TG6rlIYG
 phoenix/pointcloud/projectors.py,sha256=TQgwc9cJDjJkin1WZyZzgl3HsYrLLiyWD7Czy4jNW3U,1088
 phoenix/pointcloud/umap_parameters.py,sha256=3UQSjrysVOvq2V4KNpTMqNqNiK0BsTZnPBHWZ4fyJtQ,1708
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=FRauYpuCTsz761ekVqyXL93VkTx_xknFdMP4LqNowIk,31689
+phoenix/server/app.py,sha256=IilHYNvbUp-zJ9mF1HgTRS3QYkTR3MWgPu7wGe21Wcs,32063
 phoenix/server/bearer_auth.py,sha256=hChZ2li9qqQAuXIFtD4bp7Z60gEZhVqfCCqLeiTvD2k,5513
 phoenix/server/dml_event.py,sha256=MpjCFqljxvgb9OB5Cez9vJesb3oHb3XxXictynBfcis,2851
 phoenix/server/dml_event_handler.py,sha256=yU23-DDwXcL35p5EPwFW0oZh6mxQxJrJAQPKcYZYJz4,8310
 phoenix/server/grpc_server.py,sha256=nqeHg0FxGppjy8MqQUQ8DJtY4bkvJkR3tCxxmXGk6n0,3733
 phoenix/server/jwt_store.py,sha256=lmcTRBPoxMHzSV7WT8Ew40ag4uxeMWPLDw-vPIoWL8s,16829
-phoenix/server/main.py,sha256=pCKqOaJxAWwlXAl0AVIjHdszx3vkocMSSN9hJbsfsIs,15800
+phoenix/server/main.py,sha256=T_W34ZpBdZDTv5qZMPOfwvz5Nc2QtxUov_sFKNDUneg,15759
 phoenix/server/oauth2.py,sha256=4zt6NEh0o2a3HUThYs-FPqOeDcvYM3YKKMg-s_Awls0,2043
 phoenix/server/prometheus.py,sha256=x9zV3M8VeZ9GS32SLr5OxNdi1VaxvdHCZPx2zI8Hoa4,3549
 phoenix/server/rate_limiters.py,sha256=7UJDH9YTmBIeGFLJX1759-LpS-XNp0zN1TlzxyqTnZc,7163
@@ -306,8 +306,9 @@ phoenix/utilities/logging.py,sha256=B8t2WPULOwVyuGLRLbwKsw5N41N26vtgF-lCAYgWTEk,
 phoenix/utilities/project.py,sha256=8IJuMM4yUMoooPi37sictGj8Etu9rGmq6RFtc9848cQ,436
 phoenix/utilities/re.py,sha256=nr_B0txj_7CXc45953X6vr2KCRSWMuaXJSEkL8s8Sjc,2036
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize_phoenix-5.1.6.dist-info/METADATA,sha256=1rGHnsEZnVX5YeAnzItvZpm-JS6nqrA8-EWTr-9kDt0,11631
-arize_phoenix-5.1.6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-arize_phoenix-5.1.6.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-5.1.6.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-5.1.6.dist-info/RECORD,,
+arize_phoenix-5.2.0.dist-info/METADATA,sha256=nVtj-KDh3ofDUhefKI_lOJ6Afnbmz6kFsia44aTPSfo,11973
+arize_phoenix-5.2.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+arize_phoenix-5.2.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-5.2.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-5.2.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-5.2.0.dist-info/RECORD,,

arize_phoenix-5.2.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+arize-phoenix = phoenix.server.main:main
+phoenix = phoenix.server.main:main

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)

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()

phoenix/version.py

@@ -1 +1 @@
-__version__ = "5.1.6"
+__version__ = "5.2.0"