arize 8.0.0b0__tar.gz → 8.0.0b1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0b0
+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,47 +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: openinference-semantic-conventions<1,>=0.1.21; extra == 'datasets-experiments'
-Requires-Dist: opentelemetry-api>=1.38.0; extra == 'datasets-experiments'
-Requires-Dist: opentelemetry-exporter-otlp-proto-common>=1.38.0; extra == 'datasets-experiments'
-Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.38.0; extra == 'datasets-experiments'
-Requires-Dist: opentelemetry-proto>=1.38.0; extra == 'datasets-experiments'
-Requires-Dist: opentelemetry-sdk>=1.38.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">
@@ -130,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.
@@ -149,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/):
@@ -183,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
 
@@ -259,9 +251,8 @@ 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
 
@@ -276,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,
@@ -328,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,
@@ -356,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,
@@ -414,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
@@ -530,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
@@ -554,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:
@@ -649,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
@@ -657,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
@@ -668,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.0b0 → 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,9 +191,8 @@ 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
 
@@ -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,
@@ -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.0b0 → 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,44 +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.0b0 → 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.0b0 → 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.0b0 → 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.0b0 → 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
 

{arize-8.0.0b0 → arize-8.0.0b1}/src/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-8.0.0b0 → arize-8.0.0b1}/src/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-8.0.0b0 → arize-8.0.0b1}/src/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 = '',