arize-phoenix 8.26.0__tar.gz → 8.26.2__tar.gz

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.26.0
+Version: 8.26.2
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -53,7 +53,6 @@ Requires-Dist: strawberry-graphql>=0.262.0
 Requires-Dist: tqdm
 Requires-Dist: typing-extensions>=4.6
 Requires-Dist: uvicorn
-Requires-Dist: websockets
 Requires-Dist: wrapt>=1.17.2
 Provides-Extra: container
 Requires-Dist: aiohttp; extra == 'container'

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/pyproject.toml

@@ -57,7 +57,6 @@ dependencies = [
   "fastapi",
   "pydantic>=2.1.0", # exclude 2.0.* since it does not support the `json_encoders` configuration setting
   "authlib",
-  "websockets",
   "arize-phoenix-client",
   "email-validator",
 ]

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/src/phoenix/config.py

@@ -119,10 +119,6 @@ ENV_LOG_MIGRATIONS = "PHOENIX_LOG_MIGRATIONS"
 """
 Whether or not to log migrations. Defaults to true.
 """
-ENV_PHOENIX_ENABLE_WEBSOCKETS = "PHOENIX_ENABLE_WEBSOCKETS"
-"""
-Whether or not to enable websockets. Defaults to None.
-"""
 
 ENV_PHOENIX_DANGEROUSLY_DISABLE_MIGRATIONS = "PHOENIX_DANGEROUSLY_DISABLE_MIGRATIONS"
 """
@@ -574,10 +570,6 @@ def get_env_smtp_validate_certs() -> bool:
     return _bool_val(ENV_PHOENIX_SMTP_VALIDATE_CERTS, True)
 
 
-def get_env_enable_websockets() -> Optional[bool]:
-    return _bool_val(ENV_PHOENIX_ENABLE_WEBSOCKETS)
-
-
 @dataclass(frozen=True)
 class OAuth2ClientConfig:
     idp_name: str
@@ -1145,6 +1137,13 @@ def verify_server_environment_variables() -> None:
     get_env_phoenix_secret()
     get_env_phoenix_admin_secret()
 
+    # Notify users about deprecated environment variables if they are being used.
+    if os.getenv("PHOENIX_ENABLE_WEBSOCKETS") is not None:
+        logger.warning(
+            "The environment variable PHOENIX_ENABLE_WEBSOCKETS is deprecated "
+            "because WebSocket is no longer necessary."
+        )
+
 
 SKLEARN_VERSION = cast(tuple[int, int], tuple(map(int, version("scikit-learn").split(".", 2)[:2])))
 PLAYGROUND_PROJECT_NAME = "playground"

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/src/phoenix/server/api/mutations/user_mutations.py

@@ -143,6 +143,8 @@ class UserMutationMixin:
             if input.new_role:
                 if user.email == DEFAULT_ADMIN_EMAIL:
                     raise Unauthorized("Cannot modify role for the default admin user")
+                if user_id == requester_id:
+                    raise Unauthorized("Cannot modify own role")
                 user_role_id = await session.scalar(_select_role_id_by_name(input.new_role.value))
                 if user_role_id is None:
                     raise NotFound(f"Role {input.new_role.value} not found")

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/src/phoenix/server/app.py

@@ -25,23 +25,21 @@ from urllib.parse import urlparse
 import strawberry
 from fastapi import APIRouter, Depends, FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.middleware.gzip import GZipMiddleware
 from fastapi.utils import is_body_allowed_for_status_code
 from grpc.aio import ServerInterceptor
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker
 from starlette.datastructures import State as StarletteState
-from starlette.exceptions import HTTPException, WebSocketException
+from starlette.exceptions import HTTPException
 from starlette.middleware import Middleware
 from starlette.middleware.authentication import AuthenticationMiddleware
 from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
 from starlette.requests import Request
-from starlette.responses import JSONResponse, PlainTextResponse, Response
+from starlette.responses import PlainTextResponse, Response
 from starlette.staticfiles import StaticFiles
 from starlette.status import HTTP_401_UNAUTHORIZED
 from starlette.templating import Jinja2Templates
 from starlette.types import Scope, StatefulLifespan
-from starlette.websockets import WebSocket
 from strawberry.extensions import SchemaExtension
 from strawberry.fastapi import GraphQLRouter
 from strawberry.subscriptions import GRAPHQL_TRANSPORT_WS_PROTOCOL
@@ -123,6 +121,7 @@ from phoenix.server.dml_event_handler import DmlEventHandler
 from phoenix.server.email.types import EmailSender
 from phoenix.server.grpc_server import GrpcServer
 from phoenix.server.jwt_store import JwtStore
+from phoenix.server.middleware.gzip import GZipMiddleware
 from phoenix.server.oauth2 import OAuth2Clients
 from phoenix.server.telemetry import initialize_opentelemetry_tracer_provider
 from phoenix.server.types import (
@@ -207,7 +206,6 @@ class AppConfig(NamedTuple):
     web_manifest_path: Path
     authentication_enabled: bool
     """ Whether authentication is enabled """
-    websockets_enabled: bool
     oauth2_idps: Sequence[OAuth2Idp]
 
 
@@ -258,7 +256,6 @@ class Static(StaticFiles):
                     "manifest": self._web_manifest,
                     "authentication_enabled": self._app_config.authentication_enabled,
                     "oauth2_idps": self._app_config.oauth2_idps,
-                    "websockets_enabled": self._app_config.websockets_enabled,
                 },
             )
         except Exception as e:
@@ -723,36 +720,12 @@ async def plain_text_http_exception_handler(request: Request, exc: HTTPException
     return PlainTextResponse(str(exc.detail), status_code=exc.status_code, headers=headers)
 
 
-async def websocket_denial_response_handler(websocket: WebSocket, exc: WebSocketException) -> None:
-    """
-    Overrides the default exception handler for WebSocketException to ensure
-    that the HTTP response returned when a WebSocket connection is denied has
-    the same status code as the raised exception. This is in keeping with the
-    WebSocket Denial Response Extension of the ASGI specificiation described
-    below.
-
-    "Websocket connections start with the client sending a HTTP request
-    containing the appropriate upgrade headers. On receipt of this request a
-    server can choose to either upgrade the connection or respond with an HTTP
-    response (denying the upgrade). The core ASGI specification does not allow
-    for any control over the denial response, instead specifying that the HTTP
-    status code 403 should be returned, whereas this extension allows an ASGI
-    framework to control the denial response."
-
-    For details, see:
-    - https://asgi.readthedocs.io/en/latest/extensions.html#websocket-denial-response
-    """
-    assert isinstance(exc, WebSocketException)
-    await websocket.send_denial_response(JSONResponse(status_code=exc.code, content=exc.reason))
-
-
 def create_app(
     db: DbSessionFactory,
     export_path: Path,
     model: Model,
     authentication_enabled: bool,
     umap_params: UMAPParameters,
-    enable_websockets: bool,
     corpus: Optional[Model] = None,
     debug: bool = False,
     dev: bool = False,
@@ -900,7 +873,6 @@ def create_app(
         middleware=middlewares,
         exception_handlers={
             HTTPException: plain_text_http_exception_handler,
-            WebSocketException: websocket_denial_response_handler,  # type: ignore[dict-item]
         },
         debug=debug,
         swagger_ui_parameters={
@@ -935,7 +907,6 @@ def create_app(
                     authentication_enabled=authentication_enabled,
                     web_manifest_path=web_manifest_path,
                     oauth2_idps=oauth2_idps,
-                    websockets_enabled=enable_websockets,
                 ),
             ),
             name="static",

{arize_phoenix-8.26.0 → arize_phoenix-8.26.2}/src/phoenix/server/main.py

@@ -23,7 +23,6 @@ from phoenix.config import (
     get_env_db_logging_level,
     get_env_disable_migrations,
     get_env_enable_prometheus,
-    get_env_enable_websockets,
     get_env_grpc_port,
     get_env_host,
     get_env_host_root_path,
@@ -355,13 +354,6 @@ def main() -> None:
         None if corpus_inferences is None else create_model_from_inferences(corpus_inferences)
     )
 
-    # Get enable_websockets from environment variable or command line argument
-    enable_websockets = get_env_enable_websockets()
-    if args.enable_websockets is not None:
-        enable_websockets = args.enable_websockets.lower() == "true"
-    if enable_websockets is None:
-        enable_websockets = True
-
     allowed_origins = get_env_allowed_origins()
 
     # Print information about the server
@@ -374,7 +366,6 @@ def main() -> None:
         storage=get_printable_db_url(db_connection_str),
         schema=get_env_database_schema(),
         auth_enabled=authentication_enabled,
-        websockets_enabled=enable_websockets,
         allowed_origins=allowed_origins,
     )
     if sys.platform.startswith("win"):
@@ -405,7 +396,6 @@ def main() -> None:
         db=factory,
         export_path=export_path,
         model=model,
-        enable_websockets=enable_websockets,
         authentication_enabled=authentication_enabled,
         umap_params=umap_params,
         corpus=corpus_model,

arize_phoenix-8.26.2/src/phoenix/server/middleware/gzip.py

@@ -0,0 +1,33 @@
+from starlette.datastructures import Headers
+from starlette.middleware.gzip import GZipMiddleware as _GZipMiddleware
+from starlette.middleware.gzip import GZipResponder, IdentityResponder
+from starlette.types import ASGIApp, Receive, Scope, Send
+
+
+class GZipMiddleware(_GZipMiddleware):
+    """
+    Subclass of Starlette's GZipMiddleware that excludes multipart/mixed responses from compression.
+
+    This middleware adds a check to exclude multipart/mixed content types from compression,
+    which is important for streaming responses where compression could interfere with delivery.
+
+    The middleware will use the IdentityResponder (no compression) when:
+    1. The client doesn't support gzip compression, or
+    2. The response is a multipart/mixed content type
+    """
+
+    async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+        if scope["type"] != "http":
+            await self.app(scope, receive, send)
+            return
+
+        headers = Headers(scope=scope)
+        responder: ASGIApp
+        if "gzip" not in headers.get("Accept-Encoding", "") or "multipart/mixed" in headers.get(
+            "Accept", ""
+        ):
+            responder = IdentityResponder(self.app, self.minimum_size)
+        else:
+            responder = GZipResponder(self.app, self.minimum_size, compresslevel=self.compresslevel)
+
+        await responder(scope, receive, send)

arize_phoenix-8.26.2/src/phoenix/server/static/.vite/manifest.json

@@ -0,0 +1,87 @@
+{
+  "_components-BpgLJxZt.js": {
+    "file": "assets/components-BpgLJxZt.js",
+    "name": "components",
+    "imports": [
+      "_vendor-C77BEnnR.js",
+      "_pages-CNkfmPS4.js",
+      "_vendor-arizeai-CMLBpPBs.js",
+      "_vendor-codemirror-DT1ReytN.js",
+      "_vendor-three-C5WAXd5r.js"
+    ]
+  },
+  "_pages-CNkfmPS4.js": {
+    "file": "assets/pages-CNkfmPS4.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-C77BEnnR.js",
+      "_vendor-arizeai-CMLBpPBs.js",
+      "_components-BpgLJxZt.js",
+      "_vendor-codemirror-DT1ReytN.js",
+      "_vendor-recharts-ByEcbiML.js"
+    ]
+  },
+  "_vendor-C77BEnnR.js": {
+    "file": "assets/vendor-C77BEnnR.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-C5WAXd5r.js"
+    ],
+    "css": [
+      "assets/vendor-Cg6lcjUC.css"
+    ]
+  },
+  "_vendor-Cg6lcjUC.css": {
+    "file": "assets/vendor-Cg6lcjUC.css",
+    "src": "_vendor-Cg6lcjUC.css"
+  },
+  "_vendor-arizeai-CMLBpPBs.js": {
+    "file": "assets/vendor-arizeai-CMLBpPBs.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-C77BEnnR.js"
+    ]
+  },
+  "_vendor-codemirror-DT1ReytN.js": {
+    "file": "assets/vendor-codemirror-DT1ReytN.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-C77BEnnR.js",
+      "_vendor-shiki-DEgpVt4G.js"
+    ]
+  },
+  "_vendor-recharts-ByEcbiML.js": {
+    "file": "assets/vendor-recharts-ByEcbiML.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-C77BEnnR.js"
+    ]
+  },
+  "_vendor-shiki-DEgpVt4G.js": {
+    "file": "assets/vendor-shiki-DEgpVt4G.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-C77BEnnR.js"
+    ]
+  },
+  "_vendor-three-C5WAXd5r.js": {
+    "file": "assets/vendor-three-C5WAXd5r.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-DHuvzpC2.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-C77BEnnR.js",
+      "_vendor-arizeai-CMLBpPBs.js",
+      "_pages-CNkfmPS4.js",
+      "_components-BpgLJxZt.js",
+      "_vendor-three-C5WAXd5r.js",
+      "_vendor-codemirror-DT1ReytN.js",
+      "_vendor-shiki-DEgpVt4G.js",
+      "_vendor-recharts-ByEcbiML.js"
+    ]
+  }
+}