arize-phoenix 2.4.1rc7__py3-none-any.whl → 2.4.1rc9__py3-none-any.whl

{arize_phoenix-2.4.1rc7.dist-info → arize_phoenix-2.4.1rc9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 2.4.1rc7
+Version: 2.4.1rc9
 Summary: ML Observability in your notebook
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-2.4.1rc7.dist-info → arize_phoenix-2.4.1rc9.dist-info}/RECORD

@@ -4,7 +4,7 @@ phoenix/datetime_utils.py,sha256=D955QLrkgrrSdUM6NyqbCeAu2SMsjhR5rHVQEsVUdng,277
 phoenix/exceptions.py,sha256=igIWGAg3m8jm5YwQDeCY1p8ml_60A7zaGVXJ1yZhY9s,44
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=f6AeyKTuOpy9RCcTCjVH3gx5nYZhbTMFOuv1WSUOB5o,4992
-phoenix/version.py,sha256=6MTHjTynSYAE6HRt-H9tZfIrR0KFY9-QPoh2nbs-WnA,25
+phoenix/version.py,sha256=cl8HhQP1c4TLftr0CGm7Pdu7eOjzwAfHNDqZJusK81k,25
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/evals.py,sha256=OrHeYlh804rpcZIXTA6kan2mzSZMfgpphNNQdPMpNoM,7597
@@ -55,11 +55,11 @@ phoenix/pointcloud/pointcloud.py,sha256=ms-h1FLC0xXb3sk256zpSuZQDE2hdOAJzRNBklP0
 phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcBYPU,1057
 phoenix/pointcloud/umap_parameters.py,sha256=lJsEOrbSuSiqI7g4Yt6xj7kgYxEqoep4ZHWLr6VWBqw,1760
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=moiXf69APCeN2NRU-JtXEBsCkF-iekfTOouh2oacTxo,7480
+phoenix/server/app.py,sha256=8v1lBRZsLnmAat46xwmFLsxQjo_seObK1myukrnkXyE,6880
 phoenix/server/evaluation_handler.py,sha256=HzaoD8Cv9HbEdd0nYSTZoakKsE8Ic5lVjeuBh0vnhoA,1554
 phoenix/server/main.py,sha256=ZWV_UQOE2c5Gx4x_2UQiZIww3JMRWosza5BbzstTUsg,6814
 phoenix/server/span_handler.py,sha256=reYUDaN5bavSFjEiSfvYyAG_mpJs6S3iB-RNCkZrSUU,1295
-phoenix/server/thread_server.py,sha256=z6VwimjlHEsOavlybcQFVWubZvHDGYOH7p61nscC0lU,2164
+phoenix/server/thread_server.py,sha256=dP6cm6Cf08jNhDA1TRlVZpziu1YgtPDmaeIJMm725eI,2154
 phoenix/server/trace_handler.py,sha256=pXanrp9L21Mh7MnyJbj202NJ-Rn4bCjG0oL4DtdKcls,2074
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/context.py,sha256=02vRgyLFpDCmh97QwsjWD5cdNZkoCUtDPPs1YItbdbI,583
@@ -130,10 +130,10 @@ phoenix/server/static/index.js,sha256=dJKCIzqIRBf_IKfXUbnSZnegpgnfMV3wsnFGXWV9Ie
 phoenix/server/static/index.js.map,sha256=Pe5FBAB7cKnRik9CrPFsP4AHe3MR3LUI7OShR2pA5ys,15465655
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/templates/index.html,sha256=K6l-H24QcFpTF5uJkaL8HE9Ebtf6U9GFpYXd6ltxT38,1948
+phoenix/server/templates/index.html,sha256=DlfcGoq1V5C2QkJWqP1j4Nu6_kPfsOzOrtzYF3ogghE,1900
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/session/evaluation.py,sha256=s5OivScAMSj8qfU4IexpmbyKvcGBj_nt-GP_13_o-iY,4843
-phoenix/session/session.py,sha256=McMcxqEabSK3-U3aT9AX6kC-mMx36RofhhYT9t4y2zA,20187
+phoenix/session/session.py,sha256=2em59avmwV3lv37qMcOUg32P85de-vr0Mv8v12iAMM8,20023
 phoenix/trace/__init__.py,sha256=4d_MqzUIFmlY9WWcFeTONJ4xL5mPGoWZaPM2TJ0ZDBQ,266
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
 phoenix/trace/exporter.py,sha256=z3xrGJhIRh7XMy4Q1FkR3KmFZym-GX0XxLTZ6eSnN0Q,4347
@@ -167,8 +167,8 @@ phoenix/trace/v1/evaluation_pb2.pyi,sha256=cCbbx06gwQmaH14s3J1X25TtaARh-k1abbxQd
 phoenix/utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
-arize_phoenix-2.4.1rc7.dist-info/METADATA,sha256=idhujLgZiUC8-ChFcyXlhmYRTz3_EvlHXjpDZoxn-Ms,26482
-arize_phoenix-2.4.1rc7.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
-arize_phoenix-2.4.1rc7.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-2.4.1rc7.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-2.4.1rc7.dist-info/RECORD,,
+arize_phoenix-2.4.1rc9.dist-info/METADATA,sha256=XHhMD329fsB0kidIeZci3KdlhdlZp2ofMqXt5MQU7B8,26482
+arize_phoenix-2.4.1rc9.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
+arize_phoenix-2.4.1rc9.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-2.4.1rc9.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-2.4.1rc9.dist-info/RECORD,,

phoenix/server/app.py

@@ -1,10 +1,9 @@
 import logging
-import re
 from pathlib import Path
 from typing import Any, NamedTuple, Optional, Union
 
 from starlette.applications import Starlette
-from starlette.datastructures import Headers, QueryParams
+from starlette.datastructures import QueryParams
 from starlette.endpoints import HTTPEndpoint
 from starlette.exceptions import HTTPException
 from starlette.middleware import Middleware
@@ -51,21 +50,6 @@ def _is_databricks(url: str) -> bool:
     return "databricks" in url
 
 
-def _get_basename(request: Request) -> str:
-    """
-    Determine the basename for the API and static content.
-    If the server is running in a hosted notebook, the base URL
-    must be configured to point to the hosted notebook's proxy.
-    """
-    # TODO(mikeldking): convert sagemaker to use root_path
-    if _is_sagemaker(str(request.url)):
-        # Sagemaker sets up a proxy at /proxy/6006
-        # NB: this is the only port that is open
-        return "/proxy/6006"
-
-    return request.scope.get("root_path", "")
-
-
 class Static(StaticFiles):
     "Static file serving with a fallback to index.html"
 
@@ -92,9 +76,8 @@ class Static(StaticFiles):
                     "min_dist": self._app_config.min_dist,
                     "n_neighbors": self._app_config.n_neighbors,
                     "n_samples": self._app_config.n_samples,
-                    "basename": _get_basename(request),
+                    "basename": request.scope.get("root_path", ""),
                     "request": request,
-                    "root_path": request.scope.get("root_path", ""),
                 },
             )
         except Exception as e:

phoenix/server/templates/index.html

@@ -20,13 +20,11 @@
   </head>
   <body>
     <div id="root"></div>
-    // Request Info:
-    // URL: {{request.url}}
-    // Base URL: {{request.base_url}}
-    // Headers: {{request.headers}}
-    // Referer: {{referer}}
-    // Root Path: {{root_path}}
     <script>(function() {
+        // Request Info:
+        // URL: {{request.url}}
+        // Base URL: {{request.base_url}}
+        // Root Path: {{root_path}}
         Object.defineProperty(window, "Config", {
             // Place any server-side injected config here
             // E.g. default UMAP parameters etc. that needs to be

phoenix/server/thread_server.py

@@ -2,7 +2,7 @@ import asyncio
 import logging
 from threading import Thread
 from time import sleep, time
-from typing import Generator, Optional
+from typing import Generator
 
 from starlette.applications import Starlette
 from uvicorn import Config, Server

phoenix/session/session.py

@@ -97,7 +97,6 @@ class Session(ABC):
         default_umap_parameters: Optional[Mapping[str, Any]] = None,
         host: Optional[str] = None,
         port: Optional[int] = None,
-        root_path: Optional[str] = None,
         notebook_env: Optional[NotebookEnvironment] = None,
     ):
         self.primary_dataset = primary_dataset
@@ -127,12 +126,12 @@ class Session(ABC):
 
         self.host = host or get_env_host()
         self.port = port or get_env_port()
-        self.root_path = root_path or ""
         self.temp_dir = TemporaryDirectory()
         self.export_path = Path(self.temp_dir.name) / "exports"
         self.export_path.mkdir(parents=True, exist_ok=True)
         self.exported_data = ExportedData()
         self.notebook_env = notebook_env or _get_notebook_environment()
+        self.root_path = _get_root_path(self.notebook_env, self.port)
 
     @abstractmethod
     def end(self) -> None:
@@ -239,7 +238,6 @@ class ProcessSession(Session):
             default_umap_parameters=default_umap_parameters,
             host=host,
             port=port,
-            root_path=root_path,
             notebook_env=notebook_env,
         )
         primary_dataset.to_disc()
@@ -303,7 +301,6 @@ class ThreadSession(Session):
             default_umap_parameters=default_umap_parameters,
             host=host,
             port=port,
-            root_path=root_path,
             notebook_env=notebook_env,
         )
         # Initialize an app service that keeps the server running
@@ -410,7 +407,6 @@ def launch_app(
 
     host = host or get_env_host()
     port = port or get_env_port()
-    root_path = _get_root_path(nb_env, port)
 
     if run_in_thread:
         _session = ThreadSession(
@@ -421,7 +417,6 @@ def launch_app(
             default_umap_parameters,
             host=host,
             port=port,
-            root_path=root_path,
             notebook_env=nb_env,
         )
         # TODO: catch exceptions from thread
@@ -434,7 +429,6 @@ def launch_app(
             default_umap_parameters,
             host=host,
             port=port,
-            root_path=root_path,
             notebook_env=nb_env,
         )
 
@@ -565,11 +559,12 @@ def _get_sagemaker_notebook_base_url() -> str:
     return f"https://{notebook_instance_name}.notebook.{region}.sagemaker.aws"
 
 
-def _get_root_path(environment: Optional[NotebookEnvironment], port: int) -> str:
+def _get_root_path(environment: NotebookEnvironment, port: int) -> str:
     """
     Returns the base path for the app if the app is running behind a proxy
     """
-    print(f"getting root path for {environment}")
+    if environment == NotebookEnvironment.SAGEMAKER:
+        return f"/proxy/{port}/"
     if environment == NotebookEnvironment.DATABRICKS:
         context = _get_databricks_context()
         return f"/driver-proxy/o/{context.org_id}/{context.cluster_id}/{port}/"
@@ -585,7 +580,7 @@ class DatabricksContext(NamedTuple):
 def _get_databricks_context() -> DatabricksContext:
     """
     Returns the databricks context for constructing the base url
-    and the base_path for the app
+    and the root_path for the app
     """
     import IPython
 
@@ -606,4 +601,4 @@ def _get_databricks_notebook_base_url(context: DatabricksContext) -> str:
     """
     Returns base url of the databricks notebook by parsing the tags
     """
-    return f"https://{context.host}/driver-proxy/o/{context.org_id}/{context.cluster_id}/"
+    return f"https://{context.host}/driver-proxy/o/{context.org_id}/{context.cluster_id}"

phoenix/version.py

@@ -1 +1 @@
-__version__ = "2.4.1rc7"
+__version__ = "2.4.1rc9"