scale-gp-beta 0.1.0a30__py3-none-any.whl → 0.1.0a31__py3-none-any.whl

scale_gp_beta/_utils/_datetime_parse.py

@@ -0,0 +1,136 @@
+"""
+This file contains code from https://github.com/pydantic/pydantic/blob/main/pydantic/v1/datetime_parse.py
+without the Pydantic v1 specific errors.
+"""
+
+from __future__ import annotations
+
+import re
+from typing import Dict, Union, Optional
+from datetime import date, datetime, timezone, timedelta
+
+from .._types import StrBytesIntFloat
+
+date_expr = r"(?P<year>\d{4})-(?P<month>\d{1,2})-(?P<day>\d{1,2})"
+time_expr = (
+    r"(?P<hour>\d{1,2}):(?P<minute>\d{1,2})"
+    r"(?::(?P<second>\d{1,2})(?:\.(?P<microsecond>\d{1,6})\d{0,6})?)?"
+    r"(?P<tzinfo>Z|[+-]\d{2}(?::?\d{2})?)?$"
+)
+
+date_re = re.compile(f"{date_expr}$")
+datetime_re = re.compile(f"{date_expr}[T ]{time_expr}")
+
+
+EPOCH = datetime(1970, 1, 1)
+# if greater than this, the number is in ms, if less than or equal it's in seconds
+# (in seconds this is 11th October 2603, in ms it's 20th August 1970)
+MS_WATERSHED = int(2e10)
+# slightly more than datetime.max in ns - (datetime.max - EPOCH).total_seconds() * 1e9
+MAX_NUMBER = int(3e20)
+
+
+def _get_numeric(value: StrBytesIntFloat, native_expected_type: str) -> Union[None, int, float]:
+    if isinstance(value, (int, float)):
+        return value
+    try:
+        return float(value)
+    except ValueError:
+        return None
+    except TypeError:
+        raise TypeError(f"invalid type; expected {native_expected_type}, string, bytes, int or float") from None
+
+
+def _from_unix_seconds(seconds: Union[int, float]) -> datetime:
+    if seconds > MAX_NUMBER:
+        return datetime.max
+    elif seconds < -MAX_NUMBER:
+        return datetime.min
+
+    while abs(seconds) > MS_WATERSHED:
+        seconds /= 1000
+    dt = EPOCH + timedelta(seconds=seconds)
+    return dt.replace(tzinfo=timezone.utc)
+
+
+def _parse_timezone(value: Optional[str]) -> Union[None, int, timezone]:
+    if value == "Z":
+        return timezone.utc
+    elif value is not None:
+        offset_mins = int(value[-2:]) if len(value) > 3 else 0
+        offset = 60 * int(value[1:3]) + offset_mins
+        if value[0] == "-":
+            offset = -offset
+        return timezone(timedelta(minutes=offset))
+    else:
+        return None
+
+
+def parse_datetime(value: Union[datetime, StrBytesIntFloat]) -> datetime:
+    """
+    Parse a datetime/int/float/string and return a datetime.datetime.
+
+    This function supports time zone offsets. When the input contains one,
+    the output uses a timezone with a fixed offset from UTC.
+
+    Raise ValueError if the input is well formatted but not a valid datetime.
+    Raise ValueError if the input isn't well formatted.
+    """
+    if isinstance(value, datetime):
+        return value
+
+    number = _get_numeric(value, "datetime")
+    if number is not None:
+        return _from_unix_seconds(number)
+
+    if isinstance(value, bytes):
+        value = value.decode()
+
+    assert not isinstance(value, (float, int))
+
+    match = datetime_re.match(value)
+    if match is None:
+        raise ValueError("invalid datetime format")
+
+    kw = match.groupdict()
+    if kw["microsecond"]:
+        kw["microsecond"] = kw["microsecond"].ljust(6, "0")
+
+    tzinfo = _parse_timezone(kw.pop("tzinfo"))
+    kw_: Dict[str, Union[None, int, timezone]] = {k: int(v) for k, v in kw.items() if v is not None}
+    kw_["tzinfo"] = tzinfo
+
+    return datetime(**kw_)  # type: ignore
+
+
+def parse_date(value: Union[date, StrBytesIntFloat]) -> date:
+    """
+    Parse a date/int/float/string and return a datetime.date.
+
+    Raise ValueError if the input is well formatted but not a valid date.
+    Raise ValueError if the input isn't well formatted.
+    """
+    if isinstance(value, date):
+        if isinstance(value, datetime):
+            return value.date()
+        else:
+            return value
+
+    number = _get_numeric(value, "date")
+    if number is not None:
+        return _from_unix_seconds(number).date()
+
+    if isinstance(value, bytes):
+        value = value.decode()
+
+    assert not isinstance(value, (float, int))
+    match = date_re.match(value)
+    if match is None:
+        raise ValueError("invalid date format")
+
+    kw = {k: int(v) for k, v in match.groupdict().items()}
+
+    try:
+        return date(**kw)
+    except ValueError:
+        raise ValueError("invalid date format") from None

scale_gp_beta/_utils/_transform.py

@@ -16,18 +16,20 @@ from ._utils import (
     lru_cache,
     is_mapping,
     is_iterable,
+    is_sequence,
 )
 from .._files import is_base64_file_input
+from ._compat import get_origin, is_typeddict
 from ._typing import (
     is_list_type,
     is_union_type,
     extract_type_arg,
     is_iterable_type,
     is_required_type,
+    is_sequence_type,
     is_annotated_type,
     strip_annotated_type,
 )
-from .._compat import get_origin, model_dump, is_typeddict
 
 _T = TypeVar("_T")
 
@@ -167,6 +169,8 @@ def _transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -184,6 +188,8 @@ def _transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.
@@ -329,6 +335,8 @@ async def _async_transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -346,6 +354,8 @@ async def _async_transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.

scale_gp_beta/_utils/_typing.py

@@ -15,7 +15,7 @@ from typing_extensions import (
 
 from ._utils import lru_cache
 from .._types import InheritsGeneric
-from .._compat import is_union as _is_union
+from ._compat import is_union as _is_union
 
 
 def is_annotated_type(typ: type) -> bool:
@@ -26,6 +26,11 @@ def is_list_type(typ: type) -> bool:
     return (get_origin(typ) or typ) == list
 
 
+def is_sequence_type(typ: type) -> bool:
+    origin = get_origin(typ) or typ
+    return origin == typing_extensions.Sequence or origin == typing.Sequence or origin == _c_abc.Sequence
+
+
 def is_iterable_type(typ: type) -> bool:
     """If the given type is `typing.Iterable[T]`"""
     origin = get_origin(typ) or typ

scale_gp_beta/_utils/_utils.py

@@ -22,7 +22,6 @@ from typing_extensions import TypeGuard
 import sniffio
 
 from .._types import NotGiven, FileTypes, NotGivenOr, HeadersLike
-from .._compat import parse_date as parse_date, parse_datetime as parse_datetime
 
 _T = TypeVar("_T")
 _TupleT = TypeVar("_TupleT", bound=Tuple[object, ...])

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

scale_gp_beta/resources/__init__.py

@@ -146,18 +146,18 @@ __all__ = [
     "AsyncDatasetsResourceWithRawResponse",
     "DatasetsResourceWithStreamingResponse",
     "AsyncDatasetsResourceWithStreamingResponse",
-    "EvaluationsResource",
-    "AsyncEvaluationsResource",
-    "EvaluationsResourceWithRawResponse",
-    "AsyncEvaluationsResourceWithRawResponse",
-    "EvaluationsResourceWithStreamingResponse",
-    "AsyncEvaluationsResourceWithStreamingResponse",
     "DatasetItemsResource",
     "AsyncDatasetItemsResource",
     "DatasetItemsResourceWithRawResponse",
     "AsyncDatasetItemsResourceWithRawResponse",
     "DatasetItemsResourceWithStreamingResponse",
     "AsyncDatasetItemsResourceWithStreamingResponse",
+    "EvaluationsResource",
+    "AsyncEvaluationsResource",
+    "EvaluationsResourceWithRawResponse",
+    "AsyncEvaluationsResourceWithRawResponse",
+    "EvaluationsResourceWithStreamingResponse",
+    "AsyncEvaluationsResourceWithStreamingResponse",
     "EvaluationItemsResource",
     "AsyncEvaluationItemsResource",
     "EvaluationItemsResourceWithRawResponse",

scale_gp_beta/resources/chat/completions.py

@@ -2,12 +2,12 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, List, Union, Iterable, cast
+from typing import Any, Dict, Union, Iterable, cast
 from typing_extensions import Literal, overload
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from ..._utils import required_args, maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -62,7 +62,7 @@ class CompletionsResource(SyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -70,7 +70,7 @@ class CompletionsResource(SyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream: Literal[False] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -191,7 +191,7 @@ class CompletionsResource(SyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -199,7 +199,7 @@ class CompletionsResource(SyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -319,7 +319,7 @@ class CompletionsResource(SyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -327,7 +327,7 @@ class CompletionsResource(SyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -446,7 +446,7 @@ class CompletionsResource(SyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -454,7 +454,7 @@ class CompletionsResource(SyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream: Literal[False] | Literal[True] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -619,7 +619,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -627,7 +627,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream: Literal[False] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -748,7 +748,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -756,7 +756,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -876,7 +876,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -884,7 +884,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -1003,7 +1003,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         max_completion_tokens: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
         metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        modalities: List[str] | NotGiven = NOT_GIVEN,
+        modalities: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         n: int | NotGiven = NOT_GIVEN,
         parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
         prediction: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -1011,7 +1011,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         reasoning_effort: str | NotGiven = NOT_GIVEN,
         response_format: Dict[str, object] | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
         stream: Literal[False] | Literal[True] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,

scale_gp_beta/resources/completions.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union
+from typing import Dict, Union
 from typing_extensions import Literal, overload
 
 import httpx
 
 from ..types import completion_create_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -50,7 +50,7 @@ class CompletionsResource(SyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
@@ -60,7 +60,7 @@ class CompletionsResource(SyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream: Literal[False] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
@@ -139,7 +139,7 @@ class CompletionsResource(SyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         stream: Literal[True],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
@@ -150,7 +150,7 @@ class CompletionsResource(SyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -228,7 +228,7 @@ class CompletionsResource(SyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         stream: bool,
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
@@ -239,7 +239,7 @@ class CompletionsResource(SyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -317,7 +317,7 @@ class CompletionsResource(SyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
@@ -327,7 +327,7 @@ class CompletionsResource(SyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream: Literal[False] | Literal[True] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
@@ -402,7 +402,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
@@ -412,7 +412,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream: Literal[False] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
@@ -491,7 +491,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         stream: Literal[True],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
@@ -502,7 +502,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -580,7 +580,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         stream: bool,
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
@@ -591,7 +591,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
@@ -669,7 +669,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         self,
         *,
         model: str,
-        prompt: Union[str, List[str]],
+        prompt: Union[str, SequenceNotStr[str]],
         best_of: int | NotGiven = NOT_GIVEN,
         echo: bool | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
@@ -679,7 +679,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         n: int | NotGiven = NOT_GIVEN,
         presence_penalty: float | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
-        stop: Union[str, List[str]] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         stream: Literal[False] | Literal[True] | NotGiven = NOT_GIVEN,
         stream_options: Dict[str, object] | NotGiven = NOT_GIVEN,
         suffix: str | NotGiven = NOT_GIVEN,

scale_gp_beta/resources/datasets.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable
+from typing import Dict, Iterable
 from typing_extensions import Literal
 
 import httpx
 
 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 .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -52,7 +52,7 @@ class DatasetsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
-        tags: List[str] | NotGiven = NOT_GIVEN,
+        tags: SequenceNotStr[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,
@@ -139,7 +139,7 @@ class DatasetsResource(SyncAPIResource):
         *,
         description: str | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
-        tags: List[str] | NotGiven = NOT_GIVEN,
+        tags: SequenceNotStr[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,
@@ -188,7 +188,7 @@ class DatasetsResource(SyncAPIResource):
         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,
+        tags: SequenceNotStr[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,
@@ -292,7 +292,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
-        tags: List[str] | NotGiven = NOT_GIVEN,
+        tags: SequenceNotStr[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,7 +379,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         *,
         description: str | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
-        tags: List[str] | NotGiven = NOT_GIVEN,
+        tags: SequenceNotStr[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,
@@ -428,7 +428,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         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,
+        tags: SequenceNotStr[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,