arize-phoenix 4.27.0__py3-none-any.whl → 4.29.0__py3-none-any.whl

{arize_phoenix-4.27.0.dist-info → arize_phoenix-4.29.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 4.27.0
+Version: 4.29.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -21,6 +21,7 @@ Requires-Dist: aioitertools
 Requires-Dist: aiosqlite
 Requires-Dist: alembic<2,>=1.3.0
 Requires-Dist: arize-phoenix-evals>=0.13.1
+Requires-Dist: arize-phoenix-otel>=0.4.1
 Requires-Dist: cachetools
 Requires-Dist: fastapi
 Requires-Dist: grpcio

{arize_phoenix-4.27.0.dist-info → arize_phoenix-4.29.0.dist-info}/RECORD

@@ -1,12 +1,12 @@
 phoenix/__init__.py,sha256=TGNWqm2UW-l67yIRpOtmqGHVAmdoobSNqUsiTtip7uQ,1542
-phoenix/auth.py,sha256=N8vTFmc5BEsdX4xr6Bmh6OwBrNUQykr74LuCIkC28jA,1455
+phoenix/auth.py,sha256=ugvGZlseYX9NkpWaSqb8D2kzUBlAPqT45Dx5_VUANqk,1621
 phoenix/config.py,sha256=wYA_8GSSz5rnpfIWDjeBL9ehKuTy9jqXaMZnxUqRYEU,10131
 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=3tdrXCYXhzGl0HhTFxiRhMv5mTezDVgvqXVYnKeIJeo,23
+phoenix/version.py,sha256=i2svGkhQxPKAQeMjtRMY-YcR__KOdErWKzax-PEH1rY,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
@@ -63,13 +63,16 @@ phoenix/metrics/mixins.py,sha256=moZ5hENIKzUQt2IRhWOd5EFXnoqQkVrpqEqMH7KQzyA,744
 phoenix/metrics/retrieval_metrics.py,sha256=XFQPo66h16w7-1AJ92M1VL_BUIXIWxXHGKF_QVOABZI,4384
 phoenix/metrics/timeseries.py,sha256=Cib3E0njJzi0vZpmyADvbakFQA98rIkfDaYAOmsmBz8,6277
 phoenix/metrics/wrappers.py,sha256=umZqa_5lf1wZSFe3FgzxF-qp1xbPdKD54W628GlGCUI,8392
+phoenix/otel/__init__.py,sha256=YvEiD-3aGZs9agwLNCXU34ofV3G-Q-dolfsiinOJuT0,407
+phoenix/otel/otel.py,sha256=6tlr7VHdVzykQ3Pu5VjvDcjUBKG6GV5fJVZCluXi_d0,11793
+phoenix/otel/settings.py,sha256=Qr2-RkgLQRfLhJqtLpEkSpqns7qLjPoOvpEOTqeSohM,3026
 phoenix/pointcloud/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/pointcloud/clustering.py,sha256=IzcG67kJ2hPP7pcqVmKPSL_6gKRonKdOT3bCtbTOqnk,820
 phoenix/pointcloud/pointcloud.py,sha256=4zAIkKs2xOUbchpj4XDAV-iPMXrfAJ15TG6rlIYGrao,2145
 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=rPzpaEpTSViIP-RLbHzYfxAf7zplKOEbFHxCb40YXFc,26954
+phoenix/server/app.py,sha256=7LiZiusw_yHZBVWMk8gH4tX8yFitTJ2jdyhMWVm2DdM,26817
 phoenix/server/dml_event.py,sha256=MpjCFqljxvgb9OB5Cez9vJesb3oHb3XxXictynBfcis,2851
 phoenix/server/dml_event_handler.py,sha256=6p-PucctivelVHfO-_9zNxWZYPr_eGjDF3bKjLtc5co,8251
 phoenix/server/grpc_server.py,sha256=jllxDNkpLQxDkvej4RhTokobowbvydF-SU8gSw1MTCc,3378
@@ -79,15 +82,16 @@ phoenix/server/telemetry.py,sha256=T_2OKrxNViAeaANlNspEekg_Y5uZIFWvKAnpz8Aoqvk,2
 phoenix/server/thread_server.py,sha256=RwXQGP_QhGD7le6WB7xEygEEuwBl5Ck_Zo8xGIYGi9M,2135
 phoenix/server/types.py,sha256=S2dReLNboR2nzjRK5j3MUyUDqu6AQFD7KRwJkeKj1q4,3609
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/api/context.py,sha256=N_kqhQIlRyPS6JpN-fgCZ2U4_ZbhaknH-23g5JoRU8g,3282
+phoenix/server/api/context.py,sha256=WuhGT2549C5Yc7pWj2S7NaPeT4a-N-_mmz-Vg5bUkI8,3637
+phoenix/server/api/exceptions.py,sha256=KdAzgwNan-wQ7THDrSoeJU2k9zWQVcH6lRiB462VsRA,990
 phoenix/server/api/interceptor.py,sha256=ykDnoC_apUd-llVli3m1CW18kNSIgjz2qZ6m5JmPDu8,1294
-phoenix/server/api/queries.py,sha256=GXQzz1KmRqZCOENemdgoqBlTVA_4V2AZE-g0hBcO0Qk,23839
-phoenix/server/api/schema.py,sha256=BcxdqO5CSGqpKd-AAJHMjFlzaK9oJA8GJuxmMfcdjn4,434
+phoenix/server/api/queries.py,sha256=hUzeHOUWuBQ-kjXh13-d5LgJfkbB8XSpFaHJX4YXpC8,23875
+phoenix/server/api/schema.py,sha256=4L2m6QXhaV13YPTZCEZ3hqCPQFHZOy3QnJVLRYQFzpg,548
 phoenix/server/api/utils.py,sha256=Kl47G-1A7QKTDrc75BU2QK6HupsG6MWuXxy351FOfKQ,858
 phoenix/server/api/dataloaders/__init__.py,sha256=TrOGnU_SD_vEIxOE_dm8HrD5C2ScLFQ4xQ7f8r-E76s,3064
 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/dataset_example_revisions.py,sha256=i0g8F4akEf3kQOzAvBjO27QwXNsq-kJEM8dtzduxQgY,3720
+phoenix/server/api/dataloaders/dataset_example_revisions.py,sha256=rZhJoIYUGgYhXwVBtq5u0bqtHmIQ2Sh6HNnJsSGIXis,3767
 phoenix/server/api/dataloaders/dataset_example_spans.py,sha256=-TjdyyJv2c2JiN1OXu6MMmQ-BEKlHXucEDcuObeRVsU,1416
 phoenix/server/api/dataloaders/document_evaluation_summaries.py,sha256=5XOom2KRAmCwPmtlraiZOSl3vhfaW-eiiYkmetAEalw,5616
 phoenix/server/api/dataloaders/document_evaluations.py,sha256=V6sE34jON_qFxt7eArJbktykAsty-gnBZHlEkORcj0E,1296
@@ -139,11 +143,12 @@ phoenix/server/api/input_types/TimeRange.py,sha256=yzx-gxj8mDeGLft1FzU_x1MVEgIG5
 phoenix/server/api/input_types/TraceAnnotationSort.py,sha256=BzwiUnMh2VsgQYnhDlbJ6ljHugqIS4YDUlYzvq_tl3o,365
 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=Cu4lPgUFRAGzKO528jKepwKtfre9lkLTN059S2Shmnw,977
-phoenix/server/api/mutations/api_key_mutations.py,sha256=cv6AT6UAL55lTC_UqMdcN-1TjWAgqqZi__S9Tw12t6I,3688
+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/auth.py,sha256=8o4tTfGCPkpUauuB9ijPH84Od77UX_UrQWfmUsnujI4,524
-phoenix/server/api/mutations/dataset_mutations.py,sha256=0feBUW_07FEIx6uzepjxfRVhk5lAck0AkrqS1GVdoF4,27041
-phoenix/server/api/mutations/experiment_mutations.py,sha256=OXtLYdLA33RGy1MFctfv6ug2sODcDElhJph_J9vkIjk,3157
+phoenix/server/api/mutations/auth_mutations.py,sha256=XLCxmsjyVp1riGWxUhVUOiGhIFs_ZmOfZM77vs_RLDw,2182
+phoenix/server/api/mutations/dataset_mutations.py,sha256=8S6qjmraSBxA7ioNogTQPp6q27ZdvdAn6yt0Z4fmOI0,27096
+phoenix/server/api/mutations/experiment_mutations.py,sha256=Z2xPrK8J117l5XWN-IvdKpykWIiVXysaUhzuwbIiLtk,3226
 phoenix/server/api/mutations/export_events_mutations.py,sha256=t_wYBxaqvBJYRoHslh3Bmoxmwlzoy0u8SsBKWIKN5hE,4028
 phoenix/server/api/mutations/project_mutations.py,sha256=MLm7I97lJ85hTuc1tq8sdYA8Ps5WKMV-bGqeeN-Ey90,2279
 phoenix/server/api/mutations/span_annotations_mutations.py,sha256=DM9gzxrMSAcxwXQ6jNaNGDVgl8oP50LZsBWRYQwLaSo,5955
@@ -153,14 +158,13 @@ phoenix/server/api/openapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 phoenix/server/api/openapi/main.py,sha256=KNutA_7AvV_WlGX8cOkvvDujcJKQ7AD1HT6rTpCpR8A,616
 phoenix/server/api/openapi/schema.py,sha256=oVZoflWMfzOrLKMIrjr3iLnJ13rmN-t_DOe9g6KoN5s,471
 phoenix/server/api/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/api/routers/auth.py,sha256=dGug0NjOjW1mmIghmheAgHutG7_0-RjL-FcEReWzTHc,1806
 phoenix/server/api/routers/utils.py,sha256=M41BoH-fl37izhRuN2aX7lWm7jOC20A_3uClv9TVUUY,583
 phoenix/server/api/routers/v1/__init__.py,sha256=nb49zcOdAi3DSGuC9gUubN9Yri-o7-WFdlGak4jGuFw,1462
 phoenix/server/api/routers/v1/datasets.py,sha256=l3Hlc9AVyvX5GdT9iOXBsV-i4c_vtnCaXeSAWdNzcw8,37090
 phoenix/server/api/routers/v1/evaluations.py,sha256=FSfz9MTi8s65F07abDXlb9-y97fDZSYbqsCXpimwO7g,12628
 phoenix/server/api/routers/v1/experiment_evaluations.py,sha256=RTQnjupjmh07xowjq77ajbuAZhzIEfYxA4ZtECvGwOU,4844
 phoenix/server/api/routers/v1/experiment_runs.py,sha256=0G7GgGcZv9dzK47tsPp-p4k5O7W4F_aNRrsNuJN7mho,6393
-phoenix/server/api/routers/v1/experiments.py,sha256=3u275sGuYSiMyzC_obbjK3mf6aYb7SkY2c_wOg3z4xg,11751
+phoenix/server/api/routers/v1/experiments.py,sha256=6Ouby3jQDZjeb_nJoD5eH8h2jxINBLrHzCPMsklzcJI,11820
 phoenix/server/api/routers/v1/pydantic_compat.py,sha256=FeK8oe2brqu-djsoqRxiKL4tw5cHmi89OHVfCFxYsAo,2890
 phoenix/server/api/routers/v1/spans.py,sha256=MAkMLrONFtItQxkHJde_Wpvz0jsgydegxVZOkZkRUsU,8781
 phoenix/server/api/routers/v1/traces.py,sha256=HJDmYKMATL40dZEJro6uQ3imbCZBzk3nUun9d21jcDs,7799
@@ -237,10 +241,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=NCacyzu0qbu92qY-iwPA7JHvzK56ZJnp_usNyMD3fZw,1929
-phoenix/server/static/assets/components-1MfQimGx.js,sha256=NfJgri_ChJVeYEExpGOvfr5SLFuX_bZcvjfEVGH3HWI,187209
-phoenix/server/static/assets/index-B263sE2x.js,sha256=PZG-hlU6oncPYh7r6tfdNm5pjuL94SFLHg4fCyr5Oe8,7515
-phoenix/server/static/assets/pages-CqZDVx20.js,sha256=_Lor33vSFj6dZBGvOK4fPt1vXj9pndZFpyyi5BG_8AY,467596
+phoenix/server/static/.vite/manifest.json,sha256=1BcJFFm0NeLAAfZ6r47H6TP_44e5P36X2rORXCQE5e0,1929
+phoenix/server/static/assets/components-BYH03rjA.js,sha256=P4bMCzZcpwxqWV4NaUU4jpmay0GZusDgXbT9t9IUyeM,189918
+phoenix/server/static/assets/index-fqdjNpYm.js,sha256=SDvWOvsXmEJY3350kb2M0bRmC58xw7M3zDZ0MmdneeU,7515
+phoenix/server/static/assets/pages-DnbxgoTK.js,sha256=OLwZw5mVqnO6AelWT6ndr4kDOhf_8l6ezF5buwHBB7E,502087
 phoenix/server/static/assets/vendor-DxkFTwjz.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
 phoenix/server/static/assets/vendor-aSQri0vz.js,sha256=x_07SENutKMhtJ9HgFqkQHvwsDTfPkMmzQznY3HY7Zo,1359197
 phoenix/server/static/assets/vendor-arizeai-CsdcB1NH.js,sha256=VEn7hFJXcHV_DODmeDi9pEpF_D2NQ1bZYewbPe3BhIw,304008
@@ -291,8 +295,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.27.0.dist-info/METADATA,sha256=P6671H1bFkfBHk7FF-1pEwq0_9AoN8zR0RRwVJRqZ2w,11936
-arize_phoenix-4.27.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-arize_phoenix-4.27.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-4.27.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-4.27.0.dist-info/RECORD,,
+arize_phoenix-4.29.0.dist-info/METADATA,sha256=UJiskTOjTLRjXDFO41I1ioPp3KZDU2nmgqMucGUyuAI,11977
+arize_phoenix-4.29.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+arize_phoenix-4.29.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-4.29.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-4.29.0.dist-info/RECORD,,

phoenix/auth.py

@@ -39,7 +39,10 @@ def validate_password_format(password: str) -> None:
         raise ValueError("Password cannot contain whitespace characters")
     if not password.isascii():
         raise ValueError("Password can contain only ASCII characters")
+    if not len(password) >= MIN_PASSWORD_LENGTH:
+        raise ValueError(f"Password must be at least {MIN_PASSWORD_LENGTH} characters long")
 
 
 EMAIL_PATTERN = re.compile(r"^[^@\s]+@[^@\s]+[.][^@\s]+\Z")
 NUM_ITERATIONS = 10_000
+MIN_PASSWORD_LENGTH = 4

phoenix/otel/__init__.py

@@ -0,0 +1,22 @@
+from opentelemetry.sdk.resources import Resource
+
+from .otel import (
+    PROJECT_NAME,
+    BatchSpanProcessor,
+    GRPCSpanExporter,
+    HTTPSpanExporter,
+    SimpleSpanProcessor,
+    TracerProvider,
+    register,
+)
+
+__all__ = [
+    "TracerProvider",
+    "SimpleSpanProcessor",
+    "BatchSpanProcessor",
+    "HTTPSpanExporter",
+    "GRPCSpanExporter",
+    "Resource",
+    "PROJECT_NAME",
+    "register",
+]

phoenix/otel/otel.py

@@ -0,0 +1,284 @@
+import inspect
+import os
+import warnings
+from typing import Any, Dict, List, Optional, Tuple, Union, cast
+from urllib.parse import ParseResult, urlparse
+
+from openinference.semconv.resource import ResourceAttributes as _ResourceAttributes
+from opentelemetry import trace as trace_api
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter as _GRPCSpanExporter,
+)
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+    OTLPSpanExporter as _HTTPSpanExporter,
+)
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import SpanProcessor
+from opentelemetry.sdk.trace import TracerProvider as _TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor as _BatchSpanProcessor
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor as _SimpleSpanProcessor
+from opentelemetry.sdk.trace.export import SpanExporter
+
+from .settings import get_env_client_headers, get_env_collector_endpoint, get_env_project_name
+
+PROJECT_NAME = _ResourceAttributes.PROJECT_NAME
+
+_DEFAULT_GRPC_PORT = 4317
+
+
+def register(
+    *,
+    endpoint: Optional[str] = None,
+    project_name: Optional[str] = None,
+    batch: bool = False,
+    set_global_tracer: bool = True,
+    headers: Optional[Dict[str, str]] = None,
+    verbose: bool = True,
+) -> _TracerProvider:
+    """
+    Creates an OpenTelemetry TracerProvider for enabling OpenInference tracing.
+
+    For futher configuration, the `phoenix.otel` module provides drop-in replacements for
+    OpenTelemetry TracerProvider, SimpleSpanProcessor, BatchSpanProcessor, HTTPSpanExporter, and
+    GRPCSpanExporter objects with Phoenix-aware defaults. Documentation on how to configure tracing
+    can be found at https://opentelemetry.io/docs/specs/otel/trace/sdk/.
+
+    Args:
+        endpoint (str, optional): The collector endpoint to which spans will be exported. If not
+            provided, the `PHOENIX_OTEL_COLLECTOR_ENDPOINT` environment variable will be used. The
+            export protocol will be inferred from the endpoint.
+        project_name (str, optional): The name of the project to which spans will be associated. If
+            not provided, the `PHOENIX_PROJECT_NAME` environment variable will be used.
+        batch (bool): If True, spans will be processed using a BatchSpanprocessor. If False, spans
+            will be processed one at a time using a SimpleSpanProcessor.
+        set_global_tracer (bool): If False, the TracerProvider will not be set as the global
+            tracer provider. Defaults to True.
+        headers (dict, optional): Optional headers to include in the HTTP request to the collector.
+        verbose (bool): If True, configuration details will be printed to stdout.
+    """
+
+    project_name = project_name or get_env_project_name()
+    resource = Resource.create({PROJECT_NAME: project_name})
+    tracer_provider = TracerProvider(resource=resource, verbose=False)
+    span_processor: SpanProcessor
+    if batch:
+        span_processor = BatchSpanProcessor(endpoint=endpoint, headers=headers)
+    else:
+        span_processor = SimpleSpanProcessor(endpoint=endpoint, headers=headers)
+    tracer_provider.add_span_processor(span_processor)
+    tracer_provider._default_processor = True
+
+    if set_global_tracer:
+        trace_api.set_tracer_provider(tracer_provider)
+        global_provider_msg = (
+            "|  \n"
+            "|  `register` has set this TracerProvider as the global OpenTelemetry default.\n"
+            "|  To disable this behavior, call `register` with `set_global_tracer=False`.\n"
+        )
+    else:
+        global_provider_msg = ""
+
+    details = tracer_provider._tracing_details()
+    if verbose:
+        print(f"{details}" f"{global_provider_msg}")
+    return tracer_provider
+
+
+class TracerProvider(_TracerProvider):
+    def __init__(
+        self, *args: Any, endpoint: Optional[str] = None, verbose: bool = True, **kwargs: Any
+    ):
+        sig = inspect.signature(_TracerProvider)
+        bound_args = sig.bind_partial(*args, **kwargs)
+        bound_args.apply_defaults()
+        if bound_args.arguments.get("resource") is None:
+            bound_args.arguments["resource"] = Resource.create(
+                {PROJECT_NAME: get_env_project_name()}
+            )
+        super().__init__(**bound_args.arguments)
+
+        parsed_url, endpoint = _normalized_endpoint(endpoint)
+        self._default_processor = False
+
+        if _maybe_http_endpoint(parsed_url):
+            http_exporter: SpanExporter = HTTPSpanExporter(endpoint=endpoint)
+            self.add_span_processor(SimpleSpanProcessor(span_exporter=http_exporter))
+            self._default_processor = True
+        elif _maybe_grpc_endpoint(parsed_url):
+            grpc_exporter: SpanExporter = GRPCSpanExporter(endpoint=endpoint)
+            self.add_span_processor(SimpleSpanProcessor(span_exporter=grpc_exporter))
+            self._default_processor = True
+        if verbose:
+            print(self._tracing_details())
+
+    def add_span_processor(self, *args: Any, **kwargs: Any) -> None:
+        if self._default_processor:
+            self._active_span_processor.shutdown()
+            self._active_span_processor._span_processors = tuple()  # remove default processors
+            self._default_processor = False
+        return super().add_span_processor(*args, **kwargs)
+
+    def _tracing_details(self) -> str:
+        project = self.resource.attributes.get(PROJECT_NAME)
+        processor_name: Optional[str] = None
+        endpoint: Optional[str] = None
+        transport: Optional[str] = None
+        headers: Optional[Union[Dict[str, str], str]] = None
+
+        if self._active_span_processor:
+            if processors := self._active_span_processor._span_processors:
+                if len(processors) == 1:
+                    span_processor = self._active_span_processor._span_processors[0]
+                    if exporter := getattr(span_processor, "span_exporter"):
+                        processor_name = span_processor.__class__.__name__
+                        endpoint = exporter._endpoint
+                        transport = _exporter_transport(exporter)
+                        headers = _printable_headers(exporter._headers)
+                else:
+                    processor_name = "Multiple Span Processors"
+                    endpoint = "Multiple Span Exporters"
+                    transport = "Multiple Span Exporters"
+                    headers = "Multiple Span Exporters"
+
+        if os.name == "nt":
+            details_header = "OpenTelemetry Tracing Details"
+        else:
+            details_header = "🔭 OpenTelemetry Tracing Details 🔭"
+
+        configuration_msg = (
+            "|  Using a default SpanProcessor. `add_span_processor` will overwrite this default.\n"
+        )
+
+        details_msg = (
+            f"{details_header}\n"
+            f"|  Phoenix Project: {project}\n"
+            f"|  Span Processor: {processor_name}\n"
+            f"|  Collector Endpoint: {endpoint}\n"
+            f"|  Transport: {transport}\n"
+            f"|  Transport Headers: {headers}\n"
+            "|  \n"
+            f"{configuration_msg if self._default_processor else ''}"
+        )
+        return details_msg
+
+
+class SimpleSpanProcessor(_SimpleSpanProcessor):
+    def __init__(
+        self,
+        span_exporter: Optional[SpanExporter] = None,
+        endpoint: Optional[str] = None,
+        headers: Optional[Dict[str, str]] = None,
+    ):
+        if span_exporter is None:
+            parsed_url, endpoint = _normalized_endpoint(endpoint)
+            if _maybe_http_endpoint(parsed_url):
+                span_exporter = HTTPSpanExporter(endpoint=endpoint, headers=headers)
+            elif _maybe_grpc_endpoint(parsed_url):
+                span_exporter = GRPCSpanExporter(endpoint=endpoint, headers=headers)
+            else:
+                warnings.warn("Could not infer collector endpoint protocol, defaulting to HTTP.")
+                span_exporter = HTTPSpanExporter(endpoint=endpoint, headers=headers)
+        super().__init__(span_exporter)
+
+
+class BatchSpanProcessor(_BatchSpanProcessor):
+    def __init__(
+        self,
+        span_exporter: Optional[SpanExporter] = None,
+        endpoint: Optional[str] = None,
+        headers: Optional[Dict[str, str]] = None,
+    ):
+        if span_exporter is None:
+            parsed_url, endpoint = _normalized_endpoint(endpoint)
+            if _maybe_http_endpoint(parsed_url):
+                span_exporter = HTTPSpanExporter(endpoint=endpoint, headers=headers)
+            elif _maybe_grpc_endpoint(parsed_url):
+                span_exporter = GRPCSpanExporter(endpoint=endpoint, headers=headers)
+            else:
+                warnings.warn("Could not infer collector endpoint protocol, defaulting to HTTP.")
+                span_exporter = HTTPSpanExporter(endpoint=endpoint, headers=headers)
+        super().__init__(span_exporter)
+
+
+class HTTPSpanExporter(_HTTPSpanExporter):
+    def __init__(self, *args: Any, **kwargs: Any):
+        sig = inspect.signature(_HTTPSpanExporter)
+        bound_args = sig.bind_partial(*args, **kwargs)
+        bound_args.apply_defaults()
+
+        if not bound_args.arguments.get("headers"):
+            bound_args.arguments["headers"] = get_env_client_headers()
+
+        if bound_args.arguments.get("endpoint") is None:
+            _, endpoint = _normalized_endpoint(None)
+            bound_args.arguments["endpoint"] = endpoint
+        super().__init__(**bound_args.arguments)
+
+
+class GRPCSpanExporter(_GRPCSpanExporter):
+    def __init__(self, *args: Any, **kwargs: Any):
+        sig = inspect.signature(_GRPCSpanExporter)
+        bound_args = sig.bind_partial(*args, **kwargs)
+        bound_args.apply_defaults()
+
+        if not bound_args.arguments.get("headers"):
+            bound_args.arguments["headers"] = get_env_client_headers()
+
+        if bound_args.arguments.get("endpoint") is None:
+            _, endpoint = _normalized_endpoint(None)
+            bound_args.arguments["endpoint"] = endpoint
+        super().__init__(**bound_args.arguments)
+
+
+def _maybe_http_endpoint(parsed_endpoint: ParseResult) -> bool:
+    if parsed_endpoint.path == "/v1/traces":
+        return True
+    return False
+
+
+def _maybe_grpc_endpoint(parsed_endpoint: ParseResult) -> bool:
+    if not parsed_endpoint.path and parsed_endpoint.port == 4317:
+        return True
+    return False
+
+
+def _exporter_transport(exporter: SpanExporter) -> str:
+    if isinstance(exporter, _HTTPSpanExporter):
+        return "HTTP"
+    if isinstance(exporter, _GRPCSpanExporter):
+        return "gRPC"
+    else:
+        return exporter.__class__.__name__
+
+
+def _printable_headers(headers: Union[List[Tuple[str, str]], Dict[str, str]]) -> Dict[str, str]:
+    if isinstance(headers, dict):
+        return {key.lower(): "****" for key, _ in headers.items()}
+    return {key.lower(): "****" for key, _ in headers}
+
+
+def _construct_http_endpoint(parsed_endpoint: ParseResult) -> ParseResult:
+    return parsed_endpoint._replace(path="/v1/traces")
+
+
+def _construct_grpc_endpoint(parsed_endpoint: ParseResult) -> ParseResult:
+    return parsed_endpoint._replace(netloc=f"{parsed_endpoint.hostname}:{_DEFAULT_GRPC_PORT}")
+
+
+_KNOWN_PROVIDERS = {
+    "app.phoenix.arize.com": _construct_http_endpoint,
+}
+
+
+def _normalized_endpoint(endpoint: Optional[str]) -> Tuple[ParseResult, str]:
+    if endpoint is None:
+        base_endpoint = get_env_collector_endpoint() or "http://localhost:6006"
+        parsed = urlparse(base_endpoint)
+        if parsed.hostname in _KNOWN_PROVIDERS:
+            parsed = _KNOWN_PROVIDERS[parsed.hostname](parsed)
+        else:
+            parsed = _construct_grpc_endpoint(parsed)
+    else:
+        parsed = urlparse(endpoint)
+    parsed = cast(ParseResult, parsed)
+    return parsed, parsed.geturl()

phoenix/otel/settings.py

@@ -0,0 +1,82 @@
+import os
+import urllib
+from logging import getLogger
+from re import compile
+from typing import Dict, List, Optional
+
+_logger = getLogger(__name__)
+
+# Environment variables specific to the subpackage
+ENV_PHOENIX_COLLECTOR_ENDPOINT = "PHOENIX_COLLECTOR_ENDPOINT"
+ENV_PHOENIX_PROJECT_NAME = "PHOENIX_PROJECT_NAME"
+ENV_PHOENIX_CLIENT_HEADERS = "PHOENIX_CLIENT_HEADERS"
+
+
+def get_env_collector_endpoint() -> Optional[str]:
+    return os.getenv(ENV_PHOENIX_COLLECTOR_ENDPOINT)
+
+
+def get_env_project_name() -> str:
+    return os.getenv(ENV_PHOENIX_PROJECT_NAME, "default")
+
+
+def get_env_client_headers() -> Optional[Dict[str, str]]:
+    if headers_str := os.getenv(ENV_PHOENIX_CLIENT_HEADERS):
+        return parse_env_headers(headers_str)
+    return None
+
+
+# Optional whitespace
+_OWS = r"[ \t]*"
+# A key contains printable US-ASCII characters except: SP and "(),/:;<=>?@[\]{}
+_KEY_FORMAT = r"[\x21\x23-\x27\x2a\x2b\x2d\x2e\x30-\x39\x41-\x5a\x5e-\x7a\x7c\x7e]+"
+# A value contains a URL-encoded UTF-8 string. The encoded form can contain any
+# printable US-ASCII characters (0x20-0x7f) other than SP, DEL, and ",;/
+_VALUE_FORMAT = r"[\x21\x23-\x2b\x2d-\x3a\x3c-\x5b\x5d-\x7e]*"
+# A key-value is key=value, with optional whitespace surrounding key and value
+_KEY_VALUE_FORMAT = rf"{_OWS}{_KEY_FORMAT}{_OWS}={_OWS}{_VALUE_FORMAT}{_OWS}"
+
+_HEADER_PATTERN = compile(_KEY_VALUE_FORMAT)
+_DELIMITER_PATTERN = compile(r"[ \t]*,[ \t]*")
+
+
+def parse_env_headers(s: str) -> Dict[str, str]:
+    """
+    Parse ``s``, which is a ``str`` instance containing HTTP headers encoded
+    for use in ENV variables per the W3C Baggage HTTP header format at
+    https://www.w3.org/TR/baggage/#baggage-http-header-format, except that
+    additional semi-colon delimited metadata is not supported.
+
+    If the headers are not urlencoded, we will log a warning and attempt to urldecode them.
+    """
+    headers: Dict[str, str] = {}
+    headers_list: List[str] = _DELIMITER_PATTERN.split(s)
+
+    for header in headers_list:
+        if not header:  # empty string
+            continue
+
+        match = _HEADER_PATTERN.fullmatch(header.strip())
+        if not match:
+            parts = header.split("=", 1)
+            name, value = parts
+            encoded_header = f"{urllib.parse.quote(name)}={urllib.parse.quote(value)}"
+            match = _HEADER_PATTERN.fullmatch(encoded_header.strip())
+            if not match:
+                _logger.warning(
+                    "Header format invalid! Header values in environment variables must be "
+                    "URL encoded: %s",
+                    f"{name}: ****",
+                )
+                continue
+            _logger.warning(
+                "Header values in environment variables should be URL encoded, attempting to "
+                "URL encode header: {name}: ****"
+            )
+
+        name, value = header.split("=", 1)
+        name = urllib.parse.unquote(name).strip().lower()
+        value = urllib.parse.unquote(value).strip()
+        headers[name] = value
+
+    return headers

phoenix/server/api/context.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass
 from pathlib import Path
 from typing import Any, Optional
 
+from starlette.responses import Response as StarletteResponse
 from strawberry.fastapi import BaseContext
 
 from phoenix.core.model_schema import Model
@@ -77,7 +78,7 @@ class Context(BaseContext):
     secret: Optional[str] = None
 
     def get_secret(self) -> str:
-        """A type safe way to get the application secret. Throws an error if the secret is not set.
+        """A type-safe way to get the application secret. Throws an error if the secret is not set.
 
         Returns:
             str: the phoenix secret
@@ -88,3 +89,11 @@ class Context(BaseContext):
                 " Please set the PHOENIX_SECRET environment variable and re-deploy the application."
             )
         return self.secret
+
+    def get_response(self) -> StarletteResponse:
+        """
+        A type-safe way to get the response object. Throws an error if the response is not set.
+        """
+        if (response := self.response) is None:
+            raise ValueError("no response is set")
+        return response

phoenix/server/api/dataloaders/dataset_example_revisions.py

@@ -10,6 +10,7 @@ from strawberry.dataloader import DataLoader
 from typing_extensions import TypeAlias
 
 from phoenix.db import models
+from phoenix.server.api.exceptions import NotFound
 from phoenix.server.api.types.DatasetExampleRevision import DatasetExampleRevision
 from phoenix.server.types import DbSessionFactory
 
@@ -24,7 +25,7 @@ class DatasetExampleRevisionsDataLoader(DataLoader[Key, Result]):
         super().__init__(load_fn=self._load_fn)
         self._db = db
 
-    async def _load_fn(self, keys: List[Key]) -> List[Union[Result, ValueError]]:
+    async def _load_fn(self, keys: List[Key]) -> List[Union[Result, NotFound]]:
         # sqlalchemy has limited SQLite support for VALUES, so use UNION ALL instead.
         # For details, see https://github.com/sqlalchemy/sqlalchemy/issues/7228
         keys_subquery = union(
@@ -95,4 +96,4 @@ class DatasetExampleRevisionsDataLoader(DataLoader[Key, Result]):
                 ) in await session.stream(query)
                 if is_valid_version
             }
-        return [results.get(key, ValueError("Could not find revision.")) for key in keys]
+        return [results.get(key, NotFound("Could not find revision.")) for key in keys]

phoenix/server/api/exceptions.py

@@ -0,0 +1,41 @@
+from graphql.error import GraphQLError
+from strawberry.extensions import MaskErrors
+
+
+class CustomGraphQLError(Exception):
+    """
+    An error that represents an expected error scenario in a GraphQL resolver.
+    """
+
+
+class BadRequest(CustomGraphQLError):
+    """
+    An error raised due to a malformed or invalid request.
+    """
+
+
+class NotFound(CustomGraphQLError):
+    """
+    An error raised when the requested resource is not found.
+    """
+
+
+class Unauthorized(CustomGraphQLError):
+    """
+    An error raised when login fails or a user or other entity is not authorized
+    to access a resource.
+    """
+
+
+def get_mask_errors_extension() -> MaskErrors:
+    return MaskErrors(
+        should_mask_error=_should_mask_error,
+        error_message="an unexpected error occurred",
+    )
+
+
+def _should_mask_error(error: GraphQLError) -> bool:
+    """
+    Masks unexpected errors raised from GraphQL resolvers.
+    """
+    return not isinstance(error.original_error, CustomGraphQLError)

phoenix/server/api/mutations/__init__.py

@@ -1,6 +1,7 @@
 import strawberry
 
 from phoenix.server.api.mutations.api_key_mutations import ApiKeyMutationMixin
+from phoenix.server.api.mutations.auth_mutations import AuthMutationMixin
 from phoenix.server.api.mutations.dataset_mutations import DatasetMutationMixin
 from phoenix.server.api.mutations.experiment_mutations import ExperimentMutationMixin
 from phoenix.server.api.mutations.export_events_mutations import ExportEventsMutationMixin
@@ -12,13 +13,14 @@ from phoenix.server.api.mutations.user_mutations import UserMutationMixin
 
 @strawberry.type
 class Mutation(
-    ProjectMutationMixin,
+    ApiKeyMutationMixin,
+    AuthMutationMixin,
     DatasetMutationMixin,
     ExperimentMutationMixin,
     ExportEventsMutationMixin,
+    ProjectMutationMixin,
     SpanAnnotationMutationMixin,
     TraceAnnotationMutationMixin,
-    ApiKeyMutationMixin,
     UserMutationMixin,
 ):
     pass