arize-phoenix 8.9.0__tar.gz → 8.11.0__tar.gz

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.9.0
+Version: 8.11.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -48,7 +48,7 @@ Requires-Dist: scipy
 Requires-Dist: sqlalchemy[asyncio]<3,>=2.0.4
 Requires-Dist: sqlean-py>=3.45.1
 Requires-Dist: starlette
-Requires-Dist: strawberry-graphql==0.253.1
+Requires-Dist: strawberry-graphql>=0.262.0
 Requires-Dist: tqdm
 Requires-Dist: typing-extensions>=4.6
 Requires-Dist: uvicorn
@@ -69,7 +69,7 @@ Requires-Dist: opentelemetry-sdk; extra == 'container'
 Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
-Requires-Dist: strawberry-graphql[opentelemetry]==0.253.1; extra == 'container'
+Requires-Dist: strawberry-graphql[opentelemetry]==0.262.0; extra == 'container'
 Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
@@ -100,7 +100,7 @@ Requires-Dist: pytest-postgresql; extra == 'dev'
 Requires-Dist: pytest-xdist; extra == 'dev'
 Requires-Dist: pytest==8.3.3; extra == 'dev'
 Requires-Dist: ruff==0.6.9; extra == 'dev'
-Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.253.1; extra == 'dev'
+Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.262.0; extra == 'dev'
 Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
   "starlette",
   "uvicorn",
   "psutil",
-  "strawberry-graphql==0.253.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql>=0.262.0",  # resolve security vulnerability https://github.com/Arize-ai/phoenix/security/dependabot/45
   "pyarrow",
   "typing-extensions>=4.6",
   "scipy",
@@ -88,7 +88,7 @@ dev = [
   "pytest-postgresql",
   "asyncpg",
   "psycopg[binary,pool]",
-  "strawberry-graphql[debug-server,opentelemetry]==0.253.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[debug-server,opentelemetry]==0.262.0",  # need to pin version because we're monkey-patching
   "pre-commit",
   "arize[AutoEmbeddings, LLM_Evaluation]",
   "llama-index>=0.10.3",
@@ -135,7 +135,7 @@ container = [
   "opentelemetry-instrumentation-sqlalchemy",
   "opentelemetry-instrumentation-grpc",
   "py-grpc-prometheus",
-  "strawberry-graphql[opentelemetry]==0.253.1",  # need to pin version because we're monkey-patching
+  "strawberry-graphql[opentelemetry]==0.262.0",  # need to pin version because we're monkey-patching
   "uvloop; platform_system != 'Windows'",
   "fast-hdbscan>=0.2.0",
   "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/src/phoenix/experiments/types.py

@@ -282,6 +282,8 @@ class ExperimentEvaluationRun:
 ExperimentTask: TypeAlias = Union[
     Callable[[Example], TaskOutput],
     Callable[[Example], Awaitable[TaskOutput]],
+    Callable[..., JSONSerializable],
+    Callable[..., Awaitable[JSONSerializable]],
 ]
 
 

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/src/phoenix/server/api/helpers/playground_clients.py

@@ -761,6 +761,13 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
         import anthropic.types as anthropic_types
 
         anthropic_messages, system_prompt = self._build_anthropic_messages(messages)
+        if (
+            "thinking" in invocation_parameters
+            and isinstance(invocation_parameters["thinking"], dict)
+            and "type" in invocation_parameters["thinking"]
+            and invocation_parameters["thinking"]["type"] == "enabled"
+        ):
+            invocation_parameters.pop("top_p", None)
         anthropic_params = {
             "messages": anthropic_messages,
             "model": self.model_name,
@@ -810,9 +817,9 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
                 elif isinstance(event, anthropic_streaming._types.CitationEvent):
                     raise NotImplementedError
                 elif isinstance(event, anthropic_streaming._types.ThinkingEvent):
-                    raise NotImplementedError
+                    pass
                 elif isinstance(event, anthropic_streaming._types.SignatureEvent):
-                    raise NotImplementedError
+                    pass
                 else:
                     assert_never(event)
 

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/src/phoenix/server/api/mutations/__init__.py

@@ -13,12 +13,14 @@ from phoenix.server.api.mutations.prompt_mutations import PromptMutationMixin
 from phoenix.server.api.mutations.prompt_version_tag_mutations import PromptVersionTagMutationMixin
 from phoenix.server.api.mutations.span_annotations_mutations import SpanAnnotationMutationMixin
 from phoenix.server.api.mutations.trace_annotations_mutations import TraceAnnotationMutationMixin
+from phoenix.server.api.mutations.trace_mutations import TraceMutationMixin
 from phoenix.server.api.mutations.user_mutations import UserMutationMixin
 
 
 @strawberry.type
 class Mutation(
     ApiKeyMutationMixin,
+    ChatCompletionMutationMixin,
     DatasetMutationMixin,
     ExperimentMutationMixin,
     ExportEventsMutationMixin,
@@ -28,7 +30,7 @@ class Mutation(
     PromptLabelMutationMixin,
     SpanAnnotationMutationMixin,
     TraceAnnotationMutationMixin,
+    TraceMutationMixin,
     UserMutationMixin,
-    ChatCompletionMutationMixin,
 ):
     pass

arize_phoenix-8.11.0/src/phoenix/server/api/mutations/trace_mutations.py

@@ -0,0 +1,74 @@
+import strawberry
+from sqlalchemy import and_, delete, not_, select
+from sqlalchemy.orm import load_only
+from sqlalchemy.sql import literal
+from strawberry.relay import GlobalID
+from strawberry.types import Info
+
+from phoenix.db import models
+from phoenix.server.api.auth import IsNotReadOnly
+from phoenix.server.api.context import Context
+from phoenix.server.api.exceptions import BadRequest
+from phoenix.server.api.queries import Query
+from phoenix.server.api.types.node import from_global_id_with_expected_type
+from phoenix.server.dml_event import SpanDeleteEvent
+
+
+@strawberry.type
+class TraceMutationMixin:
+    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
+    async def delete_traces(
+        self,
+        info: Info[Context, None],
+        trace_ids: list[GlobalID],
+    ) -> Query:
+        if not trace_ids:
+            raise BadRequest("Must provide at least one trace ID to delete")
+        trace_ids = list(set(trace_ids))
+        try:
+            trace_rowids = [
+                from_global_id_with_expected_type(global_id=id, expected_type_name="Trace")
+                for id in trace_ids
+            ]
+        except ValueError as error:
+            raise BadRequest(str(error))
+        async with info.context.db() as session:
+            traces = (
+                await session.scalars(
+                    delete(models.Trace)
+                    .where(models.Trace.id.in_(trace_rowids))
+                    .returning(models.Trace)
+                    .options(
+                        load_only(models.Trace.project_rowid, models.Trace.project_session_rowid)
+                    )
+                )
+            ).all()
+            if len(traces) < len(trace_rowids):
+                await session.rollback()
+                raise BadRequest("Invalid trace IDs provided")
+            project_ids = tuple(set(trace.project_rowid for trace in traces))
+            if len(project_ids) > 1:
+                await session.rollback()
+                raise BadRequest("Cannot delete traces from multiple projects")
+            session_ids = set(
+                session_id
+                for trace in traces
+                if (session_id := trace.project_session_rowid) is not None
+            )
+            if session_ids:
+                await session.execute(
+                    delete(models.ProjectSession).where(
+                        and_(
+                            models.ProjectSession.id.in_(session_ids),
+                            not_(
+                                select(literal(1))
+                                .where(
+                                    models.Trace.project_session_rowid == models.ProjectSession.id
+                                )
+                                .exists()
+                            ),
+                        )
+                    )
+                )
+            info.context.event_queue.put(SpanDeleteEvent(project_ids))
+        return Query()

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/src/phoenix/server/api/routers/v1/prompts.py

@@ -2,13 +2,13 @@ import logging
 from typing import Any, Optional, Union
 
 from fastapi import APIRouter, HTTPException, Path, Query
-from pydantic import ValidationError
+from pydantic import ValidationError, model_validator
 from sqlalchemy import select
 from sqlalchemy.sql import Select
 from starlette.requests import Request
 from starlette.status import HTTP_404_NOT_FOUND, HTTP_422_UNPROCESSABLE_ENTITY
 from strawberry.relay import GlobalID
-from typing_extensions import TypeAlias, assert_never
+from typing_extensions import Self, TypeAlias, assert_never
 
 from phoenix.db import models
 from phoenix.db.types.identifier import Identifier
@@ -52,6 +52,18 @@ class PromptVersionData(V1RoutesBaseModel):
     tools: Optional[PromptTools] = None
     response_format: Optional[PromptResponseFormat] = None
 
+    @model_validator(mode="after")
+    def check_template_type_match(self) -> Self:
+        if self.template_type is PromptTemplateType.CHAT:
+            if self.template.type == "chat":
+                return self
+        elif self.template_type is PromptTemplateType.STRING:
+            if self.template.type == "string":
+                return self
+        else:
+            assert_never(self.template_type)
+        raise ValueError("Template type does not match template")
+
 
 class PromptVersion(PromptVersionData):
     id: str

{arize_phoenix-8.9.0 → arize_phoenix-8.11.0}/src/phoenix/server/app.py

@@ -549,7 +549,7 @@ def create_graphql_router(
     read_only: bool = False,
     secret: Optional[str] = None,
     token_store: Optional[TokenStore] = None,
-) -> GraphQLRouter:  # type: ignore[type-arg]
+) -> GraphQLRouter[Context, None]:
     """Creates the GraphQL router.
 
     Args:

arize_phoenix-8.11.0/src/phoenix/server/static/.vite/manifest.json

@@ -0,0 +1,87 @@
+{
+  "_components-CWWuhGhD.js": {
+    "file": "assets/components-CWWuhGhD.js",
+    "name": "components",
+    "imports": [
+      "_vendor-BqKqAj6V.js",
+      "_pages-D99rPTo0.js",
+      "_vendor-arizeai-BuhF1q5O.js",
+      "_vendor-codemirror-D55VwBKY.js",
+      "_vendor-three-C-AGeJYv.js"
+    ]
+  },
+  "_pages-D99rPTo0.js": {
+    "file": "assets/pages-D99rPTo0.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-BqKqAj6V.js",
+      "_vendor-arizeai-BuhF1q5O.js",
+      "_components-CWWuhGhD.js",
+      "_vendor-codemirror-D55VwBKY.js",
+      "_vendor-recharts-CJPXurMK.js"
+    ]
+  },
+  "_vendor-BqKqAj6V.js": {
+    "file": "assets/vendor-BqKqAj6V.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-C-AGeJYv.js"
+    ],
+    "css": [
+      "assets/vendor-Cg6lcjUC.css"
+    ]
+  },
+  "_vendor-Cg6lcjUC.css": {
+    "file": "assets/vendor-Cg6lcjUC.css",
+    "src": "_vendor-Cg6lcjUC.css"
+  },
+  "_vendor-arizeai-BuhF1q5O.js": {
+    "file": "assets/vendor-arizeai-BuhF1q5O.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-BqKqAj6V.js"
+    ]
+  },
+  "_vendor-codemirror-D55VwBKY.js": {
+    "file": "assets/vendor-codemirror-D55VwBKY.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-BqKqAj6V.js",
+      "_vendor-shiki-c5S6DQ8L.js"
+    ]
+  },
+  "_vendor-recharts-CJPXurMK.js": {
+    "file": "assets/vendor-recharts-CJPXurMK.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-BqKqAj6V.js"
+    ]
+  },
+  "_vendor-shiki-c5S6DQ8L.js": {
+    "file": "assets/vendor-shiki-c5S6DQ8L.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-BqKqAj6V.js"
+    ]
+  },
+  "_vendor-three-C-AGeJYv.js": {
+    "file": "assets/vendor-three-C-AGeJYv.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-B_phsWge.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-BqKqAj6V.js",
+      "_vendor-arizeai-BuhF1q5O.js",
+      "_pages-D99rPTo0.js",
+      "_components-CWWuhGhD.js",
+      "_vendor-three-C-AGeJYv.js",
+      "_vendor-codemirror-D55VwBKY.js",
+      "_vendor-shiki-c5S6DQ8L.js",
+      "_vendor-recharts-CJPXurMK.js"
+    ]
+  }
+}