arize-phoenix 4.12.1rc1__py3-none-any.whl → 4.15.0__py3-none-any.whl

phoenix/utilities/client.py

@@ -0,0 +1,116 @@
+import warnings
+from typing import Any
+
+import httpx
+
+PHOENIX_SERVER_VERSION_HEADER = "x-phoenix-server-version"
+
+
+class VersionedClient(httpx.Client):
+    """
+    A httpx.Client wrapper that warns if there is a server/client version mismatch.
+    """
+
+    def __init__(self, *args: Any, **kwargs: Any):
+        from phoenix import __version__ as phoenix_version
+
+        super().__init__(*args, **kwargs)
+        self._client_phoenix_version = phoenix_version
+        self._warned_on_minor_version_mismatch = False
+
+    def _check_version(self, response: httpx.Response) -> None:
+        server_version = response.headers.get(PHOENIX_SERVER_VERSION_HEADER)
+
+        if server_version is None:
+            warnings.warn(
+                "The Phoenix server has an unknown version and may have compatibility issues."
+            )
+            return
+
+        try:
+            client_major, client_minor, client_patch = map(
+                int, self._client_phoenix_version.split(".")
+            )
+            server_major, server_minor, server_patch = map(int, server_version.split("."))
+            if abs(server_major - client_major) >= 1:
+                warnings.warn(
+                    f"⚠️⚠️ The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are severely mismatched. Upgrade "
+                    " either the client or server to ensure API compatibility ⚠️⚠️"
+                )
+            elif (
+                abs(server_minor - client_minor) >= 1 and not self._warned_on_minor_version_mismatch
+            ):
+                self._warned_on_minor_version_mismatch = True
+                warnings.warn(
+                    f"The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are mismatched and may have "
+                    "compatibility issues."
+                )
+        except ValueError:
+            # if either the client or server version includes a suffix e.g. "rc1", check for an
+            # exact version match of the version string
+            if self._client_phoenix_version != server_version:
+                warnings.warn(
+                    f"The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are mismatched and may have "
+                    "compatibility issues."
+                )
+
+    def request(self, *args: Any, **kwargs: Any) -> httpx.Response:
+        response = super().request(*args, **kwargs)
+        self._check_version(response)
+        return response
+
+
+class VersionedAsyncClient(httpx.AsyncClient):
+    """
+    A httpx.Client wrapper that warns if there is a server/client version mismatch.
+    """
+
+    def __init__(self, *args: Any, **kwargs: Any):
+        from phoenix import __version__ as phoenix_version
+
+        super().__init__(*args, **kwargs)
+        self._client_phoenix_version = phoenix_version
+        self._warned_on_minor_version_mismatch = False
+
+    def _check_version(self, response: httpx.Response) -> None:
+        server_version = response.headers.get(PHOENIX_SERVER_VERSION_HEADER)
+
+        if server_version is None:
+            return
+
+        try:
+            client_major, client_minor, client_patch = map(
+                int, self._client_phoenix_version.split(".")
+            )
+            server_major, server_minor, server_patch = map(int, server_version.split("."))
+            if abs(server_major - client_major) >= 1:
+                warnings.warn(
+                    f"⚠️⚠️ The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are severely mismatched. Upgrade "
+                    " either the client or server to ensure API compatibility ⚠️⚠️"
+                )
+            elif (
+                abs(server_minor - client_minor) >= 1 and not self._warned_on_minor_version_mismatch
+            ):
+                self._warned_on_minor_version_mismatch = True
+                warnings.warn(
+                    f"The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are mismatched and may have "
+                    "compatibility issues."
+                )
+        except ValueError:
+            # if the version includes a suffix e.g. "rc1", check for an exact version match
+            if self._client_phoenix_version != server_version:
+                warnings.warn(
+                    f"The Phoenix server ({server_version}) and client "
+                    f"({self._client_phoenix_version}) versions are mismatched and may have "
+                    "compatibility issues."
+                )
+
+    async def request(self, *args: Any, **kwargs: Any) -> httpx.Response:
+        response = await super().request(*args, **kwargs)
+        self._check_version(response)
+        return response

phoenix/utilities/project.py

@@ -9,5 +9,5 @@ from phoenix.config import DEFAULT_PROJECT_NAME
 def get_project_name(attributes: Iterable[KeyValue]) -> str:
     for kv in attributes:
         if kv.key == ResourceAttributes.PROJECT_NAME and (v := kv.value.string_value):
-            return v
+            return str(v)
     return DEFAULT_PROJECT_NAME

phoenix/version.py

@@ -1 +1 @@
-__version__ = "4.12.1rc1"
+__version__ = "4.15.0"

phoenix/server/api/routers/v1/dataset_examples.py

@@ -1,157 +0,0 @@
-from datetime import datetime
-from typing import Any, Dict, List, Optional
-
-from fastapi import APIRouter, HTTPException, Path, Query
-from sqlalchemy import and_, func, select
-from starlette.requests import Request
-from starlette.status import HTTP_404_NOT_FOUND
-from strawberry.relay import GlobalID
-
-from phoenix.db.models import (
-    Dataset as ORMDataset,
-)
-from phoenix.db.models import (
-    DatasetExample as ORMDatasetExample,
-)
-from phoenix.db.models import (
-    DatasetExampleRevision as ORMDatasetExampleRevision,
-)
-from phoenix.db.models import (
-    DatasetVersion as ORMDatasetVersion,
-)
-
-from .pydantic_compat import V1RoutesBaseModel
-from .utils import ResponseBody, add_errors_to_responses
-
-router = APIRouter(tags=["datasets"])
-
-
-class DatasetExample(V1RoutesBaseModel):
-    id: str
-    input: Dict[str, Any]
-    output: Dict[str, Any]
-    metadata: Dict[str, Any]
-    updated_at: datetime
-
-
-class ListDatasetExamplesData(V1RoutesBaseModel):
-    dataset_id: str
-    version_id: str
-    examples: List[DatasetExample]
-
-
-class ListDatasetExamplesResponseBody(ResponseBody[ListDatasetExamplesData]):
-    pass
-
-
-@router.get(
-    "/datasets/{id}/examples",
-    operation_id="getDatasetExamples",
-    summary="Get examples from a dataset",
-    responses=add_errors_to_responses([HTTP_404_NOT_FOUND]),
-)
-async def get_dataset_examples(
-    request: Request,
-    id: str = Path(description="The ID of the dataset"),
-    version_id: Optional[str] = Query(
-        default=None,
-        description=(
-            "The ID of the dataset version " "(if omitted, returns data from the latest version)"
-        ),
-    ),
-) -> ListDatasetExamplesResponseBody:
-    dataset_gid = GlobalID.from_id(id)
-    version_gid = GlobalID.from_id(version_id) if version_id else None
-
-    if (dataset_type := dataset_gid.type_name) != "Dataset":
-        raise HTTPException(
-            detail=f"ID {dataset_gid} refers to a {dataset_type}", status_code=HTTP_404_NOT_FOUND
-        )
-
-    if version_gid and (version_type := version_gid.type_name) != "DatasetVersion":
-        raise HTTPException(
-            detail=f"ID {version_gid} refers to a {version_type}", status_code=HTTP_404_NOT_FOUND
-        )
-
-    async with request.app.state.db() as session:
-        if (
-            resolved_dataset_id := await session.scalar(
-                select(ORMDataset.id).where(ORMDataset.id == int(dataset_gid.node_id))
-            )
-        ) is None:
-            raise HTTPException(
-                detail=f"No dataset with id {dataset_gid} can be found.",
-                status_code=HTTP_404_NOT_FOUND,
-            )
-
-        # Subquery to find the maximum created_at for each dataset_example_id
-        # timestamp tiebreaks are resolved by the largest id
-        partial_subquery = select(
-            func.max(ORMDatasetExampleRevision.id).label("max_id"),
-        ).group_by(ORMDatasetExampleRevision.dataset_example_id)
-
-        if version_gid:
-            if (
-                resolved_version_id := await session.scalar(
-                    select(ORMDatasetVersion.id).where(
-                        and_(
-                            ORMDatasetVersion.dataset_id == resolved_dataset_id,
-                            ORMDatasetVersion.id == int(version_gid.node_id),
-                        )
-                    )
-                )
-            ) is None:
-                raise HTTPException(
-                    detail=f"No dataset version with id {version_id} can be found.",
-                    status_code=HTTP_404_NOT_FOUND,
-                )
-            # if a version_id is provided, filter the subquery to only include revisions from that
-            partial_subquery = partial_subquery.filter(
-                ORMDatasetExampleRevision.dataset_version_id <= resolved_version_id
-            )
-        else:
-            if (
-                resolved_version_id := await session.scalar(
-                    select(func.max(ORMDatasetVersion.id)).where(
-                        ORMDatasetVersion.dataset_id == resolved_dataset_id
-                    )
-                )
-            ) is None:
-                raise HTTPException(
-                    detail="Dataset has no versions.",
-                    status_code=HTTP_404_NOT_FOUND,
-                )
-
-        subquery = partial_subquery.subquery()
-        # Query for the most recent example revisions that are not deleted
-        query = (
-            select(ORMDatasetExample, ORMDatasetExampleRevision)
-            .join(
-                ORMDatasetExampleRevision,
-                ORMDatasetExample.id == ORMDatasetExampleRevision.dataset_example_id,
-            )
-            .join(
-                subquery,
-                (subquery.c.max_id == ORMDatasetExampleRevision.id),
-            )
-            .filter(ORMDatasetExample.dataset_id == resolved_dataset_id)
-            .filter(ORMDatasetExampleRevision.revision_kind != "DELETE")
-            .order_by(ORMDatasetExample.id.asc())
-        )
-        examples = [
-            DatasetExample(
-                id=str(GlobalID("DatasetExample", str(example.id))),
-                input=revision.input,
-                output=revision.output,
-                metadata=revision.metadata_,
-                updated_at=revision.created_at,
-            )
-            async for example, revision in await session.stream(query)
-        ]
-    return ListDatasetExamplesResponseBody(
-        data=ListDatasetExamplesData(
-            dataset_id=str(GlobalID("Dataset", str(resolved_dataset_id))),
-            version_id=str(GlobalID("DatasetVersion", str(resolved_version_id))),
-            examples=examples,
-        )
-    )

phoenix/server/static/assets/index-BEKPzgQs.js

@@ -1,100 +0,0 @@
-import{r as d,j as e,d2 as F,v as s,F as P,R as v,w as E,aN as L,d3 as R,d4 as S,d5 as a,d6 as w,d7 as z,b as A,d8 as j}from"./vendor-CUDAPm8e.js";import{Q as C,j as k,Y as _,S as $,t as I,a4 as O}from"./vendor-arizeai-Do2HOmcL.js";import{aY as D,d as T,R as N,aZ as G,a_ as M}from"./components-C8sm_r1F.js";import{L as B,E as U,h as Y,M as q,a as m,D as J,d as K,b as W,e as H,P as Q,c as V,f as Z,p as X,g as u,i as ee,j as re,k as g,l as ae,m as h,n as f,o as oe,q as te,r as ne,s as se,t as le,A as ie}from"./pages-bN7juCjh.js";import"./vendor-three-DwGkEfCM.js";import"./vendor-codemirror-CrdxOlMs.js";import"./vendor-recharts-PKRvByVe.js";(function(){const n=document.createElement("link").relList;if(n&&n.supports&&n.supports("modulepreload"))return;for(const o of document.querySelectorAll('link[rel="modulepreload"]'))c(o);new MutationObserver(o=>{for(const t of o)if(t.type==="childList")for(const l of t.addedNodes)l.tagName==="LINK"&&l.rel==="modulepreload"&&c(l)}).observe(document,{childList:!0,subtree:!0});function i(o){const t={};return o.integrity&&(t.integrity=o.integrity),o.referrerPolicy&&(t.referrerPolicy=o.referrerPolicy),o.crossOrigin==="use-credentials"?t.credentials="include":o.crossOrigin==="anonymous"?t.credentials="omit":t.credentials="same-origin",t}function c(o){if(o.ep)return;o.ep=!0;const t=i(o);fetch(o.href,t)}})();const x="arize-phoenix-feature-flags",p={__CLEAR__:!1};function ce(){const r=localStorage.getItem(x);if(!r)return p;try{const n=JSON.parse(r);return Object.assign({},p,n)}catch{return p}}const b=d.createContext(null);function de(){const r=v.useContext(b);if(r===null)throw new Error("useFeatureFlags must be used within a FeatureFlagsProvider");return r}function pe(r){const[n,i]=d.useState(ce()),c=o=>{localStorage.setItem(x,JSON.stringify(o)),i(o)};return e(b.Provider,{value:{featureFlags:n,setFeatureFlags:c},children:e(me,{children:r.children})})}function me(r){const{children:n}=r,{featureFlags:i,setFeatureFlags:c}=de(),[o,t]=d.useState(!1);return F("ctrl+shift+f",()=>t(!0)),s(P,{children:[n,e($,{type:"modal",isDismissable:!0,onDismiss:()=>t(!1),children:o&&e(C,{title:"Feature Flags",children:e(k,{height:"size-1000",padding:"size-100",children:Object.keys(i).map(l=>e(_,{isSelected:i[l],onChange:y=>c({...i,[l]:y}),children:l},l))})})})]})}function ue(){return e(L,{styles:r=>E`
-        body {
-          background-color: var(--ac-global-color-grey-75);
-          color: var(--ac-global-text-color-900);
-          font-family: "Roboto";
-          font-size: ${r.typography.sizes.medium.fontSize}px;
-          margin: 0;
-          #root,
-          #root > div[data-overlay-container="true"],
-          #root > div[data-overlay-container="true"] > .ac-theme {
-            height: 100vh;
-          }
-        }
-
-        /* Remove list styling */
-        ul {
-          display: block;
-          list-style-type: none;
-          margin-block-start: none;
-          margin-block-end: 0;
-          padding-inline-start: 0;
-          margin-block-start: 0;
-        }
-
-        /* A reset style for buttons */
-        .button--reset {
-          background: none;
-          border: none;
-          padding: 0;
-        }
-        /* this css class is added to html via modernizr @see modernizr.js */
-        .no-hiddenscroll {
-          /* Works on Firefox */
-          * {
-            scrollbar-width: thin;
-            scrollbar-color: var(--ac-global-color-grey-300)
-              var(--ac-global-color-grey-400);
-          }
-
-          /* Works on Chrome, Edge, and Safari */
-          *::-webkit-scrollbar {
-            width: 14px;
-          }
-
-          *::-webkit-scrollbar-track {
-            background: var(--ac-global-color-grey-100);
-          }
-
-          *::-webkit-scrollbar-thumb {
-            background-color: var(--ac-global-color-grey-75);
-            border-radius: 8px;
-            border: 1px solid var(--ac-global-color-grey-300);
-          }
-        }
-
-        :root {
-          --px-blue-color: ${r.colors.arizeBlue};
-
-          --px-flex-gap-sm: ${r.spacing.margin4}px;
-          --px-flex-gap-sm: ${r.spacing.margin8}px;
-
-          --px-section-background-color: ${r.colors.gray500};
-
-          /* An item is a typically something in a list */
-          --px-item-background-color: ${r.colors.gray800};
-          --px-item-border-color: ${r.colors.gray600};
-
-          --px-spacing-sm: ${r.spacing.padding4}px;
-          --px-spacing-med: ${r.spacing.padding8}px;
-          --px-spacing-lg: ${r.spacing.padding16}px;
-
-          --px-border-radius-med: ${r.borderRadius.medium}px;
-
-          --px-font-size-sm: ${r.typography.sizes.small.fontSize}px;
-          --px-font-size-med: ${r.typography.sizes.medium.fontSize}px;
-          --px-font-size-lg: ${r.typography.sizes.large.fontSize}px;
-
-          --px-gradient-bar-height: 8px;
-
-          --px-nav-collapsed-width: 45px;
-          --px-nav-expanded-width: 200px;
-        }
-
-        .ac-theme--dark {
-          --px-primary-color: #9efcfd;
-          --px-primary-color--transparent: rgb(158, 252, 253, 0.2);
-          --px-reference-color: #baa1f9;
-          --px-reference-color--transparent: #baa1f982;
-          --px-corpus-color: #92969c;
-          --px-corpus-color--transparent: #92969c63;
-        }
-        .ac-theme--light {
-          --px-primary-color: #00add0;
-          --px-primary-color--transparent: rgba(0, 173, 208, 0.2);
-          --px-reference-color: #4500d9;
-          --px-reference-color--transparent: rgba(69, 0, 217, 0.2);
-          --px-corpus-color: #92969c;
-          --px-corpus-color--transparent: #92969c63;
-        }
-      `})}const ge=R(S(s(a,{path:"/",element:e(B,{}),errorElement:e(U,{}),children:[e(a,{index:!0,loader:Y}),s(a,{path:"/model",handle:{crumb:()=>"model"},element:e(q,{}),children:[e(a,{index:!0,element:e(m,{})}),e(a,{element:e(m,{}),children:e(a,{path:"dimensions",children:e(a,{path:":dimensionId",element:e(J,{}),loader:K})})}),e(a,{path:"embeddings",children:e(a,{path:":embeddingDimensionId",element:e(W,{}),loader:H,handle:{crumb:r=>r.embedding.name}})})]}),s(a,{path:"/projects",handle:{crumb:()=>"projects"},element:e(Q,{}),children:[e(a,{index:!0,element:e(V,{})}),s(a,{path:":projectId",element:e(Z,{}),loader:X,handle:{crumb:r=>r.project.name},children:[e(a,{index:!0,element:e(u,{})}),e(a,{element:e(u,{}),children:e(a,{path:"traces/:traceId",element:e(ee,{})})})]})]}),s(a,{path:"/datasets",handle:{crumb:()=>"datasets"},children:[e(a,{index:!0,element:e(re,{})}),s(a,{path:":datasetId",loader:g,handle:{crumb:r=>r.dataset.name},children:[s(a,{element:e(ae,{}),loader:g,children:[e(a,{index:!0,element:e(h,{}),loader:f}),e(a,{path:"experiments",element:e(h,{}),loader:f}),e(a,{path:"examples",element:e(oe,{}),loader:te,children:e(a,{path:":exampleId",element:e(ne,{})})})]}),e(a,{path:"compare",handle:{crumb:()=>"compare"},loader:se,element:e(le,{})})]})]}),e(a,{path:"/apis",element:e(ie,{}),handle:{crumb:()=>"APIs"}})]})),{basename:window.Config.basename});function he(){return e(w,{router:ge})}function fe(){return e(D,{children:e(xe,{})})}function xe(){const{theme:r}=T();return e(O,{theme:r,children:e(z,{theme:I,children:s(A.RelayEnvironmentProvider,{environment:N,children:[e(ue,{}),e(pe,{children:e(G,{children:e(d.Suspense,{children:e(M,{children:e(he,{})})})})})]})})})}const be=document.getElementById("root"),ye=j.createRoot(be);ye.render(e(fe,{}));