To support tuple of string for Embeddings and AsyncEmbeddings openai#…

…1934
deepthinkerdev · Dec 26, 2024 · 2fd0e20 · 2fd0e20
1 parent 39ccc3d
commit 2fd0e20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/openai/resources/embeddings.py b/src/openai/resources/embeddings.py
@@ -46,7 +46,7 @@ def with_streaming_response(self) -> EmbeddingsWithStreamingResponse:
     def create(
         self,
         *,
-        input: Union[str, List[str], Tuple[str], Iterable[int], Iterable[Iterable[int]]],
+        input: Union[str, List[str], Tuple[str, ...], Iterable[int], Iterable[Iterable[int]]],
         model: Union[str, EmbeddingModel],
         dimensions: int | NotGiven = NOT_GIVEN,
         encoding_format: Literal["float", "base64"] | NotGiven = NOT_GIVEN,
@@ -162,7 +162,7 @@ def with_streaming_response(self) -> AsyncEmbeddingsWithStreamingResponse:
     async def create(
         self,
         *,
-        input: Union[str, List[str], Tuple[str], Iterable[int], Iterable[Iterable[int]]],
+        input: Union[str, List[str], Tuple[str, ...], Iterable[int], Iterable[Iterable[int]]],
         model: Union[str, EmbeddingModel],
         dimensions: int | NotGiven = NOT_GIVEN,
         encoding_format: Literal["float", "base64"] | NotGiven = NOT_GIVEN,