arize 8.0.0a21__py3-none-any.whl → 8.0.0a23__py3-none-any.whl

arize/_generated/api_client/models/projects_list200_response.py

@@ -0,0 +1,106 @@
+# 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
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field
+from typing import Any, ClassVar, Dict, List
+from arize._generated.api_client.models.pagination_metadata import PaginationMetadata
+from arize._generated.api_client.models.project import Project
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ProjectsList200Response(BaseModel):
+    """
+    ProjectsList200Response
+    """ # noqa: E501
+    projects: List[Project] = Field(description="A list of projects")
+    pagination: PaginationMetadata
+    __properties: ClassVar[List[str]] = ["projects", "pagination"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ProjectsList200Response from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in projects (list)
+        _items = []
+        if self.projects:
+            for _item_projects in self.projects:
+                if _item_projects:
+                    _items.append(_item_projects.to_dict())
+            _dict['projects'] = _items
+        # override the default output from pydantic by calling `to_dict()` of pagination
+        if self.pagination:
+            _dict['pagination'] = self.pagination.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ProjectsList200Response from a dict"""
+        if obj is None:
+            return None
+
+        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 ProjectsList200Response) in the input: " + _key)
+
+        _obj = cls.model_validate({
+            "projects": [Project.from_dict(_item) for _item in obj["projects"]] if obj.get("projects") is not None else None,
+            "pagination": PaginationMetadata.from_dict(obj["pagination"]) if obj.get("pagination") is not None else None
+        })
+        return _obj
+
+

arize/_generated/api_client/rest.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

arize/_generated/api_client/test/test_dataset.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -54,6 +54,8 @@ class TestDataset(unittest.TestCase):
                 id = '',
                 name = '',
                 space_id = '',
+                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
         )
         """
 

arize/_generated/api_client/test/test_dataset_example.py

@@ -0,0 +1,56 @@
+# 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.dataset_example import DatasetExample
+
+class TestDatasetExample(unittest.TestCase):
+    """DatasetExample unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> DatasetExample:
+        """Test DatasetExample
+            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 `DatasetExample`
+        """
+        model = DatasetExample()
+        if include_optional:
+            return DatasetExample(
+                id = '',
+                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f')
+            )
+        else:
+            return DatasetExample(
+                id = '',
+                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+        )
+        """
+
+    def testDatasetExample(self):
+        """Test DatasetExample"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

arize/_generated/api_client/test/test_dataset_example_update.py

@@ -0,0 +1,52 @@
+# 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.dataset_example_update import DatasetExampleUpdate
+
+class TestDatasetExampleUpdate(unittest.TestCase):
+    """DatasetExampleUpdate unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> DatasetExampleUpdate:
+        """Test DatasetExampleUpdate
+            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 `DatasetExampleUpdate`
+        """
+        model = DatasetExampleUpdate()
+        if include_optional:
+            return DatasetExampleUpdate(
+                id = ''
+            )
+        else:
+            return DatasetExampleUpdate(
+                id = '',
+        )
+        """
+
+    def testDatasetExampleUpdate(self):
+        """Test DatasetExampleUpdate"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

arize/_generated/api_client/test/test_dataset_version.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -43,6 +43,11 @@ class TestDatasetVersion(unittest.TestCase):
             )
         else:
             return DatasetVersion(
+                id = '',
+                name = '',
+                dataset_id = '',
+                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
         )
         """
 

arize/_generated/api_client/test/test_datasets_api.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -29,35 +29,49 @@ class TestDatasetsApi(unittest.TestCase):
     def test_datasets_create(self) -> None:
         """Test case for datasets_create
 
-        Create a new dataset with JSON examples
+        Create a dataset
         """
         pass
 
     def test_datasets_delete(self) -> None:
         """Test case for datasets_delete
 
-        Delete a dataset by ID
+        Delete a dataset
         """
         pass
 
-    def test_datasets_get(self) -> None:
-        """Test case for datasets_get
+    def test_datasets_examples_insert(self) -> None:
+        """Test case for datasets_examples_insert
 
-        Get dataset by ID
+        Add new examples to a dataset
         """
         pass
 
-    def test_datasets_list(self) -> None:
-        """Test case for datasets_list
+    def test_datasets_examples_list(self) -> None:
+        """Test case for datasets_examples_list
+
+        List dataset examples
+        """
+        pass
+
+    def test_datasets_examples_update(self) -> None:
+        """Test case for datasets_examples_update
 
-        List datasets the user has access to
+        Update existing examples in a dataset
         """
         pass
 
-    def test_datasets_list_examples(self) -> None:
-        """Test case for datasets_list_examples
+    def test_datasets_get(self) -> None:
+        """Test case for datasets_get
+
+        Get a dataset
+        """
+        pass
+
+    def test_datasets_list(self) -> None:
+        """Test case for datasets_list
 
-        List examples for a dataset
+        List datasets
         """
         pass
 

arize/_generated/api_client/test/test_datasets_create_request.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -38,7 +38,9 @@ class TestDatasetsCreateRequest(unittest.TestCase):
                 name = '',
                 space_id = '',
                 examples = [
-                    None
+                    {
+                        'key' : null
+                        }
                     ]
             )
         else:
@@ -46,7 +48,9 @@ class TestDatasetsCreateRequest(unittest.TestCase):
                 name = '',
                 space_id = '',
                 examples = [
-                    None
+                    {
+                        'key' : null
+                        }
                     ],
         )
         """

arize/_generated/api_client/test/{test_datasets_list_examples200_response.py → test_datasets_examples_insert_request.py}

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -14,10 +14,10 @@
 
 import unittest
 
-from arize._generated.api_client.models.datasets_list_examples200_response import DatasetsListExamples200Response
+from arize._generated.api_client.models.datasets_examples_insert_request import DatasetsExamplesInsertRequest
 
-class TestDatasetsListExamples200Response(unittest.TestCase):
-    """DatasetsListExamples200Response unit test stubs"""
+class TestDatasetsExamplesInsertRequest(unittest.TestCase):
+    """DatasetsExamplesInsertRequest unit test stubs"""
 
     def setUp(self):
         pass
@@ -25,30 +25,34 @@ class TestDatasetsListExamples200Response(unittest.TestCase):
     def tearDown(self):
         pass
 
-    def make_instance(self, include_optional) -> DatasetsListExamples200Response:
-        """Test DatasetsListExamples200Response
+    def make_instance(self, include_optional) -> DatasetsExamplesInsertRequest:
+        """Test DatasetsExamplesInsertRequest
             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 `DatasetsListExamples200Response`
+        # uncomment below to create an instance of `DatasetsExamplesInsertRequest`
         """
-        model = DatasetsListExamples200Response()
+        model = DatasetsExamplesInsertRequest()
         if include_optional:
-            return DatasetsListExamples200Response(
+            return DatasetsExamplesInsertRequest(
                 examples = [
-                    None
+                    {
+                        'key' : null
+                        }
                     ]
             )
         else:
-            return DatasetsListExamples200Response(
+            return DatasetsExamplesInsertRequest(
                 examples = [
-                    None
+                    {
+                        'key' : null
+                        }
                     ],
         )
         """
 
-    def testDatasetsListExamples200Response(self):
-        """Test DatasetsListExamples200Response"""
+    def testDatasetsExamplesInsertRequest(self):
+        """Test DatasetsExamplesInsertRequest"""
         # inst_req_only = self.make_instance(include_optional=False)
         # inst_req_and_optional = self.make_instance(include_optional=True)
 

arize/_generated/api_client/test/test_datasets_examples_list200_response.py

@@ -0,0 +1,66 @@
+# 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.datasets_examples_list200_response import DatasetsExamplesList200Response
+
+class TestDatasetsExamplesList200Response(unittest.TestCase):
+    """DatasetsExamplesList200Response unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> DatasetsExamplesList200Response:
+        """Test DatasetsExamplesList200Response
+            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 `DatasetsExamplesList200Response`
+        """
+        model = DatasetsExamplesList200Response()
+        if include_optional:
+            return DatasetsExamplesList200Response(
+                examples = [
+                    {
+                        'key' : null
+                        }
+                    ],
+                pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
+                    next_cursor = '', 
+                    has_more = True, )
+            )
+        else:
+            return DatasetsExamplesList200Response(
+                examples = [
+                    {
+                        'key' : null
+                        }
+                    ],
+                pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
+                    next_cursor = '', 
+                    has_more = True, ),
+        )
+        """
+
+    def testDatasetsExamplesList200Response(self):
+        """Test DatasetsExamplesList200Response"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

arize/_generated/api_client/test/test_datasets_examples_update_request.py

@@ -0,0 +1,61 @@
+# 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.datasets_examples_update_request import DatasetsExamplesUpdateRequest
+
+class TestDatasetsExamplesUpdateRequest(unittest.TestCase):
+    """DatasetsExamplesUpdateRequest unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> DatasetsExamplesUpdateRequest:
+        """Test DatasetsExamplesUpdateRequest
+            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 `DatasetsExamplesUpdateRequest`
+        """
+        model = DatasetsExamplesUpdateRequest()
+        if include_optional:
+            return DatasetsExamplesUpdateRequest(
+                examples = [
+                    {
+                        'key' : null
+                        }
+                    ],
+                new_version = ''
+            )
+        else:
+            return DatasetsExamplesUpdateRequest(
+                examples = [
+                    {
+                        'key' : null
+                        }
+                    ],
+        )
+        """
+
+    def testDatasetsExamplesUpdateRequest(self):
+        """Test DatasetsExamplesUpdateRequest"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

arize/_generated/api_client/test/test_datasets_list200_response.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -50,7 +50,10 @@ class TestDatasetsList200Response(unittest.TestCase):
                                 created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                 updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
                             ], )
-                    ]
+                    ],
+                pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
+                    next_cursor = '', 
+                    has_more = True, )
             )
         else:
             return DatasetsList200Response(
@@ -70,6 +73,9 @@ class TestDatasetsList200Response(unittest.TestCase):
                                 updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
                             ], )
                     ],
+                pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
+                    next_cursor = '', 
+                    has_more = True, ),
         )
         """
 

arize/_generated/api_client/test/test_experiment.py

@@ -3,9 +3,9 @@
 """
     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. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml 
+    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: 0.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -41,7 +41,6 @@ class TestExperiment(unittest.TestCase):
                 dataset_version_id = '',
                 created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
                 updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
-                created_by = '',
                 experiment_traces_project_id = ''
             )
         else:
@@ -52,7 +51,6 @@ class TestExperiment(unittest.TestCase):
                 dataset_version_id = '',
                 created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
                 updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
-                created_by = '',
         )
         """