Include delete custom metric

Author: ramon-marques

aimon/resources/datasets/datasets.py

@@ -116,7 +116,7 @@ def create(
         # multipart/form-data; boundary=---abc--
         extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
         return self._post(
-            "/v1/dataset",
+            "/v2/dataset",
             body=maybe_transform(body, dataset_create_params.DatasetCreateParams),
             files=files,
             options=make_request_options(
@@ -233,7 +233,7 @@ async def create(
         # multipart/form-data; boundary=---abc--
         extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
         return await self._post(
-            "/v1/dataset",
+            "/v2/dataset",
             body=await async_maybe_transform(body, dataset_create_params.DatasetCreateParams),
             files=files,
             options=make_request_options(

aimon/resources/metrics.py

@@ -17,6 +17,7 @@
 )
 from .._base_client import make_request_options
 from ..types.metric_list_response import MetricListResponse
+from ..types.metric_delete_response import MetricDeleteResponse
 
 __all__ = ["MetricsResource", "AsyncMetricsResource"]
 
@@ -103,6 +104,39 @@ def list(
             cast_to=MetricListResponse,
         )
 
+    def delete(
+        self,
+        uuid: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> MetricDeleteResponse:
+        """
+        Delete a custom metric
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uuid:
+            raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
+        return self._delete(
+            f"/v1/custom-metric/{uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=MetricDeleteResponse,
+        )
+
 
 class AsyncMetricsResource(AsyncAPIResource):
     @cached_property
@@ -186,6 +220,39 @@ async def list(
             cast_to=MetricListResponse,
         )
 
+    async def delete(
+        self,
+        uuid: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> MetricDeleteResponse:
+        """
+        Delete a custom metric
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uuid:
+            raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
+        return await self._delete(
+            f"/v1/custom-metric/{uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=MetricDeleteResponse,
+        )
+
 
 class MetricsResourceWithRawResponse:
     def __init__(self, metrics: MetricsResource) -> None:
@@ -197,6 +264,9 @@ def __init__(self, metrics: MetricsResource) -> None:
         self.list = to_raw_response_wrapper(
             metrics.list,
         )
+        self.delete = to_raw_response_wrapper(
+            metrics.delete,
+        )
 
 
 class AsyncMetricsResourceWithRawResponse:
@@ -209,6 +279,9 @@ def __init__(self, metrics: AsyncMetricsResource) -> None:
         self.list = async_to_raw_response_wrapper(
             metrics.list,
         )
+        self.delete = async_to_raw_response_wrapper(
+            metrics.delete,
+        )
 
 
 class MetricsResourceWithStreamingResponse:
@@ -221,6 +294,9 @@ def __init__(self, metrics: MetricsResource) -> None:
         self.list = to_streamed_response_wrapper(
             metrics.list,
         )
+        self.delete = to_streamed_response_wrapper(
+            metrics.delete,
+        )
 
 
 class AsyncMetricsResourceWithStreamingResponse:
@@ -233,3 +309,6 @@ def __init__(self, metrics: AsyncMetricsResource) -> None:
         self.list = async_to_streamed_response_wrapper(
             metrics.list,
         )
+        self.delete = async_to_streamed_response_wrapper(
+            metrics.delete,
+        )

aimon/types/__init__.py

@@ -15,6 +15,7 @@
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
 from .model_create_response import ModelCreateResponse as ModelCreateResponse
 from .model_retrieve_params import ModelRetrieveParams as ModelRetrieveParams
+from .metric_delete_response import MetricDeleteResponse as MetricDeleteResponse
 from .user_validate_response import UserValidateResponse as UserValidateResponse
 from .analyze_create_response import AnalyzeCreateResponse as AnalyzeCreateResponse
 from .inference_detect_params import InferenceDetectParams as InferenceDetectParams
@@ -31,4 +32,4 @@
 from .application_delete_response import ApplicationDeleteResponse as ApplicationDeleteResponse
 from .application_retrieve_params import ApplicationRetrieveParams as ApplicationRetrieveParams
 from .evaluation_retrieve_response import EvaluationRetrieveResponse as EvaluationRetrieveResponse
-from .application_retrieve_response import ApplicationRetrieveResponse as ApplicationRetrieveResponse
+from .application_retrieve_response import ApplicationRetrieveResponse as ApplicationRetrieveResponse

aimon/types/metric_delete_response.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from .._models import BaseModel
+
+__all__ = ["MetricDeleteResponse"]
+
+
+class MetricDeleteResponse(BaseModel):
+    message: Optional[str] = None

tests/test_low_level_api.py

@@ -286,8 +286,11 @@ def test_dataset_collection_create_retrieve(self):
             dataset_base_name = f"{self.prefix}_dataset_{i}.csv"
             try:
                 with open(temp_file_path, 'rb') as f:
-                    dataset_args = json.dumps({"name": dataset_base_name, "description": f"Test dataset {i}"})
-                    dataset = self.client.datasets.create(file=f, json_data=dataset_args)
+                    dataset = self.client.datasets.create(
+                        file=f,
+                        name=dataset_base_name,
+                        description=f"Test dataset {i}"
+                    )
                     dataset_shas.append(dataset.sha)
                     self.log_info(f"Prerequisite dataset {i} created: {dataset_base_name} (SHA: {dataset.sha})")
             except Exception as e:
@@ -339,9 +342,11 @@ def test_evaluation_create_retrieve(self):
 
             temp_file_path = self.create_temp_dataset_file()
             with open(temp_file_path, 'rb') as f:
-                 # Add description field to json_data
-                 dataset_args = json.dumps({"name": self.dataset_name, "description": f"Prereq dataset {self.timestamp}"})
-                 dataset = self.client.datasets.create(file=f, json_data=dataset_args)
+                dataset = self.client.datasets.create(
+                    file=f,
+                    name=self.dataset_name,
+                    description=f"Prereq dataset {self.timestamp}"
+                )
 
             # Add description to collection create call
             collection = self.client.datasets.collection.create(name=self.collection_name, dataset_ids=[dataset.sha], description=f"Prereq collection {self.timestamp}") 
@@ -400,9 +405,11 @@ def test_evaluation_run_create(self):
             app = self.client.applications.create(name=self.app_name, model_name=model.name, stage="evaluation", type="qa")
             temp_file_path = self.create_temp_dataset_file()
             with open(temp_file_path, 'rb') as f:
-                 # Add description field to json_data
-                 dataset_args = json.dumps({"name": self.dataset_name, "description": f"Prereq dataset {self.timestamp}"})
-                 dataset = self.client.datasets.create(file=f, json_data=dataset_args)
+                dataset = self.client.datasets.create(
+                    file=f,
+                    name=self.dataset_name,
+                    description=f"Prereq dataset {self.timestamp}"
+                )
             # Add description to collection create call
             collection = self.client.datasets.collection.create(name=self.collection_name, dataset_ids=[dataset.sha], description=f"Prereq collection {self.timestamp}") 
             evaluation = self.client.evaluations.create(