scale-gp-beta 0.1.0a15__py3-none-any.whl → 0.1.0a17__py3-none-any.whl

scale_gp_beta/_base_client.py

@@ -960,6 +960,9 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
             if self.custom_auth is not None:
                 kwargs["auth"] = self.custom_auth
 
+            if options.follow_redirects is not None:
+                kwargs["follow_redirects"] = options.follow_redirects
+
             log.debug("Sending HTTP Request: %s %s", request.method, request.url)
 
             response = None
@@ -1460,6 +1463,9 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
             if self.custom_auth is not None:
                 kwargs["auth"] = self.custom_auth
 
+            if options.follow_redirects is not None:
+                kwargs["follow_redirects"] = options.follow_redirects
+
             log.debug("Sending HTTP Request: %s %s", request.method, request.url)
 
             response = None

scale_gp_beta/_models.py

@@ -737,6 +737,7 @@ class FinalRequestOptionsInput(TypedDict, total=False):
     idempotency_key: str
     json_data: Body
     extra_json: AnyMapping
+    follow_redirects: bool
 
 
 @final
@@ -750,6 +751,7 @@ class FinalRequestOptions(pydantic.BaseModel):
     files: Union[HttpxRequestFiles, None] = None
     idempotency_key: Union[str, None] = None
     post_parser: Union[Callable[[Any], Any], NotGiven] = NotGiven()
+    follow_redirects: Union[bool, None] = None
 
     # It should be noted that we cannot use `json` here as that would override
     # a BaseModel method in an incompatible fashion.

scale_gp_beta/_types.py

@@ -100,6 +100,7 @@ class RequestOptions(TypedDict, total=False):
     params: Query
     extra_json: AnyMapping
     idempotency_key: str
+    follow_redirects: bool
 
 
 # Sentinel class used until PEP 0661 is accepted
@@ -215,3 +216,4 @@ class _GenericAlias(Protocol):
 
 class HttpxSendArgs(TypedDict, total=False):
     auth: httpx.Auth
+    follow_redirects: bool

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.15"  # x-release-please-version
+__version__ = "0.1.0-alpha.17"  # x-release-please-version

scale_gp_beta/resources/dataset_items.py

@@ -23,10 +23,11 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..pagination import SyncCursorPage, AsyncCursorPage, SyncAPIListPage, AsyncAPIListPage
+from ..pagination import SyncCursorPage, AsyncCursorPage
 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"]
 
@@ -233,7 +234,7 @@ class DatasetItemsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncAPIListPage[DatasetItem]:
+    ) -> DatasetItemBatchCreateResponse:
         """
         Batch Create Dataset Items
 
@@ -250,9 +251,8 @@ class DatasetItemsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._post(
             "/v5/dataset-items/batch",
-            page=SyncAPIListPage[DatasetItem],
             body=maybe_transform(
                 {
                     "data": data,
@@ -263,8 +263,7 @@ class DatasetItemsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            model=DatasetItem,
-            method="post",
+            cast_to=DatasetItemBatchCreateResponse,
         )
 
 
@@ -461,7 +460,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
             cast_to=DatasetItemDeleteResponse,
         )
 
-    def batch_create(
+    async def batch_create(
         self,
         *,
         data: Iterable[Dict[str, object]],
@@ -472,7 +471,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[DatasetItem, AsyncAPIListPage[DatasetItem]]:
+    ) -> DatasetItemBatchCreateResponse:
         """
         Batch Create Dataset Items
 
@@ -489,10 +488,9 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._post(
             "/v5/dataset-items/batch",
-            page=AsyncAPIListPage[DatasetItem],
-            body=maybe_transform(
+            body=await async_maybe_transform(
                 {
                     "data": data,
                     "dataset_id": dataset_id,
@@ -502,8 +500,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            model=DatasetItem,
-            method="post",
+            cast_to=DatasetItemBatchCreateResponse,
         )