arize-phoenix 7.12.2__py3-none-any.whl → 7.12.3__py3-none-any.whl

{arize_phoenix-7.12.2.dist-info → arize_phoenix-7.12.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 7.12.2
+Version: 7.12.3
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-7.12.2.dist-info → arize_phoenix-7.12.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
 phoenix/__init__.py,sha256=X3eUEwd2rG8KKWWYVNNDJoqo08ihfjgHhlP29dcdNJE,5481
 phoenix/auth.py,sha256=VVMHrWN31tln3Zo4z6ofecrV4daiqJjLd8r85mqlxek,10939
-phoenix/config.py,sha256=9cda9bRDiN3pwf7D8_erOmfbiW4Vs1jsR5Dy1ugsRV8,26466
+phoenix/config.py,sha256=clBjtpvVN8VznxfYRBgmdG7MclXfCCvyham-Qyf0uWk,27107
 phoenix/datetime_utils.py,sha256=iJzNG6YJ6V7_u8B2iA7P2Z26FyxYbOPtx0dhJ7kNDHA,3398
 phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=kpW1WL0kiB8XJsO6XycvZVJ-lBkNoenhQ7atCvBoSe8,5365
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=WmgZ_39xxpi-JHjL2z86DumX1qR0MPVwHwyeuWN_3ss,23
+phoenix/version.py,sha256=-YxTinWwN6lwwSBelQsFBd_QYUgamTbtExBGoWCHGeU,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=qBFraOtmwCCnWJltKNP18DDG0mULXigytlFsa6YOz6k,4837
@@ -135,7 +135,7 @@ phoenix/server/api/dataloaders/cache/two_tier_cache.py,sha256=cmo8FUT3E91R139IEz
 phoenix/server/api/helpers/__init__.py,sha256=m2-xaSPqUiSs91k62JaRDjFNfl-1byxBfY-m_Vxw16U,272
 phoenix/server/api/helpers/dataset_helpers.py,sha256=DoMBTg-qXTnC_K4Evx1WKpCCYgRbITpVqyY-8efJRf0,8984
 phoenix/server/api/helpers/experiment_run_filters.py,sha256=DOnVwrmn39eAkk2mwuZP8kIcAnR5jrOgllEwWSjsw94,29893
-phoenix/server/api/helpers/playground_clients.py,sha256=U9EwldYQtQrubT0cJDzUAo4Y7n18_xoU9uIKPYHiBHs,39890
+phoenix/server/api/helpers/playground_clients.py,sha256=hpXgjtrCQx1jwf6atMlW8yf6l0s2f5XCvywb141wMDE,39804
 phoenix/server/api/helpers/playground_registry.py,sha256=CPLMziFB2wmr-dfbx7VbzO2f8YIG_k5RftzvGXYGQ1w,2570
 phoenix/server/api/helpers/playground_spans.py,sha256=qGk7V7IZK7EkRE1mvZyROpLN5kgOahOZifFzUWmqYFc,16546
 phoenix/server/api/input_types/AddExamplesToDatasetInput.py,sha256=mIQz0S_z8YdrktKIY6RCvtNJ2yZF9pYvTGgasUsI-54,430
@@ -238,7 +238,7 @@ phoenix/server/api/types/ExperimentRunAnnotation.py,sha256=HEbPygBwbWg10fwY3VZyM
 phoenix/server/api/types/ExportedFile.py,sha256=e3GTn7B5LgsTbqiwjhMCQH7VsiqXitrBO4aCMS1lHsg,163
 phoenix/server/api/types/Functionality.py,sha256=tzV9xdhB8zqfsjWxP66NDC7EZsplYkYO7jRbLWJIeeg,382
 phoenix/server/api/types/GenerativeModel.py,sha256=P7eBUMXbeqaLwSSGBKdZy3a5gOLd9I0fuP8o1st6H08,193
-phoenix/server/api/types/GenerativeProvider.py,sha256=Zg2WcCPiM5xmQrOrfm8-Lp9srRiqNaVGeBnyNBTr390,3925
+phoenix/server/api/types/GenerativeProvider.py,sha256=0XJ_K6zd5-rQOh1xVwSfWMiR8fEvXMfU3hAXTg6xWsA,3941
 phoenix/server/api/types/Inferences.py,sha256=wv88PjcK-KwnzmTdukiAX9EV2KX4GqsKXVAUm1JtnDA,3383
 phoenix/server/api/types/InferencesRole.py,sha256=mLfeHpyhGUVX1-tWzT9IwC_cD18BZrD3RA4YsHYuSpA,595
 phoenix/server/api/types/LabelFraction.py,sha256=zsDxdFALrNiGA1eNykeP8o65gbA0HOhRp54MPH_iRAM,93
@@ -337,9 +337,9 @@ phoenix/utilities/project.py,sha256=auVpARXkDb-JgeX5f2aStyFIkeKvGwN9l7qrFeJMVxI,
 phoenix/utilities/re.py,sha256=6YyUWIkv0zc2SigsxfOWIHzdpjKA_TZo2iqKq7zJKvw,2081
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/template_formatters.py,sha256=gh9PJD6WEGw7TEYXfSst1UR4pWWwmjxMLrDVQ_CkpkQ,2779
-arize_phoenix-7.12.2.dist-info/METADATA,sha256=nJgYrfWkxDNnpl-0GmYDAM3LP_v8DnwehuhKeLScCb0,23390
-arize_phoenix-7.12.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-7.12.2.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-7.12.2.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-7.12.2.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-7.12.2.dist-info/RECORD,,
+arize_phoenix-7.12.3.dist-info/METADATA,sha256=SL2DnU9B8l6vwIwLxghLqZgRMguSGGKmc4ABvAFlREQ,23390
+arize_phoenix-7.12.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-7.12.3.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-7.12.3.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-7.12.3.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-7.12.3.dist-info/RECORD,,

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

phoenix/server/api/helpers/playground_clients.py

@@ -4,7 +4,6 @@ import asyncio
 import importlib.util
 import inspect
 import json
-import os
 import time
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator, Callable, Iterator
@@ -22,6 +21,7 @@ from strawberry import UNSET
 from strawberry.scalars import JSON as JSONScalarType
 from typing_extensions import TypeAlias, assert_never
 
+from phoenix.config import getenv
 from phoenix.evals.models.rate_limiters import (
     AsyncCallable,
     GenericType,
@@ -483,8 +483,8 @@ class OpenAIStreamingClient(OpenAIBaseStreamingClient):
     ) -> None:
         from openai import AsyncOpenAI
 
-        base_url = model.base_url or os.environ.get("OPENAI_BASE_URL")
-        if not (api_key := api_key or os.environ.get("OPENAI_API_KEY")):
+        base_url = model.base_url or getenv("OPENAI_BASE_URL")
+        if not (api_key := api_key or getenv("OPENAI_API_KEY")):
             if not base_url:
                 raise BadRequest("An API key is required for OpenAI models")
             api_key = "sk-fake-api-key"
@@ -656,11 +656,11 @@ class AzureOpenAIStreamingClient(OpenAIBaseStreamingClient):
     ):
         from openai import AsyncAzureOpenAI
 
-        if not (api_key := api_key or os.environ.get("AZURE_OPENAI_API_KEY")):
+        if not (api_key := api_key or getenv("AZURE_OPENAI_API_KEY")):
             raise BadRequest("An Azure API key is required for Azure OpenAI models")
-        if not (endpoint := model.endpoint or os.environ.get("AZURE_OPENAI_ENDPOINT")):
+        if not (endpoint := model.endpoint or getenv("AZURE_OPENAI_ENDPOINT")):
             raise BadRequest("An Azure endpoint is required for Azure OpenAI models")
-        if not (api_version := model.api_version or os.environ.get("OPENAI_API_VERSION")):
+        if not (api_version := model.api_version or getenv("OPENAI_API_VERSION")):
             raise BadRequest("An OpenAI API version is required for Azure OpenAI models")
         client = AsyncAzureOpenAI(
             api_key=api_key,
@@ -697,7 +697,7 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
         super().__init__(model=model, api_key=api_key)
         self._attributes[LLM_PROVIDER] = OpenInferenceLLMProviderValues.ANTHROPIC.value
         self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.ANTHROPIC.value
-        if not (api_key := api_key or os.environ.get("ANTHROPIC_API_KEY")):
+        if not (api_key := api_key or getenv("ANTHROPIC_API_KEY")):
             raise BadRequest("An API key is required for Anthropic models")
         self.client = anthropic.AsyncAnthropic(api_key=api_key)
         self.model_name = model.name
@@ -877,11 +877,7 @@ class GeminiStreamingClient(PlaygroundStreamingClient):
         super().__init__(model=model, api_key=api_key)
         self._attributes[LLM_PROVIDER] = OpenInferenceLLMProviderValues.GOOGLE.value
         self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.VERTEXAI.value
-        if not (
-            api_key := api_key
-            or os.environ.get("GEMINI_API_KEY")
-            or os.environ.get("GOOGLE_API_KEY")
-        ):
+        if not (api_key := api_key or getenv("GEMINI_API_KEY") or getenv("GOOGLE_API_KEY")):
             raise BadRequest("An API key is required for Gemini models")
         google_genai.configure(api_key=api_key)
         self.model_name = model.name

phoenix/server/api/types/GenerativeProvider.py

@@ -1,10 +1,10 @@
-import os
 from enum import Enum
 from typing import Any, ClassVar, Optional, Union
 
 import strawberry
 from openinference.semconv.trace import OpenInferenceLLMProviderValues, SpanAttributes
 
+from phoenix.config import getenv
 from phoenix.trace.attributes import get_attribute_value
 
 
@@ -72,7 +72,7 @@ class GenerativeProvider:
 
     @strawberry.field(description="Whether the credentials are set on the server for the provider")  # type: ignore
     async def api_key_set(self) -> bool:
-        return os.environ.get(self.model_provider_to_api_key_env_var_map[self.key]) is not None
+        return getenv(self.model_provider_to_api_key_env_var_map[self.key]) is not None
 
     @classmethod
     def _infer_model_provider_from_model_name(

phoenix/version.py

@@ -1 +1 @@
-__version__ = "7.12.2"
+__version__ = "7.12.3"