arize 8.0.0b1__tar.gz → 8.0.0b2__tar.gz

{arize-8.0.0b1 → arize-8.0.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0b1
+Version: 8.0.0b2
 Summary: A helper library to interact with Arize AI APIs
 Project-URL: Homepage, https://arize.com
 Project-URL: Documentation, https://docs.arize.com/arize
@@ -81,6 +81,8 @@ Description-Content-Type: text/markdown
 - [Overview](#overview)
 - [Key Features](#key-features)
 - [Installation](#installation)
+  - [Optional Dependencies](#optional-dependencies)
+  - [Migrating from Version 7](#migrating-from-version-7)
 - [Usage](#usage)
   - [Instrumentation](#instrumentation)
   - [Operations on Spans](#operations-on-spans)
@@ -140,19 +142,34 @@ We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 m
 
 # Installation
 
-Install Arize (version 8 is currently under beta release) via `pip`:
+Install the base package:
 
 ```bash
-pip install --pre arize
+pip install arize
 ```
 
-where `--pre` denotes the installation of pre-release versions. Install the
-`arize-otel` package for auto-instrumentation of your LLM library:
+## Optional Dependencies
+
+The following optional extras provide specialized functionality:
+
+> **Note:** The `otel` extra installs the `arize-otel` package, which is also available as a standalone package. If you only need auto-instrumentation without the full SDK, install `arize-otel` directly.
+
+| Extra | Install Command | What It Provides |
+|-------|----------------|------------------|
+| **otel** | `pip install arize[otel]` | OpenTelemetry auto-instrumentation package (arize-otel) for automatic tracing |
+| **embeddings** | `pip install arize[embeddings]` | Automatic embedding generation for NLP, CV, and structured data (Pillow, datasets, tokenizers, torch, transformers) |
+| **mimic** | `pip install arize[mimic]` | MIMIC explainer for model interpretability |
+
+Install multiple extras:
 
 ```bash
-pip install arize-otel
+pip install arize[otel,embeddings,mimic]
 ```
 
+## Migrating from Version 7
+
+If you're upgrading from version 7, please refer to the [Migration Guide](https://arize.com/docs/api-clients/python/version-8/migration) for detailed migration steps and breaking changes.
+
 # Usage
 
 ## Instrumentation

{arize-8.0.0b1 → arize-8.0.0b2}/README.md

@@ -21,6 +21,8 @@
 - [Overview](#overview)
 - [Key Features](#key-features)
 - [Installation](#installation)
+  - [Optional Dependencies](#optional-dependencies)
+  - [Migrating from Version 7](#migrating-from-version-7)
 - [Usage](#usage)
   - [Instrumentation](#instrumentation)
   - [Operations on Spans](#operations-on-spans)
@@ -80,19 +82,34 @@ We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 m
 
 # Installation
 
-Install Arize (version 8 is currently under beta release) via `pip`:
+Install the base package:
 
 ```bash
-pip install --pre arize
+pip install arize
 ```
 
-where `--pre` denotes the installation of pre-release versions. Install the
-`arize-otel` package for auto-instrumentation of your LLM library:
+## Optional Dependencies
+
+The following optional extras provide specialized functionality:
+
+> **Note:** The `otel` extra installs the `arize-otel` package, which is also available as a standalone package. If you only need auto-instrumentation without the full SDK, install `arize-otel` directly.
+
+| Extra | Install Command | What It Provides |
+|-------|----------------|------------------|
+| **otel** | `pip install arize[otel]` | OpenTelemetry auto-instrumentation package (arize-otel) for automatic tracing |
+| **embeddings** | `pip install arize[embeddings]` | Automatic embedding generation for NLP, CV, and structured data (Pillow, datasets, tokenizers, torch, transformers) |
+| **mimic** | `pip install arize[mimic]` | MIMIC explainer for model interpretability |
+
+Install multiple extras:
 
 ```bash
-pip install arize-otel
+pip install arize[otel,embeddings,mimic]
 ```
 
+## Migrating from Version 7
+
+If you're upgrading from version 7, please refer to the [Migration Guide](https://arize.com/docs/api-clients/python/version-8/migration) for detailed migration steps and breaking changes.
+
 # Usage
 
 ## Instrumentation

{arize-8.0.0b1 → arize-8.0.0b2}/pyproject.toml

@@ -106,7 +106,6 @@ exclude = [
 include = '\.pyi?$'
 exclude = '(_pb2\.py$|docs/source/.*\.py)'
 
-# TODO(Kiko): Revisit ruff ignores
 [tool.ruff]
 target-version = "py310"
 line-length = 80

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/__init__.py

@@ -37,7 +37,7 @@ def make_to_df(field_name: str) -> object:
         json_normalize: bool = False,
         convert_dtypes: bool = True,
     ) -> object:
-        """Convert a list of objects to a pandas DataFrame.
+        """Convert a list of objects to a :class:`pandas.DataFrame`.
 
         Behavior:
           - If an item is a Pydantic v2 model, use `.model_dump(by_alias=...)`.

arize-8.0.0b2/src/arize/_client_factory.py

@@ -0,0 +1,50 @@
+"""Factory for creating and caching the generated OpenAPI client."""
+
+from __future__ import annotations
+
+import threading
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from arize._generated.api_client.api_client import ApiClient
+    from arize.config import SDKConfiguration
+
+
+class GeneratedClientFactory:
+    """Factory for creating and caching generated OpenAPI clients.
+
+    This factory is owned by ArizeClient and provides thread-safe lazy
+    initialization of the OpenAPI client used by various subclients.
+    """
+
+    def __init__(self, sdk_config: SDKConfiguration) -> None:
+        """Initialize the factory.
+
+        Args:
+            sdk_config: SDK configuration containing API settings.
+        """
+        self._sdk_config = sdk_config
+        self._client: ApiClient | None = None
+        self._lock = threading.Lock()
+
+    def get_client(self) -> ApiClient:
+        """Get or create the generated OpenAPI client instance.
+
+        Returns:
+            The shared generated API client instance.
+        """
+        if self._client is not None:
+            return self._client
+
+        with self._lock:
+            if self._client is not None:
+                return self._client
+
+            # Import lazily to avoid extra dependencies at config time
+            from arize._generated import api_client as gen
+
+            cfg = gen.Configuration(host=self._sdk_config.api_url)
+            if self._sdk_config.api_key:
+                cfg.access_token = self._sdk_config.api_key
+            self._client = gen.ApiClient(cfg)
+            return self._client

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/_flight/client.py

@@ -333,8 +333,8 @@ class ArizeFlightClient:
                 latest version.
 
         Returns:
-            A pandas DataFrame containing the dataset examples with JSON string columns
-            converted to dict objects.
+            :class:`pandas.DataFrame`: A pandas DataFrame containing the dataset examples
+                with JSON string columns converted to dict objects.
 
         Raises:
             RuntimeError: If the Flight request fails.
@@ -374,8 +374,8 @@ class ArizeFlightClient:
             experiment_id: Experiment ID to retrieve runs from.
 
         Returns:
-            A pandas DataFrame containing the experiment runs with JSON string columns
-            converted to dict objects.
+            :class:`pandas.DataFrame`: A pandas DataFrame containing the experiment runs
+                with JSON string columns converted to dict objects.
 
         Raises:
             RuntimeError: If the Flight request fails.

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/_generated/api_client/api/datasets_api.py

@@ -940,7 +940,7 @@ class DatasetsApi:
         self,
         dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
         dataset_version_id: Annotated[Optional[StrictStr], Field(description="The unique identifier of the dataset version")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1021,7 +1021,7 @@ class DatasetsApi:
         self,
         dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
         dataset_version_id: Annotated[Optional[StrictStr], Field(description="The unique identifier of the dataset version")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1102,7 +1102,7 @@ class DatasetsApi:
         self,
         dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
         dataset_version_id: Annotated[Optional[StrictStr], Field(description="The unique identifier of the dataset version")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1861,7 +1861,7 @@ class DatasetsApi:
     def datasets_list(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -1941,7 +1941,7 @@ class DatasetsApi:
     def datasets_list_with_http_info(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -2021,7 +2021,7 @@ class DatasetsApi:
     def datasets_list_without_preload_content(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/_generated/api_client/api/experiments_api.py

@@ -898,7 +898,7 @@ class ExperimentsApi:
     def experiments_list(
         self,
         dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -978,7 +978,7 @@ class ExperimentsApi:
     def experiments_list_with_http_info(
         self,
         dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -1058,7 +1058,7 @@ class ExperimentsApi:
     def experiments_list_without_preload_content(
         self,
         dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -1211,7 +1211,7 @@ class ExperimentsApi:
     def experiments_runs_list(
         self,
         experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1288,7 +1288,7 @@ class ExperimentsApi:
     def experiments_runs_list_with_http_info(
         self,
         experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1365,7 +1365,7 @@ class ExperimentsApi:
     def experiments_runs_list_without_preload_content(
         self,
         experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
-        limit: Annotated[Optional[Annotated[int, Field(le=10000, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/_generated/api_client/api/projects_api.py

@@ -891,7 +891,7 @@ class ProjectsApi:
     def projects_list(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -970,7 +970,7 @@ class ProjectsApi:
     def projects_list_with_http_info(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,
@@ -1049,7 +1049,7 @@ class ProjectsApi:
     def projects_list_without_preload_content(
         self,
         space_id: Annotated[Optional[StrictStr], Field(description="Filter search results to a particular space ID")] = None,
-        limit: Annotated[Optional[Annotated[int, Field(le=500, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(le=100, strict=True, ge=1)]], Field(description="Maximum items to return")] = None,
         cursor: Annotated[Optional[StrictStr], Field(description="Opaque pagination cursor returned from a previous response (`pagination.next_cursor`). Treat it as an unreadable token; do not attempt to parse or construct it. ")] = None,
         _request_timeout: Union[
             None,

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/_lazy.py

@@ -1,6 +1,7 @@
 # src/arize/_lazy.py
 from __future__ import annotations
 
+import inspect
 import logging
 import sys
 import threading
@@ -8,6 +9,8 @@ from importlib import import_module
 from typing import TYPE_CHECKING, ClassVar
 
 if TYPE_CHECKING:
+    import types
+
     from arize.config import SDKConfiguration
 
 logger = logging.getLogger(__name__)
@@ -22,6 +25,11 @@ class LazySubclientsMixin:
         self._lazy_cache: dict[str, object] = {}
         self._lazy_lock = threading.Lock()
 
+        # Add generated client factory
+        from arize._client_factory import GeneratedClientFactory
+
+        self._gen_client_factory = GeneratedClientFactory(sdk_config)
+
     def __getattr__(self, name: str) -> object:
         subs = self._SUBCLIENTS
         if name not in subs:
@@ -41,12 +49,18 @@ class LazySubclientsMixin:
             module = _dynamic_import(module_path)
             klass = getattr(module, class_name)
 
-            # Pass sdk_config if the child accepts it; otherwise construct bare.
-            try:
-                instance = klass(sdk_config=self.sdk_config)
-            except TypeError:
-                instance = klass()
-
+            # Determine which parameters this subclient needs
+            # and build kwargs accordingly
+            sig = inspect.signature(klass.__init__)
+            kwargs = {}
+            if "sdk_config" in sig.parameters:
+                kwargs["sdk_config"] = self.sdk_config
+            if "generated_client" in sig.parameters:
+                kwargs["generated_client"] = (
+                    self._gen_client_factory.get_client()
+                )
+
+            instance = klass(**kwargs)
             self._lazy_cache[name] = instance
             return instance
 
@@ -82,8 +96,8 @@ def require(
         )
 
 
-def _dynamic_import(modname: str, retries: int = 2) -> object:
-    def _attempt_import(remaining_attempts: int) -> object:
+def _dynamic_import(modname: str, retries: int = 2) -> types.ModuleType:
+    def _attempt_import(remaining_attempts: int) -> types.ModuleType:
         try:
             return import_module(modname)
         except (ModuleNotFoundError, ImportError, KeyError):
@@ -92,4 +106,6 @@ def _dynamic_import(modname: str, retries: int = 2) -> object:
                 raise
             return _attempt_import(remaining_attempts - 1)
 
-    return _attempt_import(retries) if retries > 0 else None
+    if retries <= 0:
+        raise ValueError(f"retries must be > 0, got {retries}")
+    return _attempt_import(retries)

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/client.py

@@ -20,11 +20,6 @@ if TYPE_CHECKING:
 
 logger = logging.getLogger(__name__)
 
-# TODO(Kiko): InvalidAdditionalHeadersError is unused. Have we handled extra headers?
-
-# TODO(Kiko): Clean commented lines over the SDK
-# TODO(Kiko): Implement https://github.com/Arize-ai/arize/pull/59917
-
 # TODO(Kiko): Go private connect. Need a `base_domain`, such that we get:
 # - api.<base_domain>
 # - app.<base_domain>
@@ -33,29 +28,23 @@ logger = logging.getLogger(__name__)
 
 # TODO(Kiko): Enforce type checking, remove all type ignores
 
-# TODO(Kiko): Go over docstrings
-# TODO(Kiko): Missing parameter descriptions in some docstrings
-# TODO(Kiko): Missing return descriptions in some docstrings
-
 # TODO(Kiko): Go over headers on each logging call
+# TODO(Kiko): InvalidAdditionalHeadersError is unused. Have we handled extra headers?
 
 # TODO(Kiko): Need to implement 'Update existing examples in a dataset'
 
-# TODO(Kiko): why logs don't show on scripts, only on jupyter notebooks
 # TODO(Kiko): test caching in colab environment
 # TODO(Kiko): Protobuf versioning is too old
 # TODO(Kiko): Go through main APIs and add CtxAdapter where missing
 # TODO(Kiko): Search and handle other TODOs
-# TODO(Kiko): Go over **every file** and do not import anything at runtime, use `if TYPE_CHECKING`
-# with `from __future__ import annotations` (must include for Python < 3.11)
 
 
 class ArizeClient(LazySubclientsMixin):
     """Root client for the Arize SDK.
 
     The ArizeClient provides access to all Arize platform services including datasets,
-    experiments, ML models, projects, and spans. It uses SDKConfiguration internally to
-    manage configuration settings.
+    experiments, ML models, projects, and spans. It uses :class:`arize.config.SDKConfiguration`
+    internally to manage configuration settings.
 
     All parameters are optional (except api_key which must be provided via argument
     or environment variable). For each parameter, values are resolved in this order:
@@ -149,7 +138,7 @@ class ArizeClient(LazySubclientsMixin):
             api_key: Arize API key for authentication. Required - must be provided here
                 or via ARIZE_API_KEY environment variable. Raises MissingAPIKeyError if not set.
             region: Arize region (e.g., Region.US_CENTRAL, Region.EU_WEST). When specified,
-                overrides individual host/port settings. ENV: ARIZE_REGION. Default: Region.UNSPECIFIED.
+                overrides individual host/port settings. ENV: ARIZE_REGION. Default: Region.UNSET.
             api_host: Custom API endpoint host. ENV: ARIZE_API_HOST. Default: "api.arize.com".
             api_scheme: API endpoint scheme (http/https). ENV: ARIZE_API_SCHEME. Default: "https".
             otlp_host: OTLP endpoint host. ENV: ARIZE_OTLP_HOST. Default: "otlp.arize.com".
@@ -184,7 +173,8 @@ class ArizeClient(LazySubclientsMixin):
 
         Notes:
             Values provided to this class override environment variables, which in turn
-            override default values. See SDKConfiguration for detailed parameter documentation.
+            override default values. See :class:`arize.config.SDKConfiguration`
+            for detailed parameter documentation.
         """
         cfg_kwargs: dict = {}
         if api_key is not None:

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/config.py

@@ -3,10 +3,8 @@
 import logging
 import os
 import sys
-import threading
 from dataclasses import dataclass, field, fields
 from pathlib import Path
-from typing import Any
 
 from arize.constants.config import (
     DEFAULT_API_HOST,
@@ -167,6 +165,9 @@ def _parse_bool(val: bool | str | None) -> bool:
 class SDKConfiguration:
     """Configuration for the Arize SDK with endpoint and authentication settings.
 
+    This class holds pure configuration data and does not manage client lifecycle.
+    Client creation and caching is handled by :class:`arize.ArizeClient`.
+
     This class is used internally by ArizeClient to manage SDK configuration. It is not
     recommended to use this class directly; users should interact with ArizeClient
     instead.
@@ -225,13 +226,16 @@ class SDKConfiguration:
         region: Arize region (e.g., US_CENTRAL, EU_WEST). When specified, overrides
             individual host/port settings.
             Environment variable: ARIZE_REGION.
-            Default: Region.UNSPECIFIED.
+            Default: :class:`Region.UNSET`.
         single_host: Single host to use for all endpoints. Overrides individual host settings.
             Environment variable: ARIZE_SINGLE_HOST.
             Default: "" (not set).
         single_port: Single port to use for all endpoints. Overrides individual port settings (0-65535).
             Environment variable: ARIZE_SINGLE_PORT.
             Default: 0 (not set).
+
+    Raises:
+        MissingAPIKeyError: If api_key is not provided via argument or environment variable.
     """
 
     api_key: str = field(
@@ -323,17 +327,11 @@ class SDKConfiguration:
         )
     )
 
-    # Private, excluded from comparisons & repr
-    _gen_client: Any = field(default=None, repr=False, compare=False)
-    _gen_lock: threading.Lock = field(
-        default_factory=threading.Lock, repr=False, compare=False
-    )
-
     def __post_init__(self) -> None:
         """Validate and configure SDK endpoints after initialization.
 
         Raises:
-            MissingAPIKeyError: If API key is not provided.
+            MissingAPIKeyError: If api_key is not provided via argument or environment variable.
         """
         # Validate Configuration
         if not self.api_key:
@@ -341,7 +339,7 @@ class SDKConfiguration:
 
         has_single_host = bool(self.single_host)
         has_single_port = self.single_port != 0
-        has_region = self.region is not Region.UNSPECIFIED
+        has_region = self.region is not Region.UNSET
         if (has_single_host or has_single_port) and has_region:
             logger.info(
                 "Multiple endpoint override options provided. Preference order is: "
@@ -444,28 +442,3 @@ class SDKConfiguration:
             lines.append(f"  {f.name}={val!r},")
         lines.append(")")
         return "\n".join(lines)
-
-    # TODO(Kiko): This may not be well placed in this class
-    def get_generated_client(self) -> object:
-        """Get or create the generated OpenAPI client instance."""
-        # If already cached, return immediately
-        if self._gen_client is not None:
-            return self._gen_client
-
-        # Thread-safe initialization
-        with self._gen_lock:
-            if self._gen_client is not None:
-                return self._gen_client
-
-            # Import lazily so extra dependencies can be
-            # enforced outside the configuration class
-            from arize._generated import api_client as gen
-
-            cfg = gen.Configuration(host=self.api_url)
-            if self.api_key:
-                cfg.access_token = self.api_key
-            client = gen.ApiClient(cfg)
-
-            # Bypass frozen to set the cache once
-            object.__setattr__(self, "_gen_client", client)
-            return client

{arize-8.0.0b1 → arize-8.0.0b2}/src/arize/constants/ml.py

@@ -3,50 +3,43 @@
 import json
 from pathlib import Path
 
-# MAX_BYTES_PER_BULK_RECORD = 100000
-# MAX_DAYS_WITHIN_RANGE = 365
 MIN_PREDICTION_ID_LEN = 1
 MAX_PREDICTION_ID_LEN = 512
 MIN_DOCUMENT_ID_LEN = 1
 MAX_DOCUMENT_ID_LEN = 128
-# # The maximum number of character for tag values
+# The maximum number of character for tag values
 MAX_TAG_LENGTH = 20_000
 MAX_TAG_LENGTH_TRUNCATION = 1_000
-# # The maximum number of character for embedding raw data
+# The maximum number of character for embedding raw data
 MAX_RAW_DATA_CHARACTERS = 2_000_000
 MAX_RAW_DATA_CHARACTERS_TRUNCATION = 5_000
 # The maximum number of acceptable years in the past from current time for prediction_timestamps
 MAX_PAST_YEARS_FROM_CURRENT_TIME = 5
 # The maximum number of acceptable years in the future from current time for prediction_timestamps
 MAX_FUTURE_YEARS_FROM_CURRENT_TIME = 1
-# # The maximum number of character for llm model name
+# The maximum number of character for llm model name
 MAX_LLM_MODEL_NAME_LENGTH = 20_000
 MAX_LLM_MODEL_NAME_LENGTH_TRUNCATION = 50
-# # The maximum number of character for prompt template
+# The maximum number of character for prompt template
 MAX_PROMPT_TEMPLATE_LENGTH = 50_000
 MAX_PROMPT_TEMPLATE_LENGTH_TRUNCATION = 5_000
-# # The maximum number of character for prompt template version
+# The maximum number of character for prompt template version
 MAX_PROMPT_TEMPLATE_VERSION_LENGTH = 20_000
 MAX_PROMPT_TEMPLATE_VERSION_LENGTH_TRUNCATION = 50
-# # The maximum number of embeddings
+# The maximum number of embeddings
 MAX_NUMBER_OF_EMBEDDINGS = 30
 MAX_EMBEDDING_DIMENSIONALITY = 20_000
-# # The maximum number of classes for multi class
+# The maximum number of classes for multi class
 MAX_NUMBER_OF_MULTI_CLASS_CLASSES = 500
 MAX_MULTI_CLASS_NAME_LENGTH = 100
 # The maximum number of references in embedding similarity search params
 MAX_NUMBER_OF_SIMILARITY_REFERENCES = 10
-#
-# # Arize generated columns
-# GENERATED_PREDICTION_LABEL_COL = "arize_generated_prediction_label"
-# GENERATED_LLM_PARAMS_JSON_COL = "arize_generated_llm_params_json"
-#
-# # reserved columns for LLM run metadata
+# reserved columns for LLM run metadata
 LLM_RUN_METADATA_TOTAL_TOKEN_COUNT_TAG_NAME = "total_token_count"  # noqa: S105
 LLM_RUN_METADATA_PROMPT_TOKEN_COUNT_TAG_NAME = "prompt_token_count"  # noqa: S105
 LLM_RUN_METADATA_RESPONSE_TOKEN_COUNT_TAG_NAME = "response_token_count"  # noqa: S105
 LLM_RUN_METADATA_RESPONSE_LATENCY_MS_TAG_NAME = "response_latency_ms"
-#
+
 # all reserved tags
 RESERVED_TAG_COLS = [
     LLM_RUN_METADATA_TOTAL_TOKEN_COUNT_TAG_NAME,