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

scale_gp_beta/__init__.py

@@ -1,5 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+import typing as _t
+
 from . import types
 from ._types import NOT_GIVEN, Omit, NoneType, NotGiven, Transport, ProxiesTypes
 from ._utils import file_from_path
@@ -80,6 +82,9 @@ __all__ = [
     "DefaultAsyncHttpxClient",
 ]
 
+if not _t.TYPE_CHECKING:
+    from ._utils._resources_proxy import resources as resources
+
 _setup_logging()
 
 # Update the __module__ attribute for exported symbols so that

scale_gp_beta/_utils/_proxy.py

@@ -46,7 +46,10 @@ class LazyProxy(Generic[T], ABC):
     @property  # type: ignore
     @override
     def __class__(self) -> type:  # pyright: ignore
-        proxied = self.__get_proxied__()
+        try:
+            proxied = self.__get_proxied__()
+        except Exception:
+            return type(self)
         if issubclass(type(proxied), LazyProxy):
             return type(proxied)
         return proxied.__class__

scale_gp_beta/_utils/_resources_proxy.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from typing import Any
+from typing_extensions import override
+
+from ._proxy import LazyProxy
+
+
+class ResourcesProxy(LazyProxy[Any]):
+    """A proxy for the `scale_gp_beta.resources` module.
+
+    This is used so that we can lazily import `scale_gp_beta.resources` only when
+    needed *and* so that users can just import `scale_gp_beta` and reference `scale_gp_beta.resources`
+    """
+
+    @override
+    def __load__(self) -> Any:
+        import importlib
+
+        mod = importlib.import_module("scale_gp_beta.resources")
+        return mod
+
+
+resources = ResourcesProxy().__as_proxied__()

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.12"  # 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,10 +18,11 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ..._streaming import Stream, AsyncStream
-from ...types.chat import completion_create_params
+from ...types.chat import completion_create_params, completion_models_params
 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"]
 
@@ -518,6 +519,67 @@ class CompletionsResource(SyncAPIResource):
             stream_cls=Stream[ChatCompletionChunk],
         )
 
+    def models(
+        self,
+        *,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> CompletionModelsResponse:
+        """
+        Chat Completions
+
+        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
+        """
+        return self._get(
+            "/v5/chat/completions/models",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "ending_before": ending_before,
+                        "limit": limit,
+                        "model_vendor": model_vendor,
+                        "sort_order": sort_order,
+                        "starting_after": starting_after,
+                    },
+                    completion_models_params.CompletionModelsParams,
+                ),
+            ),
+            cast_to=CompletionModelsResponse,
+        )
+
 
 class AsyncCompletionsResource(AsyncAPIResource):
     @cached_property
@@ -1011,6 +1073,67 @@ class AsyncCompletionsResource(AsyncAPIResource):
             stream_cls=AsyncStream[ChatCompletionChunk],
         )
 
+    async def models(
+        self,
+        *,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> CompletionModelsResponse:
+        """
+        Chat Completions
+
+        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
+        """
+        return await self._get(
+            "/v5/chat/completions/models",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                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,
+                ),
+            ),
+            cast_to=CompletionModelsResponse,
+        )
+
 
 class CompletionsResourceWithRawResponse:
     def __init__(self, completions: CompletionsResource) -> None:
@@ -1019,6 +1142,9 @@ class CompletionsResourceWithRawResponse:
         self.create = to_raw_response_wrapper(
             completions.create,
         )
+        self.models = to_raw_response_wrapper(
+            completions.models,
+        )
 
 
 class AsyncCompletionsResourceWithRawResponse:
@@ -1028,6 +1154,9 @@ class AsyncCompletionsResourceWithRawResponse:
         self.create = async_to_raw_response_wrapper(
             completions.create,
         )
+        self.models = async_to_raw_response_wrapper(
+            completions.models,
+        )
 
 
 class CompletionsResourceWithStreamingResponse:
@@ -1037,6 +1166,9 @@ class CompletionsResourceWithStreamingResponse:
         self.create = to_streamed_response_wrapper(
             completions.create,
         )
+        self.models = to_streamed_response_wrapper(
+            completions.models,
+        )
 
 
 class AsyncCompletionsResourceWithStreamingResponse:
@@ -1046,3 +1178,6 @@ class AsyncCompletionsResourceWithStreamingResponse:
         self.create = async_to_streamed_response_wrapper(
             completions.create,
         )
+        self.models = async_to_streamed_response_wrapper(
+            completions.models,
+        )

scale_gp_beta/resources/dataset_items.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict, Iterable, Optional
+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"]
 
@@ -55,7 +55,7 @@ class DatasetItemsResource(SyncAPIResource):
         self,
         dataset_item_id: str,
         *,
-        version: Optional[int] | 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.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -132,12 +132,13 @@ class DatasetItemsResource(SyncAPIResource):
     def list(
         self,
         *,
-        dataset_id: Optional[str] | NotGiven = NOT_GIVEN,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        dataset_id: 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,
-        version: Optional[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.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -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",
         )
 
 
@@ -288,7 +292,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         self,
         dataset_item_id: str,
         *,
-        version: Optional[int] | 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.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -367,12 +371,13 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
     def list(
         self,
         *,
-        dataset_id: Optional[str] | NotGiven = NOT_GIVEN,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        dataset_id: 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,
-        version: Optional[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.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -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

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable, Optional
+from typing import Dict, List, Iterable
+from typing_extensions import Literal
 
 import httpx
 
@@ -181,10 +182,13 @@ class DatasetsResource(SyncAPIResource):
     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,
         # 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,
@@ -217,7 +221,10 @@ class DatasetsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
+                        "tags": tags,
                     },
                     dataset_list_params.DatasetListParams,
                 ),
@@ -415,10 +422,13 @@ class AsyncDatasetsResource(AsyncAPIResource):
     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,
         # 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,
@@ -451,7 +461,10 @@ class AsyncDatasetsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
+                        "tags": tags,
                     },
                     dataset_list_params.DatasetListParams,
                 ),

scale_gp_beta/resources/evaluation_items.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing_extensions import Literal
 
 import httpx
 
@@ -87,11 +87,12 @@ class EvaluationItemsResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
-        evaluation_id: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        evaluation_id: str | NotGiven = NOT_GIVEN,
         include_archived: bool | 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,
@@ -125,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,
@@ -197,11 +199,12 @@ class AsyncEvaluationItemsResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
-        evaluation_id: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        evaluation_id: str | NotGiven = NOT_GIVEN,
         include_archived: bool | 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,
@@ -235,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,