feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 27
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-37bb31ed76da599d3bded543a3765f745c8575d105c13554df7f8361c3641482.yml
-openapi_spec_hash: 15bdec12ca84042768bfb28cc48dfce3
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-2788217b7ad7d61d1a77800bc5ff12a6810f1692d4d770b72fa8f898c6a055ab.yml
+openapi_spec_hash: 4423bf747e228484547b441468a9f156
 config_hash: 64fd304bf0ff077a74053ce79a255106

api.md

@@ -75,12 +75,18 @@ Methods:
 
 ## Versions
 
+Types:
+
+```python
+from replicate.types.models import VersionCreateTrainingResponse
+```
+
 Methods:
 
 - <code title="get /models/{model_owner}/{model_name}/versions/{version_id}">client.models.versions.<a href="./src/replicate/resources/models/versions.py">retrieve</a>(version_id, \*, model_owner, model_name) -> None</code>
 - <code title="get /models/{model_owner}/{model_name}/versions">client.models.versions.<a href="./src/replicate/resources/models/versions.py">list</a>(model_name, \*, model_owner) -> None</code>
 - <code title="delete /models/{model_owner}/{model_name}/versions/{version_id}">client.models.versions.<a href="./src/replicate/resources/models/versions.py">delete</a>(version_id, \*, model_owner, model_name) -> None</code>
-- <code title="post /models/{model_owner}/{model_name}/versions/{version_id}/trainings">client.models.versions.<a href="./src/replicate/resources/models/versions.py">create_training</a>(version_id, \*, model_owner, model_name, \*\*<a href="src/replicate/types/models/version_create_training_params.py">params</a>) -> None</code>
+- <code title="post /models/{model_owner}/{model_name}/versions/{version_id}/trainings">client.models.versions.<a href="./src/replicate/resources/models/versions.py">create_training</a>(version_id, \*, model_owner, model_name, \*\*<a href="src/replicate/types/models/version_create_training_params.py">params</a>) -> <a href="./src/replicate/types/models/version_create_training_response.py">VersionCreateTrainingResponse</a></code>
 
 # Predictions
 
@@ -99,11 +105,17 @@ Methods:
 
 # Trainings
 
+Types:
+
+```python
+from replicate.types import TrainingRetrieveResponse, TrainingListResponse, TrainingCancelResponse
+```
+
 Methods:
 
-- <code title="get /trainings/{training_id}">client.trainings.<a href="./src/replicate/resources/trainings.py">retrieve</a>(training_id) -> None</code>
-- <code title="get /trainings">client.trainings.<a href="./src/replicate/resources/trainings.py">list</a>() -> None</code>
-- <code title="post /trainings/{training_id}/cancel">client.trainings.<a href="./src/replicate/resources/trainings.py">cancel</a>(training_id) -> None</code>
+- <code title="get /trainings/{training_id}">client.trainings.<a href="./src/replicate/resources/trainings.py">retrieve</a>(training_id) -> <a href="./src/replicate/types/training_retrieve_response.py">TrainingRetrieveResponse</a></code>
+- <code title="get /trainings">client.trainings.<a href="./src/replicate/resources/trainings.py">list</a>() -> <a href="./src/replicate/types/training_list_response.py">SyncCursorURLPage[TrainingListResponse]</a></code>
+- <code title="post /trainings/{training_id}/cancel">client.trainings.<a href="./src/replicate/resources/trainings.py">cancel</a>(training_id) -> <a href="./src/replicate/types/training_cancel_response.py">TrainingCancelResponse</a></code>
 
 # Webhooks
 

src/replicate/resources/models/versions.py

@@ -22,6 +22,7 @@
 )
 from ..._base_client import make_request_options
 from ...types.models import version_create_training_params
+from ...types.models.version_create_training_response import VersionCreateTrainingResponse
 
 __all__ = ["VersionsResource", "AsyncVersionsResource"]
 
@@ -281,7 +282,7 @@ def create_training(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> VersionCreateTrainingResponse:
         """
         Start a new training of the model version you specify.
 
@@ -398,7 +399,6 @@ def create_training(
             raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
         if not version_id:
             raise ValueError(f"Expected a non-empty value for `version_id` but received {version_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._post(
             f"/models/{model_owner}/{model_name}/versions/{version_id}/trainings",
             body=maybe_transform(
@@ -413,7 +413,7 @@ def create_training(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=VersionCreateTrainingResponse,
         )
 
 
@@ -672,7 +672,7 @@ async def create_training(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> VersionCreateTrainingResponse:
         """
         Start a new training of the model version you specify.
 
@@ -789,7 +789,6 @@ async def create_training(
             raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
         if not version_id:
             raise ValueError(f"Expected a non-empty value for `version_id` but received {version_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._post(
             f"/models/{model_owner}/{model_name}/versions/{version_id}/trainings",
             body=await async_maybe_transform(
@@ -804,7 +803,7 @@ async def create_training(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=VersionCreateTrainingResponse,
         )
 
 

src/replicate/resources/trainings.py

@@ -4,7 +4,7 @@
 
 import httpx
 
-from .._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -13,7 +13,11 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from .._base_client import make_request_options
+from ..pagination import SyncCursorURLPage, AsyncCursorURLPage
+from .._base_client import AsyncPaginator, make_request_options
+from ..types.training_list_response import TrainingListResponse
+from ..types.training_cancel_response import TrainingCancelResponse
+from ..types.training_retrieve_response import TrainingRetrieveResponse
 
 __all__ = ["TrainingsResource", "AsyncTrainingsResource"]
 
@@ -48,7 +52,7 @@ def retrieve(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> TrainingRetrieveResponse:
         """
         Get the current state of a training.
 
@@ -123,13 +127,12 @@ def retrieve(
         """
         if not training_id:
             raise ValueError(f"Expected a non-empty value for `training_id` but received {training_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._get(
             f"/trainings/{training_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TrainingRetrieveResponse,
         )
 
     def list(
@@ -141,7 +144,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> SyncCursorURLPage[TrainingListResponse]:
         """
         Get a paginated list of all trainings created by the user or organization
         associated with the provided API token.
@@ -207,13 +210,13 @@ def list(
 
         `version` will be the unique ID of model version used to create the training.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return self._get(
+        return self._get_api_list(
             "/trainings",
+            page=SyncCursorURLPage[TrainingListResponse],
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            model=TrainingListResponse,
         )
 
     def cancel(
@@ -226,7 +229,7 @@ def cancel(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> TrainingCancelResponse:
         """
         Cancel a training
 
@@ -241,13 +244,12 @@ def cancel(
         """
         if not training_id:
             raise ValueError(f"Expected a non-empty value for `training_id` but received {training_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._post(
             f"/trainings/{training_id}/cancel",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TrainingCancelResponse,
         )
 
 
@@ -281,7 +283,7 @@ async def retrieve(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> TrainingRetrieveResponse:
         """
         Get the current state of a training.
 
@@ -356,16 +358,15 @@ async def retrieve(
         """
         if not training_id:
             raise ValueError(f"Expected a non-empty value for `training_id` but received {training_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._get(
             f"/trainings/{training_id}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TrainingRetrieveResponse,
         )
 
-    async def list(
+    def list(
         self,
         *,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -374,7 +375,7 @@ async def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AsyncPaginator[TrainingListResponse, AsyncCursorURLPage[TrainingListResponse]]:
         """
         Get a paginated list of all trainings created by the user or organization
         associated with the provided API token.
@@ -440,13 +441,13 @@ async def list(
 
         `version` will be the unique ID of model version used to create the training.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return await self._get(
+        return self._get_api_list(
             "/trainings",
+            page=AsyncCursorURLPage[TrainingListResponse],
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            model=TrainingListResponse,
         )
 
     async def cancel(
@@ -459,7 +460,7 @@ async def cancel(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> TrainingCancelResponse:
         """
         Cancel a training
 
@@ -474,13 +475,12 @@ async def cancel(
         """
         if not training_id:
             raise ValueError(f"Expected a non-empty value for `training_id` but received {training_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._post(
             f"/trainings/{training_id}/cancel",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TrainingCancelResponse,
         )
 
 

src/replicate/types/__init__.py

@@ -9,11 +9,14 @@
 from .account_list_response import AccountListResponse as AccountListResponse
 from .hardware_list_response import HardwareListResponse as HardwareListResponse
 from .prediction_list_params import PredictionListParams as PredictionListParams
+from .training_list_response import TrainingListResponse as TrainingListResponse
 from .deployment_create_params import DeploymentCreateParams as DeploymentCreateParams
 from .deployment_list_response import DeploymentListResponse as DeploymentListResponse
 from .deployment_update_params import DeploymentUpdateParams as DeploymentUpdateParams
 from .prediction_create_params import PredictionCreateParams as PredictionCreateParams
+from .training_cancel_response import TrainingCancelResponse as TrainingCancelResponse
 from .deployment_create_response import DeploymentCreateResponse as DeploymentCreateResponse
 from .deployment_update_response import DeploymentUpdateResponse as DeploymentUpdateResponse
+from .training_retrieve_response import TrainingRetrieveResponse as TrainingRetrieveResponse
 from .deployment_retrieve_response import DeploymentRetrieveResponse as DeploymentRetrieveResponse
 from .model_create_prediction_params import ModelCreatePredictionParams as ModelCreatePredictionParams

src/replicate/types/deployment_list_response.py

@@ -49,11 +49,7 @@ class CurrentRelease(BaseModel):
     """The model identifier string in the format of `{model_owner}/{model_name}`."""
 
     number: Optional[int] = None
-    """The release number.
-
-    This is an auto-incrementing integer that starts at 1, and is set automatically
-    when a deployment is created.
-    """
+    """The release number."""
 
     version: Optional[str] = None
     """The ID of the model version used in the release."""

src/replicate/types/models/__init__.py

@@ -3,3 +3,4 @@
 from __future__ import annotations
 
 from .version_create_training_params import VersionCreateTrainingParams as VersionCreateTrainingParams
+from .version_create_training_response import VersionCreateTrainingResponse as VersionCreateTrainingResponse

src/replicate/types/models/version_create_training_response.py

@@ -0,0 +1,74 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+
+__all__ = ["VersionCreateTrainingResponse", "Metrics", "Output", "URLs"]
+
+
+class Metrics(BaseModel):
+    predict_time: Optional[float] = None
+    """The amount of CPU or GPU time, in seconds, that the training used while running"""
+
+
+class Output(BaseModel):
+    version: Optional[str] = None
+    """The version of the model created by the training"""
+
+    weights: Optional[str] = None
+    """The weights of the trained model"""
+
+
+class URLs(BaseModel):
+    cancel: Optional[str] = None
+    """URL to cancel the training"""
+
+    get: Optional[str] = None
+    """URL to get the training details"""
+
+
+class VersionCreateTrainingResponse(BaseModel):
+    id: Optional[str] = None
+    """The unique ID of the training"""
+
+    completed_at: Optional[datetime] = None
+    """The time when the training completed"""
+
+    created_at: Optional[datetime] = None
+    """The time when the training was created"""
+
+    error: Optional[str] = None
+    """Error message if the training failed"""
+
+    input: Optional[Dict[str, object]] = None
+    """The input parameters used for the training"""
+
+    logs: Optional[str] = None
+    """The logs from the training process"""
+
+    metrics: Optional[Metrics] = None
+    """Metrics about the training process"""
+
+    model: Optional[str] = None
+    """The name of the model in the format owner/name"""
+
+    output: Optional[Output] = None
+    """The output of the training process"""
+
+    source: Optional[Literal["web", "api"]] = None
+    """How the training was created"""
+
+    started_at: Optional[datetime] = None
+    """The time when the training started"""
+
+    status: Optional[Literal["starting", "processing", "succeeded", "failed", "canceled"]] = None
+    """The current status of the training"""
+
+    urls: Optional[URLs] = None
+    """URLs for interacting with the training"""
+
+    version: Optional[str] = None
+    """The ID of the model version used for training"""

src/replicate/types/prediction_output.py

@@ -6,5 +6,5 @@
 __all__ = ["PredictionOutput"]
 
 PredictionOutput: TypeAlias = Union[
-    Optional[Dict[str, object]], Optional[List[object]], Optional[str], Optional[float], Optional[bool]
+    Optional[Dict[str, object]], Optional[List[Dict[str, object]]], Optional[str], Optional[float], Optional[bool]
 ]