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

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

@@ -20,6 +20,8 @@ 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
 
@@ -105,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
-        "ml": (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,

arize/experiments/__init__.py

@@ -4,10 +4,10 @@ from arize.experiments.evaluators.types import (
     EvaluationResult,
     EvaluationResultFieldNames,
 )
-from arize.experiments.types import ExperimentTaskResultFieldNames
+from arize.experiments.types import ExperimentTaskFieldNames
 
 __all__ = [
     "EvaluationResult",
     "EvaluationResultFieldNames",
-    "ExperimentTaskResultFieldNames",
+    "ExperimentTaskFieldNames",
 ]

arize/experiments/client.py

@@ -43,7 +43,7 @@ if TYPE_CHECKING:
     from arize.experiments.evaluators.types import EvaluationResultFieldNames
     from arize.experiments.types import (
         ExperimentTask,
-        ExperimentTaskResultFieldNames,
+        ExperimentTaskFieldNames,
     )
 
 logger = logging.getLogger(__name__)
@@ -114,7 +114,7 @@ class ExperimentsClient:
         name: str,
         dataset_id: str,
         experiment_runs: list[dict[str, object]] | pd.DataFrame,
-        task_fields: ExperimentTaskResultFieldNames,
+        task_fields: ExperimentTaskFieldNames,
         evaluator_columns: dict[str, EvaluationResultFieldNames] | None = None,
         force_http: bool = False,
     ) -> models.Experiment:
@@ -175,7 +175,6 @@ class ExperimentsClient:
             from arize._generated import api_client as gen
 
             data = experiment_df.to_dict(orient="records")
-
             body = gen.ExperimentsCreateRequest(
                 name=name,
                 dataset_id=dataset_id,

arize/experiments/functions.py

@@ -56,7 +56,7 @@ from arize.experiments.types import (
     ExperimentEvaluationRun,
     ExperimentRun,
     ExperimentTask,
-    ExperimentTaskResultFieldNames,
+    ExperimentTaskFieldNames,
     _TaskSummary,
 )
 
@@ -768,7 +768,7 @@ def get_result_attr(r: object, attr: str, default: object = None) -> object:
 
 def transform_to_experiment_format(
     experiment_runs: list[dict[str, object]] | pd.DataFrame,
-    task_fields: ExperimentTaskResultFieldNames,
+    task_fields: ExperimentTaskFieldNames,
     evaluator_fields: dict[str, EvaluationResultFieldNames] | None = None,
 ) -> pd.DataFrame:
     """Transform a DataFrame to match the format returned by run_experiment().
@@ -788,7 +788,7 @@ def transform_to_experiment_format(
         else pd.DataFrame(experiment_runs)
     )
     # Validate required columns
-    required_cols = {task_fields.example_id, task_fields.result}
+    required_cols = {task_fields.example_id, task_fields.output}
     missing_cols = required_cols - set(data.columns)
     if missing_cols:
         raise ValueError(f"Missing required columns: {missing_cols}")
@@ -799,11 +799,11 @@ def transform_to_experiment_format(
     out_df["example_id"] = data[task_fields.example_id]
     if task_fields.example_id != "example_id":
         out_df.drop(task_fields.example_id, axis=1, inplace=True)
-    out_df["result"] = data[task_fields.result].apply(
+    out_df["output"] = data[task_fields.output].apply(
         lambda x: json.dumps(x) if isinstance(x, dict) else x
     )
-    if task_fields.result != "result":
-        out_df.drop(task_fields.result, axis=1, inplace=True)
+    if task_fields.output != "output":
+        out_df.drop(task_fields.output, axis=1, inplace=True)
 
     # Process evaluator results
     if evaluator_fields:

arize/experiments/types.py

@@ -397,17 +397,17 @@ def _top_string(s: pd.Series, length: int = 100) -> str | None:
 
 
 @dataclass
-class ExperimentTaskResultFieldNames:
+class ExperimentTaskFieldNames:
     """Column names for mapping experiment task results in a DataFrame.
 
     Args:
         example_id: Name of column containing example IDs.
             The ID values must match the id of the dataset rows.
-        result: Name of column containing task results
+        output: Name of column containing task results
     """
 
     example_id: str
-    result: str
+    output: str
 
 
 TaskOutput = JSONSerializable

arize/spans/client.py

@@ -92,7 +92,7 @@ class SpansClient:
                 before sending to Arize.
 
         Returns:
-            `Response` object
+            Response object from the HTTP request.
 
         """
         from arize.spans.columns import (
@@ -1011,7 +1011,7 @@ class SpansClient:
         WHERE clauses and similarity search for semantic retrieval.
 
         Returns:
-            pd.DataFrame: DataFrame containing the requested span data with columns
+            DataFrame containing the requested span data with columns
                 for span metadata, attributes, events, and any custom fields.
         """
         with ArizeFlightClient(

arize/version.py

@@ -1,3 +1,3 @@
 """Version information for the Arize SDK."""
 
-__version__ = "8.0.0b0"
+__version__ = "8.0.0b1"

{arize-8.0.0b0.dist-info → arize-8.0.0b1.dist-info}/METADATA

@@ -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.dist-info → arize-8.0.0b1.dist-info}/RECORD

@@ -1,11 +1,11 @@
 arize/__init__.py,sha256=C2NOiNcFL9TMu5NTkJwF88kL6PKAHHjBMG8HTAIQ5TU,3551
 arize/_lazy.py,sha256=aLdgSGxdfdV9QiLllxlLlkN5k44pFf_7jdKRV9Z9sgo,2864
-arize/client.py,sha256=RwnDTmxHzPfpzonYD8u_d4V-RFyehpbUADZV8Xpki0c,13660
+arize/client.py,sha256=XGOaze-rfvDscL9vpWohen9vUbTNwLsx27pF77p9zxA,13150
 arize/config.py,sha256=80iQgiaAhSqgE1Q-nKlivjQ2z3F7PifiPAsZlhQjozo,16471
 arize/logging.py,sha256=CZkgdpDKXoNNJ9fQMNwbCQEe-ipUg7fQf_CE1hRN1Qc,8574
 arize/pre_releases.py,sha256=QWZb5Kko-fHDKDKNk8HUwehMQuLrZPYnyD2c6_mnEqI,1216
 arize/regions.py,sha256=oEKJdZbW5r8X3fL3PaDVMdk-f2MoZ8OBvUvYZLMGBpw,1039
-arize/version.py,sha256=OIc4vIJor74VQDoVG5zuAmxWI0h4EEU19vIPa3y2Tjc,70
+arize/version.py,sha256=wAs5eLjhoEbU63izq_pHXyel5O0ilqn4QY6kj6xnago,70
 arize/_exporter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/_exporter/client.py,sha256=DRxdREi4k0RiRI-kPvYCwX2zo6DGkdt1XMezF4ZhR08,16425
 arize/_exporter/validation.py,sha256=9gX4oHrZcChMfkwoBpe3Qp4aK-txy_zZ4azrKf-bEQw,1022
@@ -15,8 +15,8 @@ arize/_flight/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/_flight/client.py,sha256=HpUu22Q2r-jnh_j0RaQtJLZnaXKGBNNjZlZ0dvX98xs,20391
 arize/_flight/types.py,sha256=GB_4dQu2ElIrcDGAcqhG7oI4g-b0ZdSlbrQkf0TFzVE,194
 arize/_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize/_generated/api_client_README.md,sha256=SfBbE1r1thnfwk3AVr6CZUP6kJ7HNy28gx1V7DjQhp8,7451
-arize/_generated/api_client/__init__.py,sha256=I_KGjOG-2Z5dokY6Ln9E5fFHUIe53WUvCaUvg4DtQss,4672
+arize/_generated/api_client_README.md,sha256=9UxtVJdMFIdV-_9NqUwg2ox3Many_5LIyXz7ZqYOdU0,7379
+arize/_generated/api_client/__init__.py,sha256=QjXl6DlavvyG5VRCT_-U2_eS_JeFIUVoMIWr19vroK8,4554
 arize/_generated/api_client/api_client.py,sha256=Cr9leKLpAJYQmL_U1hAPsMNFdaSO6Ti1X52jvGp2WsQ,27899
 arize/_generated/api_client/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
 arize/_generated/api_client/configuration.py,sha256=EJ6DkTKzHWqjN0BwMEHQ1D9pEDdClOsM72EcffquZ9I,19406
@@ -26,13 +26,13 @@ arize/_generated/api_client/api/__init__.py,sha256=ycVTd-FBlPyFX82wbCQiuCTbj78pZ
 arize/_generated/api_client/api/datasets_api.py,sha256=pwS_NU7puzAdoqp49D24XjgcReptPue6K2ZI_C8DwYY,102867
 arize/_generated/api_client/api/experiments_api.py,sha256=xMMH-QA80mmXXPbGKKPIONMnxNowSGlf8wVdLKqzPss,65311
 arize/_generated/api_client/api/projects_api.py,sha256=SRMimrrXNZLlKD9zbTFs-iy2SeOHiVDdlMB0Hbp0ApE,48820
-arize/_generated/api_client/models/__init__.py,sha256=qajnlyF3Q2lCuIIeCqldVFcdgogFPd2iLr5YMYv8Alw,2261
+arize/_generated/api_client/models/__init__.py,sha256=_kbyaJU4uC0Gdnc87FsdBDfb-QcH8LKID7YNsk11JSk,2183
 arize/_generated/api_client/models/dataset.py,sha256=eStUwLDPU7j3kMTdSO7RqMCScsGU9rOZtU41xnSXUUs,4332
 arize/_generated/api_client/models/dataset_example.py,sha256=avmtSuitMruY7Yphw5gDpPbfkx_GKDhePBSoG6S_sc0,3825
 arize/_generated/api_client/models/dataset_example_update.py,sha256=DMV1_IkN74tQJrnWFPsJ-7Yj9d-tBaF2rVZZlktprpo,3290
 arize/_generated/api_client/models/dataset_version.py,sha256=OlkhTlcQuNvxJdNxxU5CpHZ9a2OCro5FRPAUMCYLBdE,3709
-arize/_generated/api_client/models/datasets_create_request.py,sha256=av3O6kxoeLL1JACsDL_SrHnNKTNVMHlUyrcBhrsiOac,3673
-arize/_generated/api_client/models/datasets_examples_insert_request.py,sha256=u_9QqAgcEUVUyZoZJCWnd9Wuz00-vyRrDKL-82hXO4U,3456
+arize/_generated/api_client/models/datasets_create_request.py,sha256=jUARKZVlr8hrkuf2Y59ct466f2PdHtAD2O02d5lqE_w,3149
+arize/_generated/api_client/models/datasets_examples_insert_request.py,sha256=MVuOJ85lDpzVZssJ76d-cgK3lQkphTMwFAumwHTG6BQ,2932
 arize/_generated/api_client/models/datasets_examples_list200_response.py,sha256=fIY3yg5cbIcRBdawxBRnWqDt5dxOLN1Pxju_g6BDU30,3869
 arize/_generated/api_client/models/datasets_examples_update_request.py,sha256=awa70zUxyMPLiEddQmcigVCR4tPhqTjVL6ClV-CeqK4,3797
 arize/_generated/api_client/models/datasets_list200_response.py,sha256=KsDfQU44q7gg7TwR5MDmJfkp69-SNTo_c87d9uz_8fk,3777
@@ -43,7 +43,6 @@ arize/_generated/api_client/models/experiments_create_request.py,sha256=L0DllTYt
 arize/_generated/api_client/models/experiments_list200_response.py,sha256=AKot7N4isrGFHNr6-mXsSr8ikeo5veneFFsuGYRKSRc,3843
 arize/_generated/api_client/models/experiments_runs_list200_response.py,sha256=vkaYsUKdAnChSBKCEzxc1DrE5vs_0tG1Ya-yXhjuPBo,3979
 arize/_generated/api_client/models/pagination_metadata.py,sha256=2NgzlV140yhFqS8IURPfMrVtW2p-0Xr3DaA9GxFfInw,3262
-arize/_generated/api_client/models/primitive_value.py,sha256=iDHyARo1ILnJymsS9EaVFE5V37X0BvUB6uTX_aVhKbg,6400
 arize/_generated/api_client/models/problem.py,sha256=mbAAHZAC8mKYC0r902SGE4QcLCUMfSzFJOLvjyjO8Pw,3584
 arize/_generated/api_client/models/project.py,sha256=w4OT222655N-dd8IHauiDobgUyMnol4xzZ80Rniv3ts,3636
 arize/_generated/api_client/models/projects_create_request.py,sha256=hnnerivgBpJ4BNcdE1hfZ8EoIawtmxsx1iv8NVoHkw4,3024
@@ -54,20 +53,19 @@ arize/_generated/api_client/test/test_dataset_example.py,sha256=musVzDrorF1yf5oz
 arize/_generated/api_client/test/test_dataset_example_update.py,sha256=Z59KydV8_su5QlPdFx7Vs4WAOh-NEYvNVMsIe_5bz0k,1521
 arize/_generated/api_client/test/test_dataset_version.py,sha256=rPczEI--Htxd5tRYzCgNMYEisu7ub87EOxRE-ZIEbp0,1996
 arize/_generated/api_client/test/test_datasets_api.py,sha256=2WYqfqygypzUEB61iMzqWZRAZIyYhrZQ3LaNZDAOIhc,1707
-arize/_generated/api_client/test/test_datasets_create_request.py,sha256=wjDWgl-NUlaQZTM0eDpVN15fvJ0tH9c3RegBwOKAYCk,1873
-arize/_generated/api_client/test/test_datasets_examples_insert_request.py,sha256=N3mAB0BjNAXGdEyN6l3BmRiUeXG4WUoRxgjtvjmWX7w,1854
-arize/_generated/api_client/test/test_datasets_examples_list200_response.py,sha256=gfm4cmvhpOr0bXLgoRN1G2I9wc155F4IPDjvVySLWcw,2244
-arize/_generated/api_client/test/test_datasets_examples_update_request.py,sha256=a1m-IaI0RGUd3dd6ktQG0nADEORX_r0KLbY-9IqbwBw,1888
+arize/_generated/api_client/test/test_datasets_create_request.py,sha256=GrWJg_lguEbsJWCUw5QJJjSK-fpSBtUsVFJpntw_t20,1751
+arize/_generated/api_client/test/test_datasets_examples_insert_request.py,sha256=1vegfmE4E8kGoELac7jML3iQ1Vi2IlkUXgAKJOab-sw,1732
+arize/_generated/api_client/test/test_datasets_examples_list200_response.py,sha256=C1nYlj8ggOHmczjyPDfBRU4dTvd-f7_xwbe_0fPuR58,2122
+arize/_generated/api_client/test/test_datasets_examples_update_request.py,sha256=_BZqsFRz2lXIkgwOpWfllnZWdrhqDozYiADANQlPVKA,1766
 arize/_generated/api_client/test/test_datasets_list200_response.py,sha256=jRWD_HBWE3doypQuaexSp39F_fFdp-aJVXsosxM36M8,3905
 arize/_generated/api_client/test/test_experiment.py,sha256=sOwd8Rep47jAf7vhq1hm3rK0P7I0VpGS9cZJLOb1RSE,2080
 arize/_generated/api_client/test/test_experiment_run.py,sha256=FPmaQpEPqs6_jR-o7S-Ejaj2OKeJjO7pLOr-r51fetk,1560
 arize/_generated/api_client/test/test_experiment_run_create.py,sha256=oZorTWEiI6fTmSFw35vZe-1Npw92Ne01nTIbRmfPnAc,1583
 arize/_generated/api_client/test/test_experiments_api.py,sha256=Ez6hw0XNW9y2IDLR6_7YlgjbTXgfZr9JxBoaVxEU8N8,1414
-arize/_generated/api_client/test/test_experiments_create_request.py,sha256=7bNXzkDE3kL5VZCi1ogey9pWmWWqGi_rDG60_93bQoA,1927
+arize/_generated/api_client/test/test_experiments_create_request.py,sha256=-TFrUHrVIR6ewNtMMaVDKQA5XdC3bThr5Z4_WX4aovs,1805
 arize/_generated/api_client/test/test_experiments_list200_response.py,sha256=NlLuUOjLRXlUqFsgqSI04B2wCl_4OEZUMtGJrVGUwWA,3085
-arize/_generated/api_client/test/test_experiments_runs_list200_response.py,sha256=BtOi_RMEYJciMCazYfmCkR3BbXi8VDdYm1qs_8razkE,2246
+arize/_generated/api_client/test/test_experiments_runs_list200_response.py,sha256=M6BVVxY6wI_57yR9tkS3qZEgYEWlaPar10-rP6_zU7Q,2124
 arize/_generated/api_client/test/test_pagination_metadata.py,sha256=6I0fz9C-PeS0xGlqfnpCb95jieoCp6wzC_77pFefiI4,1546
-arize/_generated/api_client/test/test_primitive_value.py,sha256=e3Hj8RATS82D1QpAtxysqRrgccFaYXHJHsX0hkLNC90,1399
 arize/_generated/api_client/test/test_problem.py,sha256=cP3YV_YYLjkZuh7RCKUoqJy1FuxukuRj1swYKzFmz9g,1514
 arize/_generated/api_client/test/test_project.py,sha256=89LB5u8gUqrNBGWt9UfC47A9pBFpLZlJ1DqwjUGKsfU,1693
 arize/_generated/api_client/test/test_projects_api.py,sha256=oFZh9KZlPpc-Jaf5IQ1NhLOHJf0sJBHAJIrvPACYmXI,1207
@@ -106,11 +104,11 @@ arize/exceptions/parameters.py,sha256=xbF38TNZI14PruYMor6DYVMB7uVhyusJfRPUn_YPHz
 arize/exceptions/spaces.py,sha256=RZdXCnxkc-t5_jY2je7gVUDxaXhBQvKpXcUGXptO8mA,631
 arize/exceptions/types.py,sha256=ks-4B2YpiVxHVopWUh4Gv_iM2h-LhdJ_zZDXtN3Ctc0,9468
 arize/exceptions/values.py,sha256=rzT8sZbY2m2Vxc5vt5V-ZjsQyiGbxM0pxPi-bp9d-pU,27920
-arize/experiments/__init__.py,sha256=8ESMbJbtGc3VGDo4efQxxCeXmuZlT-0tPv-EBS48Xpo,358
-arize/experiments/client.py,sha256=ScUgT44GkNJ6SdwctHeLPNefev3hLCjW55PVkVf_-xI,26233
-arize/experiments/functions.py,sha256=b51Dv1qk4oiDiJv6VZgpBhfdKlo4ff8FLTtH5ruEPOA,34675
+arize/experiments/__init__.py,sha256=zQAEQggDaJqZaubqjCJfmqbhemMeY_WakWI4ul-wkk8,346
+arize/experiments/client.py,sha256=wRIVy2VtQ0LsvTRtfBz78cNnUkP1quzMCReUg2L1rHU,26220
+arize/experiments/functions.py,sha256=4rQEB7eB3fvmdjgbCPRRltWlGwXMj1qLSrnCWedP2qY,34663
 arize/experiments/tracing.py,sha256=P3Q2DTUGNsRhvxRNAx_NXY30Id8y7cw9MV9piduSbB0,10207
-arize/experiments/types.py,sha256=qwv-LV9DqfdlKwXc4SdL9p2QsI8pHoEy2_HCldzsdZs,14093
+arize/experiments/types.py,sha256=N9FuJx4atV3VxpdVXch6gTI3ZH6guVrJQzfbeJnDxwQ,14087
 arize/experiments/evaluators/__init__.py,sha256=FUTfWRir8jaY8PC3pRHUy5m4RqrKRBPD1EJ4GYkIMLY,76
 arize/experiments/evaluators/base.py,sha256=m9Gk2efWyssSYnZIdSJouB6CNx1KkqHjHUrRWft1L54,10981
 arize/experiments/evaluators/exceptions.py,sha256=iqg5-OcGfRWY354OCcdmVC39cb3jEb-RO1LnVn7BD7M,376
@@ -133,7 +131,7 @@ arize/ml/surrogate_explainer/mimic.py,sha256=Zx5idIHUyln0AxsU4K0NJoQ1HFOtKk0GoA6
 arize/projects/__init__.py,sha256=hx29_muM9ZJxH5j518QJ88-_VVRAFymF5WLRvQVpDH8,64
 arize/projects/client.py,sha256=hjyqg8yfINEIyorPIEwndurnaIfQimjxHQG6AQ_lt-U,4341
 arize/spans/__init__.py,sha256=rXRx4DshOoP9WmbVRUyYdM_QHtUmUTbfO2pZPkztKL0,57
-arize/spans/client.py,sha256=45wkvUmdxtYQKNp75RQYSkuM_Hwx_6Uf-V8osF0h8S8,48785
+arize/spans/client.py,sha256=N3qa6jAd6JXMZr5pb8XbbVl58-6BgrSRmGSFGn3RdKk,48792
 arize/spans/columns.py,sha256=Sap8UrOeeFebxsI1LbelustuirAPmKrtjW8z3vHJVFk,13378
 arize/spans/conversion.py,sha256=9GrNd9puZtJJZma2C6UMmNEcTfkb4maG1BxkRiE7zZU,4937
 arize/spans/validation/__init__.py,sha256=QSb0EFPBCI2riIrgtSaFXPDGDPQ_aYK4N1Dej499cu8,55
@@ -168,8 +166,8 @@ arize/utils/size.py,sha256=qIjUYYg-NttqK25ObX17_32MzkAepLJDDv8N7YxJwAo,805
 arize/utils/types.py,sha256=BxAQwrqRMeA5BaDZcmEczcKkd-T33mi25FL9kBroNXA,2892
 arize/utils/online_tasks/__init__.py,sha256=T7778_bgutP0U7h8e9NRPYfwBQSmHOi859M1K2fjpjw,213
 arize/utils/online_tasks/dataframe_preprocessor.py,sha256=-VcQSetJjZJamWyZryvkA1LhPOT7BdqDGOL4hP-wtik,9399
-arize-8.0.0b0.dist-info/METADATA,sha256=wRxcpzxQ06IxGjITW8n-kRBjEr_DlocmQXRAIW74oA0,29211
-arize-8.0.0b0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-arize-8.0.0b0.dist-info/licenses/LICENSE,sha256=Yseh419WQGiW16p8pS0MwNJyrAIrXSeW59aQXbijY2o,9723
-arize-8.0.0b0.dist-info/licenses/NOTICE,sha256=V9Mdpy_w2tgo5GxgjpsZFTu2WcltLQIh6wBVFwBnJuY,550
-arize-8.0.0b0.dist-info/RECORD,,
+arize-8.0.0b1.dist-info/METADATA,sha256=rkXsmAOix3klcDApv4J3sJbXyEO0cF8zsTWXzE_o_N8,27749
+arize-8.0.0b1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+arize-8.0.0b1.dist-info/licenses/LICENSE,sha256=Yseh419WQGiW16p8pS0MwNJyrAIrXSeW59aQXbijY2o,9723
+arize-8.0.0b1.dist-info/licenses/NOTICE,sha256=V9Mdpy_w2tgo5GxgjpsZFTu2WcltLQIh6wBVFwBnJuY,550
+arize-8.0.0b1.dist-info/RECORD,,

arize/_generated/api_client/models/primitive_value.py

@@ -1,172 +0,0 @@
-# coding: utf-8
-
-"""
-    Arize REST API
-
-    API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment. 
-
-    The version of the OpenAPI document: 2.0.0
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-    Do not edit the class manually.
-"""  # noqa: E501
-
-
-from __future__ import annotations
-from inspect import getfullargspec
-import json
-import pprint
-import re  # noqa: F401
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, ValidationError, field_validator
-from typing import Optional, Union
-from typing import Union, Any, List, Set, TYPE_CHECKING, Optional, Dict
-from typing_extensions import Literal, Self
-from pydantic import Field
-
-PRIMITIVEVALUE_ANY_OF_SCHEMAS = ["bool", "float", "int", "str"]
-
-class PrimitiveValue(BaseModel):
-    """
-    A JSON primitive value (string, number, integer, or boolean)
-    """
-
-    # data type: int
-    anyof_schema_1_validator: Optional[StrictInt] = None
-    # data type: float
-    anyof_schema_2_validator: Optional[Union[StrictFloat, StrictInt]] = None
-    # data type: str
-    anyof_schema_3_validator: Optional[StrictStr] = None
-    # data type: bool
-    anyof_schema_4_validator: Optional[StrictBool] = None
-    if TYPE_CHECKING:
-        actual_instance: Optional[Union[bool, float, int, str]] = None
-    else:
-        actual_instance: Any = None
-    any_of_schemas: Set[str] = { "bool", "float", "int", "str" }
-
-    model_config = {
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
-
-    def __init__(self, *args, **kwargs) -> None:
-        if args:
-            if len(args) > 1:
-                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
-            if kwargs:
-                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
-            super().__init__(actual_instance=args[0])
-        else:
-            super().__init__(**kwargs)
-
-    @field_validator('actual_instance')
-    def actual_instance_must_validate_anyof(cls, v):
-        instance = PrimitiveValue.model_construct()
-        error_messages = []
-        # validate data type: int
-        try:
-            instance.anyof_schema_1_validator = v
-            return v
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # validate data type: float
-        try:
-            instance.anyof_schema_2_validator = v
-            return v
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # validate data type: str
-        try:
-            instance.anyof_schema_3_validator = v
-            return v
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # validate data type: bool
-        try:
-            instance.anyof_schema_4_validator = v
-            return v
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        if error_messages:
-            # no match
-            raise ValueError("No match found when setting the actual_instance in PrimitiveValue with anyOf schemas: bool, float, int, str. Details: " + ", ".join(error_messages))
-        else:
-            return v
-
-    @classmethod
-    def from_dict(cls, obj: Dict[str, Any]) -> Self:
-        return cls.from_json(json.dumps(obj))
-
-    @classmethod
-    def from_json(cls, json_str: str) -> Self:
-        """Returns the object represented by the json string"""
-        instance = cls.model_construct()
-        error_messages = []
-        # deserialize data into int
-        try:
-            # validation
-            instance.anyof_schema_1_validator = json.loads(json_str)
-            # assign value to actual_instance
-            instance.actual_instance = instance.anyof_schema_1_validator
-            return instance
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # deserialize data into float
-        try:
-            # validation
-            instance.anyof_schema_2_validator = json.loads(json_str)
-            # assign value to actual_instance
-            instance.actual_instance = instance.anyof_schema_2_validator
-            return instance
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # deserialize data into str
-        try:
-            # validation
-            instance.anyof_schema_3_validator = json.loads(json_str)
-            # assign value to actual_instance
-            instance.actual_instance = instance.anyof_schema_3_validator
-            return instance
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # deserialize data into bool
-        try:
-            # validation
-            instance.anyof_schema_4_validator = json.loads(json_str)
-            # assign value to actual_instance
-            instance.actual_instance = instance.anyof_schema_4_validator
-            return instance
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-
-        if error_messages:
-            # no match
-            raise ValueError("No match found when deserializing the JSON string into PrimitiveValue with anyOf schemas: bool, float, int, str. Details: " + ", ".join(error_messages))
-        else:
-            return instance
-
-    def to_json(self) -> str:
-        """Returns the JSON representation of the actual instance"""
-        if self.actual_instance is None:
-            return "null"
-
-        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
-            return self.actual_instance.to_json()
-        else:
-            return json.dumps(self.actual_instance)
-
-    def to_dict(self) -> Optional[Union[Dict[str, Any], bool, float, int, str]]:
-        """Returns the dict representation of the actual instance"""
-        if self.actual_instance is None:
-            return None
-
-        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
-            return self.actual_instance.to_dict()
-        else:
-            return self.actual_instance
-
-    def to_str(self) -> str:
-        """Returns the string representation of the actual instance"""
-        return pprint.pformat(self.model_dump())
-
-

arize/_generated/api_client/test/test_primitive_value.py

@@ -1,50 +0,0 @@
-# coding: utf-8
-
-"""
-    Arize REST API
-
-    API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment. 
-
-    The version of the OpenAPI document: 2.0.0
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-    Do not edit the class manually.
-"""  # noqa: E501
-
-
-import unittest
-
-from arize._generated.api_client.models.primitive_value import PrimitiveValue
-
-class TestPrimitiveValue(unittest.TestCase):
-    """PrimitiveValue unit test stubs"""
-
-    def setUp(self):
-        pass
-
-    def tearDown(self):
-        pass
-
-    def make_instance(self, include_optional) -> PrimitiveValue:
-        """Test PrimitiveValue
-            include_optional is a boolean, when False only required
-            params are included, when True both required and
-            optional params are included """
-        # uncomment below to create an instance of `PrimitiveValue`
-        """
-        model = PrimitiveValue()
-        if include_optional:
-            return PrimitiveValue(
-            )
-        else:
-            return PrimitiveValue(
-        )
-        """
-
-    def testPrimitiveValue(self):
-        """Test PrimitiveValue"""
-        # inst_req_only = self.make_instance(include_optional=False)
-        # inst_req_and_optional = self.make_instance(include_optional=True)
-
-if __name__ == '__main__':
-    unittest.main()