scale-gp-beta 0.1.0a12__py3-none-any.whl → 0.1.0a14__py3-none-any.whl

scale_gp_beta/resources/evaluations.py

@@ -2,12 +2,17 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable, Optional
+from typing import Dict, List, Iterable
 from typing_extensions import Literal, overload
 
 import httpx
 
-from ..types import evaluation_list_params, evaluation_create_params, evaluation_retrieve_params
+from ..types import (
+    evaluation_list_params,
+    evaluation_create_params,
+    evaluation_update_params,
+    evaluation_retrieve_params,
+)
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -22,7 +27,7 @@ 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_archive_response import EvaluationArchiveResponse
+from ..types.evaluation_delete_response import EvaluationDeleteResponse
 
 __all__ = ["EvaluationsResource", "AsyncEvaluationsResource"]
 
@@ -246,13 +251,62 @@ class EvaluationsResource(SyncAPIResource):
             cast_to=Evaluation,
         )
 
+    def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
+        # 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 Evaluation
+
+        Args:
+          tags: The tags associated with the entity
+
+          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 evaluation_id:
+            raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
+        return self._patch(
+            f"/v5/evaluations/{evaluation_id}",
+            body=maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                    "tags": tags,
+                },
+                evaluation_update_params.EvaluationUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Evaluation,
+        )
+
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # 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.
@@ -286,7 +340,10 @@ class EvaluationsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
+                        "tags": tags,
                         "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
@@ -295,7 +352,7 @@ class EvaluationsResource(SyncAPIResource):
             model=Evaluation,
         )
 
-    def archive(
+    def delete(
         self,
         evaluation_id: str,
         *,
@@ -305,7 +362,7 @@ class EvaluationsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EvaluationArchiveResponse:
+    ) -> EvaluationDeleteResponse:
         """
         Archive Evaluation
 
@@ -325,7 +382,7 @@ class EvaluationsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationArchiveResponse,
+            cast_to=EvaluationDeleteResponse,
         )
 
 
@@ -548,13 +605,62 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             cast_to=Evaluation,
         )
 
+    async def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
+        # 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 Evaluation
+
+        Args:
+          tags: The tags associated with the entity
+
+          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 evaluation_id:
+            raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
+        return await self._patch(
+            f"/v5/evaluations/{evaluation_id}",
+            body=await async_maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                    "tags": tags,
+                },
+                evaluation_update_params.EvaluationUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Evaluation,
+        )
+
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # 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.
@@ -588,7 +694,10 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
+                        "tags": tags,
                         "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
@@ -597,7 +706,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             model=Evaluation,
         )
 
-    async def archive(
+    async def delete(
         self,
         evaluation_id: str,
         *,
@@ -607,7 +716,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EvaluationArchiveResponse:
+    ) -> EvaluationDeleteResponse:
         """
         Archive Evaluation
 
@@ -627,7 +736,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationArchiveResponse,
+            cast_to=EvaluationDeleteResponse,
         )
 
 
@@ -641,11 +750,14 @@ class EvaluationsResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = to_raw_response_wrapper(
+            evaluations.update,
+        )
         self.list = to_raw_response_wrapper(
             evaluations.list,
         )
-        self.archive = to_raw_response_wrapper(
-            evaluations.archive,
+        self.delete = to_raw_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -659,11 +771,14 @@ class AsyncEvaluationsResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = async_to_raw_response_wrapper(
+            evaluations.update,
+        )
         self.list = async_to_raw_response_wrapper(
             evaluations.list,
         )
-        self.archive = async_to_raw_response_wrapper(
-            evaluations.archive,
+        self.delete = async_to_raw_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -677,11 +792,14 @@ class EvaluationsResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = to_streamed_response_wrapper(
+            evaluations.update,
+        )
         self.list = to_streamed_response_wrapper(
             evaluations.list,
         )
-        self.archive = to_streamed_response_wrapper(
-            evaluations.archive,
+        self.delete = to_streamed_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -695,9 +813,12 @@ class AsyncEvaluationsResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = async_to_streamed_response_wrapper(
+            evaluations.update,
+        )
         self.list = async_to_streamed_response_wrapper(
             evaluations.list,
         )
-        self.archive = async_to_streamed_response_wrapper(
-            evaluations.archive,
+        self.delete = async_to_streamed_response_wrapper(
+            evaluations.delete,
         )

scale_gp_beta/resources/files/files.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict, Mapping, Optional, cast
+from typing import Dict, Mapping, cast
+from typing_extensions import Literal
 
 import httpx
 
@@ -167,9 +168,10 @@ class FilesResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         # 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,
@@ -201,6 +203,7 @@ class FilesResource(SyncAPIResource):
                     {
                         "ending_before": ending_before,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     file_list_params.FileListParams,
@@ -377,9 +380,10 @@ class AsyncFilesResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         # 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,
@@ -411,6 +415,7 @@ class AsyncFilesResource(AsyncAPIResource):
                     {
                         "ending_before": ending_before,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     file_list_params.FileListParams,

scale_gp_beta/resources/models.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Optional
+from typing import Dict
 from typing_extensions import Literal, overload
 
 import httpx
@@ -303,26 +303,25 @@ class ModelsResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        model_vendor: Optional[
-            Literal[
-                "openai",
-                "cohere",
-                "vertex_ai",
-                "anthropic",
-                "azure",
-                "gemini",
-                "launch",
-                "llmengine",
-                "model_zoo",
-                "bedrock",
-                "xai",
-            ]
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "xai",
         ]
         | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         # 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,
@@ -356,6 +355,7 @@ class ModelsResource(SyncAPIResource):
                         "limit": limit,
                         "model_vendor": model_vendor,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     model_list_params.ModelListParams,
@@ -675,26 +675,25 @@ class AsyncModelsResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        model_vendor: Optional[
-            Literal[
-                "openai",
-                "cohere",
-                "vertex_ai",
-                "anthropic",
-                "azure",
-                "gemini",
-                "launch",
-                "llmengine",
-                "model_zoo",
-                "bedrock",
-                "xai",
-            ]
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "xai",
         ]
         | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         # 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,
@@ -728,6 +727,7 @@ class AsyncModelsResource(AsyncAPIResource):
                         "limit": limit,
                         "model_vendor": model_vendor,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     model_list_params.ModelListParams,