arize 8.0.0b0__tar.gz → 8.0.0b2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0b0
+Version: 8.0.0b2
 Summary: A helper library to interact with Arize AI APIs
 Project-URL: Homepage, https://arize.com
 Project-URL: Documentation, https://docs.arize.com/arize
@@ -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">
@@ -92,6 +81,8 @@ Description-Content-Type: text/markdown
 - [Overview](#overview)
 - [Key Features](#key-features)
 - [Installation](#installation)
+  - [Optional Dependencies](#optional-dependencies)
+  - [Migrating from Version 7](#migrating-from-version-7)
 - [Usage](#usage)
   - [Instrumentation](#instrumentation)
   - [Operations on Spans](#operations-on-spans)
@@ -130,16 +121,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 +142,36 @@ 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 the base package:
 
 ```bash
-pip install arize==8.0.0ax 
+pip install arize
 ```
-where `x` denotes the specific alpha release. Install the `arize-otel` package for auto-instrumentation of your LLM library:
+
+## Optional Dependencies
+
+The following optional extras provide specialized functionality:
+
+> **Note:** The `otel` extra installs the `arize-otel` package, which is also available as a standalone package. If you only need auto-instrumentation without the full SDK, install `arize-otel` directly.
+
+| Extra | Install Command | What It Provides |
+|-------|----------------|------------------|
+| **otel** | `pip install arize[otel]` | OpenTelemetry auto-instrumentation package (arize-otel) for automatic tracing |
+| **embeddings** | `pip install arize[embeddings]` | Automatic embedding generation for NLP, CV, and structured data (Pillow, datasets, tokenizers, torch, transformers) |
+| **mimic** | `pip install arize[mimic]` | MIMIC explainer for model interpretability |
+
+Install multiple extras:
 
 ```bash
-pip install arize-otel
+pip install arize[otel,embeddings,mimic]
 ```
 
+## Migrating from Version 7
+
+If you're upgrading from version 7, please refer to the [Migration Guide](https://arize.com/docs/api-clients/python/version-8/migration) for detailed migration steps and breaking changes.
+
 # Usage
-  
+
 ## Instrumentation
 
 See [arize-otel in PyPI](https://pypi.org/project/arize-otel/):
@@ -183,9 +193,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 +268,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 +284,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 +336,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 +364,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 +422,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 +538,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 +562,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 +657,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 +665,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 +676,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.0b2}/README.md

@@ -21,6 +21,8 @@
 - [Overview](#overview)
 - [Key Features](#key-features)
 - [Installation](#installation)
+  - [Optional Dependencies](#optional-dependencies)
+  - [Migrating from Version 7](#migrating-from-version-7)
 - [Usage](#usage)
   - [Instrumentation](#instrumentation)
   - [Operations on Spans](#operations-on-spans)
@@ -59,16 +61,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 +82,36 @@ 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 the base package:
 
 ```bash
-pip install arize==8.0.0ax 
+pip install arize
 ```
-where `x` denotes the specific alpha release. Install the `arize-otel` package for auto-instrumentation of your LLM library:
+
+## Optional Dependencies
+
+The following optional extras provide specialized functionality:
+
+> **Note:** The `otel` extra installs the `arize-otel` package, which is also available as a standalone package. If you only need auto-instrumentation without the full SDK, install `arize-otel` directly.
+
+| Extra | Install Command | What It Provides |
+|-------|----------------|------------------|
+| **otel** | `pip install arize[otel]` | OpenTelemetry auto-instrumentation package (arize-otel) for automatic tracing |
+| **embeddings** | `pip install arize[embeddings]` | Automatic embedding generation for NLP, CV, and structured data (Pillow, datasets, tokenizers, torch, transformers) |
+| **mimic** | `pip install arize[mimic]` | MIMIC explainer for model interpretability |
+
+Install multiple extras:
 
 ```bash
-pip install arize-otel
+pip install arize[otel,embeddings,mimic]
 ```
 
+## Migrating from Version 7
+
+If you're upgrading from version 7, please refer to the [Migration Guide](https://arize.com/docs/api-clients/python/version-8/migration) for detailed migration steps and breaking changes.
+
 # Usage
-  
+
 ## Instrumentation
 
 See [arize-otel in PyPI](https://pypi.org/project/arize-otel/):
@@ -112,9 +133,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 +208,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 +224,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 +276,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 +304,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 +362,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 +478,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 +502,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 +597,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 +605,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 +616,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.0b2}/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",
@@ -126,7 +106,6 @@ exclude = [
 include = '\.pyi?$'
 exclude = '(_pb2\.py$|docs/source/.*\.py)'
 
-# TODO(Kiko): Revisit ruff ignores
 [tool.ruff]
 target-version = "py310"
 line-length = 80

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

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

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

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

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

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

{arize-8.0.0b0 → arize-8.0.0b2}/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