arize-phoenix 7.12.2__tar.gz → 8.0.0__tar.gz

arize_phoenix-8.0.0/.gitignore

@@ -0,0 +1,70 @@
+# Generated dirs
+.vscode
+.idea
+.coverage
+node_modules
+dist
+sdist
+*__pycache__*
+**/.pytest_cache
+**/.ipynb_checkpoints/
+**/.DS_Store
+
+# Server static files
+/src/phoenix/server/static/*
+
+# Data files
+*.csv
+!tests/**/fixtures/*.csv
+*.hdf
+*.hdf5
+
+# Testing
+coverage.xml
+.tox
+
+# devtools
+pyrightconfig.json
+!/packages/phoenix-client/src/phoenix/client/helpers/sdk/pyrightconfig.json
+!/packages/phoenix-client/tests/canary/sdk/pyrightconfig.json
+!/tests/integration/pyrightconfig.json
+
+# Demo data
+tutorials/internal/demo_llama_index/*.json
+examples/agent_framework_comparison/utils/saved_traces/*.parquet
+.env
+
+# python environments
+.conda
+.venv
+
+# Deno Notebook Environment
+js/examples/notebooks/**/deno.lock
+
+# These can be generated by code refactoring via IntelliJ,
+# but they should be excluded from their Python packages, as
+# they would clobber the same files in phoenix when they are
+# installed as submodules.
+/packages/__init__.py
+/packages/phoenix-client/__init__.py
+/packages/phoenix-client/src/__init__.py
+/packages/phoenix-client/src/phoenix/__init__.py
+/packages/phoenix-evals/__init__.py
+/packages/phoenix-evals/src/__init__.py
+/packages/phoenix-evals/src/phoenix/__init__.py
+/packages/phoenix-otel/__init__.py
+/packages/phoenix-otel/src/__init__.py
+/packages/phoenix-otel/src/phoenix/__init__.py
+
+# Symbolic links
+# Note that this can affect hatch build for those packages. That's why
+# we need to un-ignore it if it's an actual folder (i.e. with the trailing /).
+/src/phoenix/client
+!src/phoenix/client/
+/src/phoenix/evals
+!src/phoenix/evals/
+/src/phoenix/otel
+!src/phoenix/otel/
+
+# Caches
+.pnpm-store

{arize_phoenix-7.12.2 → arize_phoenix-8.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 7.12.2
+Version: 8.0.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -20,6 +20,7 @@ Requires-Python: <3.14,>=3.9
 Requires-Dist: aioitertools
 Requires-Dist: aiosqlite
 Requires-Dist: alembic<2,>=1.3.0
+Requires-Dist: arize-phoenix-client
 Requires-Dist: arize-phoenix-evals>=0.13.1
 Requires-Dist: arize-phoenix-otel>=0.5.1
 Requires-Dist: authlib
@@ -29,6 +30,7 @@ Requires-Dist: grpc-interceptor
 Requires-Dist: grpcio
 Requires-Dist: httpx
 Requires-Dist: jinja2
+Requires-Dist: jsonschema<=4.23.0,>=4.0.0
 Requires-Dist: numpy!=2.0.0
 Requires-Dist: openinference-instrumentation>=0.1.12
 Requires-Dist: openinference-semantic-conventions>=0.1.12
@@ -40,7 +42,7 @@ Requires-Dist: pandas>=1.0
 Requires-Dist: protobuf<6.0,>=3.20.2
 Requires-Dist: psutil
 Requires-Dist: pyarrow
-Requires-Dist: pydantic!=2.0.*,<3,>=1.0
+Requires-Dist: pydantic>=2.1.0
 Requires-Dist: python-multipart
 Requires-Dist: scikit-learn
 Requires-Dist: scipy
@@ -83,7 +85,7 @@ 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: litellm<1.57.5,>=1.0.3; extra == 'dev'
 Requires-Dist: llama-index>=0.10.3; extra == 'dev'
 Requires-Dist: mypy==1.12.1; extra == 'dev'
 Requires-Dist: nbqa; extra == 'dev'
@@ -103,6 +105,7 @@ Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.253.1; extra ==
 Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'
+Requires-Dist: types-jsonschema; extra == 'dev'
 Requires-Dist: types-tabulate; extra == 'dev'
 Requires-Dist: uv==0.4.8; extra == 'dev'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'dev'
@@ -179,30 +182,30 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 
 ## Features
 
-| Key Features                                                                                                     | Availability   |
-| ---------------------------------------------------------------------------------------------------------------- | -------------- |
-| [Tracing](https://docs.arize.com/phoenix/tracing/concepts-tracing/what-are-traces)                               | ✅             |
-| [Evaluation](https://docs.arize.com/phoenix/evaluation/llm-evals)                                                | ✅             |
-| [Retrieval (RAG) Analysis](https://docs.arize.com/phoenix/tracing/use-cases-tracing/rag-evaluation)              | ✅             |
-| [Datasets](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets)                            | ✅             |
-| [Fine-Tuning Export](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-datasets/exporting-datasets) | ✅             |
-| [Annotations](https://docs.arize.com/phoenix/tracing/concepts-tracing/how-to-annotate-traces)                    | ✅             |
-| [Human Feedback](https://docs.arize.com/phoenix/tracing/how-to-tracing/capture-feedback)                         | ✅             |
-| [Experiments](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-experiments/run-experiments)        | ✅             |
-| [Embeddings Analysis](https://docs.arize.com/phoenix/inferences/phoenix-inferences)                              | ✅             |
-| [Data Export](https://docs.arize.com/phoenix/tracing/how-to-tracing/extract-data-from-spans)                     | ✅             |
-| REST API                                                                                                         | ✅             |
-| GraphQL API                                                                                                      | ✅             |
-| Data Retention                                                                                                   | Customizable   |
-| [Authentication](https://docs.arize.com/phoenix/deployment/authentication)                                       | ✅             |
-| [Social Login](https://docs.arize.com/phoenix/deployment/authentication#configuring-oauth2-identity-providers)   | ✅             |
-| [RBAC](https://docs.arize.com/phoenix/deployment/authentication#permissions)                                     | ✅             |
-| Projects                                                                                                         | ✅             |
-| [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅             |
-| Jupyter Notebooks                                                                                                | ✅             |
-| [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅             |
-| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅             |
-| Prompt Management                                                                                                | Coming soon ⏱️ |
+| Key Features                                                                                                     | Availability |
+| ---------------------------------------------------------------------------------------------------------------- | ------------ |
+| [Tracing](https://docs.arize.com/phoenix/tracing/concepts-tracing/what-are-traces)                               | ✅           |
+| [Evaluation](https://docs.arize.com/phoenix/evaluation/llm-evals)                                                | ✅           |
+| [Retrieval (RAG) Analysis](https://docs.arize.com/phoenix/tracing/use-cases-tracing/rag-evaluation)              | ✅           |
+| [Datasets](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets)                            | ✅           |
+| [Fine-Tuning Export](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-datasets/exporting-datasets) | ✅           |
+| [Annotations](https://docs.arize.com/phoenix/tracing/concepts-tracing/how-to-annotate-traces)                    | ✅           |
+| [Human Feedback](https://docs.arize.com/phoenix/tracing/how-to-tracing/capture-feedback)                         | ✅           |
+| [Experiments](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-experiments/run-experiments)        | ✅           |
+| [Embeddings Analysis](https://docs.arize.com/phoenix/inferences/phoenix-inferences)                              | ✅           |
+| [Data Export](https://docs.arize.com/phoenix/tracing/how-to-tracing/extract-data-from-spans)                     | ✅           |
+| REST API                                                                                                         | ✅           |
+| GraphQL API                                                                                                      | ✅           |
+| Data Retention                                                                                                   | Customizable |
+| [Authentication](https://docs.arize.com/phoenix/deployment/authentication)                                       | ✅           |
+| [Social Login](https://docs.arize.com/phoenix/deployment/authentication#configuring-oauth2-identity-providers)   | ✅           |
+| [RBAC](https://docs.arize.com/phoenix/deployment/authentication#permissions)                                     | ✅           |
+| Projects                                                                                                         | ✅           |
+| [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅           |
+| Jupyter Notebooks                                                                                                | ✅           |
+| [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅           |
+| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅           |
+| Prompt Management                                                                                                | ✅           |
 
 ## Tracing Integrations
 
@@ -257,6 +260,6 @@ See the [migration guide](./MIGRATION.md) for a list of breaking changes.
 
 Copyright 2024 Arize AI, Inc. All Rights Reserved.
 
-Portions of this code are patent protected by one or more U.S. Patents. See [IP_NOTICE](https://github.com/Arize-ai/phoenix/blob/main/IP_NOTICE).
+Portions of this code are patent protected by one or more U.S. Patents. See the [IP_NOTICE](https://github.com/Arize-ai/phoenix/blob/main/IP_NOTICE).
 
 This software is licensed under the terms of the Elastic License 2.0 (ELv2). See [LICENSE](https://github.com/Arize-ai/phoenix/blob/main/LICENSE).

{arize_phoenix-7.12.2 → arize_phoenix-8.0.0}/README.md

@@ -53,30 +53,30 @@ Phoenix container images are available via [Docker Hub](https://hub.docker.com/r
 
 ## Features
 
-| Key Features                                                                                                     | Availability   |
-| ---------------------------------------------------------------------------------------------------------------- | -------------- |
-| [Tracing](https://docs.arize.com/phoenix/tracing/concepts-tracing/what-are-traces)                               | ✅             |
-| [Evaluation](https://docs.arize.com/phoenix/evaluation/llm-evals)                                                | ✅             |
-| [Retrieval (RAG) Analysis](https://docs.arize.com/phoenix/tracing/use-cases-tracing/rag-evaluation)              | ✅             |
-| [Datasets](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets)                            | ✅             |
-| [Fine-Tuning Export](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-datasets/exporting-datasets) | ✅             |
-| [Annotations](https://docs.arize.com/phoenix/tracing/concepts-tracing/how-to-annotate-traces)                    | ✅             |
-| [Human Feedback](https://docs.arize.com/phoenix/tracing/how-to-tracing/capture-feedback)                         | ✅             |
-| [Experiments](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-experiments/run-experiments)        | ✅             |
-| [Embeddings Analysis](https://docs.arize.com/phoenix/inferences/phoenix-inferences)                              | ✅             |
-| [Data Export](https://docs.arize.com/phoenix/tracing/how-to-tracing/extract-data-from-spans)                     | ✅             |
-| REST API                                                                                                         | ✅             |
-| GraphQL API                                                                                                      | ✅             |
-| Data Retention                                                                                                   | Customizable   |
-| [Authentication](https://docs.arize.com/phoenix/deployment/authentication)                                       | ✅             |
-| [Social Login](https://docs.arize.com/phoenix/deployment/authentication#configuring-oauth2-identity-providers)   | ✅             |
-| [RBAC](https://docs.arize.com/phoenix/deployment/authentication#permissions)                                     | ✅             |
-| Projects                                                                                                         | ✅             |
-| [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅             |
-| Jupyter Notebooks                                                                                                | ✅             |
-| [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅             |
-| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅             |
-| Prompt Management                                                                                                | Coming soon ⏱️ |
+| Key Features                                                                                                     | Availability |
+| ---------------------------------------------------------------------------------------------------------------- | ------------ |
+| [Tracing](https://docs.arize.com/phoenix/tracing/concepts-tracing/what-are-traces)                               | ✅           |
+| [Evaluation](https://docs.arize.com/phoenix/evaluation/llm-evals)                                                | ✅           |
+| [Retrieval (RAG) Analysis](https://docs.arize.com/phoenix/tracing/use-cases-tracing/rag-evaluation)              | ✅           |
+| [Datasets](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets)                            | ✅           |
+| [Fine-Tuning Export](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-datasets/exporting-datasets) | ✅           |
+| [Annotations](https://docs.arize.com/phoenix/tracing/concepts-tracing/how-to-annotate-traces)                    | ✅           |
+| [Human Feedback](https://docs.arize.com/phoenix/tracing/how-to-tracing/capture-feedback)                         | ✅           |
+| [Experiments](https://docs.arize.com/phoenix/datasets-and-experiments/how-to-experiments/run-experiments)        | ✅           |
+| [Embeddings Analysis](https://docs.arize.com/phoenix/inferences/phoenix-inferences)                              | ✅           |
+| [Data Export](https://docs.arize.com/phoenix/tracing/how-to-tracing/extract-data-from-spans)                     | ✅           |
+| REST API                                                                                                         | ✅           |
+| GraphQL API                                                                                                      | ✅           |
+| Data Retention                                                                                                   | Customizable |
+| [Authentication](https://docs.arize.com/phoenix/deployment/authentication)                                       | ✅           |
+| [Social Login](https://docs.arize.com/phoenix/deployment/authentication#configuring-oauth2-identity-providers)   | ✅           |
+| [RBAC](https://docs.arize.com/phoenix/deployment/authentication#permissions)                                     | ✅           |
+| Projects                                                                                                         | ✅           |
+| [Self-Hosting](https://docs.arize.com/phoenix/deployment)                                                        | ✅           |
+| Jupyter Notebooks                                                                                                | ✅           |
+| [Prompt Playground](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)                          | ✅           |
+| [Sessions](https://docs.arize.com/phoenix/tracing/how-to-tracing/setup-sessions)                                 | ✅           |
+| Prompt Management                                                                                                | ✅           |
 
 ## Tracing Integrations
 
@@ -131,6 +131,6 @@ See the [migration guide](./MIGRATION.md) for a list of breaking changes.
 
 Copyright 2024 Arize AI, Inc. All Rights Reserved.
 
-Portions of this code are patent protected by one or more U.S. Patents. See [IP_NOTICE](https://github.com/Arize-ai/phoenix/blob/main/IP_NOTICE).
+Portions of this code are patent protected by one or more U.S. Patents. See the [IP_NOTICE](https://github.com/Arize-ai/phoenix/blob/main/IP_NOTICE).
 
 This software is licensed under the terms of the Elastic License 2.0 (ELv2). See [LICENSE](https://github.com/Arize-ai/phoenix/blob/main/LICENSE).

{arize_phoenix-7.12.2 → arize_phoenix-8.0.0}/pyproject.toml

@@ -55,9 +55,11 @@ dependencies = [
   "arize-phoenix-evals>=0.13.1",
   "arize-phoenix-otel>=0.5.1",
   "fastapi",
-  "pydantic>=1.0,!=2.0.*,<3", # exclude 2.0.* since it does not support the `json_encoders` configuration setting
+  "pydantic>=2.1.0", # exclude 2.0.* since it does not support the `json_encoders` configuration setting
   "authlib",
   "websockets",
+  "jsonschema>=4.0.0,<=4.23.0", # the upper bound is to keep us off the bleeding edge in case there's a regression since this controls what gets written to the database
+  "arize-phoenix-client",
 ]
 dynamic = ["version"]
 
@@ -92,7 +94,7 @@ dev = [
   "arize[AutoEmbeddings, LLM_Evaluation]",
   "llama-index>=0.10.3",
   "langchain>=0.0.334",
-  "litellm>=1.0.3",
+  "litellm>=1.0.3,<1.57.5", # windows compatibility broken on 1.57.5 (https://github.com/BerriAI/litellm/issues/7677)
   "google-cloud-aiplatform>=1.3",
   "anthropic",
   "prometheus_client",
@@ -101,6 +103,7 @@ dev = [
   "portpicker",
   "uvloop; platform_system != 'Windows'",
   "grpc-interceptor[testing]",
+  "types-jsonschema",
 ]
 embeddings = [
   "fast-hdbscan>=0.2.0",

{arize_phoenix-7.12.2 → arize_phoenix-8.0.0}/src/phoenix/config.py

@@ -162,8 +162,8 @@ ENV_PHOENIX_GRPC_INTERCEPTOR_PATHS = "PHOENIX_GRPC_INTERCEPTOR_PATHS"
 
 def server_instrumentation_is_enabled() -> bool:
     return bool(
-        os.getenv(ENV_PHOENIX_SERVER_INSTRUMENTATION_OTLP_TRACE_COLLECTOR_HTTP_ENDPOINT)
-    ) or bool(os.getenv(ENV_PHOENIX_SERVER_INSTRUMENTATION_OTLP_TRACE_COLLECTOR_GRPC_ENDPOINT))
+        getenv(ENV_PHOENIX_SERVER_INSTRUMENTATION_OTLP_TRACE_COLLECTOR_HTTP_ENDPOINT)
+    ) or bool(getenv(ENV_PHOENIX_SERVER_INSTRUMENTATION_OTLP_TRACE_COLLECTOR_GRPC_ENDPOINT))
 
 
 def _get_temp_path() -> Path:
@@ -192,7 +192,7 @@ def get_working_dir() -> Path:
     """
     Get the working directory for saving, loading, and exporting datasets.
     """
-    working_dir_str = os.getenv(ENV_PHOENIX_WORKING_DIR)
+    working_dir_str = getenv(ENV_PHOENIX_WORKING_DIR)
     if working_dir_str is not None:
         return Path(working_dir_str)
     # Fall back to ~/.phoenix if PHOENIX_WORKING_DIR is not set
@@ -207,7 +207,7 @@ def _bool_val(env_var: str, default: Optional[bool] = None) -> Optional[bool]:
     """
     Parses a boolean environment variable, returning `default` if the variable is not set.
     """
-    if (value := os.environ.get(env_var)) is None:
+    if (value := getenv(env_var)) is None:
         return default
     assert (lower := value.lower()) in (
         "true",
@@ -224,7 +224,7 @@ def _float_val(env_var: str, default: Optional[float] = None) -> Optional[float]
     """
     Parses a numeric environment variable, returning `default` if the variable is not set.
     """
-    if (value := os.environ.get(env_var)) is None:
+    if (value := getenv(env_var)) is None:
         return default
     try:
         return float(value)
@@ -243,7 +243,7 @@ def _int_val(env_var: str, default: Optional[int] = None) -> Optional[int]:
     """
     Parses a numeric environment variable, returning `default` if the variable is not set.
     """
-    if (value := os.environ.get(env_var)) is None:
+    if (value := getenv(env_var)) is None:
         return default
     try:
         return int(value)
@@ -254,6 +254,33 @@ def _int_val(env_var: str, default: Optional[int] = None) -> Optional[int]:
         )
 
 
+@overload
+def getenv(key: str) -> Optional[str]: ...
+@overload
+def getenv(key: str, default: str) -> str: ...
+def getenv(key: str, default: Optional[str] = None) -> Optional[str]:
+    """
+    Retrieves the value of an environment variable.
+
+    Parameters
+    ----------
+    key : str
+        The name of the environment variable.
+    default : Optional[str], optional
+        The default value to return if the environment variable is not set, by default None.
+
+    Returns
+    -------
+    Optional[str]
+        The value of the environment variable, or `default` if the variable is not set.
+        Leading and trailing whitespaces are stripped from the value, assuming they were
+        inadvertently added.
+    """
+    if (value := os.getenv(key)) is None:
+        return default
+    return value.strip()
+
+
 def get_env_enable_auth() -> bool:
     """
     Gets the value of the PHOENIX_ENABLE_AUTH environment variable.
@@ -273,7 +300,7 @@ def get_env_phoenix_secret() -> Optional[str]:
     Gets the value of the PHOENIX_SECRET environment variable
     and performs validation.
     """
-    phoenix_secret = os.environ.get(ENV_PHOENIX_SECRET)
+    phoenix_secret = getenv(ENV_PHOENIX_SECRET)
     if phoenix_secret is None:
         return None
     from phoenix.auth import REQUIREMENTS_FOR_PHOENIX_SECRET
@@ -285,7 +312,7 @@ def get_env_phoenix_secret() -> Optional[str]:
 def get_env_default_admin_initial_password() -> str:
     from phoenix.auth import DEFAULT_ADMIN_PASSWORD
 
-    return os.environ.get(ENV_PHOENIX_DEFAULT_ADMIN_INITIAL_PASSWORD) or DEFAULT_ADMIN_PASSWORD
+    return getenv(ENV_PHOENIX_DEFAULT_ADMIN_INITIAL_PASSWORD) or DEFAULT_ADMIN_PASSWORD
 
 
 def get_env_phoenix_use_secure_cookies() -> bool:
@@ -293,7 +320,7 @@ def get_env_phoenix_use_secure_cookies() -> bool:
 
 
 def get_env_phoenix_api_key() -> Optional[str]:
-    return os.environ.get(ENV_PHOENIX_API_KEY)
+    return getenv(ENV_PHOENIX_API_KEY)
 
 
 def get_env_auth_settings() -> tuple[bool, Optional[str]]:
@@ -354,7 +381,7 @@ def get_env_refresh_token_expiry() -> timedelta:
 
 def get_env_csrf_trusted_origins() -> list[str]:
     origins: list[str] = []
-    if not (csrf_trusted_origins := os.getenv(ENV_PHOENIX_CSRF_TRUSTED_ORIGINS)):
+    if not (csrf_trusted_origins := getenv(ENV_PHOENIX_CSRF_TRUSTED_ORIGINS)):
         return origins
     for origin in csrf_trusted_origins.split(","):
         if not origin:
@@ -369,19 +396,19 @@ def get_env_csrf_trusted_origins() -> list[str]:
 
 
 def get_env_smtp_username() -> str:
-    return os.getenv(ENV_PHOENIX_SMTP_USERNAME) or ""
+    return getenv(ENV_PHOENIX_SMTP_USERNAME, "")
 
 
 def get_env_smtp_password() -> str:
-    return os.getenv(ENV_PHOENIX_SMTP_PASSWORD) or ""
+    return getenv(ENV_PHOENIX_SMTP_PASSWORD, "")
 
 
 def get_env_smtp_mail_from() -> str:
-    return os.getenv(ENV_PHOENIX_SMTP_MAIL_FROM) or "noreply@arize.com"
+    return getenv(ENV_PHOENIX_SMTP_MAIL_FROM) or "noreply@arize.com"
 
 
 def get_env_smtp_hostname() -> str:
-    return os.getenv(ENV_PHOENIX_SMTP_HOSTNAME) or ""
+    return getenv(ENV_PHOENIX_SMTP_HOSTNAME, "")
 
 
 def get_env_smtp_port() -> int:
@@ -410,16 +437,14 @@ class OAuth2ClientConfig:
     def from_env(cls, idp_name: str) -> "OAuth2ClientConfig":
         idp_name_upper = idp_name.upper()
         if not (
-            client_id := os.getenv(
-                client_id_env_var := f"PHOENIX_OAUTH2_{idp_name_upper}_CLIENT_ID"
-            )
+            client_id := getenv(client_id_env_var := f"PHOENIX_OAUTH2_{idp_name_upper}_CLIENT_ID")
         ):
             raise ValueError(
                 f"A client id must be set for the {idp_name} OAuth2 IDP "
                 f"via the {client_id_env_var} environment variable"
             )
         if not (
-            client_secret := os.getenv(
+            client_secret := getenv(
                 client_secret_env_var := f"PHOENIX_OAUTH2_{idp_name_upper}_CLIENT_SECRET"
             )
         ):
@@ -429,7 +454,7 @@ class OAuth2ClientConfig:
             )
         if not (
             oidc_config_url := (
-                os.getenv(
+                getenv(
                     oidc_config_url_env_var := f"PHOENIX_OAUTH2_{idp_name_upper}_OIDC_CONFIG_URL",
                 )
             )
@@ -447,7 +472,7 @@ class OAuth2ClientConfig:
             )
         return cls(
             idp_name=idp_name,
-            idp_display_name=os.getenv(
+            idp_display_name=getenv(
                 f"PHOENIX_OAUTH2_{idp_name_upper}_DISPLAY_NAME",
                 _get_default_idp_display_name(idp_name),
             ),
@@ -541,7 +566,7 @@ def get_exported_files(directory: Path) -> list[Path]:
 
 
 def get_env_port() -> int:
-    if not (port := os.getenv(ENV_PHOENIX_PORT)):
+    if not (port := getenv(ENV_PHOENIX_PORT)):
         return PORT
     if port.isnumeric():
         return int(port)
@@ -560,7 +585,7 @@ def get_env_port() -> int:
 
 
 def get_env_grpc_port() -> int:
-    if not (port := os.getenv(ENV_PHOENIX_GRPC_PORT)):
+    if not (port := getenv(ENV_PHOENIX_GRPC_PORT)):
         return GRPC_PORT
     if port.isnumeric():
         return int(port)
@@ -571,11 +596,11 @@ def get_env_grpc_port() -> int:
 
 
 def get_env_host() -> str:
-    return os.getenv(ENV_PHOENIX_HOST) or HOST
+    return getenv(ENV_PHOENIX_HOST) or HOST
 
 
 def get_env_host_root_path() -> str:
-    if (host_root_path := os.getenv(ENV_PHOENIX_HOST_ROOT_PATH)) is None:
+    if (host_root_path := getenv(ENV_PHOENIX_HOST_ROOT_PATH)) is None:
         return HOST_ROOT_PATH
     if not host_root_path.startswith("/"):
         raise ValueError(
@@ -591,15 +616,15 @@ def get_env_host_root_path() -> str:
 
 
 def get_env_collector_endpoint() -> Optional[str]:
-    return os.getenv(ENV_PHOENIX_COLLECTOR_ENDPOINT)
+    return getenv(ENV_PHOENIX_COLLECTOR_ENDPOINT)
 
 
 def get_env_project_name() -> str:
-    return os.getenv(ENV_PHOENIX_PROJECT_NAME) or DEFAULT_PROJECT_NAME
+    return getenv(ENV_PHOENIX_PROJECT_NAME, DEFAULT_PROJECT_NAME)
 
 
 def get_env_database_connection_str() -> str:
-    env_url = os.getenv(ENV_PHOENIX_SQL_DATABASE_URL)
+    env_url = getenv(ENV_PHOENIX_SQL_DATABASE_URL)
     if env_url is None:
         working_dir = get_working_dir()
         return f"sqlite:///{working_dir}/phoenix.db"
@@ -609,11 +634,11 @@ def get_env_database_connection_str() -> str:
 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)
+    return getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA)
 
 
 def get_env_enable_prometheus() -> bool:
-    if (enable_promotheus := os.getenv(ENV_PHOENIX_ENABLE_PROMETHEUS)) is None or (
+    if (enable_promotheus := getenv(ENV_PHOENIX_ENABLE_PROMETHEUS)) is None or (
         enable_promotheus_lower := enable_promotheus.lower()
     ) == "false":
         return False
@@ -626,7 +651,7 @@ def get_env_enable_prometheus() -> bool:
 
 
 def get_env_client_headers() -> dict[str, str]:
-    headers = parse_env_headers(os.getenv(ENV_PHOENIX_CLIENT_HEADERS))
+    headers = parse_env_headers(getenv(ENV_PHOENIX_CLIENT_HEADERS))
     if (api_key := get_env_phoenix_api_key()) and "authorization" not in [
         k.lower() for k in headers
     ]:
@@ -661,7 +686,7 @@ class LoggingMode(Enum):
 
 
 def get_env_logging_mode() -> LoggingMode:
-    if (logging_mode := os.getenv(ENV_LOGGING_MODE)) is None:
+    if (logging_mode := getenv(ENV_LOGGING_MODE)) is None:
         return LoggingMode.DEFAULT
     try:
         return LoggingMode(logging_mode.lower().strip())
@@ -688,7 +713,7 @@ def get_env_db_logging_level() -> int:
 
 
 def get_env_fastapi_middleware_paths() -> list[tuple[str, str]]:
-    env_value = os.getenv(ENV_PHOENIX_FASTAPI_MIDDLEWARE_PATHS, "")
+    env_value = getenv(ENV_PHOENIX_FASTAPI_MIDDLEWARE_PATHS, "")
     paths = []
     for entry in env_value.split(","):
         entry = entry.strip()
@@ -703,7 +728,7 @@ def get_env_fastapi_middleware_paths() -> list[tuple[str, str]]:
 
 
 def get_env_gql_extension_paths() -> list[tuple[str, str]]:
-    env_value = os.getenv(ENV_PHOENIX_GQL_EXTENSION_PATHS, "")
+    env_value = getenv(ENV_PHOENIX_GQL_EXTENSION_PATHS, "")
     paths = []
     for entry in env_value.split(","):
         entry = entry.strip()
@@ -718,7 +743,7 @@ def get_env_gql_extension_paths() -> list[tuple[str, str]]:
 
 
 def get_env_grpc_interceptor_paths() -> list[tuple[str, str]]:
-    env_value = os.getenv(ENV_PHOENIX_GRPC_INTERCEPTOR_PATHS, "")
+    env_value = getenv(ENV_PHOENIX_GRPC_INTERCEPTOR_PATHS, "")
     paths = []
     for entry in env_value.split(","):
         entry = entry.strip()
@@ -733,7 +758,7 @@ def get_env_grpc_interceptor_paths() -> list[tuple[str, str]]:
 
 
 def _get_logging_level(env_var: str, default_level: int) -> int:
-    logging_level = os.getenv(env_var)
+    logging_level = getenv(env_var)
     if not logging_level:
         return default_level
 
@@ -753,7 +778,7 @@ def _get_logging_level(env_var: str, default_level: int) -> int:
 
 
 def get_env_log_migrations() -> bool:
-    log_migrations = os.getenv(ENV_LOG_MIGRATIONS)
+    log_migrations = getenv(ENV_LOG_MIGRATIONS)
     # Default to True
     if log_migrations is None:
         return True