arize-phoenix 12.7.1__py3-none-any.whl → 12.8.0__py3-none-any.whl

{arize_phoenix-12.7.1.dist-info → arize_phoenix-12.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 12.7.1
+Version: 12.8.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://arize.com/docs/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-12.7.1.dist-info → arize_phoenix-12.8.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=ngkyKGVatX3cO2WJdo2hKdaVKP-xJCMvqthvga6kJss,5196
 phoenix/settings.py,sha256=2kHfT3BNOVd4dAO1bq-syEQbHSG8oX2-7NhOwK2QREk,896
-phoenix/version.py,sha256=3-Wpz499-Ibd-1ZddeScicQ9Iq5cO_UPhkbj_81hMY0,23
+phoenix/version.py,sha256=gI1fVVoXdBSWPUaSm3faftAdEowgaQCFGYaqnWMq664,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=qBFraOtmwCCnWJltKNP18DDG0mULXigytlFsa6YOz6k,4837
@@ -126,7 +126,7 @@ phoenix/server/api/auth_messages.py,sha256=15oqogBNS2BgJLLDecfXz62G9LdiOAEyzPv6j
 phoenix/server/api/context.py,sha256=MmbVDZuniYDQ9Xw_g371n8_HtqlIUwt_n0Li7FkGZzs,10649
 phoenix/server/api/exceptions.py,sha256=9gB4nBRNX6k4_fsQZ12yxw6Tw53h_915l06DYK-qkPQ,1442
 phoenix/server/api/interceptor.py,sha256=ykDnoC_apUd-llVli3m1CW18kNSIgjz2qZ6m5JmPDu8,1294
-phoenix/server/api/queries.py,sha256=G8FG6xzQhtt6RqVEIX3p55PxHy2QRMSfAGeDjk0gBm8,68287
+phoenix/server/api/queries.py,sha256=SQjsnCcNYve8XnQF0CMJGzSZy60SSZLZsxvR72cdWMk,68824
 phoenix/server/api/schema.py,sha256=fcs36xQwFF_Qe41_5cWR8wYpDvOrnbcyTeo5WNMbDsA,1702
 phoenix/server/api/subscriptions.py,sha256=f2o9L04BdMxh6jWHmhdokzu84mifh6u2IF4z6Pt_ZOs,26206
 phoenix/server/api/utils.py,sha256=quCBRcusc6PUq9tJq7M8PgwFZp7nXgVAxtbw8feribY,833
@@ -262,7 +262,7 @@ phoenix/server/api/mutations/__init__.py,sha256=83V5Ia8ijsajSLo_gzAuHXo4hgDlUGLI
 phoenix/server/api/mutations/annotation_config_mutations.py,sha256=ZcjWlJ88B_jevoESmOtmO0l-SUEbtAwxwP1KpWTmYqQ,15511
 phoenix/server/api/mutations/api_key_mutations.py,sha256=3djEPrVOwF7oN3C46WEROWqiBi0EZnue7dTQDAxwBKo,6526
 phoenix/server/api/mutations/chat_mutations.py,sha256=T9S6IMWVSoFVhkwyXGMg-LGKPcQcKtKb7Y4nsDbRr_0,26020
-phoenix/server/api/mutations/dataset_label_mutations.py,sha256=Paq1zsWyrAXTcfPvX5Bn53LDt1rIIrJtAo5XlRP-0Ts,11660
+phoenix/server/api/mutations/dataset_label_mutations.py,sha256=qT5JoV4gtJwctzxwGsqyzyhMUQhSrIhAeRQ5J7-FE20,9524
 phoenix/server/api/mutations/dataset_mutations.py,sha256=BsBW2epV3gtgnO2fn3lJP9uX2qOZEY6gy0UW1UEgpUA,28137
 phoenix/server/api/mutations/dataset_split_mutations.py,sha256=Mbmr6LyqqoaLfB4rwMrx2V0GICHlcJ34hyj6c1k7tvk,17415
 phoenix/server/api/mutations/experiment_mutations.py,sha256=_fWkKl3VEMF4w885vIuh8zjH9SOhlXc4QaHzGLgBnuY,3214
@@ -402,7 +402,7 @@ phoenix/server/cost_tracking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 phoenix/server/cost_tracking/cost_details_calculator.py,sha256=Tt0YcuLhgPuXKWJemWVmYQfG0xQUvH4VziIj6KcDnoA,8945
 phoenix/server/cost_tracking/cost_model_lookup.py,sha256=jhtVdnQBzrTUHeOGPWgOebk-Io5hpJ1vAgWOu8ojeJ4,6801
 phoenix/server/cost_tracking/helpers.py,sha256=Pk6ECjnYreTxrldtRwxnwFcxIPVsvDq_yAwDA_spkOc,2122
-phoenix/server/cost_tracking/model_cost_manifest.json,sha256=d8r3YEhCH1PxX8ebla-k0JAKk6e2cjkxWQ0ybQ2zrC4,71781
+phoenix/server/cost_tracking/model_cost_manifest.json,sha256=uRmDJ7GZgNSV8ehaEbFzMtn4g4YkUGPHAmRGsPWCRNg,71778
 phoenix/server/cost_tracking/regex_specificity.py,sha256=9kqWuQ68C-hlwW25hr7BhFlRt5y2Nnpy0Ax3n9UN6Xk,11622
 phoenix/server/cost_tracking/token_cost_calculator.py,sha256=2JEZnvusx2-xbhp8krp9EarjWuyGH2KO4e-ZwJX-K0s,1598
 phoenix/server/daemons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -431,16 +431,16 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
-phoenix/server/static/.vite/manifest.json,sha256=rE94qPNPdy1j50uYFL_gWnG4riXUA3O_k3g5rTq_Dng,2328
-phoenix/server/static/assets/components-BLK5vehh.js,sha256=xvsGRDvOUWHo1EqrzrYKVg5oYfjQJKkU-EpgO0O95ZY,758741
-phoenix/server/static/assets/index-BP0Shd90.js,sha256=S_-pqQqEoPn6qS5LCxTxFe3VzFeyJkZWGt9r9c8RqYE,63675
-phoenix/server/static/assets/pages-DIVgyYyy.js,sha256=_QO8oWWO8yTZMNDqU2JPE_oNacDPnt80oXb_Idwy-IM,1352811
-phoenix/server/static/assets/vendor-3BvTzoBp.js,sha256=TTjoWR-GrJf3kUBuRWOKsoL66pKco299wKSb7lgaKLA,2676807
+phoenix/server/static/.vite/manifest.json,sha256=xZi8Y0krYRlTEV7AKGciIFWUty9Nx_klwu5ZflV0Biw,2328
+phoenix/server/static/assets/components-Bem6_7MW.js,sha256=A8bCIl-w4LKbsQZbVLMVu2cD3uQQpAs4DdvOo6EFZY4,751728
+phoenix/server/static/assets/index-NdiXbuNL.js,sha256=SPndC2GMCbo-fmLgKRWW4kJQWvfJkbON2OxmaVqfkrQ,63699
+phoenix/server/static/assets/pages-CEJgMVKU.js,sha256=NGL0M3R08Za7qwAQJE3Ob_PqOkbPhclP4W75CmzimmQ,1357599
 phoenix/server/static/assets/vendor-BGzfc4EU.css,sha256=Nx5Lmx-bqYR7nsO_O4kEBcrJ8cwknWjZ6seHN3_s4UQ,3171
-phoenix/server/static/assets/vendor-arizeai-C6_oC0y8.js,sha256=meOakkfYcA4TSYjxbobysSn5Z8YdOw2EaBZP3iC977g,61555
-phoenix/server/static/assets/vendor-codemirror-DPnZGAZA.js,sha256=Ry5RV48MqAg_Q1PK0-URAbyinGajuQ728O71QBVD-u8,413334
-phoenix/server/static/assets/vendor-recharts-CjgSbsB0.js,sha256=lnJ2ZyzB_LTd0pLIp2fohK9v9UrzMLeY1TlzDIpJQok,231652
-phoenix/server/static/assets/vendor-shiki-CJyhDG0E.js,sha256=Aia7kJ1a_hjZkazx9ePQjwgI_eyjq-7geCiGHJDDnXw,305160
+phoenix/server/static/assets/vendor-D-csRHGZ.js,sha256=uYz5BUvg6gsB97JV0OZFu7rDn40gBkL3WYmrcQW5kgk,2676807
+phoenix/server/static/assets/vendor-arizeai-BJLCG_Gc.js,sha256=x0nzGpalauyWr0XeuvQBfvF6lkvs0Limu50IfT6LWeI,61555
+phoenix/server/static/assets/vendor-codemirror-Cr963DyP.js,sha256=1an8AWTBNoLFizv2VpW5c6t77gbG_YJB_lF93qyUKiQ,413334
+phoenix/server/static/assets/vendor-recharts-DgmPLgIp.js,sha256=Lo0yw1RW2uvp61KL4-uFd67ZTRpzCND7PRKNVFPmqww,231652
+phoenix/server/static/assets/vendor-shiki-wYOt1s7u.js,sha256=YoaKapH5k--O3pS9E1kUVfLxCFGKryoo9dDr2JjiXIs,305160
 phoenix/server/static/assets/vendor-three-BtCyLs1w.js,sha256=E6e1HbskKn61fWQPWmiZiPXXGHfZYn-30v0nofpDaqo,704132
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=_iKIyXEDDr5cTTnrUCjCd617U6Alc1k-IXtdKSt8g14,7215
@@ -480,9 +480,9 @@ phoenix/utilities/project.py,sha256=auVpARXkDb-JgeX5f2aStyFIkeKvGwN9l7qrFeJMVxI,
 phoenix/utilities/re.py,sha256=6YyUWIkv0zc2SigsxfOWIHzdpjKA_TZo2iqKq7zJKvw,2081
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/template_formatters.py,sha256=gh9PJD6WEGw7TEYXfSst1UR4pWWwmjxMLrDVQ_CkpkQ,2779
-arize_phoenix-12.7.1.dist-info/METADATA,sha256=EXOyuZzJpOBr9WhzzLVTFBPnApChI4TL4x_PRAtGnr8,35327
-arize_phoenix-12.7.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-12.7.1.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-12.7.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-12.7.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-12.7.1.dist-info/RECORD,,
+arize_phoenix-12.8.0.dist-info/METADATA,sha256=xRbO3Khh91zENiRVuhmmIJZywsa_498EzAAFN3WB9wY,35327
+arize_phoenix-12.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-12.8.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-12.8.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-12.8.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-12.8.0.dist-info/RECORD,,

phoenix/server/api/mutations/dataset_label_mutations.py

@@ -4,6 +4,8 @@ import sqlalchemy
 import strawberry
 from sqlalchemy import delete, select
 from sqlalchemy.exc import IntegrityError as PostgreSQLIntegrityError
+from sqlalchemy.orm import joinedload
+from sqlalchemy.sql import tuple_
 from sqlean.dbapi2 import IntegrityError as SQLiteIntegrityError  # type: ignore[import-untyped]
 from strawberry import UNSET
 from strawberry.relay.types import GlobalID
@@ -14,7 +16,7 @@ from phoenix.server.api.auth import IsLocked, IsNotReadOnly, IsNotViewer
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, Conflict, NotFound
 from phoenix.server.api.queries import Query
-from phoenix.server.api.types.Dataset import Dataset
+from phoenix.server.api.types.Dataset import Dataset, to_gql_dataset
 from phoenix.server.api.types.DatasetLabel import DatasetLabel, to_gql_dataset_label
 from phoenix.server.api.types.node import from_global_id_with_expected_type
 
@@ -24,11 +26,13 @@ class CreateDatasetLabelInput:
     name: str
     description: Optional[str] = UNSET
     color: str
+    dataset_ids: Optional[list[GlobalID]] = UNSET
 
 
 @strawberry.type
 class CreateDatasetLabelMutationPayload:
     dataset_label: DatasetLabel
+    datasets: list[Dataset]
 
 
 @strawberry.input
@@ -41,39 +45,16 @@ class DeleteDatasetLabelsMutationPayload:
     dataset_labels: list[DatasetLabel]
 
 
-@strawberry.input
-class UpdateDatasetLabelInput:
-    dataset_label_id: GlobalID
-    name: str
-    description: Optional[str] = None
-    color: str
-
-
-@strawberry.type
-class UpdateDatasetLabelMutationPayload:
-    dataset_label: DatasetLabel
-
-
 @strawberry.input
 class SetDatasetLabelsInput:
+    dataset_id: GlobalID
     dataset_label_ids: list[GlobalID]
-    dataset_ids: list[GlobalID]
 
 
 @strawberry.type
 class SetDatasetLabelsMutationPayload:
-    query: "Query"
-
-
-@strawberry.input
-class UnsetDatasetLabelsInput:
-    dataset_label_ids: list[GlobalID]
-    dataset_ids: list[GlobalID]
-
-
-@strawberry.type
-class UnsetDatasetLabelsMutationPayload:
-    query: "Query"
+    query: Query
+    dataset: Dataset
 
 
 @strawberry.type
@@ -87,50 +68,53 @@ class DatasetLabelMutationMixin:
         name = input.name
         description = input.description
         color = input.color
+        dataset_rowids: dict[
+            int, None
+        ] = {}  # use dictionary to de-duplicate while preserving order
+        if input.dataset_ids:
+            for dataset_id in input.dataset_ids:
+                try:
+                    dataset_rowid = from_global_id_with_expected_type(dataset_id, Dataset.__name__)
+                except ValueError:
+                    raise BadRequest(f"Invalid dataset ID: {dataset_id}")
+                dataset_rowids[dataset_rowid] = None
+
         async with info.context.db() as session:
             dataset_label_orm = models.DatasetLabel(name=name, description=description, color=color)
             session.add(dataset_label_orm)
             try:
-                await session.commit()
+                await session.flush()
             except (PostgreSQLIntegrityError, SQLiteIntegrityError):
                 raise Conflict(f"A dataset label named '{name}' already exists")
             except sqlalchemy.exc.StatementError as error:
                 raise BadRequest(str(error.orig))
-        return CreateDatasetLabelMutationPayload(
-            dataset_label=to_gql_dataset_label(dataset_label_orm)
-        )
-
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsNotViewer, IsLocked])  # type: ignore
-    async def update_dataset_label(
-        self, info: Info[Context, None], input: UpdateDatasetLabelInput
-    ) -> UpdateDatasetLabelMutationPayload:
-        if not input.name or not input.name.strip():
-            raise BadRequest("Dataset label name cannot be empty")
-
-        try:
-            dataset_label_id = from_global_id_with_expected_type(
-                input.dataset_label_id, DatasetLabel.__name__
-            )
-        except ValueError:
-            raise BadRequest(f"Invalid dataset label ID: {input.dataset_label_id}")
-
-        async with info.context.db() as session:
-            dataset_label_orm = await session.get(models.DatasetLabel, dataset_label_id)
-            if not dataset_label_orm:
-                raise NotFound(f"DatasetLabel with ID {input.dataset_label_id} not found")
-
-            dataset_label_orm.name = input.name.strip()
-            dataset_label_orm.description = input.description
-            dataset_label_orm.color = input.color.strip()
 
-            try:
+            datasets_by_id: dict[int, models.Dataset] = {}
+            if dataset_rowids:
+                datasets_by_id = {
+                    dataset.id: dataset
+                    for dataset in await session.scalars(
+                        select(models.Dataset).where(models.Dataset.id.in_(dataset_rowids.keys()))
+                    )
+                }
+                if len(datasets_by_id) < len(dataset_rowids):
+                    raise NotFound("One or more datasets not found")
+                session.add_all(
+                    [
+                        models.DatasetsDatasetLabel(
+                            dataset_id=dataset_rowid,
+                            dataset_label_id=dataset_label_orm.id,
+                        )
+                        for dataset_rowid in dataset_rowids
+                    ]
+                )
                 await session.commit()
-            except (PostgreSQLIntegrityError, SQLiteIntegrityError):
-                raise Conflict(f"A dataset label named '{input.name}' already exists")
-            except sqlalchemy.exc.StatementError as error:
-                raise BadRequest(str(error.orig))
-        return UpdateDatasetLabelMutationPayload(
-            dataset_label=to_gql_dataset_label(dataset_label_orm)
+
+        return CreateDatasetLabelMutationPayload(
+            dataset_label=to_gql_dataset_label(dataset_label_orm),
+            datasets=[
+                to_gql_dataset(datasets_by_id[dataset_rowid]) for dataset_rowid in dataset_rowids
+            ],
         )
 
     @strawberry.mutation(permission_classes=[IsNotReadOnly, IsNotViewer, IsLocked])  # type: ignore
@@ -170,122 +154,84 @@ class DatasetLabelMutationMixin:
     async def set_dataset_labels(
         self, info: Info[Context, None], input: SetDatasetLabelsInput
     ) -> SetDatasetLabelsMutationPayload:
-        if not input.dataset_ids:
-            raise BadRequest("No datasets provided.")
-        if not input.dataset_label_ids:
-            raise BadRequest("No dataset labels provided.")
-
-        unique_dataset_rowids: set[int] = set()
-        for dataset_gid in input.dataset_ids:
-            try:
-                dataset_rowid = from_global_id_with_expected_type(dataset_gid, Dataset.__name__)
-            except ValueError:
-                raise BadRequest(f"Invalid dataset ID: {dataset_gid}")
-            unique_dataset_rowids.add(dataset_rowid)
-        dataset_rowids = list(unique_dataset_rowids)
+        try:
+            dataset_id = from_global_id_with_expected_type(input.dataset_id, Dataset.__name__)
+        except ValueError:
+            raise BadRequest(f"Invalid dataset ID: {input.dataset_id}")
 
-        unique_dataset_label_rowids: set[int] = set()
+        dataset_label_ids: dict[
+            int, None
+        ] = {}  # use dictionary to de-duplicate while preserving order
         for dataset_label_gid in input.dataset_label_ids:
             try:
-                dataset_label_rowid = from_global_id_with_expected_type(
+                dataset_label_id = from_global_id_with_expected_type(
                     dataset_label_gid, DatasetLabel.__name__
                 )
             except ValueError:
                 raise BadRequest(f"Invalid dataset label ID: {dataset_label_gid}")
-            unique_dataset_label_rowids.add(dataset_label_rowid)
-        dataset_label_rowids = list(unique_dataset_label_rowids)
+            dataset_label_ids[dataset_label_id] = None
 
         async with info.context.db() as session:
-            existing_dataset_ids = (
-                await session.scalars(
-                    select(models.Dataset.id).where(models.Dataset.id.in_(dataset_rowids))
-                )
-            ).all()
-            if len(existing_dataset_ids) != len(dataset_rowids):
-                raise NotFound("One or more datasets not found")
+            dataset = await session.scalar(
+                select(models.Dataset)
+                .where(models.Dataset.id == dataset_id)
+                .options(joinedload(models.Dataset.datasets_dataset_labels))
+            )
 
-            existing_dataset_label_ids = (
+            if not dataset:
+                raise NotFound(f"Dataset with ID {input.dataset_id} not found")
+
+            existing_label_ids = (
                 await session.scalars(
                     select(models.DatasetLabel.id).where(
-                        models.DatasetLabel.id.in_(dataset_label_rowids)
+                        models.DatasetLabel.id.in_(dataset_label_ids.keys())
                     )
                 )
             ).all()
-            if len(existing_dataset_label_ids) != len(dataset_label_rowids):
+            if len(existing_label_ids) != len(dataset_label_ids):
                 raise NotFound("One or more dataset labels not found")
 
-            existing_dataset_label_keys = await session.execute(
-                select(
-                    models.DatasetsDatasetLabel.dataset_id,
-                    models.DatasetsDatasetLabel.dataset_label_id,
-                ).where(
-                    models.DatasetsDatasetLabel.dataset_id.in_(dataset_rowids)
-                    & models.DatasetsDatasetLabel.dataset_label_id.in_(dataset_label_rowids)
-                )
-            )
-            unique_dataset_label_keys = set(existing_dataset_label_keys.all())
+            previously_applied_dataset_label_ids = {
+                dataset_dataset_label.dataset_label_id
+                for dataset_dataset_label in dataset.datasets_dataset_labels
+            }
 
-            datasets_dataset_labels = []
-            for dataset_rowid in dataset_rowids:
-                for dataset_label_rowid in dataset_label_rowids:
-                    if (dataset_rowid, dataset_label_rowid) in unique_dataset_label_keys:
-                        continue
-                    datasets_dataset_labels.append(
-                        models.DatasetsDatasetLabel(
-                            dataset_id=dataset_rowid,
-                            dataset_label_id=dataset_label_rowid,
+            datasets_dataset_labels_to_add = [
+                models.DatasetsDatasetLabel(
+                    dataset_id=dataset_id,
+                    dataset_label_id=dataset_label_id,
+                )
+                for dataset_label_id in dataset_label_ids
+                if dataset_label_id not in previously_applied_dataset_label_ids
+            ]
+            if datasets_dataset_labels_to_add:
+                session.add_all(datasets_dataset_labels_to_add)
+                await session.flush()
+
+            datasets_dataset_labels_to_delete = [
+                dataset_dataset_label
+                for dataset_dataset_label in dataset.datasets_dataset_labels
+                if dataset_dataset_label.dataset_label_id not in dataset_label_ids
+            ]
+            if datasets_dataset_labels_to_delete:
+                await session.execute(
+                    delete(models.DatasetsDatasetLabel).where(
+                        tuple_(
+                            models.DatasetsDatasetLabel.dataset_id,
+                            models.DatasetsDatasetLabel.dataset_label_id,
+                        ).in_(
+                            [
+                                (
+                                    datasets_dataset_labels.dataset_id,
+                                    datasets_dataset_labels.dataset_label_id,
+                                )
+                                for datasets_dataset_labels in datasets_dataset_labels_to_delete
+                            ]
                         )
                     )
-            session.add_all(datasets_dataset_labels)
-
-            if datasets_dataset_labels:
-                try:
-                    await session.commit()
-                except (PostgreSQLIntegrityError, SQLiteIntegrityError) as e:
-                    raise Conflict("Failed to add dataset labels to datasets.") from e
-
-        return SetDatasetLabelsMutationPayload(
-            query=Query(),
-        )
-
-    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsNotViewer, IsLocked])  # type: ignore
-    async def unset_dataset_labels(
-        self, info: Info[Context, None], input: UnsetDatasetLabelsInput
-    ) -> UnsetDatasetLabelsMutationPayload:
-        if not input.dataset_ids:
-            raise BadRequest("No datasets provided.")
-        if not input.dataset_label_ids:
-            raise BadRequest("No dataset labels provided.")
-
-        unique_dataset_rowids: set[int] = set()
-        for dataset_gid in input.dataset_ids:
-            try:
-                dataset_rowid = from_global_id_with_expected_type(dataset_gid, Dataset.__name__)
-            except ValueError:
-                raise BadRequest(f"Invalid dataset ID: {dataset_gid}")
-            unique_dataset_rowids.add(dataset_rowid)
-        dataset_rowids = list(unique_dataset_rowids)
-
-        unique_dataset_label_rowids: set[int] = set()
-        for dataset_label_gid in input.dataset_label_ids:
-            try:
-                dataset_label_rowid = from_global_id_with_expected_type(
-                    dataset_label_gid, DatasetLabel.__name__
-                )
-            except ValueError:
-                raise BadRequest(f"Invalid dataset label ID: {dataset_label_gid}")
-            unique_dataset_label_rowids.add(dataset_label_rowid)
-        dataset_label_rowids = list(unique_dataset_label_rowids)
-
-        async with info.context.db() as session:
-            await session.execute(
-                delete(models.DatasetsDatasetLabel).where(
-                    models.DatasetsDatasetLabel.dataset_id.in_(dataset_rowids)
-                    & models.DatasetsDatasetLabel.dataset_label_id.in_(dataset_label_rowids)
                 )
-            )
-            await session.commit()
 
-        return UnsetDatasetLabelsMutationPayload(
+        return SetDatasetLabelsMutationPayload(
+            dataset=to_gql_dataset(dataset),
             query=Query(),
         )

phoenix/server/api/queries.py

@@ -188,7 +188,17 @@ class Query:
     async def generative_models(
         self,
         info: Info[Context, None],
-    ) -> list[GenerativeModel]:
+        first: Optional[int] = 50,
+        last: Optional[int] = UNSET,
+        after: Optional[CursorString] = UNSET,
+        before: Optional[CursorString] = UNSET,
+    ) -> Connection[GenerativeModel]:
+        args = ConnectionArgs(
+            first=first,
+            after=after if isinstance(after, CursorString) else None,
+            last=last,
+            before=before if isinstance(before, CursorString) else None,
+        )
         async with info.context.db() as session:
             result = await session.scalars(
                 select(models.GenerativeModel)
@@ -200,8 +210,8 @@ class Query:
                 )
                 .options(joinedload(models.GenerativeModel.token_prices))
             )
-
-        return [to_gql_generative_model(model) for model in result.unique()]
+            data = [to_gql_generative_model(model) for model in result.unique()]
+        return connection_from_list(data=data, args=args)
 
     @strawberry.field
     async def playground_models(self, input: Optional[ModelsInput] = None) -> list[PlaygroundModel]:
@@ -1191,7 +1201,9 @@ class Query:
             before=before if isinstance(before, CursorString) else None,
         )
         async with info.context.db() as session:
-            dataset_labels = await session.scalars(select(models.DatasetLabel))
+            dataset_labels = await session.scalars(
+                select(models.DatasetLabel).order_by(models.DatasetLabel.name.asc())
+            )
         data = [to_gql_dataset_label(dataset_label) for dataset_label in dataset_labels]
         return connection_from_list(data=data, args=args)
 

phoenix/server/cost_tracking/model_cost_manifest.json

@@ -784,7 +784,7 @@
           "token_type": "output"
         },
         {
-          "base_rate": 7.5e-8,
+          "base_rate": 3e-8,
           "is_prompt": true,
           "token_type": "cache_read"
         },
@@ -1027,7 +1027,7 @@
           "token_type": "output"
         },
         {
-          "base_rate": 3.125e-7,
+          "base_rate": 1.25e-7,
           "is_prompt": true,
           "token_type": "cache_read"
         }

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

@@ -1,28 +1,32 @@
 {
-  "_components-BLK5vehh.js": {
-    "file": "assets/components-BLK5vehh.js",
+  "_components-Bem6_7MW.js": {
+    "file": "assets/components-Bem6_7MW.js",
     "name": "components",
     "imports": [
-      "_vendor-3BvTzoBp.js",
-      "_pages-DIVgyYyy.js",
-      "_vendor-arizeai-C6_oC0y8.js",
-      "_vendor-codemirror-DPnZGAZA.js",
+      "_vendor-D-csRHGZ.js",
+      "_pages-CEJgMVKU.js",
+      "_vendor-arizeai-BJLCG_Gc.js",
+      "_vendor-codemirror-Cr963DyP.js",
       "_vendor-three-BtCyLs1w.js"
     ]
   },
-  "_pages-DIVgyYyy.js": {
-    "file": "assets/pages-DIVgyYyy.js",
+  "_pages-CEJgMVKU.js": {
+    "file": "assets/pages-CEJgMVKU.js",
     "name": "pages",
     "imports": [
-      "_vendor-3BvTzoBp.js",
-      "_components-BLK5vehh.js",
-      "_vendor-arizeai-C6_oC0y8.js",
-      "_vendor-codemirror-DPnZGAZA.js",
-      "_vendor-recharts-CjgSbsB0.js"
+      "_vendor-D-csRHGZ.js",
+      "_components-Bem6_7MW.js",
+      "_vendor-arizeai-BJLCG_Gc.js",
+      "_vendor-codemirror-Cr963DyP.js",
+      "_vendor-recharts-DgmPLgIp.js"
     ]
   },
-  "_vendor-3BvTzoBp.js": {
-    "file": "assets/vendor-3BvTzoBp.js",
+  "_vendor-BGzfc4EU.css": {
+    "file": "assets/vendor-BGzfc4EU.css",
+    "src": "_vendor-BGzfc4EU.css"
+  },
+  "_vendor-D-csRHGZ.js": {
+    "file": "assets/vendor-D-csRHGZ.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-BtCyLs1w.js"
@@ -31,43 +35,39 @@
       "assets/vendor-BGzfc4EU.css"
     ]
   },
-  "_vendor-BGzfc4EU.css": {
-    "file": "assets/vendor-BGzfc4EU.css",
-    "src": "_vendor-BGzfc4EU.css"
-  },
-  "_vendor-arizeai-C6_oC0y8.js": {
-    "file": "assets/vendor-arizeai-C6_oC0y8.js",
+  "_vendor-arizeai-BJLCG_Gc.js": {
+    "file": "assets/vendor-arizeai-BJLCG_Gc.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-3BvTzoBp.js"
+      "_vendor-D-csRHGZ.js"
     ]
   },
-  "_vendor-codemirror-DPnZGAZA.js": {
-    "file": "assets/vendor-codemirror-DPnZGAZA.js",
+  "_vendor-codemirror-Cr963DyP.js": {
+    "file": "assets/vendor-codemirror-Cr963DyP.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-3BvTzoBp.js",
-      "_vendor-shiki-CJyhDG0E.js"
+      "_vendor-D-csRHGZ.js",
+      "_vendor-shiki-wYOt1s7u.js"
     ],
     "dynamicImports": [
-      "_vendor-shiki-CJyhDG0E.js",
-      "_vendor-shiki-CJyhDG0E.js",
-      "_vendor-shiki-CJyhDG0E.js"
+      "_vendor-shiki-wYOt1s7u.js",
+      "_vendor-shiki-wYOt1s7u.js",
+      "_vendor-shiki-wYOt1s7u.js"
     ]
   },
-  "_vendor-recharts-CjgSbsB0.js": {
-    "file": "assets/vendor-recharts-CjgSbsB0.js",
+  "_vendor-recharts-DgmPLgIp.js": {
+    "file": "assets/vendor-recharts-DgmPLgIp.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-3BvTzoBp.js"
+      "_vendor-D-csRHGZ.js"
     ]
   },
-  "_vendor-shiki-CJyhDG0E.js": {
-    "file": "assets/vendor-shiki-CJyhDG0E.js",
+  "_vendor-shiki-wYOt1s7u.js": {
+    "file": "assets/vendor-shiki-wYOt1s7u.js",
     "name": "vendor-shiki",
     "isDynamicEntry": true,
     "imports": [
-      "_vendor-3BvTzoBp.js"
+      "_vendor-D-csRHGZ.js"
     ]
   },
   "_vendor-three-BtCyLs1w.js": {
@@ -75,19 +75,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-BP0Shd90.js",
+    "file": "assets/index-NdiXbuNL.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-3BvTzoBp.js",
-      "_vendor-arizeai-C6_oC0y8.js",
-      "_pages-DIVgyYyy.js",
-      "_components-BLK5vehh.js",
+      "_vendor-D-csRHGZ.js",
+      "_vendor-arizeai-BJLCG_Gc.js",
+      "_pages-CEJgMVKU.js",
+      "_components-Bem6_7MW.js",
       "_vendor-three-BtCyLs1w.js",
-      "_vendor-codemirror-DPnZGAZA.js",
-      "_vendor-shiki-CJyhDG0E.js",
-      "_vendor-recharts-CjgSbsB0.js"
+      "_vendor-codemirror-Cr963DyP.js",
+      "_vendor-shiki-wYOt1s7u.js",
+      "_vendor-recharts-DgmPLgIp.js"
     ]
   }
 }