arize-phoenix 4.32.0__py3-none-any.whl → 4.33.1__py3-none-any.whl

{arize_phoenix-4.32.0.dist-info → arize_phoenix-4.33.1.dist-info}/METADATA

@@ -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.dist-info → arize_phoenix-4.33.1.dist-info}/RECORD

@@ -1,25 +1,25 @@
 phoenix/__init__.py,sha256=TGNWqm2UW-l67yIRpOtmqGHVAmdoobSNqUsiTtip7uQ,1542
 phoenix/auth.py,sha256=ugvGZlseYX9NkpWaSqb8D2kzUBlAPqT45Dx5_VUANqk,1621
-phoenix/config.py,sha256=RfS-0kzEOhirX9EmVMr8cu7qCDtkq8HpxhhctEgh0Yo,10795
+phoenix/config.py,sha256=3ZvppWHfrSdeIG8sFGl6kEygqhbS9KlEqLQTphM3iq0,11197
 phoenix/datetime_utils.py,sha256=yDKjwX2Vtqw9h5F_ProtP-TsXidM43uIvmJ_pOzYc9A,3405
 phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=OyML4t2XGnlqF0JXA9_uccL8HslTABxep9Ci7MViKEU,5216
 phoenix/settings.py,sha256=cO-qgis_S27nHirTobYI9hHPfZH18R--WMmxNdsVUwc,273
-phoenix/version.py,sha256=_QUskFVbFLd7O3BXo2PEGlMc717xpkYLlK-r6lVv4_I,23
+phoenix/version.py,sha256=s4RWPFqvsL2oe4Cl03br7jJCnnt6JizPJJwAK64hlhw,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=km_a--PBHOuA337ClRw9xqhOHhrUT6Rl9pz_zV0JYkQ,4843
 phoenix/core/model_schema.py,sha256=F2dbbVnkDLsPYoyZDv1q03uhvP8LcU1wXp0g-exiWs0,50551
 phoenix/core/model_schema_adapter.py,sha256=0Tm_Y_gV-WED8fKBCaFXAEFwE3CTEZS1dowqnTZ7x7g,8426
-phoenix/db/README.md,sha256=IvKaZyf9ECbGBYYePaRhBveKZwDbxAc-c7BMxJYZh6Q,595
+phoenix/db/README.md,sha256=Mjrq0tG68uKQ81mN1Vmu3C795EEpZndmnE6CJXsPCXE,1909
 phoenix/db/__init__.py,sha256=pDjEFXukHmJBM-1D8RjmXkvLsz85YWNxMQczt81ec3A,118
 phoenix/db/alembic.ini,sha256=p8DjVqGUs_tTx8oU56JP7qj-rMUebNFizItUSv_hPhs,3763
 phoenix/db/bulk_inserter.py,sha256=qgg8pt5k4VnHKOE0-KoReXVAfXRhLt-sMZihI-b4X9I,12761
-phoenix/db/engines.py,sha256=l9Zl7mPd1q4RTXpThzYzc4Lo7TuuBwaGrC-zK0SMnn4,5300
+phoenix/db/engines.py,sha256=uWOZhIAspPCReFNqKtHsmZnY8H2DSi1kCOpQtmU3F1U,5793
 phoenix/db/helpers.py,sha256=2zSc4n5IJfu-CaOFoBfqTB35M1nTFcAc8tqLsNtF2Jw,3488
-phoenix/db/migrate.py,sha256=NNcci4LHw0wFR7U6quWrA-sw_A4h2lAA1_LePMLkb4w,2629
-phoenix/db/models.py,sha256=lYWJYtD2asQwKU1B8JKyteWpHVYjhr1j0tmZdf9CQ5Y,23686
+phoenix/db/migrate.py,sha256=emiwHiFXijMw6P6gAvLcAw3qTDOuI6wmblD7sEnFLa8,2654
+phoenix/db/models.py,sha256=o97p2H4LeX8YHkTWFsbSGQANW9jHWVf6mJvE9KyFybU,23754
 phoenix/db/insertion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/db/insertion/constants.py,sha256=8wifm7X-1XvroZ__R2Gc96NsgLhTDn0zXl4lehlXtcA,70
 phoenix/db/insertion/dataset.py,sha256=vBIroaSIWz5ZXE0Hul8osbebjS40gDCbsSP-YnazEwk,7212
@@ -40,7 +40,7 @@ phoenix/db/migrations/versions/10460e46d750_datasets.py,sha256=eZAyz720DmpOd7Rnu
 phoenix/db/migrations/versions/3be8647b87d8_add_token_columns_to_spans_table.py,sha256=bV7urfCPkrXhkyg8oaUfLYN093UoRecLQXPkXJ8SIOs,3666
 phoenix/db/migrations/versions/cf03bd6bae1d_init.py,sha256=09cpofqje8zi4eQFfUn-i21x7VcsUYOfLKKUlrtKrGc,8662
 phoenix/experiments/__init__.py,sha256=6JGwgUd7xCbGpuHqYZlsmErmYvVgv7N_j43bn3dUqsk,123
-phoenix/experiments/functions.py,sha256=B60_JT0jmLFQbkweZc2Px1q1vzE3Q_w2EI5jRgT19Rk,32589
+phoenix/experiments/functions.py,sha256=xxxBziRBdv45H8-LY3yow9XVs4bQLYMd_4Ks5ou4u84,32591
 phoenix/experiments/tracing.py,sha256=wVpt8Ie9WNPoi1djJdcrkwCokHdTO0bicXViLg3O-1Y,2831
 phoenix/experiments/types.py,sha256=HegeRSLyx2lESlzAc2PXYQJVHBFbCLwqQQmPcDMdUnM,23433
 phoenix/experiments/utils.py,sha256=wLu5Kvt1b4a8rGPRWq5G8RQ9XSiV8fCIVm51zWBI3-g,758
@@ -72,11 +72,11 @@ phoenix/pointcloud/pointcloud.py,sha256=4zAIkKs2xOUbchpj4XDAV-iPMXrfAJ15TG6rlIYG
 phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcBYPU,1057
 phoenix/pointcloud/umap_parameters.py,sha256=3UQSjrysVOvq2V4KNpTMqNqNiK0BsTZnPBHWZ4fyJtQ,1708
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=i5VhDdqPoQ8i5WlwjnUZAjedx8xOiL16R2OzATg9LQE,28333
+phoenix/server/app.py,sha256=15zHrHy0dl-RuBDapU45t6pNSS4mXg21gHVhnRE4GM4,28469
 phoenix/server/dml_event.py,sha256=MpjCFqljxvgb9OB5Cez9vJesb3oHb3XxXictynBfcis,2851
 phoenix/server/dml_event_handler.py,sha256=yU23-DDwXcL35p5EPwFW0oZh6mxQxJrJAQPKcYZYJz4,8310
-phoenix/server/grpc_server.py,sha256=jllxDNkpLQxDkvej4RhTokobowbvydF-SU8gSw1MTCc,3378
-phoenix/server/main.py,sha256=FnETeVWwcsnA4z67gHE9Br8BUCmW8cMQC-YpjCq-Zz0,14634
+phoenix/server/grpc_server.py,sha256=XWC2pjSefBBOhM6IXz0oQmjPtnonBvkG9ccUWEfHiiw,3446
+phoenix/server/main.py,sha256=169n7AIYKjcVw_Taw6dDRUBuVT3uqViehHOLSZGB4zE,14872
 phoenix/server/prometheus.py,sha256=j9DHB2fERuq_ZKmwVaqR-9wx5WcPPuU1Cm5Bhg5241Y,2996
 phoenix/server/telemetry.py,sha256=T_2OKrxNViAeaANlNspEekg_Y5uZIFWvKAnpz8Aoqvk,2762
 phoenix/server/thread_server.py,sha256=RwXQGP_QhGD7le6WB7xEygEEuwBl5Ck_Zo8xGIYGi9M,2135
@@ -90,25 +90,25 @@ phoenix/server/api/schema.py,sha256=4L2m6QXhaV13YPTZCEZ3hqCPQFHZOy3QnJVLRYQFzpg,
 phoenix/server/api/utils.py,sha256=Kl47G-1A7QKTDrc75BU2QK6HupsG6MWuXxy351FOfKQ,858
 phoenix/server/api/dataloaders/__init__.py,sha256=D1xEKunv8ZUBEPk2eStX-JhoPR3_NaQcNOp47H-oChM,3161
 phoenix/server/api/dataloaders/annotation_summaries.py,sha256=Wv8AORZoGd5TJ4Y-em8iqJu87AMpZP7lWOTr-SML-x8,5560
-phoenix/server/api/dataloaders/average_experiment_run_latency.py,sha256=q091UmkXx37OBKh7L-GJ5LXHyRXfX2w4XTk1NMHtPpw,1827
+phoenix/server/api/dataloaders/average_experiment_run_latency.py,sha256=mb9TXSrecIhJWQYi_JStWfNkh69izzXw0nAuDH3MbRM,1817
 phoenix/server/api/dataloaders/dataset_example_revisions.py,sha256=_X2CwCc6lAkD4VuwAiyo4DfSVS8uuzub05sKyKiEXm8,5149
-phoenix/server/api/dataloaders/dataset_example_spans.py,sha256=-TjdyyJv2c2JiN1OXu6MMmQ-BEKlHXucEDcuObeRVsU,1416
+phoenix/server/api/dataloaders/dataset_example_spans.py,sha256=2CzCV94_HyFTmm-PhsFxYMFo733c0yHEZRol0v8Fle8,1409
 phoenix/server/api/dataloaders/document_evaluation_summaries.py,sha256=5XOom2KRAmCwPmtlraiZOSl3vhfaW-eiiYkmetAEalw,5616
 phoenix/server/api/dataloaders/document_evaluations.py,sha256=V6sE34jON_qFxt7eArJbktykAsty-gnBZHlEkORcj0E,1296
 phoenix/server/api/dataloaders/document_retrieval_metrics.py,sha256=JqDqkUuoeG6WfcmWSrmQptfF6IPX8XgzYzyECXTAjgg,4202
-phoenix/server/api/dataloaders/experiment_annotation_summaries.py,sha256=qDEBRo0XKJMYBYlXBgJ-69jwPK-0r4pi9ZhbjC_vx6M,2813
-phoenix/server/api/dataloaders/experiment_error_rates.py,sha256=wWTFOO2UxGIYUu53nnzzLk04-mJxw-BQgJELA9gT5bY,1949
-phoenix/server/api/dataloaders/experiment_run_annotations.py,sha256=Ey42FYy6WIASWlO-KKwHewfvH_RBL0Ci3W3xrv4oyaE,1361
-phoenix/server/api/dataloaders/experiment_run_counts.py,sha256=8VlKRaXwD56-7q_uonRr7L6QlEkyEQubkFKmKx6vuz0,1661
-phoenix/server/api/dataloaders/experiment_sequence_number.py,sha256=hSpKyOz3ES4UTfOzOfHeQ1ZXat-bzcdKybVSbYmhrBw,1563
+phoenix/server/api/dataloaders/experiment_annotation_summaries.py,sha256=gW0rvM4WTEXmZoxVBGn32oEVloaQVN-ycKVGWkXpSbM,2803
+phoenix/server/api/dataloaders/experiment_error_rates.py,sha256=9lVppbGoPZDhqtEvQu6p0M8-Xl0OPzpA-us1jKUe7y4,1939
+phoenix/server/api/dataloaders/experiment_run_annotations.py,sha256=O2M6HE_UF0Y1Ko9LnYXE-q68w_3IRhauqjqUHyJFdcE,1346
+phoenix/server/api/dataloaders/experiment_run_counts.py,sha256=r8aeDQdX-4ME5jFumW_jkszlTlj84iM98Zcy0dIqFSY,1651
+phoenix/server/api/dataloaders/experiment_sequence_number.py,sha256=yF8AmCY7DpiDN18A5bLs9Pk0HX3f3DmH5IT1Vl3AE0Q,1553
 phoenix/server/api/dataloaders/latency_ms_quantile.py,sha256=5Y2OQ_GeH1My2573eOm7zPbqpwDZ_WeMLoZMq3KDoNQ,7403
 phoenix/server/api/dataloaders/min_start_or_max_end_times.py,sha256=k82jVllBWkGI6xu2MAltqxdIqyw3DqoVCQcKbi-JsmM,2834
 phoenix/server/api/dataloaders/project_by_name.py,sha256=9p6DW_zXrsB9YGILFV0wkRwiapyHbZ1IgWK9HuHxl30,1105
 phoenix/server/api/dataloaders/record_counts.py,sha256=64OsyiQRDZASlibpQAXtRkzyA_ch89A3TvhSUByxoTI,4215
 phoenix/server/api/dataloaders/span_annotations.py,sha256=v3vGkLGTn-n4khxXZY2btgvfXitvytWgBahzrrWEz0I,1078
-phoenix/server/api/dataloaders/span_dataset_examples.py,sha256=BtLZp11fyyeaWGGBPZj2StzFM0m5jxt52zB2nFMVybo,1306
+phoenix/server/api/dataloaders/span_dataset_examples.py,sha256=06QqLBJYEHNxrDs3IIjBDlmg-oF7o76QAI9jzLFdPEg,1299
 phoenix/server/api/dataloaders/span_descendants.py,sha256=b7jGTn0Hi22gv2yskloLnf3BG3upS9z5hnKLMT9Sxac,2094
-phoenix/server/api/dataloaders/span_projects.py,sha256=LbQWiboCFqq4CHS18OzvRUwL9yORqP26fh5p7JbpFdg,1244
+phoenix/server/api/dataloaders/span_projects.py,sha256=QhlIyUJdVa29dGOuolyXFRm5WEcLz4mJMM217PwNLp8,1240
 phoenix/server/api/dataloaders/token_counts.py,sha256=6gDVely8BYiCBdmiq1ECO0lMChUYPIlsZbB34rmL1xM,4684
 phoenix/server/api/dataloaders/trace_row_ids.py,sha256=RODX4NULlBzMxHMrsq0dp1ij6ZlLH4ZzQBnafGdxOvU,1100
 phoenix/server/api/dataloaders/cache/__init__.py,sha256=SYoOM9n8FJaMdQarma5d1blu-jIg2GB8Shqg5ezSzZ8,106
@@ -145,7 +145,7 @@ phoenix/server/api/input_types/TraceAnnotationSort.py,sha256=BzwiUnMh2VsgQYnhDlb
 phoenix/server/api/input_types/UserRoleInput.py,sha256=xxhFe0ITZOgRVEJbVem_W6F1Ip_H6xDENdQqMMx-kKE,129
 phoenix/server/api/input_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/mutations/__init__.py,sha256=JS3WRqYxNeoaLsKjODFvJnZb6CF19IFW-lfOsUq3rtM,1074
-phoenix/server/api/mutations/api_key_mutations.py,sha256=5GDsP0gURgXokbajc3GJEor3AGt6BGd1IMWMg054E7U,4683
+phoenix/server/api/mutations/api_key_mutations.py,sha256=TjwPrVeiu7fP0bqAD1yO99aJ1P8TrndpwCgCwrwX6tU,4715
 phoenix/server/api/mutations/auth.py,sha256=8o4tTfGCPkpUauuB9ijPH84Od77UX_UrQWfmUsnujI4,524
 phoenix/server/api/mutations/auth_mutations.py,sha256=XLCxmsjyVp1riGWxUhVUOiGhIFs_ZmOfZM77vs_RLDw,2182
 phoenix/server/api/mutations/dataset_mutations.py,sha256=8S6qjmraSBxA7ioNogTQPp6q27ZdvdAn6yt0Z4fmOI0,27096
@@ -242,10 +242,10 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
-phoenix/server/static/.vite/manifest.json,sha256=2equEiHXZGCivKbAQc9U9NATOUVltsQoiM0lhdD4TZE,1929
-phoenix/server/static/assets/components-Ddr-5VpM.js,sha256=kKqWpR-S-axVFd3tD_bfc3OD9HEleq0Pdn_cofQuzDo,244415
-phoenix/server/static/assets/index-BvNNAF8n.js,sha256=-keFoN2koW7IT6Iv_kbUpDDqv-TkpjSccukpYvuI8yo,7602
-phoenix/server/static/assets/pages-CU_QMYRP.js,sha256=iiwCelLTc7KsQeRr5mRW0lnDd8fNFPwWo4NjSKp6J2A,511373
+phoenix/server/static/.vite/manifest.json,sha256=THlBb4tvLxAwoqYJ3NRH1WqVcwqp7KQ_TQ7qpiw5zoY,1929
+phoenix/server/static/assets/components-r5FGhZ3q.js,sha256=NWjiMcYyfM7SuQkd0zqPnbp-7qNI_-qOK1MQAs6lLmI,248754
+phoenix/server/static/assets/index-D97e4SiJ.js,sha256=0HZXgZL-N_L80RFq-qfyBtqLhObO7ovKt3qdJaHxcME,7602
+phoenix/server/static/assets/pages-Dtpf5TRQ.js,sha256=LEPj9rfWhwQPlT4ZF9SNnrbut5n0hmDJMh_x7-q-8w4,515005
 phoenix/server/static/assets/vendor-Dpio9q2e.js,sha256=1CMQ_vGa7qQyJoLj1WdL23CVQ6y9E94cv1cy_KHNi84,1435263
 phoenix/server/static/assets/vendor-DxkFTwjz.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
 phoenix/server/static/assets/vendor-arizeai-DDzwX9A6.js,sha256=ePHP1b8n-zwCMGEwgfR00aOTSEe8Ogp8gx3FyZ1-SyA,304401
@@ -296,8 +296,8 @@ phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,
 phoenix/utilities/project.py,sha256=8IJuMM4yUMoooPi37sictGj8Etu9rGmq6RFtc9848cQ,436
 phoenix/utilities/re.py,sha256=PDve_OLjRTM8yQQJHC8-n3HdIONi7aNils3ZKRZ5uBM,2045
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize_phoenix-4.32.0.dist-info/METADATA,sha256=h2RQifXqJMNa4wFC16Cz8CsS4H_rRUxrmnb_tta0lmY,11741
-arize_phoenix-4.32.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-arize_phoenix-4.32.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-4.32.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-4.32.0.dist-info/RECORD,,
+arize_phoenix-4.33.1.dist-info/METADATA,sha256=rO1Dk5WxRQ8xT3nvGOm46PQ_6nO_EBZjhgqZgnngYYE,11938
+arize_phoenix-4.33.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+arize_phoenix-4.33.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-4.33.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-4.33.1.dist-info/RECORD,,

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()

phoenix/db/README.md

@@ -8,14 +8,40 @@ This module is responsible for the database connection and the migrations.
 All migrations are managed by Alembic. Migrations are applied to the database automatically when the application starts.
 
 
-### Applying Migrations
+### Manually Applying Migrations
 
-To manually apply the migrations, run the following command:
+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
 

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
 

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:

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,

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.

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
         ]

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]

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
         ]

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
         ]

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
         ]

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
         ]

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]

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]

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]

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(

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",

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