feat(types): replace List[str] with SequenceNotStr in params

Author: stainless-app[bot]

src/maisa/_utils/_transform.py

@@ -16,6 +16,7 @@
     lru_cache,
     is_mapping,
     is_iterable,
+    is_sequence,
 )
 from .._files import is_base64_file_input
 from ._typing import (
@@ -24,6 +25,7 @@
     extract_type_arg,
     is_iterable_type,
     is_required_type,
+    is_sequence_type,
     is_annotated_type,
     strip_annotated_type,
 )
@@ -184,6 +186,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.
@@ -346,6 +350,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.

src/maisa/resources/kpu.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List, Mapping, Optional, cast
+from typing import Mapping, Optional, cast
 from typing_extensions import Literal
 
 import httpx
 
 from ..types import kpu_run_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes, SequenceNotStr
 from .._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -49,7 +49,7 @@ def run(
         query: str,
         explain_steps: bool | NotGiven = NOT_GIVEN,
         retries: int | NotGiven = NOT_GIVEN,
-        file: List[FileTypes] | NotGiven = NOT_GIVEN,
+        file: SequenceNotStr[FileTypes] | NotGiven = NOT_GIVEN,
         reasoner_model: Optional[
             Literal[
                 "gpt-4-turbo",
@@ -156,7 +156,7 @@ async def run(
         query: str,
         explain_steps: bool | NotGiven = NOT_GIVEN,
         retries: int | NotGiven = NOT_GIVEN,
-        file: List[FileTypes] | NotGiven = NOT_GIVEN,
+        file: SequenceNotStr[FileTypes] | NotGiven = NOT_GIVEN,
         reasoner_model: Optional[
             Literal[
                 "gpt-4-turbo",

src/maisa/resources/models/embeddings.py

@@ -2,11 +2,9 @@
 
 from __future__ import annotations
 
-from typing import List
-
 import httpx
 
-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
@@ -46,7 +44,7 @@ def with_streaming_response(self) -> EmbeddingsResourceWithStreamingResponse:
     def create(
         self,
         *,
-        texts: List[str],
+        texts: SequenceNotStr[str],
         # 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,
@@ -101,7 +99,7 @@ def with_streaming_response(self) -> AsyncEmbeddingsResourceWithStreamingRespons
     async def create(
         self,
         *,
-        texts: List[str],
+        texts: SequenceNotStr[str],
         # 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,

src/maisa/types/kpu_run_params.py

@@ -2,10 +2,10 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Literal, Required, TypedDict
 
-from .._types import FileTypes
+from .._types import FileTypes, SequenceNotStr
 
 __all__ = ["KpuRunParams"]
 
@@ -28,7 +28,7 @@ class KpuRunParams(TypedDict, total=False):
     intent. This feature is experimental.
     """
 
-    file: List[FileTypes]
+    file: SequenceNotStr[FileTypes]
     """Files to be used in the KPU execution. Files can be of any type."""
 
     reasoner_model: Optional[

src/maisa/types/models/embedding_create_params.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["EmbeddingCreateParams"]
 
 
 class EmbeddingCreateParams(TypedDict, total=False):
-    texts: Required[List[str]]
+    texts: Required[SequenceNotStr[str]]
     """A list of texts from which we will generate the embeddings."""