scale-gp-beta 0.1.0a32__py3-none-any.whl → 0.1.0a34__py3-none-any.whl

scale_gp_beta/resources/evaluations.py

@@ -27,7 +27,6 @@ from ..pagination import SyncCursorPage, AsyncCursorPage
 from .._base_client import AsyncPaginator, make_request_options
 from ..types.evaluation import Evaluation
 from ..types.evaluation_task_param import EvaluationTaskParam
-from ..types.evaluation_delete_response import EvaluationDeleteResponse
 
 __all__ = ["EvaluationsResource", "AsyncEvaluationsResource"]
 
@@ -115,7 +114,7 @@ class EvaluationsResource(SyncAPIResource):
         Args:
           dataset_id: The ID of the dataset containing the items referenced by the `data` field
 
-          data: Items to be evaluated, including references to the input dataset items
+          data: Items to be evaluated, including references to the input dataset
 
           metadata: Optional metadata key-value pairs for the evaluation
 
@@ -262,6 +261,7 @@ class EvaluationsResource(SyncAPIResource):
             cast_to=Evaluation,
         )
 
+    @overload
     def update(
         self,
         evaluation_id: str,
@@ -277,7 +277,7 @@ class EvaluationsResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Evaluation:
         """
-        Update Evaluation
+        Update or Restore Evaluation
 
         Args:
           tags: The tags associated with the entity
@@ -290,6 +290,52 @@ class EvaluationsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        ...
+
+    @overload
+    def update(
+        self,
+        evaluation_id: str,
+        *,
+        restore: Literal[True],
+        # 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,
+    ) -> Evaluation:
+        """
+        Update or Restore Evaluation
+
+        Args:
+          restore: Set to true to restore the entity from the database.
+
+          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
+        """
+        ...
+
+    def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | Omit = omit,
+        name: str | Omit = omit,
+        tags: SequenceNotStr[str] | Omit = omit,
+        restore: Literal[True] | Omit = omit,
+        # 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,
+    ) -> Evaluation:
         if not evaluation_id:
             raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
         return self._patch(
@@ -299,6 +345,7 @@ class EvaluationsResource(SyncAPIResource):
                     "description": description,
                     "name": name,
                     "tags": tags,
+                    "restore": restore,
                 },
                 evaluation_update_params.EvaluationUpdateParams,
             ),
@@ -373,7 +420,7 @@ class EvaluationsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> EvaluationDeleteResponse:
+    ) -> Evaluation:
         """
         Archive Evaluation
 
@@ -393,7 +440,7 @@ class EvaluationsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationDeleteResponse,
+            cast_to=Evaluation,
         )
 
 
@@ -480,7 +527,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         Args:
           dataset_id: The ID of the dataset containing the items referenced by the `data` field
 
-          data: Items to be evaluated, including references to the input dataset items
+          data: Items to be evaluated, including references to the input dataset
 
           metadata: Optional metadata key-value pairs for the evaluation
 
@@ -627,6 +674,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             cast_to=Evaluation,
         )
 
+    @overload
     async def update(
         self,
         evaluation_id: str,
@@ -642,7 +690,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Evaluation:
         """
-        Update Evaluation
+        Update or Restore Evaluation
 
         Args:
           tags: The tags associated with the entity
@@ -655,6 +703,52 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        ...
+
+    @overload
+    async def update(
+        self,
+        evaluation_id: str,
+        *,
+        restore: Literal[True],
+        # 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,
+    ) -> Evaluation:
+        """
+        Update or Restore Evaluation
+
+        Args:
+          restore: Set to true to restore the entity from the database.
+
+          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
+        """
+        ...
+
+    async def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | Omit = omit,
+        name: str | Omit = omit,
+        tags: SequenceNotStr[str] | Omit = omit,
+        restore: Literal[True] | Omit = omit,
+        # 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,
+    ) -> Evaluation:
         if not evaluation_id:
             raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
         return await self._patch(
@@ -664,6 +758,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                     "description": description,
                     "name": name,
                     "tags": tags,
+                    "restore": restore,
                 },
                 evaluation_update_params.EvaluationUpdateParams,
             ),
@@ -738,7 +833,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> EvaluationDeleteResponse:
+    ) -> Evaluation:
         """
         Archive Evaluation
 
@@ -758,7 +853,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationDeleteResponse,
+            cast_to=Evaluation,
         )