Skip to content

Commit 77c9c21

Browse files
chore: update OpenAPI spec and rebuild SDKs
1 parent d5ed889 commit 77c9c21

File tree

2 files changed

+40
-4
lines changed

2 files changed

+40
-4
lines changed

src/replicate/resources/models/predictions.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def create(
5656
webhook: str | Omit = omit,
5757
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
5858
prefer: str | Omit = omit,
59+
replicate_max_lifetime: str | Omit = omit,
5960
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
6061
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
6162
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -169,7 +170,15 @@ def create(
169170
raise ValueError(f"Expected a non-empty value for `model_owner` but received {model_owner!r}")
170171
if not model_name:
171172
raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
172-
extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
173+
extra_headers = {
174+
**strip_not_given(
175+
{
176+
"Prefer": prefer,
177+
"Replicate-Max-Lifetime": replicate_max_lifetime,
178+
}
179+
),
180+
**(extra_headers or {}),
181+
}
173182
return self._post(
174183
f"/models/{model_owner}/{model_name}/predictions",
175184
body=maybe_transform(
@@ -218,6 +227,7 @@ async def create(
218227
webhook: str | Omit = omit,
219228
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
220229
prefer: str | Omit = omit,
230+
replicate_max_lifetime: str | Omit = omit,
221231
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
222232
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
223233
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -331,7 +341,15 @@ async def create(
331341
raise ValueError(f"Expected a non-empty value for `model_owner` but received {model_owner!r}")
332342
if not model_name:
333343
raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
334-
extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
344+
extra_headers = {
345+
**strip_not_given(
346+
{
347+
"Prefer": prefer,
348+
"Replicate-Max-Lifetime": replicate_max_lifetime,
349+
}
350+
),
351+
**(extra_headers or {}),
352+
}
335353
return await self._post(
336354
f"/models/{model_owner}/{model_name}/predictions",
337355
body=await async_maybe_transform(

src/replicate/resources/predictions.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def create(
6767
webhook: str | Omit = omit,
6868
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
6969
prefer: str | Omit = omit,
70+
replicate_max_lifetime: str | Omit = omit,
7071
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
7172
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
7273
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -186,7 +187,15 @@ def create(
186187
187188
timeout: Override the client-level default timeout for this request, in seconds
188189
"""
189-
extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
190+
extra_headers = {
191+
**strip_not_given(
192+
{
193+
"Prefer": prefer,
194+
"Replicate-Max-Lifetime": replicate_max_lifetime,
195+
}
196+
),
197+
**(extra_headers or {}),
198+
}
190199
return self._post(
191200
"/predictions",
192201
body=maybe_transform(
@@ -520,6 +529,7 @@ async def create(
520529
webhook: str | Omit = omit,
521530
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
522531
prefer: str | Omit = omit,
532+
replicate_max_lifetime: str | Omit = omit,
523533
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
524534
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
525535
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -639,7 +649,15 @@ async def create(
639649
640650
timeout: Override the client-level default timeout for this request, in seconds
641651
"""
642-
extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
652+
extra_headers = {
653+
**strip_not_given(
654+
{
655+
"Prefer": prefer,
656+
"Replicate-Max-Lifetime": replicate_max_lifetime,
657+
}
658+
),
659+
**(extra_headers or {}),
660+
}
643661
return await self._post(
644662
"/predictions",
645663
body=await async_maybe_transform(

0 commit comments

Comments
 (0)