arize-phoenix 12.0.0__py3-none-any.whl → 12.3.0__py3-none-any.whl

phoenix/server/api/types/Dataset.py

@@ -18,6 +18,7 @@ from phoenix.server.api.types.DatasetExample import DatasetExample
 from phoenix.server.api.types.DatasetExperimentAnnotationSummary import (
     DatasetExperimentAnnotationSummary,
 )
+from phoenix.server.api.types.DatasetLabel import DatasetLabel, to_gql_dataset_label
 from phoenix.server.api.types.DatasetVersion import DatasetVersion
 from phoenix.server.api.types.Experiment import Experiment, to_gql_experiment
 from phoenix.server.api.types.node import from_global_id_with_expected_type
@@ -303,6 +304,13 @@ class Dataset(Node):
                 async for scores_tuple in await session.stream(query)
             ]
 
+    @strawberry.field
+    async def labels(self, info: Info[Context, None]) -> list[DatasetLabel]:
+        return [
+            to_gql_dataset_label(label)
+            for label in await info.context.data_loaders.dataset_labels.load(self.id_attr)
+        ]
+
     @strawberry.field
     def last_updated_at(self, info: Info[Context, None]) -> Optional[datetime]:
         return info.context.last_updated_at.get(self._table, self.id_attr)

phoenix/server/api/types/DatasetExample.py

@@ -142,3 +142,10 @@ class DatasetExample(Node):
             to_gql_dataset_split(split)
             for split in await info.context.data_loaders.dataset_example_splits.load(self.id_attr)
         ]
+
+
+def to_gql_dataset_example(example: models.DatasetExample) -> DatasetExample:
+    return DatasetExample(
+        id_attr=example.id,
+        created_at=example.created_at,
+    )

phoenix/server/api/types/DatasetLabel.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+import strawberry
+from strawberry.relay import Node, NodeID
+
+from phoenix.db import models
+
+
+@strawberry.type
+class DatasetLabel(Node):
+    id_attr: NodeID[int]
+    name: str
+    description: Optional[str]
+    color: str
+
+
+def to_gql_dataset_label(dataset_label: models.DatasetLabel) -> DatasetLabel:
+    return DatasetLabel(
+        id_attr=dataset_label.id,
+        name=dataset_label.name,
+        description=dataset_label.description,
+        color=dataset_label.color,
+    )

phoenix/server/api/types/Prompt.py

@@ -9,6 +9,7 @@ from strawberry.relay import Connection, GlobalID, Node, NodeID
 from strawberry.types import Info
 
 from phoenix.db import models
+from phoenix.db.types.identifier import Identifier as IdentifierModel
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import NotFound
 from phoenix.server.api.types.Identifier import Identifier
@@ -37,7 +38,10 @@ class Prompt(Node):
 
     @strawberry.field
     async def version(
-        self, info: Info[Context, None], version_id: Optional[GlobalID] = None
+        self,
+        info: Info[Context, None],
+        version_id: Optional[GlobalID] = None,
+        tag_name: Optional[Identifier] = None,
     ) -> PromptVersion:
         async with info.context.db() as session:
             if version_id:
@@ -50,6 +54,19 @@ class Prompt(Node):
                 )
                 if not version:
                     raise NotFound(f"Prompt version not found: {version_id}")
+            elif tag_name:
+                try:
+                    name = IdentifierModel(tag_name)
+                except ValueError:
+                    raise NotFound(f"Prompt version tag not found: {tag_name}")
+                version = await session.scalar(
+                    select(models.PromptVersion)
+                    .where(models.PromptVersion.prompt_id == self.id_attr)
+                    .join_from(models.PromptVersion, models.PromptVersionTag)
+                    .where(models.PromptVersionTag.name == name)
+                )
+                if not version:
+                    raise NotFound(f"This prompt has no associated versions by tag {tag_name}")
             else:
                 stmt = (
                     select(models.PromptVersion)

phoenix/server/app.py

@@ -67,7 +67,6 @@ from phoenix.config import (
     get_env_gql_extension_paths,
     get_env_grpc_interceptor_paths,
     get_env_host,
-    get_env_host_root_path,
     get_env_max_spans_queue_size,
     get_env_port,
     get_env_support_email,
@@ -82,6 +81,7 @@ from phoenix.db.facilitator import Facilitator
 from phoenix.db.helpers import SupportedSQLDialect
 from phoenix.exceptions import PhoenixMigrationError
 from phoenix.pointcloud.umap_parameters import UMAPParameters
+from phoenix.server.api.auth_messages import AUTH_ERROR_MESSAGES, AuthErrorCode
 from phoenix.server.api.context import Context, DataLoaders
 from phoenix.server.api.dataloaders import (
     AnnotationConfigsByProjectDataLoader,
@@ -145,6 +145,7 @@ from phoenix.server.api.dataloaders import (
     UserRolesDataLoader,
     UsersDataLoader,
 )
+from phoenix.server.api.dataloaders.dataset_labels import DatasetLabelsDataLoader
 from phoenix.server.api.routers import (
     auth_router,
     create_embeddings_router,
@@ -175,6 +176,7 @@ from phoenix.server.types import (
     LastUpdatedAt,
     TokenStore,
 )
+from phoenix.server.utils import get_root_path, prepend_root_path
 from phoenix.settings import Settings
 from phoenix.trace.fixtures import (
     TracesFixture,
@@ -249,6 +251,8 @@ class AppConfig(NamedTuple):
     web_manifest_path: Path
     authentication_enabled: bool
     """ Whether authentication is enabled """
+    auth_error_messages: dict[AuthErrorCode, str]
+    """ Mapping of auth error codes to user-friendly messages """
     oauth2_idps: Sequence[OAuth2Idp]
     basic_auth_disabled: bool = False
     auto_login_idp_name: Optional[str] = None
@@ -283,9 +287,6 @@ class Static(StaticFiles):
                 return {}
             raise e
 
-    def _sanitize_basename(self, basename: str) -> str:
-        return basename[:-1] if basename.endswith("/") else basename
-
     async def get_response(self, path: str, scope: Scope) -> Response:
         # Redirect to the oauth2 login page if basic auth is disabled and auto_login is enabled
         # TODO: this needs to be refactored to be cleaner
@@ -294,14 +295,10 @@ class Static(StaticFiles):
             and self._app_config.basic_auth_disabled
             and self._app_config.auto_login_idp_name
         ):
-            request = Request(scope)
-            url = URL(
-                str(
-                    Path(get_env_host_root_path())
-                    / f"oauth2/{self._app_config.auto_login_idp_name}/login"
-                )
+            redirect_path = prepend_root_path(
+                scope, f"oauth2/{self._app_config.auto_login_idp_name}/login"
             )
-            url = url.include_query_params(**request.query_params)
+            url = URL(redirect_path).include_query_params(**Request(scope).query_params)
             return RedirectResponse(url=url)
         try:
             response = await super().get_response(path, scope)
@@ -318,7 +315,7 @@ 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": self._sanitize_basename(request.scope.get("root_path", "")),
+                    "basename": get_root_path(scope),
                     "platform_version": phoenix_version,
                     "request": request,
                     "is_development": self._app_config.is_development,
@@ -332,6 +329,7 @@ class Static(StaticFiles):
                     "support_email": self._app_config.support_email,
                     "has_db_threshold": self._app_config.has_db_threshold,
                     "allow_external_resources": self._app_config.allow_external_resources,
+                    "auth_error_messages": self._app_config.auth_error_messages,
                 },
             )
         except Exception as e:
@@ -718,6 +716,7 @@ def create_graphql_router(
                     db
                 ),
                 dataset_example_splits=DatasetExampleSplitsDataLoader(db),
+                dataset_labels=DatasetLabelsDataLoader(db),
                 document_evaluation_summaries=DocumentEvaluationSummaryDataLoader(
                     db,
                     cache_map=(
@@ -1151,6 +1150,7 @@ def create_app(
                         and get_env_database_usage_insertion_blocking_threshold_percentage()
                     ),
                     allow_external_resources=get_env_allow_external_resources(),
+                    auth_error_messages=dict(AUTH_ERROR_MESSAGES) if authentication_enabled else {},
                 ),
             ),
             name="static",

phoenix/server/cost_tracking/model_cost_manifest.json

@@ -449,6 +449,60 @@
         }
       ]
     },
+    {
+      "name": "claude-sonnet-4-5",
+      "name_pattern": "claude-sonnet-4-5",
+      "source": "litellm",
+      "token_prices": [
+        {
+          "base_rate": 3e-6,
+          "is_prompt": true,
+          "token_type": "input"
+        },
+        {
+          "base_rate": 0.000015,
+          "is_prompt": false,
+          "token_type": "output"
+        },
+        {
+          "base_rate": 3e-7,
+          "is_prompt": true,
+          "token_type": "cache_read"
+        },
+        {
+          "base_rate": 3.75e-6,
+          "is_prompt": true,
+          "token_type": "cache_write"
+        }
+      ]
+    },
+    {
+      "name": "claude-sonnet-4-5-20250929",
+      "name_pattern": "claude-sonnet-4-5-20250929",
+      "source": "litellm",
+      "token_prices": [
+        {
+          "base_rate": 3e-6,
+          "is_prompt": true,
+          "token_type": "input"
+        },
+        {
+          "base_rate": 0.000015,
+          "is_prompt": false,
+          "token_type": "output"
+        },
+        {
+          "base_rate": 3e-7,
+          "is_prompt": true,
+          "token_type": "cache_read"
+        },
+        {
+          "base_rate": 3.75e-6,
+          "is_prompt": true,
+          "token_type": "cache_write"
+        }
+      ]
+    },
     {
       "name": "gemini-2.0-flash",
       "name_pattern": "gemini-2.0-flash(@[a-zA-Z0-9]+)?",
@@ -1028,60 +1082,6 @@
         }
       ]
     },
-    {
-      "name": "gemini-flash-latest",
-      "name_pattern": "gemini-flash-latest",
-      "source": "litellm",
-      "token_prices": [
-        {
-          "base_rate": 3e-7,
-          "is_prompt": true,
-          "token_type": "input"
-        },
-        {
-          "base_rate": 2.5e-6,
-          "is_prompt": false,
-          "token_type": "output"
-        },
-        {
-          "base_rate": 7.5e-8,
-          "is_prompt": true,
-          "token_type": "cache_read"
-        },
-        {
-          "base_rate": 1e-6,
-          "is_prompt": true,
-          "token_type": "audio"
-        }
-      ]
-    },
-    {
-      "name": "gemini-flash-lite-latest",
-      "name_pattern": "gemini-flash-lite-latest",
-      "source": "litellm",
-      "token_prices": [
-        {
-          "base_rate": 1e-7,
-          "is_prompt": true,
-          "token_type": "input"
-        },
-        {
-          "base_rate": 4e-7,
-          "is_prompt": false,
-          "token_type": "output"
-        },
-        {
-          "base_rate": 2.5e-8,
-          "is_prompt": true,
-          "token_type": "cache_read"
-        },
-        {
-          "base_rate": 3e-7,
-          "is_prompt": true,
-          "token_type": "audio"
-        }
-      ]
-    },
     {
       "name": "gpt-3.5-turbo",
       "name_pattern": "gpt-(35|3.5)-turbo",

phoenix/server/oauth2.py

@@ -83,10 +83,8 @@ class OAuth2Clients:
             self._auto_login_client = client
         self._clients[config.idp_name] = client
 
-    def get_client(self, idp_name: str) -> OAuth2Client:
-        if (client := self._clients.get(idp_name)) is None:
-            raise ValueError(f"unknown or unregistered OAuth2 client: {idp_name}")
-        return client
+    def get_client(self, idp_name: str) -> Optional[OAuth2Client]:
+        return self._clients.get(idp_name)
 
     @classmethod
     def from_configs(cls, configs: Iterable[OAuth2ClientConfig]) -> "OAuth2Clients":

phoenix/server/static/.vite/manifest.json

@@ -1,32 +1,32 @@
 {
-  "_components-Dl9SUw1U.js": {
-    "file": "assets/components-Dl9SUw1U.js",
+  "_components-Bs8eJEpU.js": {
+    "file": "assets/components-Bs8eJEpU.js",
     "name": "components",
     "imports": [
-      "_vendor-CtbHQYl8.js",
-      "_vendor-arizeai-D-lWOwIS.js",
-      "_pages-DKSjVA_E.js",
-      "_vendor-codemirror-BRBpy3_z.js",
+      "_vendor-D2eEI-6h.js",
+      "_pages-D-n2pkoG.js",
+      "_vendor-arizeai-kfOei7nf.js",
+      "_vendor-codemirror-1bq_t1Ec.js",
       "_vendor-three-BLWp5bic.js"
     ]
   },
-  "_pages-DKSjVA_E.js": {
-    "file": "assets/pages-DKSjVA_E.js",
+  "_pages-D-n2pkoG.js": {
+    "file": "assets/pages-D-n2pkoG.js",
     "name": "pages",
     "imports": [
-      "_vendor-CtbHQYl8.js",
-      "_vendor-arizeai-D-lWOwIS.js",
-      "_components-Dl9SUw1U.js",
-      "_vendor-codemirror-BRBpy3_z.js",
-      "_vendor-recharts--KdSwB3m.js"
+      "_vendor-D2eEI-6h.js",
+      "_components-Bs8eJEpU.js",
+      "_vendor-arizeai-kfOei7nf.js",
+      "_vendor-codemirror-1bq_t1Ec.js",
+      "_vendor-recharts-DQ4xfrf4.js"
     ]
   },
   "_vendor-BGzfc4EU.css": {
     "file": "assets/vendor-BGzfc4EU.css",
     "src": "_vendor-BGzfc4EU.css"
   },
-  "_vendor-CtbHQYl8.js": {
-    "file": "assets/vendor-CtbHQYl8.js",
+  "_vendor-D2eEI-6h.js": {
+    "file": "assets/vendor-D2eEI-6h.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-BLWp5bic.js"
@@ -35,39 +35,39 @@
       "assets/vendor-BGzfc4EU.css"
     ]
   },
-  "_vendor-arizeai-D-lWOwIS.js": {
-    "file": "assets/vendor-arizeai-D-lWOwIS.js",
+  "_vendor-arizeai-kfOei7nf.js": {
+    "file": "assets/vendor-arizeai-kfOei7nf.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-CtbHQYl8.js"
+      "_vendor-D2eEI-6h.js"
     ]
   },
-  "_vendor-codemirror-BRBpy3_z.js": {
-    "file": "assets/vendor-codemirror-BRBpy3_z.js",
+  "_vendor-codemirror-1bq_t1Ec.js": {
+    "file": "assets/vendor-codemirror-1bq_t1Ec.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-CtbHQYl8.js",
-      "_vendor-shiki-CvRzZnIo.js"
+      "_vendor-D2eEI-6h.js",
+      "_vendor-shiki-GGmcIQxA.js"
     ],
     "dynamicImports": [
-      "_vendor-shiki-CvRzZnIo.js",
-      "_vendor-shiki-CvRzZnIo.js",
-      "_vendor-shiki-CvRzZnIo.js"
+      "_vendor-shiki-GGmcIQxA.js",
+      "_vendor-shiki-GGmcIQxA.js",
+      "_vendor-shiki-GGmcIQxA.js"
     ]
   },
-  "_vendor-recharts--KdSwB3m.js": {
-    "file": "assets/vendor-recharts--KdSwB3m.js",
+  "_vendor-recharts-DQ4xfrf4.js": {
+    "file": "assets/vendor-recharts-DQ4xfrf4.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-CtbHQYl8.js"
+      "_vendor-D2eEI-6h.js"
     ]
   },
-  "_vendor-shiki-CvRzZnIo.js": {
-    "file": "assets/vendor-shiki-CvRzZnIo.js",
+  "_vendor-shiki-GGmcIQxA.js": {
+    "file": "assets/vendor-shiki-GGmcIQxA.js",
     "name": "vendor-shiki",
     "isDynamicEntry": true,
     "imports": [
-      "_vendor-CtbHQYl8.js"
+      "_vendor-D2eEI-6h.js"
     ]
   },
   "_vendor-three-BLWp5bic.js": {
@@ -75,19 +75,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-CqQS0dTo.js",
+    "file": "assets/index-C6WEu5UP.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-CtbHQYl8.js",
-      "_vendor-arizeai-D-lWOwIS.js",
-      "_pages-DKSjVA_E.js",
-      "_components-Dl9SUw1U.js",
+      "_vendor-D2eEI-6h.js",
+      "_vendor-arizeai-kfOei7nf.js",
+      "_pages-D-n2pkoG.js",
+      "_components-Bs8eJEpU.js",
       "_vendor-three-BLWp5bic.js",
-      "_vendor-codemirror-BRBpy3_z.js",
-      "_vendor-shiki-CvRzZnIo.js",
-      "_vendor-recharts--KdSwB3m.js"
+      "_vendor-codemirror-1bq_t1Ec.js",
+      "_vendor-shiki-GGmcIQxA.js",
+      "_vendor-recharts-DQ4xfrf4.js"
     ]
   }
 }