arize 8.0.0a23__tar.gz → 8.0.0b1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0a23
+Version: 8.0.0b1
 Summary: A helper library to interact with Arize AI APIs
 Project-URL: Homepage, https://arize.com
 Project-URL: Documentation, https://docs.arize.com/arize
@@ -26,41 +26,36 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Logging
 Classifier: Topic :: System :: Monitoring
 Requires-Python: >=3.10
-Requires-Dist: lazy-imports
 Requires-Dist: numpy>=2.0.0
-Provides-Extra: auto-embeddings
-Requires-Dist: datasets!=2.14.*,<3,>=2.8; extra == 'auto-embeddings'
-Requires-Dist: pandas<3,>=1.0.0; extra == 'auto-embeddings'
-Requires-Dist: pillow<11,>=8.4.0; extra == 'auto-embeddings'
-Requires-Dist: tokenizers<1,>=0.13; extra == 'auto-embeddings'
-Requires-Dist: torch<3,>=1.13; extra == 'auto-embeddings'
-Requires-Dist: transformers<5,>=4.25; extra == 'auto-embeddings'
-Provides-Extra: datasets-experiments
-Requires-Dist: numpy>=2.0.0; extra == 'datasets-experiments'
-Requires-Dist: pydantic; extra == 'datasets-experiments'
-Requires-Dist: wrapt<2.0.0,>=1.0.0; extra == 'datasets-experiments'
+Requires-Dist: openinference-semantic-conventions<1,>=0.1.25
+Requires-Dist: opentelemetry-exporter-otlp-proto-common>=1.38.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.38.0
+Requires-Dist: opentelemetry-sdk>=1.38.0
+Requires-Dist: opentelemetry-semantic-conventions<1,>=0.43b0
+Requires-Dist: pandas<3,>=2.0.0
+Requires-Dist: protobuf<6,>=4.21.0
+Requires-Dist: pyarrow>=0.15.0
+Requires-Dist: pydantic<3,>=2
+Requires-Dist: python-dateutil<3,>=2.8.2
+Requires-Dist: requests-futures<2,>=1.0.0
+Requires-Dist: requests<3,>=2.0.0
+Requires-Dist: tqdm<5,>4
+Requires-Dist: typing-extensions<5,>=4.7.1
+Requires-Dist: urllib3<3,>=2.1.0
+Requires-Dist: wrapt<2.0.0,>=1.0.0
 Provides-Extra: dev
 Requires-Dist: pytest==8.4.2; extra == 'dev'
 Requires-Dist: ruff==0.13.2; extra == 'dev'
-Provides-Extra: mimic-explainer
-Requires-Dist: interpret-community[mimic]<1,>=0.22.0; extra == 'mimic-explainer'
-Provides-Extra: ml-batch
-Requires-Dist: pandas<3,>=1.0.0; extra == 'ml-batch'
-Requires-Dist: protobuf<6,>=4.21.0; extra == 'ml-batch'
-Requires-Dist: pyarrow>=0.15.0; extra == 'ml-batch'
-Requires-Dist: requests<3,>=2.0.0; extra == 'ml-batch'
-Requires-Dist: tqdm; extra == 'ml-batch'
-Provides-Extra: ml-stream
-Requires-Dist: protobuf<6,>=4.21.0; extra == 'ml-stream'
-Requires-Dist: requests-futures<2,>=1.0.0; extra == 'ml-stream'
-Provides-Extra: spans
-Requires-Dist: openinference-semantic-conventions<1,>=0.1.21; extra == 'spans'
-Requires-Dist: opentelemetry-semantic-conventions<1,>=0.43b0; extra == 'spans'
-Requires-Dist: pandas<3,>=1.0.0; extra == 'spans'
-Requires-Dist: protobuf<6,>=4.21.0; extra == 'spans'
-Requires-Dist: pyarrow>=0.15.0; extra == 'spans'
-Requires-Dist: requests<3,>=2.0.0; extra == 'spans'
-Requires-Dist: tqdm; extra == 'spans'
+Provides-Extra: embeddings
+Requires-Dist: datasets!=2.14.*,<3,>=2.8; extra == 'embeddings'
+Requires-Dist: pillow<11,>=8.4.0; extra == 'embeddings'
+Requires-Dist: tokenizers<1,>=0.13; extra == 'embeddings'
+Requires-Dist: torch<3,>=1.13; extra == 'embeddings'
+Requires-Dist: transformers<5,>=4.25; extra == 'embeddings'
+Provides-Extra: mimic
+Requires-Dist: interpret-community[mimic]<1,>=0.22.0; extra == 'mimic'
+Provides-Extra: otel
+Requires-Dist: arize-otel<1,>=0.11.0; extra == 'otel'
 Description-Content-Type: text/markdown
 
 <p align="center">
@@ -124,16 +119,18 @@ Description-Content-Type: text/markdown
 
 A helper package to interact with Arize AI APIs.
 
-Arize is an AI engineering platform. It helps engineers develop, evaluate, and observe AI applications and agents. 
+Arize is an AI engineering platform. It helps engineers develop, evaluate, and observe AI applications and agents.
+
+Arize has both Enterprise and OSS products to support this goal:
 
-Arize has both Enterprise and OSS products to support this goal: 
 - [Arize AX](https://arize.com/) — an enterprise AI engineering platform from development to production, with an embedded AI Copilot
 - [Phoenix](https://github.com/Arize-ai/phoenix) — a lightweight, open-source project for tracing, prompt engineering, and evaluation
 - [OpenInference](https://github.com/Arize-ai/openinference) — an open-source instrumentation package to trace LLM applications across models and frameworks
 
-We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 million OSS downloads every month. 
+We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 million OSS downloads every month.
 
 # Key Features
+
 - [**_Tracing_**](https://docs.arize.com/arize/observe/tracing) - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
 - [**_Evaluation_**](https://docs.arize.com/arize/evaluate/online-evals) - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
 - [**_Datasets_**](https://docs.arize.com/arize/develop/datasets) - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
@@ -143,19 +140,21 @@ We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 m
 
 # Installation
 
-Install Arize (version 8 is currently under alpha release) via `pip` or `conda`:
+Install Arize (version 8 is currently under beta release) via `pip`:
 
 ```bash
-pip install arize==8.0.0ax 
+pip install --pre arize
 ```
-where `x` denotes the specific alpha release. Install the `arize-otel` package for auto-instrumentation of your LLM library:
+
+where `--pre` denotes the installation of pre-release versions. Install the
+`arize-otel` package for auto-instrumentation of your LLM library:
 
 ```bash
 pip install arize-otel
 ```
 
 # Usage
-  
+
 ## Instrumentation
 
 See [arize-otel in PyPI](https://pypi.org/project/arize-otel/):
@@ -177,9 +176,8 @@ OpenAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
 ## Operations on Spans
 
-Use `arize.spans` to interact with spans: log spans into Arize, update the span's evaluations, annotations and metadata in bulk. 
-
-> **WARNING**: This is currently under an alpha release. Install with `pip install arize==8.0.0ax` where the `x` denotes the specific alpha version. Check the [pre-releases](https://pypi.org/project/arize/#history) page in PyPI.
+Use `arize.spans` to interact with spans: log spans into Arize, update the span's
+evaluations, annotations and metadata in bulk.
 
 ### Logging spans
 
@@ -253,15 +251,14 @@ df = client.spans.export_to_df(
 
 ## Operations on ML Models
 
-Use `arize.models` to interact with ML models: log ML data (traininv, validation, production) into Arize, either streaming or in batches.
-
-> **WARNING**: This is currently under an alpha release. Install with `pip install arize==8.0.0ax` where the `x` denotes the specific alpha version. Check the [pre-releases](https://pypi.org/project/arize/#history) page in PyPI.
+Use `arize.ml` to interact with ML models: log ML data (training, validation, production)
+into Arize, either streaming or in batches.
 
 ### Stream log ML Data for a Classification use-case
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
@@ -270,7 +267,7 @@ MODEL_NAME = "<your-model-name>"
 features=...
 embedding_features=...
 
-response = client.models.log_stream(
+response = client.ml.log_stream(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     model_type=ModelTypes.SCORE_CATEGORICAL,
@@ -286,14 +283,14 @@ response = client.models.log_stream(
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
 MODEL_NAME = "<your-model-name>"
 MODEL_VERSION = "1.0"
 
-from arize.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
+from arize.ml.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
 
 tags = ["drift_type"]
 embedding_feature_column_names = {
@@ -322,7 +319,7 @@ schema = Schema(
 )
 
 # Logging Production DataFrame
-response = client.models.log_batch(
+response = client.ml.log_batch(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     model_type=ModelTypes.OBJECT_DETECTION,
@@ -350,7 +347,7 @@ SPACE_ID = "<your-space-id>"
 MODEL_NAME = "<your-model-name>"
 MODEL_VERSION = "1.0"
 
-df = client.models.export_to_df(
+df = client.ml.export_to_df(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     environment=Environments.TRAINING,
@@ -408,7 +405,7 @@ The response is an object of type `DatasetsList200Response`, and you can access
 
 ```python
 # Get the list of datasets from the response
-dataset_list = resp.datasets 
+dataset_list = resp.datasets
 # Get the response as a dictionary
 resp_dict = resp.to_dict()
 # Get the response in JSON format
@@ -524,7 +521,7 @@ The response is an object of type `ExperimentsList200Response`, and you can acce
 
 ```python
 # Get the list of datasets from the response
-experiment_list = resp.experiments 
+experiment_list = resp.experiments
 # Get the response as a dictionary
 resp_dict = resp.to_dict()
 # Get the response in JSON format
@@ -548,7 +545,7 @@ experiment, experiment_df = client.run_experiment(
     concurrency=..., # The number of concurrent tasks to run. Defaults to 3.
     set_global_tracer_provider=..., # If True, sets the global tracer provider for the experiment. Defaults to False
     exit_on_error=..., # If True, the experiment will stop running on first occurrence of an error. Defaults to False
-) 
+)
 ```
 
 The `Experiment` object also counts with convenience method similar to `List***` objects:
@@ -643,7 +640,7 @@ resp_df = resp.to_df()
 
 ### In Code
 
-You can use `configure_logging` to set up the logging behavior of the Arize package to your needs. 
+You can use `configure_logging` to set up the logging behavior of the Arize package to your needs.
 
 ```python
 from arize.logging import configure_logging
@@ -651,7 +648,7 @@ from arize.logging import configure_logging
 configure_logging(
     level=..., # Defaults to logging.INFO
     structured=..., # if True, emit JSON logs. Defaults to False
-) 
+)
 ```
 
 ### Via Environment Variables
@@ -662,11 +659,11 @@ Configure the same options as the section above, via:
 import os
 
 # Whether or not you want to disable logging altogether
-os.environ["ARIZE_LOG_ENABLE"] = "true" 
+os.environ["ARIZE_LOG_ENABLE"] = "true"
 # Set up the logging level
-os.environ["ARIZE_LOG_LEVEL"] = "debug" 
+os.environ["ARIZE_LOG_LEVEL"] = "debug"
 # Whether or not you want structured JSON logs
-os.environ["ARIZE_LOG_STRUCTURED"] = "false" 
+os.environ["ARIZE_LOG_STRUCTURED"] = "false"
 ```
 
 The default behavior of Arize's logs is: enabled, `INFO` level, and not structured.

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

@@ -59,16 +59,18 @@
 
 A helper package to interact with Arize AI APIs.
 
-Arize is an AI engineering platform. It helps engineers develop, evaluate, and observe AI applications and agents. 
+Arize is an AI engineering platform. It helps engineers develop, evaluate, and observe AI applications and agents.
+
+Arize has both Enterprise and OSS products to support this goal:
 
-Arize has both Enterprise and OSS products to support this goal: 
 - [Arize AX](https://arize.com/) — an enterprise AI engineering platform from development to production, with an embedded AI Copilot
 - [Phoenix](https://github.com/Arize-ai/phoenix) — a lightweight, open-source project for tracing, prompt engineering, and evaluation
 - [OpenInference](https://github.com/Arize-ai/openinference) — an open-source instrumentation package to trace LLM applications across models and frameworks
 
-We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 million OSS downloads every month. 
+We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 million OSS downloads every month.
 
 # Key Features
+
 - [**_Tracing_**](https://docs.arize.com/arize/observe/tracing) - Trace your LLM application's runtime using OpenTelemetry-based instrumentation.
 - [**_Evaluation_**](https://docs.arize.com/arize/evaluate/online-evals) - Leverage LLMs to benchmark your application's performance using response and retrieval evals.
 - [**_Datasets_**](https://docs.arize.com/arize/develop/datasets) - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
@@ -78,19 +80,21 @@ We log over 1 trillion inferences and spans, 10 million evaluation runs, and 2 m
 
 # Installation
 
-Install Arize (version 8 is currently under alpha release) via `pip` or `conda`:
+Install Arize (version 8 is currently under beta release) via `pip`:
 
 ```bash
-pip install arize==8.0.0ax 
+pip install --pre arize
 ```
-where `x` denotes the specific alpha release. Install the `arize-otel` package for auto-instrumentation of your LLM library:
+
+where `--pre` denotes the installation of pre-release versions. Install the
+`arize-otel` package for auto-instrumentation of your LLM library:
 
 ```bash
 pip install arize-otel
 ```
 
 # Usage
-  
+
 ## Instrumentation
 
 See [arize-otel in PyPI](https://pypi.org/project/arize-otel/):
@@ -112,9 +116,8 @@ OpenAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
 ## Operations on Spans
 
-Use `arize.spans` to interact with spans: log spans into Arize, update the span's evaluations, annotations and metadata in bulk. 
-
-> **WARNING**: This is currently under an alpha release. Install with `pip install arize==8.0.0ax` where the `x` denotes the specific alpha version. Check the [pre-releases](https://pypi.org/project/arize/#history) page in PyPI.
+Use `arize.spans` to interact with spans: log spans into Arize, update the span's
+evaluations, annotations and metadata in bulk.
 
 ### Logging spans
 
@@ -188,15 +191,14 @@ df = client.spans.export_to_df(
 
 ## Operations on ML Models
 
-Use `arize.models` to interact with ML models: log ML data (traininv, validation, production) into Arize, either streaming or in batches.
-
-> **WARNING**: This is currently under an alpha release. Install with `pip install arize==8.0.0ax` where the `x` denotes the specific alpha version. Check the [pre-releases](https://pypi.org/project/arize/#history) page in PyPI.
+Use `arize.ml` to interact with ML models: log ML data (training, validation, production)
+into Arize, either streaming or in batches.
 
 ### Stream log ML Data for a Classification use-case
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
@@ -205,7 +207,7 @@ MODEL_NAME = "<your-model-name>"
 features=...
 embedding_features=...
 
-response = client.models.log_stream(
+response = client.ml.log_stream(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     model_type=ModelTypes.SCORE_CATEGORICAL,
@@ -221,14 +223,14 @@ response = client.models.log_stream(
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
 MODEL_NAME = "<your-model-name>"
 MODEL_VERSION = "1.0"
 
-from arize.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
+from arize.ml.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
 
 tags = ["drift_type"]
 embedding_feature_column_names = {
@@ -257,7 +259,7 @@ schema = Schema(
 )
 
 # Logging Production DataFrame
-response = client.models.log_batch(
+response = client.ml.log_batch(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     model_type=ModelTypes.OBJECT_DETECTION,
@@ -285,7 +287,7 @@ SPACE_ID = "<your-space-id>"
 MODEL_NAME = "<your-model-name>"
 MODEL_VERSION = "1.0"
 
-df = client.models.export_to_df(
+df = client.ml.export_to_df(
     space_id=SPACE_ID,
     model_name=MODEL_NAME,
     environment=Environments.TRAINING,
@@ -343,7 +345,7 @@ The response is an object of type `DatasetsList200Response`, and you can access
 
 ```python
 # Get the list of datasets from the response
-dataset_list = resp.datasets 
+dataset_list = resp.datasets
 # Get the response as a dictionary
 resp_dict = resp.to_dict()
 # Get the response in JSON format
@@ -459,7 +461,7 @@ The response is an object of type `ExperimentsList200Response`, and you can acce
 
 ```python
 # Get the list of datasets from the response
-experiment_list = resp.experiments 
+experiment_list = resp.experiments
 # Get the response as a dictionary
 resp_dict = resp.to_dict()
 # Get the response in JSON format
@@ -483,7 +485,7 @@ experiment, experiment_df = client.run_experiment(
     concurrency=..., # The number of concurrent tasks to run. Defaults to 3.
     set_global_tracer_provider=..., # If True, sets the global tracer provider for the experiment. Defaults to False
     exit_on_error=..., # If True, the experiment will stop running on first occurrence of an error. Defaults to False
-) 
+)
 ```
 
 The `Experiment` object also counts with convenience method similar to `List***` objects:
@@ -578,7 +580,7 @@ resp_df = resp.to_df()
 
 ### In Code
 
-You can use `configure_logging` to set up the logging behavior of the Arize package to your needs. 
+You can use `configure_logging` to set up the logging behavior of the Arize package to your needs.
 
 ```python
 from arize.logging import configure_logging
@@ -586,7 +588,7 @@ from arize.logging import configure_logging
 configure_logging(
     level=..., # Defaults to logging.INFO
     structured=..., # if True, emit JSON logs. Defaults to False
-) 
+)
 ```
 
 ### Via Environment Variables
@@ -597,11 +599,11 @@ Configure the same options as the section above, via:
 import os
 
 # Whether or not you want to disable logging altogether
-os.environ["ARIZE_LOG_ENABLE"] = "true" 
+os.environ["ARIZE_LOG_ENABLE"] = "true"
 # Set up the logging level
-os.environ["ARIZE_LOG_LEVEL"] = "debug" 
+os.environ["ARIZE_LOG_LEVEL"] = "debug"
 # Whether or not you want structured JSON logs
-os.environ["ARIZE_LOG_STRUCTURED"] = "false" 
+os.environ["ARIZE_LOG_STRUCTURED"] = "false"
 ```
 
 The default behavior of Arize's logs is: enabled, `INFO` level, and not structured.

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

@@ -35,14 +35,23 @@ classifiers = [
     "Topic :: System :: Monitoring",
 ]
 dependencies = [
-    "numpy>=2.0.0", # For vector embeddings
-    "lazy-imports",
-    # "requests_futures==1.0.0",
-    # "googleapis_common_protos>=1.51.0,<2",
-    # "protobuf>=4.21.0,<7",
-    # "pyarrow>=0.15.0",
-    # "tqdm>=4.60.0,<5",
-    # "pydantic>=2.0.0,<3",
+    "numpy>=2.0.0",
+    "openinference-semantic-conventions>=0.1.25, <1",
+    "opentelemetry-exporter-otlp-proto-common>=1.38.0",
+    "opentelemetry-exporter-otlp-proto-grpc>=1.38.0",
+    "opentelemetry-sdk>=1.38.0",
+    "opentelemetry-semantic-conventions>=0.43b0, <1",
+    "pandas>=2.0.0, <3",
+    "protobuf>=4.21.0, <6",
+    "pyarrow>=0.15.0",
+    "pydantic>=2, <3",
+    "python-dateutil>=2.8.2, <3",
+    "requests>=2.0.0, <3",
+    "requests_futures>=1.0.0, <2",
+    "tqdm>4, <5", # For progress bars
+    "typing-extensions>=4.7.1, <5",
+    "urllib3>=2.1.0, <3",
+    "wrapt>=1.0.0, <2.0.0",
 ]
 dynamic = ["version"]
 
@@ -51,45 +60,15 @@ dev = [
     "pytest==8.4.2",
     "ruff==0.13.2"
 ]
-spans = [
-    "openinference-semantic-conventions>=0.1.21, <1",
-    "opentelemetry-semantic-conventions>=0.43b0, <1",
-    "pandas>=1.0.0,<3",
-    "protobuf>=4.21.0,<6",
-    "pyarrow>=0.15.0",
-    "requests>=2.0.0, <3", # For posting pyarrow files
-    "tqdm", # For export progress bars
-]
-ml-stream = [
-    "requests_futures>=1.0.0, <2",
-    "protobuf>=4.21.0,<6"
-]
-ml-batch = [
-    "pandas>=1.0.0,<3",
-    "protobuf>=4.21.0,<6",
-    "pyarrow>=0.15.0",
-    "requests>=2.0.0, <3", # For posting pyarrow files
-    "tqdm", # For export progress bars
-]
-datasets-experiments = [
-    "pydantic",
-    "numpy>=2.0.0",
-    "wrapt>=1.0.0,<2.0.0",
-    # "openinference-semantic-conventions>=0.1.21, <1",
-    # "opentelemetry-exporter-otlp-proto-common>=1.38.0",
-    # "opentelemetry-exporter-otlp-proto-grpc>=1.38.0",
-    # "opentelemetry-sdk>=1.38.0",
-    #
-    # "opentelemetry-api>=1.38.0",
-    # "opentelemetry-proto>=1.38.0",
+otel = [
+    "arize-otel>=0.11.0, <1",
 ]
-mimic-explainer = [
+mimic = [
     "interpret-community[mimic]>=0.22.0,<1"
 ]
-auto-embeddings = [
+embeddings = [
     "Pillow>=8.4.0, <11",
     "datasets>=2.8, <3, !=2.14.*",
-    "pandas>=1.0.0,<3",
     "tokenizers>=0.13, <1",
     "torch>=1.13, <3",
     "transformers>=4.25, <5",

{arize-8.0.0a23 → arize-8.0.0b1}/src/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-8.0.0a23 → arize-8.0.0b1}/src/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-8.0.0a23 → arize-8.0.0b1}/src/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-8.0.0a23 → arize-8.0.0b1}/src/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-8.0.0a23 → arize-8.0.0b1}/src/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-8.0.0a23 → arize-8.0.0b1}/src/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