Skip to content

Commit c070514

Browse files
narengoginarengogi
committed
fix: add missing parameters from OpenAI SDK v2.16.0
- Add prompt_cache_retention to responses create/stream/parse methods - Add known_speaker_names and known_speaker_references to audio transcriptions - Add description parameter to vector_stores.create - Make file_ids optional in vector_stores.file_batches.create - Update videos model type to VideoModelParam
1 parent fe9c430 commit c070514

4 files changed

Lines changed: 65 additions & 7 deletions

File tree

‎portkey_ai/api_resources/apis/audio.py‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ def create(
4141
temperature: Union[float, Omit] = omit,
4242
timestamp_granularities: Union[List[str], Omit] = omit,
4343
stream: Union[bool, Omit] = omit,
44+
known_speaker_names: Union[List[str], Omit] = omit,
45+
known_speaker_references: Union[List[str], Omit] = omit,
4446
**kwargs,
4547
) -> Union[Transcription, TranscriptionVerbose, str]:
4648
extra_headers = kwargs.pop("extra_headers", {})
@@ -58,6 +60,8 @@ def create(
5860
temperature=temperature,
5961
timestamp_granularities=timestamp_granularities,
6062
stream=stream,
63+
known_speaker_names=known_speaker_names,
64+
known_speaker_references=known_speaker_references,
6165
extra_headers=extra_headers,
6266
extra_body=kwargs,
6367
)
@@ -70,6 +74,8 @@ def create(
7074
response_format=response_format,
7175
temperature=temperature,
7276
timestamp_granularities=timestamp_granularities,
77+
known_speaker_names=known_speaker_names,
78+
known_speaker_references=known_speaker_references,
7379
extra_headers=extra_headers,
7480
extra_body=kwargs,
7581
)
@@ -179,6 +185,8 @@ async def create(
179185
temperature: Union[float, Omit] = omit,
180186
timestamp_granularities: Union[List[str], Omit] = omit,
181187
stream: Union[bool, Omit] = omit,
188+
known_speaker_names: Union[List[str], Omit] = omit,
189+
known_speaker_references: Union[List[str], Omit] = omit,
182190
**kwargs,
183191
) -> Union[Transcription, TranscriptionVerbose, str]:
184192
extra_headers = kwargs.pop("extra_headers", {})
@@ -196,6 +204,8 @@ async def create(
196204
temperature=temperature,
197205
timestamp_granularities=timestamp_granularities,
198206
stream=stream,
207+
known_speaker_names=known_speaker_names,
208+
known_speaker_references=known_speaker_references,
199209
extra_headers=extra_headers,
200210
extra_body=kwargs,
201211
)
@@ -209,6 +219,8 @@ async def create(
209219
response_format=response_format,
210220
temperature=temperature,
211221
timestamp_granularities=timestamp_granularities,
222+
known_speaker_names=known_speaker_names,
223+
known_speaker_references=known_speaker_references,
212224
extra_headers=extra_headers,
213225
extra_body=kwargs,
214226
)

‎portkey_ai/api_resources/apis/responses.py‎

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,9 @@ def create(
6868
metadata: Union[Optional[Metadata], Omit] = omit,
6969
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
7070
previous_response_id: Union[Optional[str], Omit] = omit,
71+
prompt_cache_retention: Union[
72+
Optional[Literal["in-memory", "24h"]], Omit
73+
] = omit,
7174
reasoning: Union[Optional[Reasoning], Omit] = omit,
7275
store: Union[Optional[bool], Omit] = omit,
7376
stream: Union[Literal[False], Omit] = omit,
@@ -95,6 +98,9 @@ def create(
9598
metadata: Union[Optional[Metadata], Omit] = omit,
9699
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
97100
previous_response_id: Union[Optional[str], Omit] = omit,
101+
prompt_cache_retention: Union[
102+
Optional[Literal["in-memory", "24h"]], Omit
103+
] = omit,
98104
reasoning: Union[Optional[Reasoning], Omit] = omit,
99105
store: Union[Optional[bool], Omit] = omit,
100106
temperature: Union[Optional[float], Omit] = omit,
@@ -121,6 +127,9 @@ def create(
121127
metadata: Union[Optional[Metadata], Omit] = omit,
122128
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
123129
previous_response_id: Union[Optional[str], Omit] = omit,
130+
prompt_cache_retention: Union[
131+
Optional[Literal["in-memory", "24h"]], Omit
132+
] = omit,
124133
reasoning: Union[Optional[Reasoning], Omit] = omit,
125134
store: Union[Optional[bool], Omit] = omit,
126135
temperature: Union[Optional[float], Omit] = omit,
@@ -145,6 +154,9 @@ def create(
145154
metadata: Union[Optional[Metadata], Omit] = omit,
146155
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
147156
previous_response_id: Union[Optional[str], Omit] = omit,
157+
prompt_cache_retention: Union[
158+
Optional[Literal["in-memory", "24h"]], Omit
159+
] = omit,
148160
reasoning: Union[Optional[Reasoning], Omit] = omit,
149161
store: Union[Optional[bool], Omit] = omit,
150162
stream: Union[Optional[Literal[False]], Literal[True], Omit] = omit,
@@ -171,6 +183,7 @@ def create(
171183
metadata=metadata,
172184
parallel_tool_calls=parallel_tool_calls,
173185
previous_response_id=previous_response_id,
186+
prompt_cache_retention=prompt_cache_retention,
174187
reasoning=reasoning,
175188
store=store,
176189
stream=stream, # type: ignore[arg-type]
@@ -244,6 +257,9 @@ def stream(
244257
metadata: Union[Metadata, Omit] = omit,
245258
parallel_tool_calls: Union[bool, Omit] = omit,
246259
previous_response_id: Union[str, Omit] = omit,
260+
prompt_cache_retention: Union[
261+
Optional[Literal["in-memory", "24h"]], Omit
262+
] = omit,
247263
reasoning: Union[Reasoning, Omit] = omit,
248264
store: Union[bool, Omit] = omit,
249265
stream_options: Union[response_create_params.StreamOptions, Omit] = omit,
@@ -270,6 +286,7 @@ def stream(
270286
metadata=metadata,
271287
parallel_tool_calls=parallel_tool_calls,
272288
previous_response_id=previous_response_id,
289+
prompt_cache_retention=prompt_cache_retention,
273290
reasoning=reasoning,
274291
store=store,
275292
stream_options=stream_options,
@@ -298,6 +315,9 @@ def parse(
298315
metadata: Union[Metadata, Omit] = omit,
299316
parallel_tool_calls: Union[bool, Omit] = omit,
300317
previous_response_id: Union[str, Omit] = omit,
318+
prompt_cache_retention: Union[
319+
Optional[Literal["in-memory", "24h"]], Omit
320+
] = omit,
301321
reasoning: Union[Reasoning, Omit] = omit,
302322
store: Union[bool, Omit] = omit,
303323
stream: Union[Literal[False], Literal[True], Omit] = omit,
@@ -325,6 +345,7 @@ def parse(
325345
metadata=metadata,
326346
parallel_tool_calls=parallel_tool_calls,
327347
previous_response_id=previous_response_id,
348+
prompt_cache_retention=prompt_cache_retention,
328349
reasoning=reasoning,
329350
store=store,
330351
stream=stream,
@@ -490,6 +511,9 @@ async def create(
490511
metadata: Union[Optional[Metadata], Omit] = omit,
491512
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
492513
previous_response_id: Union[Optional[str], Omit] = omit,
514+
prompt_cache_retention: Union[
515+
Optional[Literal["in-memory", "24h"]], Omit
516+
] = omit,
493517
reasoning: Union[Optional[Reasoning], Omit] = omit,
494518
store: Union[Optional[bool], Omit] = omit,
495519
stream: Union[Literal[False], Omit] = omit,
@@ -517,6 +541,9 @@ async def create(
517541
metadata: Union[Optional[Metadata], Omit] = omit,
518542
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
519543
previous_response_id: Union[Optional[str], Omit] = omit,
544+
prompt_cache_retention: Union[
545+
Optional[Literal["in-memory", "24h"]], Omit
546+
] = omit,
520547
reasoning: Union[Optional[Reasoning], Omit] = omit,
521548
store: Union[Optional[bool], Omit] = omit,
522549
temperature: Union[Optional[float], Omit] = omit,
@@ -543,6 +570,9 @@ async def create(
543570
metadata: Union[Optional[Metadata], Omit] = omit,
544571
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
545572
previous_response_id: Union[Optional[str], Omit] = omit,
573+
prompt_cache_retention: Union[
574+
Optional[Literal["in-memory", "24h"]], Omit
575+
] = omit,
546576
reasoning: Union[Optional[Reasoning], Omit] = omit,
547577
store: Union[Optional[bool], Omit] = omit,
548578
temperature: Union[Optional[float], Omit] = omit,
@@ -567,6 +597,9 @@ async def create(
567597
metadata: Union[Optional[Metadata], Omit] = omit,
568598
parallel_tool_calls: Union[Optional[bool], Omit] = omit,
569599
previous_response_id: Union[Optional[str], Omit] = omit,
600+
prompt_cache_retention: Union[
601+
Optional[Literal["in-memory", "24h"]], Omit
602+
] = omit,
570603
reasoning: Union[Optional[Reasoning], Omit] = omit,
571604
store: Union[Optional[bool], Omit] = omit,
572605
stream: Union[Optional[Literal[False]], Literal[True], Omit] = omit,
@@ -593,6 +626,7 @@ async def create(
593626
metadata=metadata,
594627
parallel_tool_calls=parallel_tool_calls,
595628
previous_response_id=previous_response_id,
629+
prompt_cache_retention=prompt_cache_retention,
596630
reasoning=reasoning,
597631
store=store,
598632
stream=stream, # type: ignore[arg-type]
@@ -666,6 +700,9 @@ def stream(
666700
metadata: Union[Metadata, Omit] = omit,
667701
parallel_tool_calls: Union[bool, Omit] = omit,
668702
previous_response_id: Union[str, Omit] = omit,
703+
prompt_cache_retention: Union[
704+
Optional[Literal["in-memory", "24h"]], Omit
705+
] = omit,
669706
reasoning: Union[Reasoning, Omit] = omit,
670707
store: Union[bool, Omit] = omit,
671708
stream_options: Union[response_create_params.StreamOptions, Omit] = omit,
@@ -692,6 +729,7 @@ def stream(
692729
metadata=metadata,
693730
parallel_tool_calls=parallel_tool_calls,
694731
previous_response_id=previous_response_id,
732+
prompt_cache_retention=prompt_cache_retention,
695733
reasoning=reasoning,
696734
store=store,
697735
stream_options=stream_options,
@@ -720,6 +758,9 @@ async def parse(
720758
metadata: Union[Metadata, Omit] = omit,
721759
parallel_tool_calls: Union[bool, Omit] = omit,
722760
previous_response_id: Union[str, Omit] = omit,
761+
prompt_cache_retention: Union[
762+
Optional[Literal["in-memory", "24h"]], Omit
763+
] = omit,
723764
reasoning: Union[Reasoning, Omit] = omit,
724765
store: Union[bool, Omit] = omit,
725766
stream: Union[Literal[False], Literal[True], Omit] = omit,
@@ -747,6 +788,7 @@ async def parse(
747788
metadata=metadata,
748789
parallel_tool_calls=parallel_tool_calls,
749790
previous_response_id=previous_response_id,
791+
prompt_cache_retention=prompt_cache_retention,
750792
reasoning=reasoning,
751793
store=store,
752794
stream=stream,

‎portkey_ai/api_resources/apis/vector_stores.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,15 @@ def create(
3232
self,
3333
*,
3434
chunking_strategy: Union[Any, Omit] = omit,
35+
description: Union[str, Omit] = omit,
3536
expires_after: Union[vector_store_create_params.ExpiresAfter, Omit] = omit,
3637
file_ids: Union[List[str], Omit] = omit,
3738
metadata: Union[Optional[Metadata], Omit] = omit,
3839
name: Union[str, Omit] = omit,
3940
) -> VectorStore:
4041
response = self.openai_client.with_raw_response.vector_stores.create(
4142
chunking_strategy=chunking_strategy,
43+
description=description,
4244
expires_after=expires_after,
4345
file_ids=file_ids,
4446
metadata=metadata,
@@ -308,7 +310,7 @@ def create(
308310
self,
309311
vector_store_id: str,
310312
*,
311-
file_ids: List[str],
313+
file_ids: Union[List[str], Omit] = omit,
312314
chunking_strategy: Union[Any, Omit] = omit,
313315
**kwargs,
314316
) -> VectorStoreFileBatch:
@@ -460,13 +462,15 @@ async def create(
460462
self,
461463
*,
462464
chunking_strategy: Union[Any, Omit] = omit,
465+
description: Union[str, Omit] = omit,
463466
expires_after: Union[vector_store_create_params.ExpiresAfter, Omit] = omit,
464467
file_ids: Union[List[str], Omit] = omit,
465468
metadata: Union[Optional[Metadata], Omit] = omit,
466469
name: Union[str, Omit] = omit,
467470
) -> VectorStore:
468471
response = await self.openai_client.with_raw_response.vector_stores.create(
469472
chunking_strategy=chunking_strategy,
473+
description=description,
470474
expires_after=expires_after,
471475
file_ids=file_ids,
472476
metadata=metadata,
@@ -736,7 +740,7 @@ async def create(
736740
self,
737741
vector_store_id: str,
738742
*,
739-
file_ids: List[str],
743+
file_ids: Union[List[str], Omit] = omit,
740744
chunking_strategy: Union[Any, Omit] = omit,
741745
**kwargs,
742746
) -> VectorStoreFileBatch:

‎portkey_ai/api_resources/apis/videos.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from typing import Any, Literal, Optional, Union
33

44
import httpx
5-
from portkey_ai._vendor.openai.types.video_model import VideoModel
5+
from portkey_ai._vendor.openai.types.video_model_param import VideoModelParam
66
from portkey_ai._vendor.openai.types.video_seconds import VideoSeconds
77
from portkey_ai._vendor.openai.types.video_size import VideoSize
88
from portkey_ai.api_resources.apis.api_resource import APIResource, AsyncAPIResource
@@ -26,7 +26,7 @@ def create(
2626
*,
2727
prompt: str,
2828
input_reference: Union[FileTypes, Omit] = omit,
29-
model: Union[VideoModel, Omit] = omit,
29+
model: Union[VideoModelParam, Omit] = omit,
3030
seconds: Union[VideoSeconds, Omit] = omit,
3131
size: Union[VideoSize, Omit] = omit,
3232
extra_headers: Optional[Headers] = None,
@@ -55,7 +55,7 @@ def create_and_poll(
5555
*,
5656
prompt: str,
5757
input_reference: Union[FileTypes, Omit] = omit,
58-
model: Union[VideoModel, Omit] = omit,
58+
model: Union[VideoModelParam, Omit] = omit,
5959
seconds: Union[VideoSeconds, Omit] = omit,
6060
size: Union[VideoSize, Omit] = omit,
6161
poll_interval_ms: Union[int, Omit] = omit,
@@ -208,7 +208,7 @@ async def create(
208208
*,
209209
prompt: str,
210210
input_reference: Union[FileTypes, Omit] = omit,
211-
model: Union[VideoModel, Omit] = omit,
211+
model: Union[VideoModelParam, Omit] = omit,
212212
seconds: Union[VideoSeconds, Omit] = omit,
213213
size: Union[VideoSize, Omit] = omit,
214214
extra_headers: Optional[Headers] = None,
@@ -237,7 +237,7 @@ async def create_and_poll(
237237
*,
238238
prompt: str,
239239
input_reference: Union[FileTypes, Omit] = omit,
240-
model: Union[VideoModel, Omit] = omit,
240+
model: Union[VideoModelParam, Omit] = omit,
241241
seconds: Union[VideoSeconds, Omit] = omit,
242242
size: Union[VideoSize, Omit] = omit,
243243
poll_interval_ms: Union[int, Omit] = omit,

0 commit comments

Comments
 (0)