|
1 | 1 | import json |
2 | 2 | from typing import Any, Dict, Iterable, List, Optional, Union |
3 | 3 | import typing |
| 4 | +from portkey_ai._vendor.openai.types.vector_stores import file_batch_create_params |
4 | 5 | from portkey_ai.api_resources.apis.api_resource import APIResource, AsyncAPIResource |
5 | 6 | from portkey_ai.api_resources.client import AsyncPortkey, Portkey |
6 | 7 | from portkey_ai.api_resources.types.shared_types import Metadata |
@@ -311,7 +312,7 @@ def create( |
311 | 312 | vector_store_id: str, |
312 | 313 | *, |
313 | 314 | file_ids: Union[List[str], Omit] = omit, |
314 | | - files: Union[Iterable[FileTypes], Omit] = omit, |
| 315 | + files: Union[Iterable[file_batch_create_params.File], Omit] = omit, |
315 | 316 | chunking_strategy: Union[Any, Omit] = omit, |
316 | 317 | **kwargs, |
317 | 318 | ) -> VectorStoreFileBatch: |
@@ -743,7 +744,7 @@ async def create( |
743 | 744 | vector_store_id: str, |
744 | 745 | *, |
745 | 746 | file_ids: Union[List[str], Omit] = omit, |
746 | | - files: Union[Iterable[FileTypes], Omit] = omit, |
| 747 | + files: Union[Iterable[file_batch_create_params.File], Omit] = omit, |
747 | 748 | chunking_strategy: Union[Any, Omit] = omit, |
748 | 749 | **kwargs, |
749 | 750 | ) -> VectorStoreFileBatch: |
|
0 commit comments