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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.8.0
+Version: 8.9.0
 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-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/config.py

@@ -86,6 +86,13 @@ ENV_PHOENIX_SQL_DATABASE_SCHEMA = "PHOENIX_SQL_DATABASE_SCHEMA"
 The schema to use for the PostgresSQL database. (This is ignored for SQLite.)
 See e.g. https://www.postgresql.org/docs/current/ddl-schemas.html
 """
+ENV_PHOENIX_DATABASE_ALLOCATED_STORAGE_CAPACITY_GIBIBYTES = (
+    "PHOENIX_DATABASE_ALLOCATED_STORAGE_CAPACITY_GIBIBYTES"
+)
+"""
+The allocated storage capacity for the Phoenix database in gibibytes (2^30 bytes). Use float for
+fractional value. This is currently used only by the UI for informational displays.
+"""
 ENV_PHOENIX_ENABLE_PROMETHEUS = "PHOENIX_ENABLE_PROMETHEUS"
 """
 Whether to enable Prometheus. Defaults to false.
@@ -690,6 +697,10 @@ def get_env_database_schema() -> Optional[str]:
     return getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA)
 
 
+def get_env_database_allocated_storage_capacity_gibibytes() -> Optional[float]:
+    return _float_val(ENV_PHOENIX_DATABASE_ALLOCATED_STORAGE_CAPACITY_GIBIBYTES)
+
+
 def get_env_enable_prometheus() -> bool:
     if (enable_promotheus := getenv(ENV_PHOENIX_ENABLE_PROMETHEUS)) is None or (
         enable_promotheus_lower := enable_promotheus.lower()

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

@@ -862,6 +862,27 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
         return content
 
 
+@register_llm_client(
+    provider_key=GenerativeProviderKey.ANTHROPIC,
+    model_names=[
+        "claude-3-7-sonnet-latest",
+        "claude-3-7-sonnet-20250219",
+    ],
+)
+class AnthropicReasoningStreamingClient(AnthropicStreamingClient):
+    @classmethod
+    def supported_invocation_parameters(cls) -> list[InvocationParameter]:
+        invocation_params = super().supported_invocation_parameters()
+        invocation_params.append(
+            JSONInvocationParameter(
+                invocation_name="thinking",
+                canonical_name=CanonicalParameterName.ANTHROPIC_EXTENDED_THINKING,
+                label="Thinking Budget",
+            )
+        )
+        return invocation_params
+
+
 @register_llm_client(
     provider_key=GenerativeProviderKey.GOOGLE,
     model_names=[

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/helpers/prompts/models.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 from enum import Enum
 from typing import Any, Literal, Mapping, Optional, Union
 
-from pydantic import BaseModel, ConfigDict, Field, RootModel
-from typing_extensions import Annotated, TypeAlias, TypeGuard, assert_never
+from pydantic import BaseModel, ConfigDict, Field, RootModel, model_validator
+from typing_extensions import Annotated, Self, TypeAlias, TypeGuard, assert_never
 
 from phoenix.db.types.model_provider import ModelProvider
 from phoenix.server.api.helpers.prompts.conversions.anthropic import AnthropicToolChoiceConversion
@@ -370,11 +370,32 @@ class PromptAzureOpenAIInvocationParameters(PromptModel):
     azure_openai: PromptAzureOpenAIInvocationParametersContent
 
 
+class PromptAnthropicThinkingConfigDisabled(PromptModel):
+    type: Literal["disabled"]
+
+
+class PromptAnthropicThinkingConfigEnabled(PromptModel):
+    type: Literal["enabled"]
+    budget_tokens: int = Field(..., ge=1024)
+
+
 class PromptAnthropicInvocationParametersContent(PromptModel):
     max_tokens: int
     temperature: float = UNDEFINED
     top_p: float = UNDEFINED
     stop_sequences: list[str] = UNDEFINED
+    thinking: Annotated[
+        Union[PromptAnthropicThinkingConfigDisabled, PromptAnthropicThinkingConfigEnabled],
+        Field(..., discriminator="type"),
+    ] = UNDEFINED
+
+    @model_validator(mode="after")
+    def check_thinking_budget_tokens_lt_max_tokens(self) -> Self:
+        if self.thinking is UNDEFINED:
+            return self
+        if self.thinking.type == "enabled" and self.thinking.budget_tokens >= self.max_tokens:
+            raise ValueError("The thinking budget must be less than max tokens.")
+        return self
 
 
 class PromptAnthropicInvocationParameters(PromptModel):

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/input_types/InvocationParameters.py

@@ -16,6 +16,7 @@ class CanonicalParameterName(str, Enum):
     TOOL_CHOICE = "tool_choice"
     RESPONSE_FORMAT = "response_format"
     REASONING_EFFORT = "reasoning_effort"
+    ANTHROPIC_EXTENDED_THINKING = "anthropic_extended_thinking"
 
 
 @strawberry.enum

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/mutations/prompt_label_mutations.py

@@ -11,6 +11,7 @@ from strawberry.types import Info
 
 from phoenix.db import models
 from phoenix.db.types.identifier import Identifier as IdentifierModel
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import Conflict, NotFound
 from phoenix.server.api.queries import Query
@@ -58,7 +59,7 @@ class PromptLabelMutationPayload:
 
 @strawberry.type
 class PromptLabelMutationMixin:
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_prompt_label(
         self, info: Info[Context, None], input: CreatePromptLabelInput
     ) -> PromptLabelMutationPayload:
@@ -77,7 +78,7 @@ class PromptLabelMutationMixin:
                 query=Query(),
             )
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_prompt_label(
         self, info: Info[Context, None], input: PatchPromptLabelInput
     ) -> PromptLabelMutationPayload:
@@ -106,7 +107,7 @@ class PromptLabelMutationMixin:
                 query=Query(),
             )
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
     async def delete_prompt_label(
         self, info: Info[Context, None], input: DeletePromptLabelInput
     ) -> PromptLabelMutationPayload:
@@ -130,7 +131,7 @@ class PromptLabelMutationMixin:
                 query=Query(),
             )
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def set_prompt_label(
         self, info: Info[Context, None], input: SetPromptLabelInput
     ) -> PromptLabelMutationPayload:
@@ -160,7 +161,7 @@ class PromptLabelMutationMixin:
                 query=Query(),
             )
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
     async def unset_prompt_label(
         self, info: Info[Context, None], input: UnsetPromptLabelInput
     ) -> PromptLabelMutationPayload:

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/mutations/prompt_mutations.py

@@ -13,6 +13,7 @@ from strawberry.types import Info
 from phoenix.db import models
 from phoenix.db.types.identifier import Identifier as IdentifierModel
 from phoenix.db.types.model_provider import ModelProvider
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, Conflict, NotFound
 from phoenix.server.api.helpers.prompts.models import (
@@ -74,7 +75,7 @@ class DeletePromptMutationPayload:
 
 @strawberry.type
 class PromptMutationMixin:
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_chat_prompt(
         self, info: Info[Context, None], input: CreateChatPromptInput
     ) -> Prompt:
@@ -141,7 +142,7 @@ class PromptMutationMixin:
                 raise Conflict(f"A prompt named '{input.name}' already exists")
         return to_gql_prompt_from_orm(prompt)
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def create_chat_prompt_version(
         self,
         info: Info[Context, None],
@@ -219,7 +220,7 @@ class PromptMutationMixin:
 
         return to_gql_prompt_from_orm(prompt)
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
     async def delete_prompt(
         self, info: Info[Context, None], input: DeletePromptInput
     ) -> DeletePromptMutationPayload:
@@ -236,7 +237,7 @@ class PromptMutationMixin:
             await session.commit()
         return DeletePromptMutationPayload(query=Query())
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def clone_prompt(self, info: Info[Context, None], input: ClonePromptInput) -> Prompt:
         prompt_id = from_global_id_with_expected_type(
             global_id=input.prompt_id, expected_type_name=Prompt.__name__
@@ -289,7 +290,7 @@ class PromptMutationMixin:
                 raise Conflict(f"A prompt named '{input.name}' already exists")
         return to_gql_prompt_from_orm(new_prompt)
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def patch_prompt(self, info: Info[Context, None], input: PatchPromptInput) -> Prompt:
         prompt_id = from_global_id_with_expected_type(
             global_id=input.prompt_id, expected_type_name=Prompt.__name__

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/mutations/prompt_version_tag_mutations.py

@@ -10,6 +10,7 @@ from strawberry.types import Info
 
 from phoenix.db import models
 from phoenix.db.types.identifier import Identifier as IdentifierModel
+from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, Conflict, NotFound
 from phoenix.server.api.queries import Query
@@ -41,7 +42,7 @@ class PromptVersionTagMutationPayload:
 
 @strawberry.type
 class PromptVersionTagMutationMixin:
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly])  # type: ignore
     async def delete_prompt_version_tag(
         self, info: Info[Context, None], input: DeletePromptVersionTagInput
     ) -> PromptVersionTagMutationPayload:
@@ -77,7 +78,7 @@ class PromptVersionTagMutationMixin:
                 prompt_version_tag=None, query=Query(), prompt=to_gql_prompt_from_orm(prompt)
             )
 
-    @strawberry.mutation
+    @strawberry.mutation(permission_classes=[IsNotReadOnly, IsLocked])  # type: ignore
     async def set_prompt_version_tag(
         self, info: Info[Context, None], input: SetPromptVersionTagInput
     ) -> PromptVersionTagMutationPayload:

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/queries.py

@@ -13,7 +13,11 @@ from strawberry.relay import Connection, GlobalID, Node
 from strawberry.types import Info
 from typing_extensions import Annotated, TypeAlias, assert_never
 
-from phoenix.config import ENV_PHOENIX_SQL_DATABASE_SCHEMA, getenv
+from phoenix.config import (
+    ENV_PHOENIX_SQL_DATABASE_SCHEMA,
+    get_env_database_allocated_storage_capacity_gibibytes,
+    getenv,
+)
 from phoenix.db import enums, models
 from phoenix.db.helpers import SupportedSQLDialect
 from phoenix.db.models import DatasetExample as OrmExample
@@ -789,6 +793,15 @@ class Query:
             clustered_events=clustered_events,
         )
 
+    @strawberry.field(
+        description="The allocated storage capacity of the database in bytes. "
+        "Return None if this information is unavailable.",
+    )  # type: ignore
+    async def db_storage_capacity_bytes(self) -> Optional[float]:
+        if gibibytes := get_env_database_allocated_storage_capacity_gibibytes():
+            return gibibytes * 2**30
+        return None
+
     @strawberry.field
     async def db_table_stats(
         self,

{arize_phoenix-8.8.0 → arize_phoenix-8.9.0}/src/phoenix/server/api/routers/v1/experiments.py

@@ -1,11 +1,13 @@
+import json
 from datetime import datetime
 from random import getrandbits
 from typing import Any, Optional
 
-from fastapi import APIRouter, HTTPException, Path
+from fastapi import APIRouter, HTTPException, Path, Response
 from pydantic import Field
-from sqlalchemy import select
+from sqlalchemy import and_, func, select
 from starlette.requests import Request
+from starlette.responses import PlainTextResponse
 from starlette.status import HTTP_404_NOT_FOUND
 from strawberry.relay import GlobalID
 
@@ -306,3 +308,102 @@ async def list_experiments(
         ]
 
         return ListExperimentsResponseBody(data=data)
+
+
+@router.get(
+    "/experiments/{experiment_id}/json",
+    operation_id="getExperimentJSON",
+    summary="Download experiment runs as a JSON file",
+    response_class=PlainTextResponse,
+    responses=add_errors_to_responses(
+        [
+            {"status_code": HTTP_404_NOT_FOUND, "description": "Experiment not found"},
+        ]
+    ),
+)
+async def get_experiment_jsonl(
+    request: Request,
+    response: Response,
+    experiment_id: str = Path(..., title="Experiment ID"),
+) -> str:
+    experiment_globalid = GlobalID.from_id(experiment_id)
+    try:
+        experiment_rowid = from_global_id_with_expected_type(experiment_globalid, "Experiment")
+    except ValueError:
+        raise HTTPException(
+            detail=f"Experiment with ID {experiment_globalid} does not exist",
+            status_code=HTTP_404_NOT_FOUND,
+        )
+
+    async with request.app.state.db() as session:
+        experiment = await session.get(models.Experiment, experiment_rowid)
+        if not experiment:
+            raise HTTPException(
+                detail=f"Experiment with ID {experiment_globalid} does not exist",
+                status_code=HTTP_404_NOT_FOUND,
+            )
+        revision_ids = (
+            select(func.max(models.DatasetExampleRevision.id))
+            .join(
+                models.DatasetExample,
+                models.DatasetExample.id == models.DatasetExampleRevision.dataset_example_id,
+            )
+            .where(
+                and_(
+                    models.DatasetExampleRevision.dataset_version_id
+                    <= experiment.dataset_version_id,
+                    models.DatasetExample.dataset_id == experiment.dataset_id,
+                )
+            )
+            .group_by(models.DatasetExampleRevision.dataset_example_id)
+            .scalar_subquery()
+        )
+        runs_and_revisions = (
+            await session.execute(
+                select(models.ExperimentRun, models.DatasetExampleRevision)
+                .join(
+                    models.DatasetExample,
+                    models.DatasetExample.id == models.ExperimentRun.dataset_example_id,
+                )
+                .join(
+                    models.DatasetExampleRevision,
+                    and_(
+                        models.DatasetExample.id
+                        == models.DatasetExampleRevision.dataset_example_id,
+                        models.DatasetExampleRevision.id.in_(revision_ids),
+                        models.DatasetExampleRevision.revision_kind != "DELETE",
+                    ),
+                )
+                .where(models.ExperimentRun.experiment_id == experiment_rowid)
+                .order_by(
+                    models.ExperimentRun.dataset_example_id, models.ExperimentRun.repetition_number
+                )
+            )
+        ).all()
+        if not runs_and_revisions:
+            raise HTTPException(
+                detail=f"Experiment with ID {experiment_globalid} has no runs",
+                status_code=HTTP_404_NOT_FOUND,
+            )
+    records = []
+    for run, revision in runs_and_revisions:
+        record = {
+            "example_id": str(
+                GlobalID(models.DatasetExample.__name__, str(run.dataset_example_id))
+            ),
+            "repetition_number": run.repetition_number,
+            "input": revision.input,
+            "reference_output": revision.output,
+            "output": run.output,
+            "error": run.error,
+            "latency_ms": run.latency_ms,
+            "start_time": run.start_time.isoformat(),
+            "end_time": run.end_time.isoformat(),
+            "trace_id": run.trace_id,
+            "prompt_token_count": run.prompt_token_count,
+            "completion_token_count": run.completion_token_count,
+        }
+        records.append(record)
+
+    response.headers["content-disposition"] = f'attachment; filename="{experiment.name}.json"'
+    return json.dumps(records, ensure_ascii=False, indent=2)

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

@@ -0,0 +1,87 @@
+{
+  "_components-C4AzQH6p.js": {
+    "file": "assets/components-C4AzQH6p.js",
+    "name": "components",
+    "imports": [
+      "_vendor-VJCVsFqd.js",
+      "_pages-DEHszGOs.js",
+      "_vendor-arizeai-CLxvyk4B.js",
+      "_vendor-codemirror-0mUqu36F.js",
+      "_vendor-three-C-AGeJYv.js"
+    ]
+  },
+  "_pages-DEHszGOs.js": {
+    "file": "assets/pages-DEHszGOs.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-VJCVsFqd.js",
+      "_vendor-arizeai-CLxvyk4B.js",
+      "_components-C4AzQH6p.js",
+      "_vendor-codemirror-0mUqu36F.js",
+      "_vendor-recharts-CfKQSOmq.js"
+    ]
+  },
+  "_vendor-Cg6lcjUC.css": {
+    "file": "assets/vendor-Cg6lcjUC.css",
+    "src": "_vendor-Cg6lcjUC.css"
+  },
+  "_vendor-VJCVsFqd.js": {
+    "file": "assets/vendor-VJCVsFqd.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-C-AGeJYv.js"
+    ],
+    "css": [
+      "assets/vendor-Cg6lcjUC.css"
+    ]
+  },
+  "_vendor-arizeai-CLxvyk4B.js": {
+    "file": "assets/vendor-arizeai-CLxvyk4B.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-VJCVsFqd.js"
+    ]
+  },
+  "_vendor-codemirror-0mUqu36F.js": {
+    "file": "assets/vendor-codemirror-0mUqu36F.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-VJCVsFqd.js",
+      "_vendor-shiki-BsgMOuir.js"
+    ]
+  },
+  "_vendor-recharts-CfKQSOmq.js": {
+    "file": "assets/vendor-recharts-CfKQSOmq.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-VJCVsFqd.js"
+    ]
+  },
+  "_vendor-shiki-BsgMOuir.js": {
+    "file": "assets/vendor-shiki-BsgMOuir.js",
+    "name": "vendor-shiki",
+    "imports": [
+      "_vendor-VJCVsFqd.js"
+    ]
+  },
+  "_vendor-three-C-AGeJYv.js": {
+    "file": "assets/vendor-three-C-AGeJYv.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-DYSOLesW.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-VJCVsFqd.js",
+      "_vendor-arizeai-CLxvyk4B.js",
+      "_pages-DEHszGOs.js",
+      "_components-C4AzQH6p.js",
+      "_vendor-three-C-AGeJYv.js",
+      "_vendor-codemirror-0mUqu36F.js",
+      "_vendor-shiki-BsgMOuir.js",
+      "_vendor-recharts-CfKQSOmq.js"
+    ]
+  }
+}