arize 8.0.0a12__py3-none-any.whl → 8.0.0a13__py3-none-any.whl

arize/_generated/api_client/models/dataset.py

@@ -34,7 +34,6 @@ class Dataset(BaseModel):
     created_at: Optional[datetime] = Field(default=None, description="Timestamp for when the dataset was created", alias="createdAt")
     updated_at: Optional[datetime] = Field(default=None, description="Timestamp for the last update of the dataset", alias="updatedAt")
     versions: Optional[List[DatasetVersion]] = Field(default=None, description="List of versions associated with this dataset")
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["id", "name", "spaceId", "createdAt", "updatedAt", "versions"]
 
     model_config = ConfigDict(
@@ -67,10 +66,8 @@ class Dataset(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -85,11 +82,6 @@ class Dataset(BaseModel):
                 if _item_versions:
                     _items.append(_item_versions.to_dict())
             _dict['versions'] = _items
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         return _dict
 
     @classmethod
@@ -101,6 +93,11 @@ class Dataset(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in Dataset) in the input: " + _key)
+
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
@@ -109,11 +106,6 @@ class Dataset(BaseModel):
             "updatedAt": obj.get("updatedAt"),
             "versions": [DatasetVersion.from_dict(_item) for _item in obj["versions"]] if obj.get("versions") is not None else None
         })
-        # store additional fields in additional_properties
-        for _key in obj.keys():
-            if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
-
         return _obj
 
 

arize/_generated/api_client/models/dataset_version.py

@@ -32,7 +32,6 @@ class DatasetVersion(BaseModel):
     dataset_id: Optional[StrictStr] = Field(default=None, description="Unique identifier for the dataset this version belongs to", alias="datasetId")
     created_at: Optional[datetime] = Field(default=None, description="Timestamp for when the dataset version was created", alias="createdAt")
     updated_at: Optional[datetime] = Field(default=None, description="Timestamp for the last update of the dataset version", alias="updatedAt")
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["id", "name", "datasetId", "createdAt", "updatedAt"]
 
     model_config = ConfigDict(
@@ -65,10 +64,8 @@ class DatasetVersion(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -76,11 +73,6 @@ class DatasetVersion(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         return _dict
 
     @classmethod
@@ -92,6 +84,11 @@ class DatasetVersion(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in DatasetVersion) in the input: " + _key)
+
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
@@ -99,11 +96,6 @@ class DatasetVersion(BaseModel):
             "createdAt": obj.get("createdAt"),
             "updatedAt": obj.get("updatedAt")
         })
-        # store additional fields in additional_properties
-        for _key in obj.keys():
-            if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
-
         return _obj
 
 

arize/_generated/api_client/models/datasets_create_request.py

@@ -29,7 +29,6 @@ class DatasetsCreateRequest(BaseModel):
     name: StrictStr
     space_id: Optional[StrictStr] = Field(default=None, alias="spaceId")
     examples: List[Dict[str, Any]]
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["name", "spaceId", "examples"]
 
     model_config = ConfigDict(
@@ -62,10 +61,8 @@ class DatasetsCreateRequest(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -73,11 +70,6 @@ class DatasetsCreateRequest(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         return _dict
 
     @classmethod
@@ -89,16 +81,16 @@ class DatasetsCreateRequest(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in DatasetsCreateRequest) in the input: " + _key)
+
         _obj = cls.model_validate({
             "name": obj.get("name"),
             "spaceId": obj.get("spaceId"),
             "examples": obj.get("examples")
         })
-        # store additional fields in additional_properties
-        for _key in obj.keys():
-            if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
-
         return _obj
 
 

arize/_generated/api_client/models/datasets_list200_response.py

@@ -87,6 +87,11 @@ class DatasetsList200Response(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in DatasetsList200Response) in the input: " + _key)
+
         _obj = cls.model_validate({
             "datasets": [Dataset.from_dict(_item) for _item in obj["datasets"]] if obj.get("datasets") is not None else None
         })

arize/_generated/api_client/models/datasets_list_examples200_response.py

@@ -26,8 +26,8 @@ class DatasetsListExamples200Response(BaseModel):
     """
     DatasetsListExamples200Response
     """ # noqa: E501
-    data: List[Dict[str, Any]] = Field(description="Array of example objects from the dataset")
-    __properties: ClassVar[List[str]] = ["data"]
+    examples: List[Dict[str, Any]] = Field(description="Array of example objects from the dataset")
+    __properties: ClassVar[List[str]] = ["examples"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -79,8 +79,13 @@ class DatasetsListExamples200Response(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in DatasetsListExamples200Response) in the input: " + _key)
+
         _obj = cls.model_validate({
-            "data": obj.get("data")
+            "examples": obj.get("examples")
         })
         return _obj
 

arize/_generated/api_client/models/error.py

@@ -27,7 +27,6 @@ class Error(BaseModel):
     Error
     """ # noqa: E501
     message: Optional[StrictStr] = None
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["message"]
 
     model_config = ConfigDict(
@@ -60,10 +59,8 @@ class Error(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -71,11 +68,6 @@ class Error(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         return _dict
 
     @classmethod
@@ -87,14 +79,14 @@ class Error(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "message": obj.get("message")
-        })
-        # store additional fields in additional_properties
+        # raise errors for additional fields in the input
         for _key in obj.keys():
             if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
+                raise ValueError("Error due to additional fields (not defined in Error) in the input: " + _key)
 
+        _obj = cls.model_validate({
+            "message": obj.get("message")
+        })
         return _obj
 
 

arize/_generated/api_client/models/experiment.py

@@ -35,7 +35,6 @@ class Experiment(BaseModel):
     updated_at: Optional[datetime] = Field(default=None, description="Timestamp for the last update of the experiment", alias="updatedAt")
     created_by: Optional[StrictStr] = Field(default=None, description="Unique identifier for the user who created the experiment", alias="createdBy")
     experiment_traces_project_id: Optional[StrictStr] = Field(default=None, description="Unique identifier for the experiment traces project this experiment belongs to (if it exists)", alias="experimentTracesProjectId")
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["id", "name", "datasetId", "datasetVersionId", "createdAt", "updatedAt", "createdBy", "experimentTracesProjectId"]
 
     model_config = ConfigDict(
@@ -68,10 +67,8 @@ class Experiment(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -79,11 +76,6 @@ class Experiment(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         return _dict
 
     @classmethod
@@ -95,6 +87,11 @@ class Experiment(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in Experiment) in the input: " + _key)
+
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
@@ -105,11 +102,6 @@ class Experiment(BaseModel):
             "createdBy": obj.get("createdBy"),
             "experimentTracesProjectId": obj.get("experimentTracesProjectId")
         })
-        # store additional fields in additional_properties
-        for _key in obj.keys():
-            if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
-
         return _obj
 
 

arize/_generated/api_client/models/experiments_list200_response.py

@@ -27,8 +27,8 @@ class ExperimentsList200Response(BaseModel):
     """
     ExperimentsList200Response
     """ # noqa: E501
-    data: List[Experiment] = Field(description="A list of experiments")
-    __properties: ClassVar[List[str]] = ["data"]
+    experiments: List[Experiment] = Field(description="A list of experiments")
+    __properties: ClassVar[List[str]] = ["experiments"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -69,13 +69,13 @@ class ExperimentsList200Response(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in data (list)
+        # override the default output from pydantic by calling `to_dict()` of each item in experiments (list)
         _items = []
-        if self.data:
-            for _item_data in self.data:
-                if _item_data:
-                    _items.append(_item_data.to_dict())
-            _dict['data'] = _items
+        if self.experiments:
+            for _item_experiments in self.experiments:
+                if _item_experiments:
+                    _items.append(_item_experiments.to_dict())
+            _dict['experiments'] = _items
         return _dict
 
     @classmethod
@@ -87,8 +87,13 @@ class ExperimentsList200Response(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
+        # raise errors for additional fields in the input
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                raise ValueError("Error due to additional fields (not defined in ExperimentsList200Response) in the input: " + _key)
+
         _obj = cls.model_validate({
-            "data": [Experiment.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
+            "experiments": [Experiment.from_dict(_item) for _item in obj["experiments"]] if obj.get("experiments") is not None else None
         })
         return _obj
 

arize/_generated/api_client/test/test_datasets_list_examples200_response.py

@@ -35,13 +35,13 @@ class TestDatasetsListExamples200Response(unittest.TestCase):
         model = DatasetsListExamples200Response()
         if include_optional:
             return DatasetsListExamples200Response(
-                data = [
+                examples = [
                     None
                     ]
             )
         else:
             return DatasetsListExamples200Response(
-                data = [
+                examples = [
                     None
                     ],
         )

arize/_generated/api_client/test/test_experiments_api.py

@@ -26,6 +26,13 @@ class TestExperimentsApi(unittest.TestCase):
     def tearDown(self) -> None:
         pass
 
+    def test_experiments_delete(self) -> None:
+        """Test case for experiments_delete
+
+        Delete an experiment by ID
+        """
+        pass
+
     def test_experiments_list(self) -> None:
         """Test case for experiments_list
 

arize/_generated/api_client/test/test_experiments_list200_response.py

@@ -35,7 +35,7 @@ class TestExperimentsList200Response(unittest.TestCase):
         model = ExperimentsList200Response()
         if include_optional:
             return ExperimentsList200Response(
-                data = [
+                experiments = [
                     arize._generated.api_client.models.experiment.Experiment(
                         id = '', 
                         name = '', 
@@ -49,7 +49,7 @@ class TestExperimentsList200Response(unittest.TestCase):
             )
         else:
             return ExperimentsList200Response(
-                data = [
+                experiments = [
                     arize._generated.api_client.models.experiment.Experiment(
                         id = '', 
                         name = '', 

arize/_generated/api_client_README.md

@@ -5,7 +5,7 @@ The `arize._generated.api_client` package is automatically generated by the [Ope
 
 - API version: 0.0.1
 - Package version: 1.0.0
-- Generator version: 7.15.0
+- Generator version: 7.16.0
 - Build package: org.openapitools.codegen.languages.PythonClientCodegen
 
 ## Requirements.
@@ -79,6 +79,7 @@ Class | Method | HTTP request | Description
 *DatasetsApi* | [**datasets_get**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_get) | **GET** /api/v1/datasets/{datasetId} | Get dataset by ID
 *DatasetsApi* | [**datasets_list**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_list) | **GET** /api/v1/datasets | List datasets the user has access to
 *DatasetsApi* | [**datasets_list_examples**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_list_examples) | **GET** /api/v1/datasets/{datasetId}/examples | List examples for a dataset
+*ExperimentsApi* | [**experiments_delete**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_delete) | **DELETE** /api/v1/experiments/{experimentId} | Delete an experiment by ID
 *ExperimentsApi* | [**experiments_list**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_list) | **GET** /api/v1/datasets/{datasetId}/experiments | List experiments for a given dataset
 
 
@@ -86,7 +87,6 @@ Class | Method | HTTP request | Description
 
  - [Dataset](arize/_generated/api_client/docs/Dataset.md)
  - [DatasetVersion](arize/_generated/api_client/docs/DatasetVersion.md)
- - [DatasetsCreate201Response](arize/_generated/api_client/docs/DatasetsCreate201Response.md)
  - [DatasetsCreateRequest](arize/_generated/api_client/docs/DatasetsCreateRequest.md)
  - [DatasetsList200Response](arize/_generated/api_client/docs/DatasetsList200Response.md)
  - [DatasetsListExamples200Response](arize/_generated/api_client/docs/DatasetsListExamples200Response.md)

arize/_generated/protocol/flight/export_pb2.py

@@ -16,7 +16,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x65xport.proto\x12\x06public\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xa2\x04\n\x15RecordQueryDescriptor\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12>\n\x0b\x65nvironment\x18\x03 \x01(\x0e\x32).public.RecordQueryDescriptor.Environment\x12\x15\n\rmodel_version\x18\x04 \x01(\t\x12\x10\n\x08\x62\x61tch_id\x18\x05 \x01(\t\x12.\n\nstart_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0finclude_actuals\x18\x08 \x01(\x08\x12\x19\n\x11\x66ilter_expression\x18\t \x01(\t\x12@\n\x18similarity_search_params\x18\n \x01(\x0b\x32\x1e.public.SimilaritySearchParams\x12\x19\n\x11projected_columns\x18\x0b \x03(\t\x12\x36\n\x11stream_chunk_size\x18\x0c \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"U\n\x0b\x45nvironment\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08TRAINING\x10\x01\x12\x0e\n\nVALIDATION\x10\x02\x12\x0e\n\nPRODUCTION\x10\x03\x12\x0b\n\x07TRACING\x10\x04\"\x93\x02\n\x16SimilaritySearchParams\x12<\n\nreferences\x18\x01 \x03(\x0b\x32(.public.SimilaritySearchParams.Reference\x12\x1a\n\x12search_column_name\x18\x02 \x01(\t\x12\x11\n\tthreshold\x18\x03 \x01(\x01\x1a\x8b\x01\n\tReference\x12\x15\n\rprediction_id\x18\x01 \x01(\t\x12\x1d\n\x15reference_column_name\x18\x02 \x01(\t\x12\x38\n\x14prediction_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0e\n\x06vector\x18\x04 \x03(\x01\x42?Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/publicb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x65xport.proto\x12\x06public\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xdb\x04\n\x15RecordQueryDescriptor\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12>\n\x0b\x65nvironment\x18\x03 \x01(\x0e\x32).public.RecordQueryDescriptor.Environment\x12\x15\n\rmodel_version\x18\x04 \x01(\t\x12\x10\n\x08\x62\x61tch_id\x18\x05 \x01(\t\x12.\n\nstart_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0finclude_actuals\x18\x08 \x01(\x08\x12\x19\n\x11\x66ilter_expression\x18\t \x01(\t\x12@\n\x18similarity_search_params\x18\n \x01(\x0b\x32\x1e.public.SimilaritySearchParams\x12\x19\n\x11projected_columns\x18\x0b \x03(\t\x12\x36\n\x11stream_chunk_size\x18\x0c \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12\x37\n\x13parallelize_exports\x18\r \x01(\x0b\x32\x1a.google.protobuf.BoolValue\"U\n\x0b\x45nvironment\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08TRAINING\x10\x01\x12\x0e\n\nVALIDATION\x10\x02\x12\x0e\n\nPRODUCTION\x10\x03\x12\x0b\n\x07TRACING\x10\x04\"\x93\x02\n\x16SimilaritySearchParams\x12<\n\nreferences\x18\x01 \x03(\x0b\x32(.public.SimilaritySearchParams.Reference\x12\x1a\n\x12search_column_name\x18\x02 \x01(\t\x12\x11\n\tthreshold\x18\x03 \x01(\x01\x1a\x8b\x01\n\tReference\x12\x15\n\rprediction_id\x18\x01 \x01(\t\x12\x1d\n\x15reference_column_name\x18\x02 \x01(\t\x12\x38\n\x14prediction_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0e\n\x06vector\x18\x04 \x03(\x01\x42?Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/publicb\x06proto3')
 
 
 
@@ -51,11 +51,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/public'
   _RECORDQUERYDESCRIPTOR._serialized_start=90
-  _RECORDQUERYDESCRIPTOR._serialized_end=636
-  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_start=551
-  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_end=636
-  _SIMILARITYSEARCHPARAMS._serialized_start=639
-  _SIMILARITYSEARCHPARAMS._serialized_end=914
-  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_start=775
-  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_end=914
+  _RECORDQUERYDESCRIPTOR._serialized_end=693
+  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_start=608
+  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_end=693
+  _SIMILARITYSEARCHPARAMS._serialized_start=696
+  _SIMILARITYSEARCHPARAMS._serialized_end=971
+  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_start=832
+  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_end=971
 # @@protoc_insertion_point(module_scope)

arize/_lazy.py

@@ -1,6 +1,7 @@
 # src/arize/_lazy.py
 from __future__ import annotations
 
+import logging
 import sys
 import threading
 from importlib import import_module
@@ -9,6 +10,8 @@ from typing import TYPE_CHECKING, Any, Dict, Tuple
 if TYPE_CHECKING:
     from arize.config import SDKConfiguration
 
+logger = logging.getLogger(__name__)
+
 
 class LazySubclientsMixin:
     _SUBCLIENTS: Dict[str, Tuple[str, str]] = {}
@@ -30,6 +33,7 @@ class LazySubclientsMixin:
             if name in self._lazy_cache:
                 return self._lazy_cache[name]
 
+            logger.debug(f"Lazily loading subclient {name!r}")
             module_path, class_name = subs[name]
             extra_key, required = self._EXTRAS.get(name, (None, ()))
             require(extra_key, required)

arize/client.py

@@ -12,6 +12,7 @@ if TYPE_CHECKING:
     from arize.spans.client import SpansClient
 
 
+# TODO(Kiko): Add flight max_chunksize opt to write_table. In config?
 # TODO(Kiko): experimental/datasets must be adapted into the datasets subclient
 # TODO(Kiko): experimental/prompt hub is missing
 # TODO(Kiko): exporter/utils/schema_parser is missing
@@ -43,10 +44,22 @@ class ArizeClient(LazySubclientsMixin):
     """
 
     _SUBCLIENTS = {
-        "datasets": ("arize.datasets.client", "DatasetsClient"),
-        "experiments": ("arize.experiments.client", "ExperimentsClient"),
-        "spans": ("arize.spans.client", "SpansClient"),
-        "models": ("arize.models.client", "MLModelsClient"),
+        "datasets": (
+            "arize.datasets.client",
+            "DatasetsClient",
+        ),
+        "experiments": (
+            "arize.experiments.client",
+            "ExperimentsClient",
+        ),
+        "spans": (
+            "arize.spans.client",
+            "SpansClient",
+        ),
+        "models": (
+            "arize.models.client",
+            "MLModelsClient",
+        ),
     }
     _EXTRAS = {
         # Gate only the generated-backed ones

arize/config.py

@@ -1,3 +1,4 @@
+import logging
 import os
 import sys
 import threading
@@ -26,6 +27,8 @@ from arize.constants.config import (
 from arize.exceptions.auth import MissingAPIKeyError
 from arize.version import __version__
 
+logger = logging.getLogger(__name__)
+
 
 def _parse_bool(val: bool | str | None) -> bool:
     if isinstance(val, bool):
@@ -81,8 +84,11 @@ def _mask_secret(secret: str, N: int = 4) -> str:
     return f"{secret[:N]}***"
 
 
-def _endpoint(scheme: str, base: str, path: str) -> str:
-    return scheme + "://" + base.rstrip("/") + "/" + path.lstrip("/")
+def _endpoint(scheme: str, base: str, path: str = "") -> str:
+    endpoint = scheme + "://" + base.rstrip("/")
+    if path:
+        endpoint += "/" + path.lstrip("/")
+    return endpoint
 
 
 @dataclass(frozen=True)
@@ -111,6 +117,10 @@ class SDKConfiguration:
         if not self.api_key:
             raise MissingAPIKeyError()
 
+    @property
+    def api_url(self) -> str:
+        return _endpoint(self.api_scheme, self.api_host)
+
     @property
     def files_url(self) -> str:
         return _endpoint(self.api_scheme, self.api_host, "/v1/pandas_arrow")
@@ -170,7 +180,7 @@ class SDKConfiguration:
             # Import lazily so extras can be enforced outside
             from arize._generated import api_client as gen
 
-            cfg = gen.Configuration(host=self.api_host)
+            cfg = gen.Configuration(host=self.api_url)
             if self.api_key:
                 cfg.api_key["ApiKeyAuth"] = self.api_key
             client = gen.ApiClient(cfg)

arize/datasets/__init__.py

@@ -0,0 +1,70 @@
+from collections.abc import Mapping
+
+from arize._generated.api_client import models
+
+
+def make_to_df(field_name: str):
+    def to_df(
+        self,
+        by_alias: bool = False,
+        exclude_none: str | bool = False,
+        json_normalize: bool = False,
+        convert_dtypes: bool = True,
+    ):
+        """
+        Convert a list of objects to a pandas DataFrame.
+
+        Behavior:
+          - If an item is a Pydantic v2 model, use `.model_dump(by_alias=...)`.
+          - If an item is a mapping (dict-like), use it as-is.
+          - Otherwise, raise a ValueError (unsupported row type).
+
+        Parameters:
+          by_alias: Use field aliases when dumping Pydantic models.
+          exclude_none:
+            - False: keep Nones as-is
+            - "all": drop columns where *all* values are None/NaN
+            - "any": drop columns where *any* value is None/NaN
+            - True: alias for "all"
+          json_normalize: If True, flatten nested dicts via `pandas.json_normalize`.
+          convert_dtypes: If True, call `DataFrame.convert_dtypes()` at the end.
+
+        Returns:
+          pandas.DataFrame
+        """
+        import pandas as pd
+
+        items = getattr(self, field_name, []) or []
+
+        rows = []
+        for it in items:
+            if hasattr(it, "model_dump"):  # Pydantic v2 object
+                rows.append(it.model_dump(by_alias=by_alias))
+
+            elif isinstance(it, Mapping):  # Plain mapping
+                rows.append(it)
+            else:
+                raise ValueError(
+                    f"Cannot convert item of type {type(it)} to DataFrame row"
+                )
+
+        df = (
+            pd.json_normalize(rows, sep=".")
+            if json_normalize
+            else pd.DataFrame(rows)
+        )
+
+        # Drop None/NaN columns if requested
+        if exclude_none in ("any", "all", True):
+            drop_how = "all" if exclude_none is True else exclude_none
+            df.dropna(axis=1, how=drop_how, inplace=True)
+
+        if convert_dtypes:
+            df = df.convert_dtypes()
+        return df
+
+    return to_df
+
+
+models.DatasetsList200Response.to_df = make_to_df("datasets")  # type: ignore[attr-defined]
+models.DatasetsListExamples200Response.to_df = make_to_df("examples")  # type: ignore[attr-defined]