@@ -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 (
0 commit comments