arize-phoenix 7.7.2__tar.gz → 7.8.1__tar.gz

{arize_phoenix-7.7.2 → arize_phoenix-7.8.1}/PKG-INFO

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

{arize_phoenix-7.7.2 → arize_phoenix-7.8.1}/src/phoenix/server/api/dataloaders/dataset_example_revisions.py

@@ -1,6 +1,6 @@
 from typing import Optional, Union
 
-from sqlalchemy import and_, case, func, null, or_, select
+from sqlalchemy import Integer, case, func, or_, select, union
 from sqlalchemy.sql.expression import literal
 from strawberry.dataloader import DataLoader
 from typing_extensions import TypeAlias
@@ -20,74 +20,50 @@ class DatasetExampleRevisionsDataLoader(DataLoader[Key, Result]):
     def __init__(self, db: DbSessionFactory) -> None:
         super().__init__(
             load_fn=self._load_fn,
-            max_batch_size=200,  # needed to prevent the size of the query from getting too large
+            # Setting max_batch_size to prevent the size of the query from getting too large.
+            # The maximum number of terms is SQLITE_MAX_COMPOUND_SELECT which defaults to 500.
+            # This is needed because of the compound select query below used in transferring
+            # the input data to the database. SQLite in fact has better ways to transfer data,
+            # but unfortunately they're not made available in sqlalchemy yet.
+            max_batch_size=200,
         )
         self._db = db
 
     async def _load_fn(self, keys: list[Key]) -> list[Union[Result, NotFound]]:
-        example_and_version_ids = tuple(
-            set(
-                (example_id, version_id)
-                for example_id, version_id in keys
-                if version_id is not None
-            )
-        )
-        versionless_example_ids = tuple(
-            set(example_id for example_id, version_id in keys if version_id is None)
-        )
-        resolved_example_and_version_ids = (
-            (
+        # sqlalchemy has limited SQLite support for VALUES, so use UNION ALL instead.
+        # For details, see https://github.com/sqlalchemy/sqlalchemy/issues/7228
+        keys_subquery = union(
+            *(
                 select(
-                    models.DatasetExample.id.label("example_id"),
-                    models.DatasetVersion.id.label("version_id"),
-                )
-                .select_from(models.DatasetExample)
-                .join(
-                    models.DatasetVersion,
-                    onclause=literal(True),  # cross join
-                )
-                .where(
-                    or_(
-                        *(
-                            and_(
-                                models.DatasetExample.id == example_id,
-                                models.DatasetVersion.id == version_id,
-                            )
-                            for example_id, version_id in example_and_version_ids
-                        )
-                    )
+                    literal(example_id, Integer).label("example_id"),
+                    literal(version_id, Integer).label("version_id"),
                 )
+                for example_id, version_id in keys
             )
-            .union(
-                select(
-                    models.DatasetExample.id.label("example_id"), null().label("version_id")
-                ).where(models.DatasetExample.id.in_(versionless_example_ids))
-            )
-            .subquery()
-        )
+        ).subquery()
         revision_ids = (
             select(
-                resolved_example_and_version_ids.c.example_id,
-                resolved_example_and_version_ids.c.version_id,
+                keys_subquery.c.example_id,
+                keys_subquery.c.version_id,
                 func.max(models.DatasetExampleRevision.id).label("revision_id"),
             )
-            .select_from(resolved_example_and_version_ids)
+            .select_from(keys_subquery)
             .join(
                 models.DatasetExampleRevision,
-                onclause=resolved_example_and_version_ids.c.example_id
+                onclause=keys_subquery.c.example_id
                 == models.DatasetExampleRevision.dataset_example_id,
             )
             .where(
                 or_(
-                    resolved_example_and_version_ids.c.version_id.is_(None),
-                    models.DatasetExampleRevision.dataset_version_id
-                    <= resolved_example_and_version_ids.c.version_id,
+                    # This query gets the latest `revision_id` for each example:
+                    # - If `version_id` is NOT given, it finds the maximum `revision_id`.
+                    # - If `version_id` is given, it finds the highest `revision_id` whose
+                    #   `version_id` is less than or equal to the one specified.
+                    keys_subquery.c.version_id.is_(None),
+                    models.DatasetExampleRevision.dataset_version_id <= keys_subquery.c.version_id,
                 )
             )
-            .group_by(
-                resolved_example_and_version_ids.c.example_id,
-                resolved_example_and_version_ids.c.version_id,
-            )
+            .group_by(keys_subquery.c.example_id, keys_subquery.c.version_id)
         ).subquery()
         query = (
             select(

{arize_phoenix-7.7.2 → arize_phoenix-7.8.1}/src/phoenix/server/api/helpers/playground_clients.py

@@ -466,7 +466,6 @@ class OpenAIBaseStreamingClient(PlaygroundStreamingClient):
         "gpt-3.5-turbo-0125",
         "gpt-3.5-turbo",
         "gpt-3.5-turbo-1106",
-        "gpt-3.5-turbo-instruct",
     ],
 )
 class OpenAIStreamingClient(OpenAIBaseStreamingClient):

{arize_phoenix-7.7.2 → arize_phoenix-7.8.1}/src/phoenix/server/static/.vite/manifest.json

@@ -1,22 +1,22 @@
 {
-  "_components-Cy9nwIvF.js": {
-    "file": "assets/components-Cy9nwIvF.js",
+  "_components-Dn-PoB4S.js": {
+    "file": "assets/components-Dn-PoB4S.js",
     "name": "components",
     "imports": [
       "_vendor-DvC8cT4X.js",
-      "_pages-CUi2xCVQ.js",
+      "_pages-DRhPpbPi.js",
       "_vendor-arizeai-Do1793cv.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-three-DwGkEfCM.js"
     ]
   },
-  "_pages-CUi2xCVQ.js": {
-    "file": "assets/pages-CUi2xCVQ.js",
+  "_pages-DRhPpbPi.js": {
+    "file": "assets/pages-DRhPpbPi.js",
     "name": "pages",
     "imports": [
       "_vendor-DvC8cT4X.js",
       "_vendor-arizeai-Do1793cv.js",
-      "_components-Cy9nwIvF.js",
+      "_components-Dn-PoB4S.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-recharts-_Jb7JjhG.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-BKvHIxkk.js",
+    "file": "assets/index-Cky21rQb.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-DvC8cT4X.js",
       "_vendor-arizeai-Do1793cv.js",
-      "_pages-CUi2xCVQ.js",
-      "_components-Cy9nwIvF.js",
+      "_pages-DRhPpbPi.js",
+      "_components-Dn-PoB4S.js",
       "_vendor-three-DwGkEfCM.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-shiki-Cl9QBraO.js",