arize-phoenix 4.32.0__tar.gz → 4.33.1__tar.gz

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/.gitignore

@@ -28,3 +28,4 @@ pyrightconfig.json
 
 # Demo data
 tutorials/internal/demo_llama_index/*.json
+.env

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 4.32.0
+Version: 4.33.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -77,12 +77,13 @@ Requires-Dist: asyncpg; extra == 'dev'
 Requires-Dist: faker>=26.0.0; extra == 'dev'
 Requires-Dist: gcsfs; extra == 'dev'
 Requires-Dist: google-cloud-aiplatform>=1.3; extra == 'dev'
+Requires-Dist: grpc-interceptor[testing]; extra == 'dev'
 Requires-Dist: hatch; extra == 'dev'
 Requires-Dist: jupyter; extra == 'dev'
 Requires-Dist: langchain>=0.0.334; extra == 'dev'
 Requires-Dist: litellm>=1.0.3; extra == 'dev'
 Requires-Dist: llama-index>=0.10.3; extra == 'dev'
-Requires-Dist: mypy==1.11.1; extra == 'dev'
+Requires-Dist: mypy==1.11.2; extra == 'dev'
 Requires-Dist: nbqa; extra == 'dev'
 Requires-Dist: pandas-stubs==2.0.3.230814; (python_version < '3.9') and extra == 'dev'
 Requires-Dist: pandas-stubs==2.2.2.240603; (python_version >= '3.9') and extra == 'dev'
@@ -96,13 +97,13 @@ Requires-Dist: pytest-cov; extra == 'dev'
 Requires-Dist: pytest-postgresql; extra == 'dev'
 Requires-Dist: pytest-xdist; extra == 'dev'
 Requires-Dist: pytest==8.3.2; extra == 'dev'
-Requires-Dist: ruff==0.6.1; extra == 'dev'
+Requires-Dist: ruff==0.6.3; extra == 'dev'
 Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.236.0; extra == 'dev'
 Requires-Dist: tabulate; extra == 'dev'
-Requires-Dist: tox; extra == 'dev'
-Requires-Dist: tox-uv; extra == 'dev'
+Requires-Dist: tox-uv==1.11.2; extra == 'dev'
+Requires-Dist: tox==4.18.0; extra == 'dev'
 Requires-Dist: types-tabulate; extra == 'dev'
-Requires-Dist: uv; extra == 'dev'
+Requires-Dist: uv==0.4.2; extra == 'dev'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'dev'
 Provides-Extra: evals
 Provides-Extra: experimental
@@ -149,15 +150,17 @@ Description-Content-Type: text/markdown
 
 Phoenix is an open-source AI observability platform designed for experimentation, evaluation, and troubleshooting. It provides:
 
--   **_Tracing_** - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
--   **_Evaluation_** - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
--   **_Datasets_** -  Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
--   **_Experiments_** -  Track and evaluate changes to prompts, LLMs, and retrieval.
+- **_Tracing_** - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
+- **_Evaluation_** - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
+- **_Datasets_** - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
+- **_Experiments_** - Track and evaluate changes to prompts, LLMs, and retrieval.
 
 Phoenix is vendor and language agnostic with out-of-the-box support for popular frameworks (🦙LlamaIndex, 🦜⛓LangChain, Haystack, 🧩DSPy) and LLM providers (OpenAI, Bedrock, and more). For details on auto-instrumentation, check out the [OpenInference](https://github.com/Arize-ai/openinference) project.
 
 Phoenix runs practically anywhere, including your Jupyter notebook, local machine, containerized deployment, or in the cloud.
 
+<img alt="phoenix_overview.gif" width="100%" src="https://github.com/Arize-ai/phoenix-assets/blob/main/gifs/phoenix_overview.gif?raw=true">
+
 ## Installation
 
 Install Phoenix via `pip` or `conda`
@@ -172,13 +175,13 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 
 Join our community to connect with thousands of AI builders.
 
--   🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
--   💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
--   🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
--   🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
--   𝕏 Follow us on [𝕏](https://twitter.com/ArizePhoenix).
--   💌️ Sign up for our [mailing list](https://phoenix.arize.com/#updates).
--   🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
+- 🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
+- 💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
+- 🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
+- 🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
+- 𝕏 Follow us on [𝕏](https://twitter.com/ArizePhoenix).
+- 💌️ Sign up for our [mailing list](https://phoenix.arize.com/#updates).
+- 🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
 
 ## Breaking Changes
 

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/README.md

@@ -29,15 +29,17 @@
 
 Phoenix is an open-source AI observability platform designed for experimentation, evaluation, and troubleshooting. It provides:
 
--   **_Tracing_** - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
--   **_Evaluation_** - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
--   **_Datasets_** -  Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
--   **_Experiments_** -  Track and evaluate changes to prompts, LLMs, and retrieval.
+- **_Tracing_** - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
+- **_Evaluation_** - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
+- **_Datasets_** - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
+- **_Experiments_** - Track and evaluate changes to prompts, LLMs, and retrieval.
 
 Phoenix is vendor and language agnostic with out-of-the-box support for popular frameworks (🦙LlamaIndex, 🦜⛓LangChain, Haystack, 🧩DSPy) and LLM providers (OpenAI, Bedrock, and more). For details on auto-instrumentation, check out the [OpenInference](https://github.com/Arize-ai/openinference) project.
 
 Phoenix runs practically anywhere, including your Jupyter notebook, local machine, containerized deployment, or in the cloud.
 
+<img alt="phoenix_overview.gif" width="100%" src="https://github.com/Arize-ai/phoenix-assets/blob/main/gifs/phoenix_overview.gif?raw=true">
+
 ## Installation
 
 Install Phoenix via `pip` or `conda`
@@ -52,13 +54,13 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 
 Join our community to connect with thousands of AI builders.
 
--   🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
--   💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
--   🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
--   🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
--   𝕏 Follow us on [𝕏](https://twitter.com/ArizePhoenix).
--   💌️ Sign up for our [mailing list](https://phoenix.arize.com/#updates).
--   🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
+- 🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
+- 💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
+- 🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
+- 🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
+- 𝕏 Follow us on [𝕏](https://twitter.com/ArizePhoenix).
+- 💌️ Sign up for our [mailing list](https://phoenix.arize.com/#updates).
+- 🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
 
 ## Breaking Changes
 

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/pyproject.toml

@@ -70,13 +70,13 @@ dynamic = ["version"]
 dev = [
   "gcsfs",
   "hatch",
-  "tox",
-  "tox-uv",
-  "uv",
+  "tox==4.18.0",
+  "tox-uv==1.11.2",
+  "uv==0.4.2",
   "jupyter",
   "nbqa",
-  "ruff==0.6.1",
-  "mypy==1.11.1",
+  "ruff==0.6.3",
+  "mypy==1.11.2",
   "pandas>=1.0",
   "tabulate",  # used by DataFrame.to_markdown()
   "types-tabulate",
@@ -102,6 +102,7 @@ dev = [
   "Faker>=26.0.0",
   "portpicker",
   "uvloop; platform_system != 'Windows'",
+  "grpc-interceptor[testing]",
 ]
 evals = []
 experimental = []
@@ -190,7 +191,7 @@ dependencies = [
 
 [tool.hatch.envs.type]
 dependencies = [
-  "mypy==1.11.1",
+  "mypy==1.11.2",
   "tenacity",
   "pandas>=1.0",
   "pandas-stubs==2.0.3.230814",
@@ -225,7 +226,7 @@ python = ["3.8", "3.9", "3.12"]
 [tool.hatch.envs.style]
 detached = true
 dependencies = [
-  "ruff==0.6.1",
+  "ruff==0.6.3",
 ]
 
 [[tool.hatch.envs.style.matrix]]
@@ -259,6 +260,7 @@ addopts = [
   "--doctest-modules",
   "--new-first",
   "--showlocals",
+  "--exitfirst",
 ]
 testpaths = [
   "tests",

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/config.py

@@ -47,7 +47,11 @@ Phoenix supports two types of database URLs:
 Note that if you plan on using SQLite, it's advised to to use a persistent volume
 and simply point the PHOENIX_WORKING_DIR to that volume.
 """
-
+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_ENABLE_PROMETHEUS = "PHOENIX_ENABLE_PROMETHEUS"
 """
 Whether to enable Prometheus. Defaults to false.
@@ -281,6 +285,12 @@ def get_env_database_connection_str() -> str:
     return env_url
 
 
+def get_env_database_schema() -> Optional[str]:
+    if get_env_database_connection_str().startswith("sqlite"):
+        return None
+    return os.getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA)
+
+
 def get_env_enable_prometheus() -> bool:
     if (enable_promotheus := os.getenv(ENV_PHOENIX_ENABLE_PROMETHEUS)) is None or (
         enable_promotheus_lower := enable_promotheus.lower()

arize_phoenix-4.33.1/src/phoenix/db/README.md

@@ -0,0 +1,54 @@
+# Database
+
+This module is responsible for the database connection and the migrations.
+
+
+## Migrations
+
+All migrations are managed by Alembic. Migrations are applied to the database automatically when the application starts.
+
+
+### Manually Applying Migrations
+
+Sometimes, it's necessary to manually apply migrations, e.g., to recover from a failed migration. To manually apply migrations, you must first clone the Phoenix repository.
+
+```bash
+git clone https://github.com/Arize-ai/phoenix.git
+```
+
+Then navigate to the current directory.
+
+```bash
+cd phoenix/src/phoenix/db
+```
+
+If you are using a non-default SQL database URL (e.g., for running PostgreSQL), ensure your `PHOENIX_SQL_DATABASE_URL` is set. This is not needed if you are running Phoenix with the default SQLite URL.
+
+```bash
+export PHOENIX_SQL_DATABASE_URL=<sql-database-url>
+```
+
+
+To manually run up-migrations, run the following command:
+
+```bash
+alembic upgrade head
+```
+
+If the above command fails, it may be necessary to undo partially applied changes from a failed migration by first running down-migrations. This can be accomplished by identifying the ID of the migration revision you wish to return to. Revisions are defined [here](./migrations/versions/).
+
+⚠️ Running down-migrations can result in lost data. Only run down-migrations if you know what you are doing and consider backing up your database first. If you have any questions or doubts, contact the Phoenix team in the `#phoenix-support` channel of the [Arize AI Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q) or via GitHub.
+
+```bash
+alembic downgrade <revision-id>
+```
+
+### Creating a Migration
+
+All migrations are stored in the `migrations` folder. To create a new migration, run the following command:
+
+```bash
+alembic revision -m "your_revision_name"
+```
+
+Then fill the migration file with the necessary changes.

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/db/engines.py

@@ -3,7 +3,7 @@ import json
 from datetime import datetime
 from enum import Enum
 from sqlite3 import Connection
-from typing import Any
+from typing import Any, Callable
 
 import aiosqlite
 import numpy as np
@@ -13,6 +13,7 @@ from sqlalchemy import URL, StaticPool, event, make_url
 from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
 from typing_extensions import assert_never
 
+from phoenix.config import get_env_database_schema
 from phoenix.db.helpers import SupportedSQLDialect
 from phoenix.db.migrate import migrate_in_thread
 from phoenix.db.models import init_models
@@ -130,6 +131,15 @@ def aio_sqlite_engine(
     return engine
 
 
+def set_postgresql_search_path(schema: str) -> Callable[[Connection, Any], None]:
+    def _(connection: Connection, _: Any) -> None:
+        cursor = connection.cursor()
+        cursor.execute(f"CREATE SCHEMA IF NOT EXISTS {schema};")
+        cursor.execute(f"SET search_path TO {schema};")
+
+    return _
+
+
 def aio_postgresql_engine(
     url: URL,
     migrate: bool = True,
@@ -143,6 +153,8 @@ def aio_postgresql_engine(
         echo=Settings.log_migrations,
         json_serializer=_dumps,
     )
+    if schema := get_env_database_schema():
+        event.listen(sync_engine, "connect", set_postgresql_search_path(schema))
     migrate_in_thread(sync_engine)
     return engine
 

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/db/migrate.py

@@ -50,6 +50,7 @@ def migrate(
         alembic_cfg.set_main_option("sqlalchemy.url", url)
         alembic_cfg.attributes["connection"] = engine.connect()
         command.upgrade(alembic_cfg, "head")
+        engine.dispose()
         printif(log_migrations, "---------------------------")
         printif(log_migrations, "✅ Migrations complete.")
     except BaseException as e:

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/db/models.py

@@ -34,7 +34,7 @@ from sqlalchemy.orm import (
 )
 from sqlalchemy.sql import expression
 
-from phoenix.config import ENABLE_AUTH
+from phoenix.config import ENABLE_AUTH, get_env_database_schema
 from phoenix.datetime_utils import normalize_datetime
 
 
@@ -100,13 +100,14 @@ class Base(DeclarativeBase):
     # Enforce best practices for naming constraints
     # https://alembic.sqlalchemy.org/en/latest/naming.html#integration-of-naming-conventions-into-operations-autogenerate
     metadata = MetaData(
+        schema=get_env_database_schema(),
         naming_convention={
             "ix": "ix_%(table_name)s_%(column_0_N_name)s",
             "uq": "uq_%(table_name)s_%(column_0_N_name)s",
             "ck": "ck_%(table_name)s_`%(constraint_name)s`",
             "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
             "pk": "pk_%(table_name)s",
-        }
+        },
     )
     type_annotation_map = {
         Dict[str, Any]: JsonDict,

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/experiments/functions.py

@@ -153,7 +153,7 @@ def run_experiment(
             experiment. Defaults to None.
         rate_limit_errors (Optional[BaseException | Sequence[BaseException]]): An exception or
             sequence of exceptions to adaptively throttle on. Defaults to None.
-        dry_run (bool | int): R the experiment in dry-run mode. When set, experiment results will
+        dry_run (bool | int): Run the experiment in dry-run mode. When set, experiment results will
             not be recorded in Phoenix. If True, the experiment will run on a random dataset
             example. If an integer, the experiment will run on a random sample of the dataset
             examples of the given size. Defaults to False.

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/average_experiment_run_latency.py

@@ -50,5 +50,5 @@ class AverageExperimentRunLatencyDataLoader(DataLoader[Key, Result]):
             }
         return [
             avg_latencies.get(experiment_id, ValueError(f"Unknown experiment: {experiment_id}"))
-            for experiment_id in experiment_ids
+            for experiment_id in keys
         ]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/dataset_example_spans.py

@@ -38,4 +38,4 @@ class DatasetExampleSpansDataLoader(DataLoader[Key, Result]):
                     )
                 )
             }
-        return [spans.get(example_id) for example_id in example_ids]
+        return [spans.get(example_id) for example_id in keys]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/experiment_annotation_summaries.py

@@ -79,5 +79,5 @@ class ExperimentAnnotationSummaryDataLoader(DataLoader[Key, Result]):
                 )
         return [
             sorted(summaries[experiment_id], key=lambda summary: summary.annotation_name)
-            for experiment_id in experiment_ids
+            for experiment_id in keys
         ]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/experiment_error_rates.py

@@ -57,5 +57,5 @@ class ExperimentErrorRatesDataLoader(DataLoader[Key, Result]):
             }
         return [
             error_rates.get(experiment_id, ValueError(f"Unknown experiment ID: {experiment_id}"))
-            for experiment_id in experiment_ids
+            for experiment_id in keys
         ]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/experiment_run_annotations.py

@@ -25,7 +25,7 @@ class ExperimentRunAnnotations(DataLoader[Key, Result]):
         self._db = db
 
     async def _load_fn(self, keys: List[Key]) -> List[Result]:
-        run_ids = tuple(set(keys))
+        run_ids = keys
         annotations: DefaultDict[Key, Result] = defaultdict(list)
         async with self._db() as session:
             async for run_id, annotation in await session.stream(
@@ -36,5 +36,5 @@ class ExperimentRunAnnotations(DataLoader[Key, Result]):
                 annotations[run_id].append(annotation)
         return [
             sorted(annotations[run_id], key=lambda annotation: annotation.name, reverse=True)
-            for run_id in run_ids
+            for run_id in keys
         ]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/experiment_run_counts.py

@@ -49,5 +49,5 @@ class ExperimentRunCountsDataLoader(DataLoader[Key, Result]):
             }
         return [
             run_counts.get(experiment_id, ValueError(f"Unknown experiment: {experiment_id}"))
-            for experiment_id in experiment_ids
+            for experiment_id in keys
         ]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/experiment_sequence_number.py

@@ -44,4 +44,4 @@ class ExperimentSequenceNumberDataLoader(DataLoader[Key, Result]):
                 experiment_id: sequence_number
                 async for experiment_id, sequence_number in await session.stream(stmt)
             }
-        return [result.get(experiment_id) for experiment_id in experiment_ids]
+        return [result.get(experiment_id) for experiment_id in keys]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/span_dataset_examples.py

@@ -33,4 +33,4 @@ class SpanDatasetExamplesDataLoader(DataLoader[Key, Result]):
                 .where(models.Span.id.in_(span_rowids))
             ):
                 dataset_examples[span_rowid].append(dataset_example)
-        return [dataset_examples.get(span_rowid, []) for span_rowid in span_rowids]
+        return [dataset_examples.get(span_rowid, []) for span_rowid in keys]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/dataloaders/span_projects.py

@@ -30,4 +30,4 @@ class SpanProjectsDataLoader(DataLoader[Key, Result]):
                     .where(models.Span.id.in_(span_ids))
                 )
             }
-        return [projects.get(span_id) or ValueError("Invalid span ID") for span_id in span_ids]
+        return [projects.get(span_id) or ValueError("Invalid span ID") for span_id in keys]

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/api/mutations/api_key_mutations.py

@@ -39,6 +39,7 @@ class DeleteApiKeyInput:
 @strawberry.type
 class DeleteSystemApiKeyMutationPayload:
     id: GlobalID
+    query: Query
 
 
 @strawberry.type
@@ -106,7 +107,7 @@ class ApiKeyMutationMixin:
 
             await session.delete(api_key)
 
-        return DeleteSystemApiKeyMutationPayload(id=input.id)
+        return DeleteSystemApiKeyMutationPayload(id=input.id, query=Query())
 
 
 def create_jwt(

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/app.py

@@ -134,6 +134,7 @@ If so, demo data is automatically ingested upon initial boot up to populate the
 NEW_DB_AGE_THRESHOLD_MINUTES = 2
 
 ProjectName: TypeAlias = str
+_Callback: TypeAlias = Callable[[], Union[None, Awaitable[None]]]
 
 
 class AppConfig(NamedTuple):
@@ -377,13 +378,16 @@ def _lifespan(
     dml_event_handler: DmlEventHandler,
     tracer_provider: Optional["TracerProvider"] = None,
     enable_prometheus: bool = False,
-    startup_callbacks: Iterable[Callable[[], None]] = (),
-    shutdown_callbacks: Iterable[Callable[[], None]] = (),
+    startup_callbacks: Iterable[_Callback] = (),
+    shutdown_callbacks: Iterable[_Callback] = (),
     read_only: bool = False,
     scaffolder_config: Optional[ScaffolderConfig] = None,
 ) -> StatefulLifespan[FastAPI]:
     @contextlib.asynccontextmanager
     async def lifespan(_: FastAPI) -> AsyncIterator[Dict[str, Any]]:
+        for callback in startup_callbacks:
+            if isinstance((res := callback()), Awaitable):
+                await res
         global DB_MUTEX
         DB_MUTEX = asyncio.Lock() if db.dialect is SupportedSQLDialect.SQLITE else None
         async with AsyncExitStack() as stack:
@@ -408,8 +412,6 @@ def _lifespan(
                     queue_evaluation=queue_evaluation,
                 )
                 await stack.enter_async_context(scaffolder)
-            for callback in startup_callbacks:
-                callback()
             yield {
                 "event_queue": dml_event_handler,
                 "enqueue": enqueue,
@@ -418,7 +420,8 @@ def _lifespan(
                 "enqueue_operation": enqueue_operation,
             }
         for callback in shutdown_callbacks:
-            callback()
+            if isinstance((res := callback()), Awaitable):
+                await res
 
     return lifespan
 
@@ -596,13 +599,13 @@ def create_app(
     initial_spans: Optional[Iterable[Union[Span, Tuple[Span, str]]]] = None,
     initial_evaluations: Optional[Iterable[pb.Evaluation]] = None,
     serve_ui: bool = True,
-    startup_callbacks: Iterable[Callable[[], None]] = (),
-    shutdown_callbacks: Iterable[Callable[[], None]] = (),
+    startup_callbacks: Iterable[_Callback] = (),
+    shutdown_callbacks: Iterable[_Callback] = (),
     secret: Optional[str] = None,
     scaffolder_config: Optional[ScaffolderConfig] = None,
 ) -> FastAPI:
-    startup_callbacks_list: List[Callable[[], None]] = list(startup_callbacks)
-    shutdown_callbacks_list: List[Callable[[], None]] = list(shutdown_callbacks)
+    startup_callbacks_list: List[_Callback] = list(startup_callbacks)
+    shutdown_callbacks_list: List[_Callback] = list(shutdown_callbacks)
     initial_batch_of_spans: Iterable[Tuple[Span, str]] = (
         ()
         if initial_spans is None
@@ -616,6 +619,7 @@ def create_app(
         CacheForDataLoaders() if db.dialect is SupportedSQLDialect.SQLITE else None
     )
     last_updated_at = LastUpdatedAt()
+    middlewares: List[Middleware] = [Middleware(HeadersMiddleware)]
     dml_event_handler = DmlEventHandler(
         db=db,
         cache_for_dataloaders=cache_for_dataloaders,
@@ -669,10 +673,7 @@ def create_app(
     if enable_prometheus:
         from phoenix.server.prometheus import PrometheusMiddleware
 
-        prometheus_middlewares = [Middleware(PrometheusMiddleware)]
-    else:
-        prometheus_middlewares = []
-
+        middlewares.append(Middleware(PrometheusMiddleware))
     app = FastAPI(
         title="Arize-Phoenix REST API",
         version=REST_API_VERSION,
@@ -687,10 +688,7 @@ def create_app(
             startup_callbacks=startup_callbacks_list,
             scaffolder_config=scaffolder_config,
         ),
-        middleware=[
-            Middleware(HeadersMiddleware),
-            *prometheus_middlewares,
-        ],
+        middleware=middlewares,
         exception_handlers={HTTPException: plain_text_http_exception_handler},
         debug=debug,
         swagger_ui_parameters={
@@ -703,7 +701,8 @@ def create_app(
     app.include_router(router)
     app.include_router(graphql_router)
     app.add_middleware(GZipMiddleware)
-    if serve_ui:
+    web_manifest_path = SERVER_DIR / "static" / ".vite" / "manifest.json"
+    if serve_ui and web_manifest_path.is_file():
         app.mount(
             "/",
             app=Static(
@@ -716,7 +715,7 @@ def create_app(
                     n_samples=umap_params.n_samples,
                     is_development=dev,
                     authentication_enabled=authentication_enabled,
-                    web_manifest_path=SERVER_DIR / "static" / ".vite" / "manifest.json",
+                    web_manifest_path=web_manifest_path,
                 ),
             ),
             name="static",

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/grpc_server.py

@@ -23,7 +23,7 @@ if TYPE_CHECKING:
 ProjectName: TypeAlias = str
 
 
-class Servicer(TraceServiceServicer):  # type:ignore
+class Servicer(TraceServiceServicer):  # type: ignore[misc,unused-ignore]
     def __init__(
         self,
         callback: Callable[[Span, ProjectName], Awaitable[None]],
@@ -78,7 +78,7 @@ class GrpcServer:
             interceptors=interceptors,
         )
         server.add_insecure_port(f"[::]:{get_env_grpc_port()}")
-        add_TraceServiceServicer_to_server(Servicer(self._callback), server)
+        add_TraceServiceServicer_to_server(Servicer(self._callback), server)  # type: ignore[no-untyped-call,unused-ignore]
         await server.start()
         self._server = server
 

{arize_phoenix-4.32.0 → arize_phoenix-4.33.1}/src/phoenix/server/main.py

@@ -11,6 +11,7 @@ from time import sleep, time
 from typing import List, Optional
 from urllib.parse import urljoin
 
+from jinja2 import BaseLoader, Environment
 from uvicorn import Config, Server
 
 import phoenix.trace.v1 as pb
@@ -18,6 +19,7 @@ from phoenix.config import (
     EXPORT_DIR,
     get_auth_settings,
     get_env_database_connection_str,
+    get_env_database_schema,
     get_env_enable_prometheus,
     get_env_grpc_port,
     get_env_host,
@@ -60,14 +62,14 @@ from phoenix.trace.schemas import Span
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())
 
-_WELCOME_MESSAGE = """
+_WELCOME_MESSAGE = Environment(loader=BaseLoader()).from_string("""
 
 ██████╗ ██╗  ██╗ ██████╗ ███████╗███╗   ██╗██╗██╗  ██╗
 ██╔══██╗██║  ██║██╔═══██╗██╔════╝████╗  ██║██║╚██╗██╔╝
 ██████╔╝███████║██║   ██║█████╗  ██╔██╗ ██║██║ ╚███╔╝
 ██╔═══╝ ██╔══██║██║   ██║██╔══╝  ██║╚██╗██║██║ ██╔██╗
 ██║     ██║  ██║╚██████╔╝███████╗██║ ╚████║██║██╔╝ ██╗
-╚═╝     ╚═╝  ╚═╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝╚═╝╚═╝  ╚═╝ v{version}
+╚═╝     ╚═╝  ╚═╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝╚═╝╚═╝  ╚═╝ v{{ version }}
 
 |
 |  🌎 Join our Community 🌎
@@ -80,12 +82,15 @@ _WELCOME_MESSAGE = """
 |  https://docs.arize.com/phoenix
 |
 |  🚀 Phoenix Server 🚀
-|  Phoenix UI: {ui_path}
+|  Phoenix UI: {{ ui_path }}
 |  Log traces:
-|    - gRPC: {grpc_path}
-|    - HTTP: {http_path}
-|  Storage: {storage}
-"""
+|    - gRPC: {{ grpc_path }}
+|    - HTTP: {{ http_path }}
+|  Storage: {{ storage }}
+{% if schema -%}
+|    - Schema: {{ schema }}
+{% endif -%}
+""")
 
 _EXPERIMENTAL_WARNING = """
 🚨 WARNING: Phoenix is running in experimental mode. 🚨
@@ -346,12 +351,13 @@ if __name__ == "__main__":
     )
     # Print information about the server
     root_path = urljoin(f"http://{host}:{port}", host_root_path)
-    msg = _WELCOME_MESSAGE.format(
+    msg = _WELCOME_MESSAGE.render(
         version=version("arize-phoenix"),
         ui_path=root_path,
         grpc_path=f"http://{host}:{get_env_grpc_port()}",
         http_path=urljoin(root_path, "v1/traces"),
         storage=get_printable_db_url(db_connection_str),
+        schema=get_env_database_schema(),
     )
     if authentication_enabled:
         msg += _EXPERIMENTAL_WARNING.format(auth_enabled=True)