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

scale_gp_beta/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "scale_gp_beta"
-__version__ = "0.1.0-alpha.13"  # x-release-please-version
+__version__ = "0.1.0-alpha.14"  # x-release-please-version

scale_gp_beta/pagination.py

@@ -5,7 +5,7 @@ from typing_extensions import Protocol, override, runtime_checkable
 
 from ._base_client import BasePage, PageInfo, BaseSyncPage, BaseAsyncPage
 
-__all__ = ["SyncCursorPage", "AsyncCursorPage"]
+__all__ = ["SyncCursorPage", "AsyncCursorPage", "SyncAPIListPage", "AsyncAPIListPage"]
 
 _T = TypeVar("_T")
 
@@ -81,3 +81,41 @@ class AsyncCursorPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
                 return None
 
             return PageInfo(params={"ending_before": item.id})
+
+
+class SyncAPIListPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    items: List[_T]
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> None:
+        """
+        This page represents a response that isn't actually paginated at the API level
+        so there will never be a next page.
+        """
+        return None
+
+
+class AsyncAPIListPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    items: List[_T]
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> None:
+        """
+        This page represents a response that isn't actually paginated at the API level
+        so there will never be a next page.
+        """
+        return None

scale_gp_beta/resources/chat/completions.py

@@ -18,12 +18,11 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ..._streaming import Stream, AsyncStream
-from ...pagination import SyncCursorPage, AsyncCursorPage
 from ...types.chat import completion_create_params, completion_models_params
-from ..._base_client import AsyncPaginator, make_request_options
-from ...types.chat.model_definition import ModelDefinition
+from ..._base_client import make_request_options
 from ...types.chat.chat_completion_chunk import ChatCompletionChunk
 from ...types.chat.completion_create_response import CompletionCreateResponse
+from ...types.chat.completion_models_response import CompletionModelsResponse
 
 __all__ = ["CompletionsResource", "AsyncCompletionsResource"]
 
@@ -539,6 +538,7 @@ class CompletionsResource(SyncAPIResource):
             "xai",
         ]
         | 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.
@@ -546,7 +546,7 @@ class CompletionsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncCursorPage[ModelDefinition]:
+    ) -> CompletionModelsResponse:
         """
         Chat Completions
 
@@ -559,9 +559,8 @@ class CompletionsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v5/chat/completions/models",
-            page=SyncCursorPage[ModelDefinition],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -572,12 +571,13 @@ class CompletionsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "limit": limit,
                         "model_vendor": model_vendor,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     completion_models_params.CompletionModelsParams,
                 ),
             ),
-            model=ModelDefinition,
+            cast_to=CompletionModelsResponse,
         )
 
 
@@ -1073,7 +1073,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
             stream_cls=AsyncStream[ChatCompletionChunk],
         )
 
-    def models(
+    async def models(
         self,
         *,
         ending_before: str | NotGiven = NOT_GIVEN,
@@ -1092,6 +1092,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
             "xai",
         ]
         | 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.
@@ -1099,7 +1100,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[ModelDefinition, AsyncCursorPage[ModelDefinition]]:
+    ) -> CompletionModelsResponse:
         """
         Chat Completions
 
@@ -1112,25 +1113,25 @@ class AsyncCompletionsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v5/chat/completions/models",
-            page=AsyncCursorPage[ModelDefinition],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "ending_before": ending_before,
                         "limit": limit,
                         "model_vendor": model_vendor,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     completion_models_params.CompletionModelsParams,
                 ),
             ),
-            model=ModelDefinition,
+            cast_to=CompletionModelsResponse,
         )
 
 

scale_gp_beta/resources/dataset_items.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Dict, Iterable
+from typing_extensions import Literal
 
 import httpx
 
@@ -22,11 +23,10 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..pagination import SyncCursorPage, AsyncCursorPage
+from ..pagination import SyncCursorPage, AsyncCursorPage, SyncAPIListPage, AsyncAPIListPage
 from .._base_client import AsyncPaginator, make_request_options
 from ..types.dataset_item import DatasetItem
 from ..types.dataset_item_delete_response import DatasetItemDeleteResponse
-from ..types.dataset_item_batch_create_response import DatasetItemBatchCreateResponse
 
 __all__ = ["DatasetItemsResource", "AsyncDatasetItemsResource"]
 
@@ -136,6 +136,7 @@ class DatasetItemsResource(SyncAPIResource):
         ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         version: int | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -178,6 +179,7 @@ class DatasetItemsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "version": version,
                     },
@@ -231,7 +233,7 @@ class DatasetItemsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> DatasetItemBatchCreateResponse:
+    ) -> SyncAPIListPage[DatasetItem]:
         """
         Batch Create Dataset Items
 
@@ -248,8 +250,9 @@ class DatasetItemsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._post(
+        return self._get_api_list(
             "/v5/dataset-items/batch",
+            page=SyncAPIListPage[DatasetItem],
             body=maybe_transform(
                 {
                     "data": data,
@@ -260,7 +263,8 @@ class DatasetItemsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=DatasetItemBatchCreateResponse,
+            model=DatasetItem,
+            method="post",
         )
 
 
@@ -371,6 +375,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         version: int | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -413,6 +418,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "version": version,
                     },
@@ -455,7 +461,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
             cast_to=DatasetItemDeleteResponse,
         )
 
-    async def batch_create(
+    def batch_create(
         self,
         *,
         data: Iterable[Dict[str, object]],
@@ -466,7 +472,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> DatasetItemBatchCreateResponse:
+    ) -> AsyncPaginator[DatasetItem, AsyncAPIListPage[DatasetItem]]:
         """
         Batch Create Dataset Items
 
@@ -483,9 +489,10 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._post(
+        return self._get_api_list(
             "/v5/dataset-items/batch",
-            body=await async_maybe_transform(
+            page=AsyncAPIListPage[DatasetItem],
+            body=maybe_transform(
                 {
                     "data": data,
                     "dataset_id": dataset_id,
@@ -495,7 +502,8 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=DatasetItemBatchCreateResponse,
+            model=DatasetItem,
+            method="post",
         )
 
 

scale_gp_beta/resources/datasets.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Dict, List, Iterable
+from typing_extensions import Literal
 
 import httpx
 
@@ -185,6 +186,7 @@ class DatasetsResource(SyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -220,6 +222,7 @@ class DatasetsResource(SyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
                     },
@@ -423,6 +426,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -458,6 +462,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
                     },

scale_gp_beta/resources/evaluation_items.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from typing_extensions import Literal
+
 import httpx
 
 from ..types import evaluation_item_list_params, evaluation_item_retrieve_params
@@ -89,6 +91,7 @@ class EvaluationItemsResource(SyncAPIResource):
         evaluation_id: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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.
@@ -123,6 +126,7 @@ class EvaluationItemsResource(SyncAPIResource):
                         "evaluation_id": evaluation_id,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     evaluation_item_list_params.EvaluationItemListParams,
@@ -199,6 +203,7 @@ class AsyncEvaluationItemsResource(AsyncAPIResource):
         evaluation_id: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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.
@@ -233,6 +238,7 @@ class AsyncEvaluationItemsResource(AsyncAPIResource):
                         "evaluation_id": evaluation_id,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
                     evaluation_item_list_params.EvaluationItemListParams,

scale_gp_beta/resources/evaluations.py

@@ -304,6 +304,7 @@ class EvaluationsResource(SyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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,
@@ -340,6 +341,7 @@ class EvaluationsResource(SyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
                         "views": views,
@@ -656,6 +658,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | 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,
@@ -692,6 +695,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
                         "views": views,

scale_gp_beta/resources/files/files.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Dict, Mapping, cast
+from typing_extensions import Literal
 
 import httpx
 
@@ -169,6 +170,7 @@ class FilesResource(SyncAPIResource):
         *,
         ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | 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.
@@ -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,
@@ -379,6 +382,7 @@ class AsyncFilesResource(AsyncAPIResource):
         *,
         ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | 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.
@@ -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

@@ -320,6 +320,7 @@ class ModelsResource(SyncAPIResource):
         ]
         | 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.
@@ -354,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,
@@ -690,6 +692,7 @@ class AsyncModelsResource(AsyncAPIResource):
         ]
         | 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.
@@ -724,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,