scale-gp-beta 0.1.0a9__py3-none-any.whl → 0.1.0a11__py3-none-any.whl

scale_gp_beta/_client.py

@@ -19,12 +19,9 @@ from ._types import (
     ProxiesTypes,
     RequestOptions,
 )
-from ._utils import (
-    is_given,
-    get_async_library,
-)
+from ._utils import is_given, get_async_library
 from ._version import __version__
-from .resources import models, datasets, inference, completions, evaluations, dataset_items, evaluation_items
+from .resources import spans, models, datasets, inference, completions, evaluations, dataset_items, evaluation_items
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import APIStatusError, SGPClientError
 from ._base_client import (
@@ -63,6 +60,7 @@ class SGPClient(SyncAPIClient):
     evaluations: evaluations.EvaluationsResource
     dataset_items: dataset_items.DatasetItemsResource
     evaluation_items: evaluation_items.EvaluationItemsResource
+    spans: spans.SpansResource
     with_raw_response: SGPClientWithRawResponse
     with_streaming_response: SGPClientWithStreamedResponse
 
@@ -165,6 +163,7 @@ class SGPClient(SyncAPIClient):
         self.evaluations = evaluations.EvaluationsResource(self)
         self.dataset_items = dataset_items.DatasetItemsResource(self)
         self.evaluation_items = evaluation_items.EvaluationItemsResource(self)
+        self.spans = spans.SpansResource(self)
         self.with_raw_response = SGPClientWithRawResponse(self)
         self.with_streaming_response = SGPClientWithStreamedResponse(self)
 
@@ -288,6 +287,7 @@ class AsyncSGPClient(AsyncAPIClient):
     evaluations: evaluations.AsyncEvaluationsResource
     dataset_items: dataset_items.AsyncDatasetItemsResource
     evaluation_items: evaluation_items.AsyncEvaluationItemsResource
+    spans: spans.AsyncSpansResource
     with_raw_response: AsyncSGPClientWithRawResponse
     with_streaming_response: AsyncSGPClientWithStreamedResponse
 
@@ -390,6 +390,7 @@ class AsyncSGPClient(AsyncAPIClient):
         self.evaluations = evaluations.AsyncEvaluationsResource(self)
         self.dataset_items = dataset_items.AsyncDatasetItemsResource(self)
         self.evaluation_items = evaluation_items.AsyncEvaluationItemsResource(self)
+        self.spans = spans.AsyncSpansResource(self)
         self.with_raw_response = AsyncSGPClientWithRawResponse(self)
         self.with_streaming_response = AsyncSGPClientWithStreamedResponse(self)
 
@@ -514,6 +515,7 @@ class SGPClientWithRawResponse:
         self.evaluations = evaluations.EvaluationsResourceWithRawResponse(client.evaluations)
         self.dataset_items = dataset_items.DatasetItemsResourceWithRawResponse(client.dataset_items)
         self.evaluation_items = evaluation_items.EvaluationItemsResourceWithRawResponse(client.evaluation_items)
+        self.spans = spans.SpansResourceWithRawResponse(client.spans)
 
 
 class AsyncSGPClientWithRawResponse:
@@ -527,6 +529,7 @@ class AsyncSGPClientWithRawResponse:
         self.evaluations = evaluations.AsyncEvaluationsResourceWithRawResponse(client.evaluations)
         self.dataset_items = dataset_items.AsyncDatasetItemsResourceWithRawResponse(client.dataset_items)
         self.evaluation_items = evaluation_items.AsyncEvaluationItemsResourceWithRawResponse(client.evaluation_items)
+        self.spans = spans.AsyncSpansResourceWithRawResponse(client.spans)
 
 
 class SGPClientWithStreamedResponse:
@@ -540,6 +543,7 @@ class SGPClientWithStreamedResponse:
         self.evaluations = evaluations.EvaluationsResourceWithStreamingResponse(client.evaluations)
         self.dataset_items = dataset_items.DatasetItemsResourceWithStreamingResponse(client.dataset_items)
         self.evaluation_items = evaluation_items.EvaluationItemsResourceWithStreamingResponse(client.evaluation_items)
+        self.spans = spans.SpansResourceWithStreamingResponse(client.spans)
 
 
 class AsyncSGPClientWithStreamedResponse:
@@ -555,6 +559,7 @@ class AsyncSGPClientWithStreamedResponse:
         self.evaluation_items = evaluation_items.AsyncEvaluationItemsResourceWithStreamingResponse(
             client.evaluation_items
         )
+        self.spans = spans.AsyncSpansResourceWithStreamingResponse(client.spans)
 
 
 Client = SGPClient

scale_gp_beta/_models.py

@@ -19,7 +19,6 @@ from typing_extensions import (
 )
 
 import pydantic
-import pydantic.generics
 from pydantic.fields import FieldInfo
 
 from ._types import (
@@ -627,8 +626,8 @@ def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any,
                 # Note: if one variant defines an alias then they all should
                 discriminator_alias = field_info.alias
 
-                if field_info.annotation and is_literal_type(field_info.annotation):
-                    for entry in get_args(field_info.annotation):
+                if (annotation := getattr(field_info, "annotation", None)) and is_literal_type(annotation):
+                    for entry in get_args(annotation):
                         if isinstance(entry, str):
                             mapping[entry] = variant
 

scale_gp_beta/_response.py

@@ -235,7 +235,7 @@ class BaseAPIResponse(Generic[R]):
         # split is required to handle cases where additional information is included
         # in the response, e.g. application/json; charset=utf-8
         content_type, *_ = response.headers.get("content-type", "*").split(";")
-        if content_type != "application/json":
+        if not content_type.endswith("json"):
             if is_basemodel(cast_to):
                 try:
                     data = response.json()

scale_gp_beta/_utils/_transform.py

@@ -5,13 +5,15 @@ import base64
 import pathlib
 from typing import Any, Mapping, TypeVar, cast
 from datetime import date, datetime
-from typing_extensions import Literal, get_args, override, get_type_hints
+from typing_extensions import Literal, get_args, override, get_type_hints as _get_type_hints
 
 import anyio
 import pydantic
 
 from ._utils import (
     is_list,
+    is_given,
+    lru_cache,
     is_mapping,
     is_iterable,
 )
@@ -108,6 +110,7 @@ def transform(
     return cast(_T, transformed)
 
 
+@lru_cache(maxsize=8096)
 def _get_annotated_type(type_: type) -> type | None:
     """If the given type is an `Annotated` type then it is returned, if not `None` is returned.
 
@@ -142,6 +145,10 @@ def _maybe_transform_key(key: str, type_: type) -> str:
     return key
 
 
+def _no_transform_needed(annotation: type) -> bool:
+    return annotation == float or annotation == int
+
+
 def _transform_recursive(
     data: object,
     *,
@@ -184,6 +191,15 @@ def _transform_recursive(
             return cast(object, data)
 
         inner_type = extract_type_arg(stripped_type, 0)
+        if _no_transform_needed(inner_type):
+            # for some types there is no need to transform anything, so we can get a small
+            # perf boost from skipping that work.
+            #
+            # but we still need to convert to a list to ensure the data is json-serializable
+            if is_list(data):
+                return data
+            return list(data)
+
         return [_transform_recursive(d, annotation=annotation, inner_type=inner_type) for d in data]
 
     if is_union_type(stripped_type):
@@ -245,6 +261,11 @@ def _transform_typeddict(
     result: dict[str, object] = {}
     annotations = get_type_hints(expected_type, include_extras=True)
     for key, value in data.items():
+        if not is_given(value):
+            # we don't need to include `NotGiven` values here as they'll
+            # be stripped out before the request is sent anyway
+            continue
+
         type_ = annotations.get(key)
         if type_ is None:
             # we do not have a type annotation for this field, leave it as is
@@ -332,6 +353,15 @@ async def _async_transform_recursive(
             return cast(object, data)
 
         inner_type = extract_type_arg(stripped_type, 0)
+        if _no_transform_needed(inner_type):
+            # for some types there is no need to transform anything, so we can get a small
+            # perf boost from skipping that work.
+            #
+            # but we still need to convert to a list to ensure the data is json-serializable
+            if is_list(data):
+                return data
+            return list(data)
+
         return [await _async_transform_recursive(d, annotation=annotation, inner_type=inner_type) for d in data]
 
     if is_union_type(stripped_type):
@@ -393,6 +423,11 @@ async def _async_transform_typeddict(
     result: dict[str, object] = {}
     annotations = get_type_hints(expected_type, include_extras=True)
     for key, value in data.items():
+        if not is_given(value):
+            # we don't need to include `NotGiven` values here as they'll
+            # be stripped out before the request is sent anyway
+            continue
+
         type_ = annotations.get(key)
         if type_ is None:
             # we do not have a type annotation for this field, leave it as is
@@ -400,3 +435,13 @@ async def _async_transform_typeddict(
         else:
             result[_maybe_transform_key(key, type_)] = await _async_transform_recursive(value, annotation=type_)
     return result
+
+
+@lru_cache(maxsize=8096)
+def get_type_hints(
+    obj: Any,
+    globalns: dict[str, Any] | None = None,
+    localns: Mapping[str, Any] | None = None,
+    include_extras: bool = False,
+) -> dict[str, Any]:
+    return _get_type_hints(obj, globalns=globalns, localns=localns, include_extras=include_extras)

scale_gp_beta/_utils/_typing.py

@@ -13,6 +13,7 @@ from typing_extensions import (
     get_origin,
 )
 
+from ._utils import lru_cache
 from .._types import InheritsGeneric
 from .._compat import is_union as _is_union
 
@@ -66,6 +67,7 @@ def is_type_alias_type(tp: Any, /) -> TypeIs[typing_extensions.TypeAliasType]:
 
 
 # Extracts T from Annotated[T, ...] or from Required[Annotated[T, ...]]
+@lru_cache(maxsize=8096)
 def strip_annotated_type(typ: type) -> type:
     if is_required_type(typ) or is_annotated_type(typ):
         return strip_annotated_type(cast(type, get_args(typ)[0]))
@@ -108,7 +110,7 @@ def extract_type_var_from_base(
     ```
     """
     cls = cast(object, get_origin(typ) or typ)
-    if cls in generic_bases:
+    if cls in generic_bases:  # pyright: ignore[reportUnnecessaryContains]
         # we're given the class directly
         return extract_type_arg(typ, index)
 

scale_gp_beta/_utils/_utils.py

@@ -72,8 +72,16 @@ def _extract_items(
         from .._files import assert_is_file_content
 
         # We have exhausted the path, return the entry we found.
-        assert_is_file_content(obj, key=flattened_key)
         assert flattened_key is not None
+
+        if is_list(obj):
+            files: list[tuple[str, FileTypes]] = []
+            for entry in obj:
+                assert_is_file_content(entry, key=flattened_key + "[]" if flattened_key else "")
+                files.append((flattened_key + "[]", cast(FileTypes, entry)))
+            return files
+
+        assert_is_file_content(obj, key=flattened_key)
         return [(flattened_key, cast(FileTypes, obj))]
 
     index += 1

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

scale_gp_beta/resources/__init__.py

@@ -16,6 +16,14 @@ from .files import (
     FilesResourceWithStreamingResponse,
     AsyncFilesResourceWithStreamingResponse,
 )
+from .spans import (
+    SpansResource,
+    AsyncSpansResource,
+    SpansResourceWithRawResponse,
+    AsyncSpansResourceWithRawResponse,
+    SpansResourceWithStreamingResponse,
+    AsyncSpansResourceWithStreamingResponse,
+)
 from .models import (
     ModelsResource,
     AsyncModelsResource,
@@ -128,4 +136,10 @@ __all__ = [
     "AsyncEvaluationItemsResourceWithRawResponse",
     "EvaluationItemsResourceWithStreamingResponse",
     "AsyncEvaluationItemsResourceWithStreamingResponse",
+    "SpansResource",
+    "AsyncSpansResource",
+    "SpansResourceWithRawResponse",
+    "AsyncSpansResourceWithRawResponse",
+    "SpansResourceWithStreamingResponse",
+    "AsyncSpansResourceWithStreamingResponse",
 ]

scale_gp_beta/resources/chat/completions.py

@@ -8,11 +8,7 @@ from typing_extensions import Literal, overload
 import httpx
 
 from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from ..._utils import (
-    required_args,
-    maybe_transform,
-    async_maybe_transform,
-)
+from ..._utils import required_args, maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -508,7 +504,9 @@ class CompletionsResource(SyncAPIResource):
                     "top_logprobs": top_logprobs,
                     "top_p": top_p,
                 },
-                completion_create_params.CompletionCreateParams,
+                completion_create_params.CompletionCreateParamsStreaming
+                if stream
+                else completion_create_params.CompletionCreateParamsNonStreaming,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -999,7 +997,9 @@ class AsyncCompletionsResource(AsyncAPIResource):
                     "top_logprobs": top_logprobs,
                     "top_p": top_p,
                 },
-                completion_create_params.CompletionCreateParams,
+                completion_create_params.CompletionCreateParamsStreaming
+                if stream
+                else completion_create_params.CompletionCreateParamsNonStreaming,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout

scale_gp_beta/resources/completions.py

@@ -9,11 +9,7 @@ import httpx
 
 from ..types import completion_create_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    required_args,
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -368,7 +364,9 @@ class CompletionsResource(SyncAPIResource):
                     "top_p": top_p,
                     "user": user,
                 },
-                completion_create_params.CompletionCreateParams,
+                completion_create_params.CompletionCreateParamsStreaming
+                if stream
+                else completion_create_params.CompletionCreateParamsNonStreaming,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -718,7 +716,9 @@ class AsyncCompletionsResource(AsyncAPIResource):
                     "top_p": top_p,
                     "user": user,
                 },
-                completion_create_params.CompletionCreateParams,
+                completion_create_params.CompletionCreateParamsStreaming
+                if stream
+                else completion_create_params.CompletionCreateParamsNonStreaming,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout

scale_gp_beta/resources/dataset_items.py

@@ -13,10 +13,7 @@ from ..types import (
     dataset_item_batch_create_params,
 )
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (

scale_gp_beta/resources/datasets.py

@@ -2,21 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Dict, Iterable, Optional
+from typing import Dict, List, Iterable, Optional
 
 import httpx
 
-from ..types import (
-    dataset_list_params,
-    dataset_create_params,
-    dataset_update_params,
-    dataset_retrieve_params,
-)
+from ..types import dataset_list_params, dataset_create_params, dataset_update_params, dataset_retrieve_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -59,6 +51,7 @@ class DatasetsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: 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,
@@ -72,6 +65,8 @@ class DatasetsResource(SyncAPIResource):
         Args:
           data: Items to be included in the dataset
 
+          tags: The tags associated with the entity
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -87,6 +82,7 @@ class DatasetsResource(SyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                 },
                 dataset_create_params.DatasetCreateParams,
             ),
@@ -142,6 +138,7 @@ class DatasetsResource(SyncAPIResource):
         *,
         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,
@@ -153,6 +150,8 @@ class DatasetsResource(SyncAPIResource):
         Update Dataset
 
         Args:
+          tags: The tags associated with the entity
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -169,6 +168,7 @@ class DatasetsResource(SyncAPIResource):
                 {
                     "description": description,
                     "name": name,
+                    "tags": tags,
                 },
                 dataset_update_params.DatasetUpdateParams,
             ),
@@ -285,6 +285,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: 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,
@@ -298,6 +299,8 @@ class AsyncDatasetsResource(AsyncAPIResource):
         Args:
           data: Items to be included in the dataset
 
+          tags: The tags associated with the entity
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -313,6 +316,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                 },
                 dataset_create_params.DatasetCreateParams,
             ),
@@ -368,6 +372,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         *,
         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,
@@ -379,6 +384,8 @@ class AsyncDatasetsResource(AsyncAPIResource):
         Update Dataset
 
         Args:
+          tags: The tags associated with the entity
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -395,6 +402,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                 {
                     "description": description,
                     "name": name,
+                    "tags": tags,
                 },
                 dataset_update_params.DatasetUpdateParams,
             ),

scale_gp_beta/resources/evaluation_items.py

@@ -8,10 +8,7 @@ import httpx
 
 from ..types import evaluation_item_list_params, evaluation_item_retrieve_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (

scale_gp_beta/resources/evaluations.py

@@ -9,11 +9,7 @@ import httpx
 
 from ..types import evaluation_list_params, evaluation_create_params, evaluation_retrieve_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    required_args,
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -58,6 +54,7 @@ class EvaluationsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -72,6 +69,8 @@ class EvaluationsResource(SyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -92,6 +91,7 @@ class EvaluationsResource(SyncAPIResource):
         name: str,
         data: Iterable[evaluation_create_params.EvaluationFromDatasetCreateRequestData] | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -108,6 +108,8 @@ class EvaluationsResource(SyncAPIResource):
 
           data: Items to be evaluated, including references to the input dataset items
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -128,6 +130,7 @@ class EvaluationsResource(SyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -144,6 +147,8 @@ class EvaluationsResource(SyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -165,6 +170,7 @@ class EvaluationsResource(SyncAPIResource):
         | NotGiven = NOT_GIVEN,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         dataset_id: str | NotGiven = NOT_GIVEN,
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset | NotGiven = NOT_GIVEN,
@@ -182,6 +188,7 @@ class EvaluationsResource(SyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                     "tasks": tasks,
                     "dataset_id": dataset_id,
                     "dataset": dataset,
@@ -349,6 +356,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -363,6 +371,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -383,6 +393,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         name: str,
         data: Iterable[evaluation_create_params.EvaluationFromDatasetCreateRequestData] | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -399,6 +410,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           data: Items to be evaluated, including references to the input dataset items
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -419,6 +432,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | 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.
@@ -435,6 +449,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -456,6 +472,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         | NotGiven = NOT_GIVEN,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         dataset_id: str | NotGiven = NOT_GIVEN,
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset | NotGiven = NOT_GIVEN,
@@ -473,6 +490,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                     "tasks": tasks,
                     "dataset_id": dataset_id,
                     "dataset": dataset,

scale_gp_beta/resources/files/files.py

@@ -16,12 +16,7 @@ from .content import (
     AsyncContentResourceWithStreamingResponse,
 )
 from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes
-from ..._utils import (
-    extract_files,
-    maybe_transform,
-    deepcopy_minimal,
-    async_maybe_transform,
-)
+from ..._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (

scale_gp_beta/resources/inference.py

@@ -8,10 +8,7 @@ import httpx
 
 from ..types import inference_create_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (