arize 8.0.0a23__py3-none-any.whl → 8.0.0b1__py3-none-any.whl

arize/__init__.py

@@ -44,18 +44,19 @@ def make_to_df(field_name: str) -> object:
           - If an item is a mapping (dict-like), use it as-is.
           - Otherwise, raise a ValueError (unsupported row type).
 
-        Parameters:
-          by_alias: Use field aliases when dumping Pydantic models.
-          exclude_none:
-            - False: keep Nones as-is
-            - "all": drop columns where *all* values are None/NaN
-            - "any": drop columns where *any* value is None/NaN
-            - True: alias for "all"
-          json_normalize: If True, flatten nested dicts via `pandas.json_normalize`.
-          convert_dtypes: If True, call `DataFrame.convert_dtypes()` at the end.
+        Args:
+            self (object): The object instance containing the field to convert.
+            by_alias (bool): Use field aliases when dumping Pydantic models.
+            exclude_none (str | bool): Control None/NaN column dropping.
+                - False: keep Nones as-is
+                - "all": drop columns where all values are None/NaN
+                - "any": drop columns where any value is None/NaN
+                - True: alias for "all"
+            json_normalize (bool): If True, flatten nested dicts via `pandas.json_normalize`.
+            convert_dtypes (bool): If True, call `DataFrame.convert_dtypes()` at the end.
 
         Returns:
-          pandas.DataFrame
+            pandas.DataFrame: The converted DataFrame.
         """
         import pandas as pd
 

arize/_exporter/client.py

@@ -17,7 +17,7 @@ from arize._exporter.validation import (
 )
 from arize._generated.protocol.flight import flight_pb2
 from arize.logging import CtxAdapter
-from arize.types import Environments, SimilaritySearchParams
+from arize.ml.types import Environments, SimilaritySearchParams
 from arize.utils.dataframe import reset_dataframe_index
 
 logger = logging.getLogger(__name__)

arize/_generated/api_client/__init__.py

@@ -46,7 +46,6 @@ __all__ = [
     "ExperimentsList200Response",
     "ExperimentsRunsList200Response",
     "PaginationMetadata",
-    "PrimitiveValue",
     "Problem",
     "Project",
     "ProjectsCreateRequest",
@@ -86,7 +85,6 @@ from arize._generated.api_client.models.experiments_create_request import Experi
 from arize._generated.api_client.models.experiments_list200_response import ExperimentsList200Response as ExperimentsList200Response
 from arize._generated.api_client.models.experiments_runs_list200_response import ExperimentsRunsList200Response as ExperimentsRunsList200Response
 from arize._generated.api_client.models.pagination_metadata import PaginationMetadata as PaginationMetadata
-from arize._generated.api_client.models.primitive_value import PrimitiveValue as PrimitiveValue
 from arize._generated.api_client.models.problem import Problem as Problem
 from arize._generated.api_client.models.project import Project as Project
 from arize._generated.api_client.models.projects_create_request import ProjectsCreateRequest as ProjectsCreateRequest

arize/_generated/api_client/models/__init__.py

@@ -29,7 +29,6 @@ from arize._generated.api_client.models.experiments_create_request import Experi
 from arize._generated.api_client.models.experiments_list200_response import ExperimentsList200Response
 from arize._generated.api_client.models.experiments_runs_list200_response import ExperimentsRunsList200Response
 from arize._generated.api_client.models.pagination_metadata import PaginationMetadata
-from arize._generated.api_client.models.primitive_value import PrimitiveValue
 from arize._generated.api_client.models.problem import Problem
 from arize._generated.api_client.models.project import Project
 from arize._generated.api_client.models.projects_create_request import ProjectsCreateRequest

arize/_generated/api_client/models/datasets_create_request.py

@@ -19,7 +19,6 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
-from arize._generated.api_client.models.primitive_value import PrimitiveValue
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,7 +28,7 @@ class DatasetsCreateRequest(BaseModel):
     """ # noqa: E501
     name: StrictStr = Field(description="Name of the new dataset")
     space_id: StrictStr = Field(description="ID of the space the dataset will belong to")
-    examples: List[Dict[str, PrimitiveValue]] = Field(description="Array of examples for the new dataset")
+    examples: List[Dict[str, Any]] = Field(description="Array of examples for the new dataset")
     __properties: ClassVar[List[str]] = ["name", "space_id", "examples"]
 
     model_config = ConfigDict(
@@ -71,13 +70,6 @@ class DatasetsCreateRequest(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in examples (list)
-        _items = []
-        if self.examples:
-            for _item_examples in self.examples:
-                if _item_examples:
-                    _items.append(_item_examples.to_dict())
-            _dict['examples'] = _items
         return _dict
 
     @classmethod
@@ -97,7 +89,7 @@ class DatasetsCreateRequest(BaseModel):
         _obj = cls.model_validate({
             "name": obj.get("name"),
             "space_id": obj.get("space_id"),
-            "examples": [Dict[str, PrimitiveValue].from_dict(_item) for _item in obj["examples"]] if obj.get("examples") is not None else None
+            "examples": obj.get("examples")
         })
         return _obj
 

arize/_generated/api_client/models/datasets_examples_insert_request.py

@@ -19,7 +19,6 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field
 from typing import Any, ClassVar, Dict, List
-from arize._generated.api_client.models.primitive_value import PrimitiveValue
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,7 +26,7 @@ class DatasetsExamplesInsertRequest(BaseModel):
     """
     DatasetsExamplesInsertRequest
     """ # noqa: E501
-    examples: List[Dict[str, PrimitiveValue]] = Field(description="Array of examples to append to the dataset version")
+    examples: List[Dict[str, Any]] = Field(description="Array of examples to append to the dataset version")
     __properties: ClassVar[List[str]] = ["examples"]
 
     model_config = ConfigDict(
@@ -69,13 +68,6 @@ class DatasetsExamplesInsertRequest(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in examples (list)
-        _items = []
-        if self.examples:
-            for _item_examples in self.examples:
-                if _item_examples:
-                    _items.append(_item_examples.to_dict())
-            _dict['examples'] = _items
         return _dict
 
     @classmethod
@@ -93,7 +85,7 @@ class DatasetsExamplesInsertRequest(BaseModel):
                 raise ValueError("Error due to additional fields (not defined in DatasetsExamplesInsertRequest) in the input: " + _key)
 
         _obj = cls.model_validate({
-            "examples": [Dict[str, PrimitiveValue].from_dict(_item) for _item in obj["examples"]] if obj.get("examples") is not None else None
+            "examples": obj.get("examples")
         })
         return _obj
 

arize/_generated/api_client/test/test_datasets_create_request.py

@@ -38,9 +38,7 @@ class TestDatasetsCreateRequest(unittest.TestCase):
                 name = '',
                 space_id = '',
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ]
             )
         else:
@@ -48,9 +46,7 @@ class TestDatasetsCreateRequest(unittest.TestCase):
                 name = '',
                 space_id = '',
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
         )
         """

arize/_generated/api_client/test/test_datasets_examples_insert_request.py

@@ -36,17 +36,13 @@ class TestDatasetsExamplesInsertRequest(unittest.TestCase):
         if include_optional:
             return DatasetsExamplesInsertRequest(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ]
             )
         else:
             return DatasetsExamplesInsertRequest(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
         )
         """

arize/_generated/api_client/test/test_datasets_examples_list200_response.py

@@ -36,9 +36,7 @@ class TestDatasetsExamplesList200Response(unittest.TestCase):
         if include_optional:
             return DatasetsExamplesList200Response(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
                 pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
                     next_cursor = '', 
@@ -47,9 +45,7 @@ class TestDatasetsExamplesList200Response(unittest.TestCase):
         else:
             return DatasetsExamplesList200Response(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
                 pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
                     next_cursor = '', 

arize/_generated/api_client/test/test_datasets_examples_update_request.py

@@ -36,18 +36,14 @@ class TestDatasetsExamplesUpdateRequest(unittest.TestCase):
         if include_optional:
             return DatasetsExamplesUpdateRequest(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
                 new_version = ''
             )
         else:
             return DatasetsExamplesUpdateRequest(
                 examples = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
         )
         """

arize/_generated/api_client/test/test_experiments_create_request.py

@@ -38,9 +38,7 @@ class TestExperimentsCreateRequest(unittest.TestCase):
                 name = '',
                 dataset_id = '',
                 experiment_runs = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ]
             )
         else:
@@ -48,9 +46,7 @@ class TestExperimentsCreateRequest(unittest.TestCase):
                 name = '',
                 dataset_id = '',
                 experiment_runs = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
         )
         """

arize/_generated/api_client/test/test_experiments_runs_list200_response.py

@@ -36,9 +36,7 @@ class TestExperimentsRunsList200Response(unittest.TestCase):
         if include_optional:
             return ExperimentsRunsList200Response(
                 experiment_runs = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
                 pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
                     next_cursor = '', 
@@ -47,9 +45,7 @@ class TestExperimentsRunsList200Response(unittest.TestCase):
         else:
             return ExperimentsRunsList200Response(
                 experiment_runs = [
-                    {
-                        'key' : null
-                        }
+                    { }
                     ],
                 pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
                     next_cursor = '', 

arize/_generated/api_client_README.md

@@ -109,7 +109,6 @@ Class | Method | HTTP request | Description
  - [ExperimentsList200Response](arize/_generated/api_client/docs/ExperimentsList200Response.md)
  - [ExperimentsRunsList200Response](arize/_generated/api_client/docs/ExperimentsRunsList200Response.md)
  - [PaginationMetadata](arize/_generated/api_client/docs/PaginationMetadata.md)
- - [PrimitiveValue](arize/_generated/api_client/docs/PrimitiveValue.md)
  - [Problem](arize/_generated/api_client/docs/Problem.md)
  - [Project](arize/_generated/api_client/docs/Project.md)
  - [ProjectsCreateRequest](arize/_generated/api_client/docs/ProjectsCreateRequest.md)

arize/client.py

@@ -13,13 +13,23 @@ from arize.config import SDKConfiguration
 if TYPE_CHECKING:
     from arize.datasets.client import DatasetsClient
     from arize.experiments.client import ExperimentsClient
-    from arize.models.client import MLModelsClient
+    from arize.ml.client import MLModelsClient
     from arize.projects.client import ProjectsClient
     from arize.regions import Region
     from arize.spans.client import SpansClient
 
 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>
+# - flight.<base_domain>
+# - otlp.<base_domain>
 
 # TODO(Kiko): Enforce type checking, remove all type ignores
 
@@ -44,133 +54,35 @@ class ArizeClient(LazySubclientsMixin):
     """Root client for the Arize SDK.
 
     The ArizeClient provides access to all Arize platform services including datasets,
-    experiments, models, projects, and spans. It uses SDKConfiguration internally to
+    experiments, ML models, projects, and spans. It uses SDKConfiguration internally to
     manage configuration settings.
 
-    Configuration Precedence
-    ------------------------
     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:
-        1. Explicit value passed to constructor (highest priority)
-        2. Environment variable value (see SDKConfiguration for variable names)
-        3. Built-in default value (lowest priority)
-
-    Parameters
-    ----------
-    api_key : str
-        Arize API key for authentication. Required - must be provided via this parameter
-        or the ARIZE_API_KEY environment variable.
-        Environment variable: ARIZE_API_KEY
-        Default: None (raises MissingAPIKeyError if not provided)
-
-    region : Region, optional
-        Arize region (e.g., Region.US_CENTRAL, Region.EU_WEST). When specified,
-        overrides individual host/port settings with region-specific endpoints.
-        Environment variable: ARIZE_REGION
-        Default: Region.UNSPECIFIED
-
-    api_host : str, optional
-        Custom API endpoint host override.
-        Environment variable: ARIZE_API_HOST
-        Default: "api.arize.com"
-
-    api_scheme : str, optional
-        API endpoint scheme (http/https).
-        Environment variable: ARIZE_API_SCHEME
-        Default: "https"
-
-    otlp_host : str, optional
-        OTLP (OpenTelemetry Protocol) endpoint host override.
-        Environment variable: ARIZE_OTLP_HOST
-        Default: "otlp.arize.com"
-
-    otlp_scheme : str, optional
-        OTLP endpoint scheme (http/https).
-        Environment variable: ARIZE_OTLP_SCHEME
-        Default: "https"
-
-    flight_host : str, optional
-        Apache Arrow Flight endpoint host override.
-        Environment variable: ARIZE_FLIGHT_HOST
-        Default: "flight.arize.com"
-
-    flight_port : int, optional
-        Apache Arrow Flight endpoint port (1-65535).
-        Environment variable: ARIZE_FLIGHT_PORT
-        Default: 443
-
-    flight_scheme : str, optional
-        Apache Arrow Flight endpoint scheme.
-        Environment variable: ARIZE_FLIGHT_SCHEME
-        Default: "grpc+tls"
-
-    pyarrow_max_chunksize : int, optional
-        Maximum chunk size for PyArrow operations (1 to MAX_CHUNKSIZE).
-        Environment variable: ARIZE_MAX_CHUNKSIZE
-        Default: 10_000
-
-    request_verify : bool, optional
-        Whether to verify SSL certificates for HTTP requests.
-        Environment variable: ARIZE_REQUEST_VERIFY
-        Default: True
-
-    stream_max_workers : int, optional
-        Maximum number of worker threads for streaming operations (minimum: 1).
-        Environment variable: ARIZE_STREAM_MAX_WORKERS
-        Default: 8
-
-    stream_max_queue_bound : int, optional
-        Maximum queue size for streaming operations (minimum: 1).
-        Environment variable: ARIZE_STREAM_MAX_QUEUE_BOUND
-        Default: 5000
-
-    max_http_payload_size_mb : float, optional
-        Maximum HTTP payload size in megabytes (minimum: 1).
-        Environment variable: ARIZE_MAX_HTTP_PAYLOAD_SIZE_MB
-        Default: 100
-
-    arize_directory : str, optional
-        Directory for Arize SDK files (cache, logs, etc.).
-        Environment variable: ARIZE_DIRECTORY
-        Default: "~/.arize"
-
-    enable_caching : bool, optional
-        Whether to enable local caching.
-        Environment variable: ARIZE_ENABLE_CACHING
-        Default: True
-
-    single_host : str, optional
-        Single host to use for all endpoints. Overrides individual host settings.
-        Environment variable: ARIZE_SINGLE_HOST
-        Default: None (not set)
-
-    single_port : int, optional
-        Single port to use for all endpoints. Overrides individual port settings (0-65535).
-        Environment variable: ARIZE_SINGLE_PORT
-        Default: 0 (not set)
 
-    Examples:
-    --------
-    Initialize with API key only (other settings use defaults):
+    1. Explicit value passed to constructor (highest priority)
+    2. Environment variable value (see SDKConfiguration for variable names)
+    3. Built-in default value (lowest priority)
 
-        >>> client = ArizeClient(api_key="your-api-key")
+    Examples:
+        Initialize with API key only (other settings use defaults):
 
-    Initialize with custom endpoints:
+            >>> client = ArizeClient(api_key="your-api-key")
 
-        >>> client = ArizeClient(
-        ...     api_key="your-api-key",
-        ...     api_host="custom.api.com",
-        ...     flight_port=8443,
-        ... )
+        Initialize with custom endpoints:
 
-    Initialize with region (overrides host/port settings):
+            >>> client = ArizeClient(
+            ...     api_key="your-api-key",
+            ...     api_host="custom.api.com",
+            ...     flight_port=8443,
+            ... )
 
-        >>> from arize.regions import Region
-        >>> client = ArizeClient(api_key="your-api-key", region=Region.EU_WEST)
+        Initialize with region (overrides host/port settings):
 
-    See Also:
-    --------
-    SDKConfiguration : Internal configuration class with detailed parameter documentation
+            >>> from arize.regions import Region
+            >>> client = ArizeClient(
+            ...     api_key="your-api-key", region=Region.EU_WEST
+            ... )
     """
 
     _SUBCLIENTS: ClassVar[dict[str, tuple[str, str]]] = {
@@ -186,8 +98,8 @@ class ArizeClient(LazySubclientsMixin):
             "arize.projects.client",
             "ProjectsClient",
         ),
-        "models": (
-            "arize.models.client",
+        "ml": (
+            "arize.ml.client",
             "MLModelsClient",
         ),
         "spans": (
@@ -195,44 +107,16 @@ class ArizeClient(LazySubclientsMixin):
             "SpansClient",
         ),
     }
-    _EXTRAS: ClassVar[dict[str, tuple[str | None, tuple[str, ...]]]] = {
-        # Gate only the generated-backed ones
-        "datasets": (
-            "datasets-experiments",
-            (
-                "pydantic",
-                "openinference.semconv",
-            ),
-        ),
-        "experiments": (
-            "datasets-experiments",
-            (
-                "pydantic",
-                "wrapt",
-                # "numpy",
-                # "openinference.semconv",
-                # "opentelemetry.sdk",
-                # "opentelemetry.exporter.otlp.proto.grpc.trace_exporter",
-            ),
-        ),
-        "spans": (
-            "spans",
-            (
-                "google.protobuf",
-                "numpy",
-                "openinference.semconv",
-                "opentelemetry",
-                "pandas",
-                "pyarrow",
-                "requests",
-                "tqdm",
-            ),
-        ),
-        # Imports are gated in each method of the models client
-        # This is to allow for very lean package install if people only
-        # want to stream ML records
-        "models": (None, ()),
-    }
+    # DISABLED: Optional dependency gating system
+    # This dict would map subclients to their optional dependencies and extra names.
+    # When enabled, it prevents loading subclients if dependencies aren't installed,
+    # showing: "Install via: pip install arize[extra-name]"
+    #
+    # To re-enable, populate with entries like:
+    # "subclient_name": ("extra-name", ("package1", "package2", "package3")),
+    # "another_subclient": (None, ()),  # No requirements
+    #
+    _EXTRAS: ClassVar[dict[str, tuple[str | None, tuple[str, ...]]]] = {}
 
     def __init__(
         self,
@@ -299,8 +183,8 @@ class ArizeClient(LazySubclientsMixin):
             MissingAPIKeyError: If api_key is not provided via argument or environment variable.
 
         Notes:
-            Values provided here override environment variables, which in turn override
-            default values. See SDKConfiguration for detailed parameter documentation.
+            Values provided to this class override environment variables, which in turn
+            override default values. See SDKConfiguration for detailed parameter documentation.
         """
         cfg_kwargs: dict = {}
         if api_key is not None:
@@ -356,9 +240,9 @@ class ArizeClient(LazySubclientsMixin):
         return self.__getattr__("experiments")
 
     @property
-    def models(self) -> MLModelsClient:
-        """Access the models client for ML model operations (lazy-loaded)."""
-        return self.__getattr__("models")
+    def ml(self) -> MLModelsClient:
+        """Access the ML models client for ML model operations (lazy-loaded)."""
+        return self.__getattr__("ml")
 
     @property
     def projects(self) -> ProjectsClient:
@@ -385,7 +269,7 @@ class ArizeClient(LazySubclientsMixin):
         #     'datasets': lazy,
         #     'experiments': lazy,
         #     'spans': lazy,
-        #     'models': lazy,
+        #     'ml': lazy,
         #   }
         # )
         lines = [f"{self.__class__.__name__}("]