arize-phoenix 11.2.1__tar.gz → 11.3.0__tar.gz

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 11.2.1
+Version: 11.3.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://arize.com/docs/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -165,6 +165,7 @@ Description-Content-Type: text/markdown
     <a target="_blank" href="https://github.com/Arize-ai/phoenix/tree/main/js/packages/phoenix-mcp">
         <img src="https://badge.mcpx.dev?status=on" title="MCP Enabled"/>
     </a>
+    <a href="https://cursor.com/install-mcp?name=phoenix&config=eyJjb21tYW5kIjoibnB4IC15IEBhcml6ZWFpL3Bob2VuaXgtbWNwQGxhdGVzdCAtLWJhc2VVcmwgaHR0cDovL2xvY2FsaG9zdDo2MDA2IC0tYXBpS2V5IHlvdXItYXBpLWtleSJ9"><img src="https://cursor.com/deeplink/mcp-install-dark.svg" alt="Add Arize Phoenix MCP server to Cursor" height=20 /></a>
 </p>
 
 Phoenix is an open-source AI observability platform designed for experimentation, evaluation, and troubleshooting. It provides:

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/README.md

@@ -34,6 +34,7 @@
     <a target="_blank" href="https://github.com/Arize-ai/phoenix/tree/main/js/packages/phoenix-mcp">
         <img src="https://badge.mcpx.dev?status=on" title="MCP Enabled"/>
     </a>
+    <a href="https://cursor.com/install-mcp?name=phoenix&config=eyJjb21tYW5kIjoibnB4IC15IEBhcml6ZWFpL3Bob2VuaXgtbWNwQGxhdGVzdCAtLWJhc2VVcmwgaHR0cDovL2xvY2FsaG9zdDo2MDA2IC0tYXBpS2V5IHlvdXItYXBpLWtleSJ9"><img src="https://cursor.com/deeplink/mcp-install-dark.svg" alt="Add Arize Phoenix MCP server to Cursor" height=20 /></a>
 </p>
 
 Phoenix is an open-source AI observability platform designed for experimentation, evaluation, and troubleshooting. It provides:

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/config.py

@@ -223,6 +223,12 @@ Examples:
     - With a sub-path: "https://example.com/phoenix"
     - Without a sub-path: "https://phoenix.example.com"
 """
+ENV_PHOENIX_MANAGEMENT_URL = "PHOENIX_MANAGEMENT_URL"
+"""
+The URL to use for redirecting to a management interface that may be hosting Phoenix. If set, and
+the current user is within PHOENIX_ADMINS, a link will be added to the navigation menu to return to
+this URL.
+"""
 
 
 # SMTP settings
@@ -818,7 +824,7 @@ def get_env_csrf_trusted_origins() -> list[str]:
 
 def get_env_admins() -> dict[str, str]:
     """
-    Parse the PHOENIX_ADMINS environment variable to extract the comma separated pairs of
+    Parse the PHOENIX_ADMINS environment variable to extract the semicolon separated pairs of
     username and email. The last equal sign (=) in each pair is used to separate the username from
     the email.
 
@@ -1540,6 +1546,13 @@ def get_env_fullstory_org() -> Optional[str]:
     return getenv(ENV_PHOENIX_FULLSTORY_ORG)
 
 
+def get_env_management_url() -> Optional[str]:
+    """
+    Gets the value of the PHOENIX_MANAGEMENT_URL environment variable.
+    """
+    return getenv(ENV_PHOENIX_MANAGEMENT_URL)
+
+
 def verify_server_environment_variables() -> None:
     """Verify that the environment variables are set correctly. Raises an error otherwise."""
     get_env_root_url()

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/db/facilitator.py

@@ -519,4 +519,5 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
             sa.update(models.GenerativeModel)
             .values(deleted_at=sa.func.now())
             .where(models.GenerativeModel.id.in_([m.id for m in remaining_models]))
+            .where(~sa.exists().where(models.GenerativeModel.id == models.SpanCost.model_id))
         )

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/api/context.py

@@ -15,6 +15,7 @@ from phoenix.auth import (
 from phoenix.core.model_schema import Model
 from phoenix.db import models
 from phoenix.server.api.dataloaders import (
+    AnnotationConfigsByProjectDataLoader,
     AnnotationSummaryDataLoader,
     AverageExperimentRunLatencyDataLoader,
     CacheForDataLoaders,
@@ -82,6 +83,7 @@ from phoenix.server.types import (
 
 @dataclass
 class DataLoaders:
+    annotation_configs_by_project: AnnotationConfigsByProjectDataLoader
     annotation_summaries: AnnotationSummaryDataLoader
     average_experiment_run_latency: AverageExperimentRunLatencyDataLoader
     dataset_example_revisions: DatasetExampleRevisionsDataLoader

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/api/dataloaders/__init__.py

@@ -4,6 +4,7 @@ from phoenix.server.api.dataloaders.span_cost_detail_summary_entries_by_project_
     SpanCostDetailSummaryEntriesByProjectSessionDataLoader,
 )
 
+from .annotation_configs_by_project import AnnotationConfigsByProjectDataLoader
 from .annotation_summaries import AnnotationSummaryCache, AnnotationSummaryDataLoader
 from .average_experiment_run_latency import AverageExperimentRunLatencyDataLoader
 from .dataset_example_revisions import DatasetExampleRevisionsDataLoader
@@ -61,6 +62,7 @@ from .user_roles import UserRolesDataLoader
 from .users import UsersDataLoader
 
 __all__ = [
+    "AnnotationConfigsByProjectDataLoader",
     "AnnotationSummaryDataLoader",
     "AverageExperimentRunLatencyDataLoader",
     "CacheForDataLoaders",

arize_phoenix-11.3.0/src/phoenix/server/api/dataloaders/annotation_configs_by_project.py

@@ -0,0 +1,31 @@
+from collections import defaultdict
+
+from sqlalchemy import select
+from strawberry.dataloader import DataLoader
+from typing_extensions import TypeAlias
+
+from phoenix.db import models
+from phoenix.server.types import DbSessionFactory
+
+ProjectId: TypeAlias = int
+Key: TypeAlias = ProjectId
+Result: TypeAlias = tuple[models.AnnotationConfig, ...]
+
+
+class AnnotationConfigsByProjectDataLoader(DataLoader[Key, Result]):
+    def __init__(self, db: DbSessionFactory) -> None:
+        super().__init__(load_fn=self._load_fn)
+        self._db = db
+
+    async def _load_fn(self, keys: list[Key]) -> list[Result]:
+        stmt = (
+            select(models.ProjectAnnotationConfig.project_id, models.AnnotationConfig)
+            .join_from(models.ProjectAnnotationConfig, models.AnnotationConfig)
+            .where(models.ProjectAnnotationConfig.project_id.in_(keys))
+        )
+        results: defaultdict[Key, list[models.AnnotationConfig]] = defaultdict(list)
+        async with self._db() as session:
+            data = await session.stream(stmt)
+            async for id_, config in data:
+                results[id_].append(config)
+        return [tuple(results[k]) for k in keys]

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/api/dataloaders/span_cost_summary_by_project.py

@@ -119,20 +119,18 @@ def _get_stmt(
             coalesce(func.sum(models.SpanCost.completion_tokens), 0).label("completion_tokens"),
             coalesce(func.sum(models.SpanCost.total_tokens), 0).label("total_tokens"),
         )
-        .select_from(models.Trace)
-        .join(models.Span, models.Span.trace_rowid == models.Trace.id)
-        .join(models.SpanCost, models.Span.id == models.SpanCost.span_rowid)
+        .join_from(models.SpanCost, models.Trace)
         .group_by(pid)
     )
 
     if start_time:
-        stmt = stmt.where(start_time <= models.Span.start_time)
+        stmt = stmt.where(start_time <= models.Trace.start_time)
     if end_time:
-        stmt = stmt.where(models.Span.start_time < end_time)
+        stmt = stmt.where(models.Trace.start_time < end_time)
 
     if filter_condition:
         sf = SpanFilter(filter_condition)
-        stmt = sf(stmt)
+        stmt = sf(stmt.join_from(models.SpanCost, models.Span))
 
     project_ids = [rowid for rowid in params]
     stmt = stmt.where(pid.in_(project_ids))

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/api/types/Project.py

@@ -655,18 +655,9 @@ class Project(Node):
             last=last,
             before=before if isinstance(before, CursorString) else None,
         )
-        async with info.context.db() as session:
-            annotation_configs = await session.stream_scalars(
-                select(models.AnnotationConfig)
-                .join(
-                    models.ProjectAnnotationConfig,
-                    models.AnnotationConfig.id
-                    == models.ProjectAnnotationConfig.annotation_config_id,
-                )
-                .where(models.ProjectAnnotationConfig.project_id == self.project_rowid)
-                .order_by(models.AnnotationConfig.name)
-            )
-            data = [to_gql_annotation_config(config) async for config in annotation_configs]
+        loader = info.context.data_loaders.annotation_configs_by_project
+        configs = await loader.load(self.project_rowid)
+        data = [to_gql_annotation_config(config) for config in configs]
         return connection_from_list(data=data, args=args)
 
     @strawberry.field

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/api/types/User.py

@@ -7,6 +7,7 @@ from strawberry import Private
 from strawberry.relay import Node, NodeID
 from strawberry.types import Info
 
+from phoenix.config import get_env_admins
 from phoenix.db import models
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import NotFound
@@ -42,6 +43,15 @@ class User(Node):
             )
         return [to_gql_api_key(api_key) for api_key in api_keys]
 
+    @strawberry.field
+    async def is_management_user(self) -> bool:
+        initial_admins = get_env_admins()
+        # this field is only visible to initial admins as they are the ones likely to have access to
+        # a management interface / the phoenix environment.
+        if self.email in initial_admins or self.email == "admin@localhost":
+            return True
+        return False
+
 
 def to_gql_user(user: models.User, api_keys: Optional[list[models.ApiKey]] = None) -> User:
     """

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/app.py

@@ -72,6 +72,7 @@ from phoenix.exceptions import PhoenixMigrationError
 from phoenix.pointcloud.umap_parameters import UMAPParameters
 from phoenix.server.api.context import Context, DataLoaders
 from phoenix.server.api.dataloaders import (
+    AnnotationConfigsByProjectDataLoader,
     AnnotationSummaryDataLoader,
     AverageExperimentRunLatencyDataLoader,
     CacheForDataLoaders,
@@ -231,6 +232,8 @@ class AppConfig(NamedTuple):
     auto_login_idp_name: Optional[str] = None
     fullstory_org: Optional[str] = None
     """ FullStory organization ID for web analytics tracking """
+    management_url: Optional[str] = None
+    """ URL for a phoenix management interface, only visible to management users """
 
 
 class Static(StaticFiles):
@@ -297,6 +300,7 @@ class Static(StaticFiles):
                     "basic_auth_disabled": self._app_config.basic_auth_disabled,
                     "auto_login_idp_name": self._app_config.auto_login_idp_name,
                     "fullstory_org": self._app_config.fullstory_org,
+                    "management_url": self._app_config.management_url,
                 },
             )
         except Exception as e:
@@ -642,6 +646,7 @@ def create_graphql_router(
             last_updated_at=last_updated_at,
             event_queue=event_queue,
             data_loaders=DataLoaders(
+                annotation_configs_by_project=AnnotationConfigsByProjectDataLoader(db),
                 average_experiment_run_latency=AverageExperimentRunLatencyDataLoader(db),
                 dataset_example_revisions=DatasetExampleRevisionsDataLoader(db),
                 dataset_example_spans=DatasetExampleSpansDataLoader(db),
@@ -847,6 +852,7 @@ def create_app(
     basic_auth_disabled: bool = False,
     bulk_inserter_factory: Optional[Callable[..., BulkInserter]] = None,
     allowed_origins: Optional[list[str]] = None,
+    management_url: Optional[str] = None,
 ) -> FastAPI:
     verify_server_environment_variables()
     if model.embedding_dimensions:
@@ -1026,6 +1032,7 @@ def create_app(
                     basic_auth_disabled=basic_auth_disabled,
                     auto_login_idp_name=auto_login_idp_name,
                     fullstory_org=Settings.fullstory_org,
+                    management_url=management_url,
                 ),
             ),
             name="static",

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/main.py

@@ -32,6 +32,7 @@ from phoenix.config import (
     get_env_log_migrations,
     get_env_logging_level,
     get_env_logging_mode,
+    get_env_management_url,
     get_env_oauth2_settings,
     get_env_password_reset_token_expiry,
     get_env_port,
@@ -377,6 +378,7 @@ def main() -> None:
     )
 
     allowed_origins = get_env_allowed_origins()
+    management_url = get_env_management_url()
 
     # Get TLS configuration
     tls_enabled_for_http = get_env_tls_enabled_for_http()
@@ -455,6 +457,7 @@ def main() -> None:
         email_sender=email_sender,
         oauth2_client_configs=get_env_oauth2_settings(),
         allowed_origins=allowed_origins,
+        management_url=management_url,
     )
 
     # Configure server with TLS if enabled

{arize_phoenix-11.2.1 → arize_phoenix-11.3.0}/src/phoenix/server/static/.vite/manifest.json

@@ -1,22 +1,22 @@
 {
-  "_components-PP4ssj43.js": {
-    "file": "assets/components-PP4ssj43.js",
+  "_components-DlL7ybQ2.js": {
+    "file": "assets/components-DlL7ybQ2.js",
     "name": "components",
     "imports": [
       "_vendor-DqQvHbPa.js",
-      "_pages-DozDSxRj.js",
+      "_pages-B7wCtpad.js",
       "_vendor-arizeai-CLX44PFA.js",
       "_vendor-codemirror-Du3XyJnB.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-DozDSxRj.js": {
-    "file": "assets/pages-DozDSxRj.js",
+  "_pages-B7wCtpad.js": {
+    "file": "assets/pages-B7wCtpad.js",
     "name": "pages",
     "imports": [
       "_vendor-DqQvHbPa.js",
       "_vendor-arizeai-CLX44PFA.js",
-      "_components-PP4ssj43.js",
+      "_components-DlL7ybQ2.js",
       "_vendor-codemirror-Du3XyJnB.js",
       "_vendor-recharts-B2PJDrnX.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-COyfeNsl.js",
+    "file": "assets/index-QP9R8k34.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-DqQvHbPa.js",
       "_vendor-arizeai-CLX44PFA.js",
-      "_pages-DozDSxRj.js",
-      "_components-PP4ssj43.js",
+      "_pages-B7wCtpad.js",
+      "_components-DlL7ybQ2.js",
       "_vendor-three-C5WAXd5r.js",
       "_vendor-codemirror-Du3XyJnB.js",
       "_vendor-shiki-CNbrFjf9.js",