chore: update OpenAPI spec and rebuild SDKs

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 36
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-c0e1943df75f6bdf6b856cf51b703e203606877b0ae96879632039bfc071fe6a.yml
-openapi_spec_hash: 4fcc989f63cf3b4c36a0b0573002956c
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-1029a7d9615d8a4fa72ca7bb2c0762d98cef9919c6f35079f87cc2db2b8ce9f5.yml
+openapi_spec_hash: f8b9ef3c211e9113e141f1894436eba6
 config_hash: 407acf62c906ee301314f2d23cdb58b1

src/replicate/resources/deployments/predictions.py

@@ -54,6 +54,7 @@ def create(
         webhook: str | Omit = omit,
         webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
         prefer: str | Omit = omit,
+        replicate_max_lifetime: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -162,7 +163,15 @@ def create(
             raise ValueError(f"Expected a non-empty value for `deployment_owner` but received {deployment_owner!r}")
         if not deployment_name:
             raise ValueError(f"Expected a non-empty value for `deployment_name` but received {deployment_name!r}")
-        extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Prefer": prefer,
+                    "Replicate-Max-Lifetime": replicate_max_lifetime,
+                }
+            ),
+            **(extra_headers or {}),
+        }
         return self._post(
             f"/deployments/{deployment_owner}/{deployment_name}/predictions",
             body=maybe_transform(
@@ -211,6 +220,7 @@ async def create(
         webhook: str | Omit = omit,
         webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
         prefer: str | Omit = omit,
+        replicate_max_lifetime: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -319,7 +329,15 @@ async def create(
             raise ValueError(f"Expected a non-empty value for `deployment_owner` but received {deployment_owner!r}")
         if not deployment_name:
             raise ValueError(f"Expected a non-empty value for `deployment_name` but received {deployment_name!r}")
-        extra_headers = {**strip_not_given({"Prefer": prefer}), **(extra_headers or {})}
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Prefer": prefer,
+                    "Replicate-Max-Lifetime": replicate_max_lifetime,
+                }
+            ),
+            **(extra_headers or {}),
+        }
         return await self._post(
             f"/deployments/{deployment_owner}/{deployment_name}/predictions",
             body=await async_maybe_transform(

src/replicate/types/deployments/prediction_create_params.py

@@ -94,3 +94,5 @@ class PredictionCreateParams(TypedDict, total=False):
     """
 
     prefer: Annotated[str, PropertyInfo(alias="Prefer")]
+
+    replicate_max_lifetime: Annotated[str, PropertyInfo(alias="Replicate-Max-Lifetime")]

src/replicate/types/models/prediction_create_params.py

@@ -94,3 +94,5 @@ class PredictionCreateParams(TypedDict, total=False):
     """
 
     prefer: Annotated[str, PropertyInfo(alias="Prefer")]
+
+    replicate_max_lifetime: Annotated[str, PropertyInfo(alias="Replicate-Max-Lifetime")]

src/replicate/types/prediction.py

@@ -77,6 +77,12 @@ class Prediction(BaseModel):
     completed_at: Optional[datetime] = None
     """The time that the model completed the prediction and all outputs were uploaded"""
 
+    deadline: Optional[datetime] = None
+    """
+    The absolute time at which the prediction will be automatically canceled if it
+    has not completed
+    """
+
     deployment: Optional[str] = None
     """The name of the deployment that created the prediction"""
 

src/replicate/types/prediction_create_params.py

@@ -92,6 +92,8 @@ class PredictionCreateParamsWithoutVersion(TypedDict, total=False):
 
     prefer: Annotated[str, PropertyInfo(alias="Prefer")]
 
+    replicate_max_lifetime: Annotated[str, PropertyInfo(alias="Replicate-Max-Lifetime")]
+
 
 class PredictionCreateParams(PredictionCreateParamsWithoutVersion):
     version: Required[str]

tests/api_resources/deployments/test_predictions.py

@@ -44,6 +44,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])
 
@@ -139,6 +140,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])
 

tests/api_resources/models/test_predictions.py

@@ -44,6 +44,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])
 
@@ -139,6 +140,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])
 

tests/api_resources/test_predictions.py

@@ -38,6 +38,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])
 
@@ -215,6 +216,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
+            replicate_max_lifetime="5m",
         )
         assert_matches_type(Prediction, prediction, path=["response"])