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-a7e12a7c7df8c6596c7ca7d1ee26992bf3dcca90b553a66e15118fcd892e27f6.yml
-openapi_spec_hash: 33ca30d22504063d2248459550f9cb3b
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-a6a1d53117b661ccba3fa9893e4598b8b21448cda3edbc907c4b8085d352203a.yml
+openapi_spec_hash: c1d5f3fe94e8f93f3d4247a5df7e3e7b
 config_hash: 407acf62c906ee301314f2d23cdb58b1

src/replicate/resources/predictions.py

@@ -448,7 +448,9 @@ def get(
         Terminated predictions (with a status of `succeeded`, `failed`, or `canceled`)
         will include a `metrics` object with a `predict_time` property showing the
         amount of CPU or GPU time, in seconds, that the prediction used while running.
-        It won't include time waiting for the prediction to start.
+        It won't include time waiting for the prediction to start. The `metrics` object
+        will also include a `total_time` property showing the total time, in seconds,
+        that the prediction took to complete.
 
         All input parameters, output values, and logs are automatically removed after an
         hour, by default, for predictions created through the API.
@@ -901,7 +903,9 @@ async def get(
         Terminated predictions (with a status of `succeeded`, `failed`, or `canceled`)
         will include a `metrics` object with a `predict_time` property showing the
         amount of CPU or GPU time, in seconds, that the prediction used while running.
-        It won't include time waiting for the prediction to start.
+        It won't include time waiting for the prediction to start. The `metrics` object
+        will also include a `total_time` property showing the total time, in seconds,
+        that the prediction took to complete.
 
         All input parameters, output values, and logs are automatically removed after an
         hour, by default, for predictions created through the API.

src/replicate/resources/trainings.py

@@ -390,7 +390,9 @@ def get(
         Terminated trainings (with a status of `succeeded`, `failed`, or `canceled`)
         will include a `metrics` object with a `predict_time` property showing the
         amount of CPU or GPU time, in seconds, that the training used while running. It
-        won't include time waiting for the training to start.
+        won't include time waiting for the training to start. The `metrics` object will
+        also include a `total_time` property showing the total time, in seconds, that
+        the training took to complete.
 
         Args:
           extra_headers: Send extra headers
@@ -774,7 +776,9 @@ async def get(
         Terminated trainings (with a status of `succeeded`, `failed`, or `canceled`)
         will include a `metrics` object with a `predict_time` property showing the
         amount of CPU or GPU time, in seconds, that the training used while running. It
-        won't include time waiting for the training to start.
+        won't include time waiting for the training to start. The `metrics` object will
+        also include a `total_time` property showing the total time, in seconds, that
+        the training took to complete.
 
         Args:
           extra_headers: Send extra headers

src/replicate/types/prediction.py

@@ -1,12 +1,14 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Dict, Union, Optional
+from typing import TYPE_CHECKING, Dict, Union, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
+from pydantic import Field as FieldInfo
+
 from .._models import BaseModel
 
-__all__ = ["Prediction", "URLs"]
+__all__ = ["Prediction", "URLs", "Metrics"]
 
 
 class URLs(BaseModel):
@@ -23,6 +25,23 @@ class URLs(BaseModel):
     """An event source to stream the output of the prediction via API"""
 
 
+class Metrics(BaseModel):
+    total_time: Optional[float] = None
+    """The total time, in seconds, that the prediction took to complete"""
+
+    if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
+        # Stub to indicate that arbitrary properties are accepted.
+        # To access properties that are not valid identifiers you can use `getattr`, e.g.
+        # `getattr(obj, '$type')`
+        def __getattr__(self, attr: str) -> object: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
+
+
 class Prediction(BaseModel):
     id: str
 
@@ -64,7 +83,7 @@ class Prediction(BaseModel):
     logs: Optional[str] = None
     """The log output from the model"""
 
-    metrics: Optional[Dict[str, object]] = None
+    metrics: Optional[Metrics] = None
     """Additional metrics associated with the prediction"""
 
     started_at: Optional[datetime] = None

src/replicate/types/training_cancel_response.py

@@ -13,6 +13,9 @@ class Metrics(BaseModel):
     predict_time: Optional[float] = None
     """The amount of CPU or GPU time, in seconds, that the training used while running"""
 
+    total_time: Optional[float] = None
+    """The total time, in seconds, that the training took to complete"""
+
 
 class Output(BaseModel):
     version: Optional[str] = None

src/replicate/types/training_create_response.py

@@ -13,6 +13,9 @@ class Metrics(BaseModel):
     predict_time: Optional[float] = None
     """The amount of CPU or GPU time, in seconds, that the training used while running"""
 
+    total_time: Optional[float] = None
+    """The total time, in seconds, that the training took to complete"""
+
 
 class Output(BaseModel):
     version: Optional[str] = None

src/replicate/types/training_get_response.py

@@ -13,6 +13,9 @@ class Metrics(BaseModel):
     predict_time: Optional[float] = None
     """The amount of CPU or GPU time, in seconds, that the training used while running"""
 
+    total_time: Optional[float] = None
+    """The total time, in seconds, that the training took to complete"""
+
 
 class Output(BaseModel):
     version: Optional[str] = None

src/replicate/types/training_list_response.py

@@ -13,6 +13,9 @@ class Metrics(BaseModel):
     predict_time: Optional[float] = None
     """The amount of CPU or GPU time, in seconds, that the training used while running"""
 
+    total_time: Optional[float] = None
+    """The total time, in seconds, that the training took to complete"""
+
 
 class Output(BaseModel):
     version: Optional[str] = None