scale-gp-beta 0.1.0a35__py3-none-any.whl → 0.1.0a37__py3-none-any.whl

scale_gp_beta/_models.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import os
 import inspect
+import weakref
 from typing import TYPE_CHECKING, Any, Type, Union, Generic, TypeVar, Callable, Optional, cast
 from datetime import date, datetime
 from typing_extensions import (
@@ -573,6 +574,9 @@ class CachedDiscriminatorType(Protocol):
     __discriminator__: DiscriminatorDetails
 
 
+DISCRIMINATOR_CACHE: weakref.WeakKeyDictionary[type, DiscriminatorDetails] = weakref.WeakKeyDictionary()
+
+
 class DiscriminatorDetails:
     field_name: str
     """The name of the discriminator field in the variant class, e.g.
@@ -615,8 +619,9 @@ class DiscriminatorDetails:
 
 
 def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any, ...]) -> DiscriminatorDetails | None:
-    if isinstance(union, CachedDiscriminatorType):
-        return union.__discriminator__
+    cached = DISCRIMINATOR_CACHE.get(union)
+    if cached is not None:
+        return cached
 
     discriminator_field_name: str | None = None
 
@@ -669,7 +674,7 @@ def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any,
         discriminator_field=discriminator_field_name,
         discriminator_alias=discriminator_alias,
     )
-    cast(CachedDiscriminatorType, union).__discriminator__ = details
+    DISCRIMINATOR_CACHE.setdefault(union, details)
     return details
 
 

scale_gp_beta/_streaming.py

@@ -57,9 +57,8 @@ class Stream(Generic[_T]):
         for sse in iterator:
             yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -121,9 +120,8 @@ class AsyncStream(Generic[_T]):
         async for sse in iterator:
             yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        async for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

scale_gp_beta/_utils/_sync.py

@@ -1,10 +1,8 @@
 from __future__ import annotations
 
-import sys
 import asyncio
 import functools
-import contextvars
-from typing import Any, TypeVar, Callable, Awaitable
+from typing import TypeVar, Callable, Awaitable
 from typing_extensions import ParamSpec
 
 import anyio
@@ -15,34 +13,11 @@ T_Retval = TypeVar("T_Retval")
 T_ParamSpec = ParamSpec("T_ParamSpec")
 
 
-if sys.version_info >= (3, 9):
-    _asyncio_to_thread = asyncio.to_thread
-else:
-    # backport of https://docs.python.org/3/library/asyncio-task.html#asyncio.to_thread
-    # for Python 3.8 support
-    async def _asyncio_to_thread(
-        func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
-    ) -> Any:
-        """Asynchronously run function *func* in a separate thread.
-
-        Any *args and **kwargs supplied for this function are directly passed
-        to *func*. Also, the current :class:`contextvars.Context` is propagated,
-        allowing context variables from the main thread to be accessed in the
-        separate thread.
-
-        Returns a coroutine that can be awaited to get the eventual result of *func*.
-        """
-        loop = asyncio.events.get_running_loop()
-        ctx = contextvars.copy_context()
-        func_call = functools.partial(ctx.run, func, *args, **kwargs)
-        return await loop.run_in_executor(None, func_call)
-
-
 async def to_thread(
     func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
 ) -> T_Retval:
     if sniffio.current_async_library() == "asyncio":
-        return await _asyncio_to_thread(func, *args, **kwargs)
+        return await asyncio.to_thread(func, *args, **kwargs)
 
     return await anyio.to_thread.run_sync(
         functools.partial(func, *args, **kwargs),
@@ -53,10 +28,7 @@ async def to_thread(
 def asyncify(function: Callable[T_ParamSpec, T_Retval]) -> Callable[T_ParamSpec, Awaitable[T_Retval]]:
     """
     Take a blocking function and create an async one that receives the same
-    positional and keyword arguments. For python version 3.9 and above, it uses
-    asyncio.to_thread to run the function in a separate thread. For python version
-    3.8, it uses locally defined copy of the asyncio.to_thread function which was
-    introduced in python 3.9.
+    positional and keyword arguments.
 
     Usage:
 

scale_gp_beta/_utils/_utils.py

@@ -133,7 +133,7 @@ def is_given(obj: _T | NotGiven | Omit) -> TypeGuard[_T]:
 # Type safe methods for narrowing types with TypeVars.
 # The default narrowing for isinstance(obj, dict) is dict[unknown, unknown],
 # however this cause Pyright to rightfully report errors. As we know we don't
-# care about the contained types we can safely use `object` in it's place.
+# care about the contained types we can safely use `object` in its place.
 #
 # There are two separate functions defined, `is_*` and `is_*_t` for different use cases.
 # `is_*` is for when you're dealing with an unknown input

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

scale_gp_beta/resources/dataset_items.py

@@ -98,6 +98,7 @@ class DatasetItemsResource(SyncAPIResource):
         dataset_item_id: str,
         *,
         data: Dict[str, object],
+        files: Dict[str, str] | 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,
@@ -111,6 +112,8 @@ class DatasetItemsResource(SyncAPIResource):
         Args:
           data: Updated dataset item data
 
+          files: Files to be associated to the dataset
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -123,7 +126,13 @@ class DatasetItemsResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `dataset_item_id` but received {dataset_item_id!r}")
         return self._patch(
             f"/v5/dataset-items/{dataset_item_id}",
-            body=maybe_transform({"data": data}, dataset_item_update_params.DatasetItemUpdateParams),
+            body=maybe_transform(
+                {
+                    "data": data,
+                    "files": files,
+                },
+                dataset_item_update_params.DatasetItemUpdateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -228,6 +237,7 @@ class DatasetItemsResource(SyncAPIResource):
         *,
         data: Iterable[Dict[str, object]],
         dataset_id: str,
+        files: Iterable[Dict[str, str]] | 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,
@@ -243,6 +253,8 @@ class DatasetItemsResource(SyncAPIResource):
 
           dataset_id: Identifier of the target dataset
 
+          files: Files to be associated to the dataset
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -257,6 +269,7 @@ class DatasetItemsResource(SyncAPIResource):
                 {
                     "data": data,
                     "dataset_id": dataset_id,
+                    "files": files,
                 },
                 dataset_item_batch_create_params.DatasetItemBatchCreateParams,
             ),
@@ -335,6 +348,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         dataset_item_id: str,
         *,
         data: Dict[str, object],
+        files: Dict[str, str] | 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,
@@ -348,6 +362,8 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         Args:
           data: Updated dataset item data
 
+          files: Files to be associated to the dataset
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -360,7 +376,13 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `dataset_item_id` but received {dataset_item_id!r}")
         return await self._patch(
             f"/v5/dataset-items/{dataset_item_id}",
-            body=await async_maybe_transform({"data": data}, dataset_item_update_params.DatasetItemUpdateParams),
+            body=await async_maybe_transform(
+                {
+                    "data": data,
+                    "files": files,
+                },
+                dataset_item_update_params.DatasetItemUpdateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -465,6 +487,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         *,
         data: Iterable[Dict[str, object]],
         dataset_id: str,
+        files: Iterable[Dict[str, str]] | 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,
@@ -480,6 +503,8 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
 
           dataset_id: Identifier of the target dataset
 
+          files: Files to be associated to the dataset
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -494,6 +519,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
                 {
                     "data": data,
                     "dataset_id": dataset_id,
+                    "files": files,
                 },
                 dataset_item_batch_create_params.DatasetItemBatchCreateParams,
             ),

scale_gp_beta/resources/datasets.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import Dict, Iterable
-from typing_extensions import Literal
+from typing_extensions import Literal, overload
 
 import httpx
 
@@ -52,6 +52,7 @@ class DatasetsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         tags: SequenceNotStr[str] | 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.
@@ -66,6 +67,8 @@ class DatasetsResource(SyncAPIResource):
         Args:
           data: Items to be included in the dataset
 
+          files: Files to be associated to the dataset
+
           tags: The tags associated with the entity
 
           extra_headers: Send extra headers
@@ -83,6 +86,7 @@ class DatasetsResource(SyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "files": files,
                     "tags": tags,
                 },
                 dataset_create_params.DatasetCreateParams,
@@ -133,6 +137,7 @@ class DatasetsResource(SyncAPIResource):
             cast_to=Dataset,
         )
 
+    @overload
     def update(
         self,
         dataset_id: str,
@@ -148,7 +153,7 @@ class DatasetsResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Dataset:
         """
-        Update Dataset
+        Update or Restore Dataset
 
         Args:
           tags: The tags associated with the entity
@@ -161,6 +166,52 @@ class DatasetsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        ...
+
+    @overload
+    def update(
+        self,
+        dataset_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,
+    ) -> Dataset:
+        """
+        Update or Restore Dataset
+
+        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,
+        dataset_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,
+    ) -> Dataset:
         if not dataset_id:
             raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}")
         return self._patch(
@@ -170,6 +221,7 @@ class DatasetsResource(SyncAPIResource):
                     "description": description,
                     "name": name,
                     "tags": tags,
+                    "restore": restore,
                 },
                 dataset_update_params.DatasetUpdateParams,
             ),
@@ -292,6 +344,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         tags: SequenceNotStr[str] | 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.
@@ -306,6 +359,8 @@ class AsyncDatasetsResource(AsyncAPIResource):
         Args:
           data: Items to be included in the dataset
 
+          files: Files to be associated to the dataset
+
           tags: The tags associated with the entity
 
           extra_headers: Send extra headers
@@ -323,6 +378,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "files": files,
                     "tags": tags,
                 },
                 dataset_create_params.DatasetCreateParams,
@@ -373,6 +429,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
             cast_to=Dataset,
         )
 
+    @overload
     async def update(
         self,
         dataset_id: str,
@@ -388,7 +445,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Dataset:
         """
-        Update Dataset
+        Update or Restore Dataset
 
         Args:
           tags: The tags associated with the entity
@@ -401,6 +458,52 @@ class AsyncDatasetsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        ...
+
+    @overload
+    async def update(
+        self,
+        dataset_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,
+    ) -> Dataset:
+        """
+        Update or Restore Dataset
+
+        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,
+        dataset_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,
+    ) -> Dataset:
         if not dataset_id:
             raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}")
         return await self._patch(
@@ -410,6 +513,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                     "description": description,
                     "name": name,
                     "tags": tags,
+                    "restore": restore,
                 },
                 dataset_update_params.DatasetUpdateParams,
             ),

scale_gp_beta/resources/evaluations.py

@@ -58,6 +58,7 @@ class EvaluationsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -74,6 +75,8 @@ class EvaluationsResource(SyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          files: Files to be associated to the evaluation
+
           metadata: Optional metadata key-value pairs for the evaluation
 
           tags: The tags associated with the entity
@@ -140,6 +143,7 @@ class EvaluationsResource(SyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -158,6 +162,8 @@ class EvaluationsResource(SyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          files: Files to be associated to the evaluation
+
           metadata: Optional metadata key-value pairs for the evaluation
 
           tags: The tags associated with the entity
@@ -183,6 +189,7 @@ class EvaluationsResource(SyncAPIResource):
         | Omit = omit,
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -202,6 +209,7 @@ class EvaluationsResource(SyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "files": files,
                     "metadata": metadata,
                     "tags": tags,
                     "tasks": tasks,
@@ -471,6 +479,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -487,6 +496,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          files: Files to be associated to the evaluation
+
           metadata: Optional metadata key-value pairs for the evaluation
 
           tags: The tags associated with the entity
@@ -553,6 +564,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -571,6 +583,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          files: Files to be associated to the evaluation
+
           metadata: Optional metadata key-value pairs for the evaluation
 
           tags: The tags associated with the entity
@@ -596,6 +610,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         | Omit = omit,
         name: str,
         description: str | Omit = omit,
+        files: Iterable[Dict[str, str]] | Omit = omit,
         metadata: Dict[str, object] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         tasks: Iterable[EvaluationTaskParam] | Omit = omit,
@@ -615,6 +630,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "files": files,
                     "metadata": metadata,
                     "tags": tags,
                     "tasks": tasks,