feat(api): manual updates (#20)

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-mpatankar%2Freplicate-client-b45f922f6a041550870a96f5acec02aa6d8830046fc98b95a275c6486f7586fc.yml
 openapi_spec_hash: ef7fddfb49b4d9c440b0635d2c86f341
-config_hash: 0927aba56566832d6080b89e0dc97d6a
+config_hash: 5691e71e792eb8d38c4c91fa7efff22c

api.md

@@ -10,8 +10,6 @@ Types:
 
 ```python
 from replicate.types import (
-    PredictionRequest,
-    PredictionResponse,
     DeploymentCreateResponse,
     DeploymentRetrieveResponse,
     DeploymentUpdateResponse,
@@ -32,7 +30,7 @@ Methods:
 
 Methods:
 
-- <code title="post /deployments/{deployment_owner}/{deployment_name}/predictions">client.deployments.predictions.<a href="./src/replicate/resources/deployments/predictions.py">create</a>(deployment_name, \*, deployment_owner, \*\*<a href="src/replicate/types/deployments/prediction_create_params.py">params</a>) -> <a href="./src/replicate/types/prediction_response.py">PredictionResponse</a></code>
+- <code title="post /deployments/{deployment_owner}/{deployment_name}/predictions">client.deployments.predictions.<a href="./src/replicate/resources/deployments/predictions.py">create</a>(deployment_name, \*, deployment_owner, \*\*<a href="src/replicate/types/deployments/prediction_create_params.py">params</a>) -> <a href="./src/replicate/types/prediction.py">Prediction</a></code>
 
 # Hardware
 
@@ -67,7 +65,7 @@ Methods:
 - <code title="get /models/{model_owner}/{model_name}">client.models.<a href="./src/replicate/resources/models/models.py">retrieve</a>(model_name, \*, model_owner) -> None</code>
 - <code title="get /models">client.models.<a href="./src/replicate/resources/models/models.py">list</a>() -> None</code>
 - <code title="delete /models/{model_owner}/{model_name}">client.models.<a href="./src/replicate/resources/models/models.py">delete</a>(model_name, \*, model_owner) -> None</code>
-- <code title="post /models/{model_owner}/{model_name}/predictions">client.models.<a href="./src/replicate/resources/models/models.py">create_prediction</a>(model_name, \*, model_owner, \*\*<a href="src/replicate/types/model_create_prediction_params.py">params</a>) -> <a href="./src/replicate/types/prediction_response.py">PredictionResponse</a></code>
+- <code title="post /models/{model_owner}/{model_name}/predictions">client.models.<a href="./src/replicate/resources/models/models.py">create_prediction</a>(model_name, \*, model_owner, \*\*<a href="src/replicate/types/model_create_prediction_params.py">params</a>) -> <a href="./src/replicate/types/prediction.py">Prediction</a></code>
 
 ## Versions
 
@@ -80,12 +78,18 @@ Methods:
 
 # Predictions
 
+Types:
+
+```python
+from replicate.types import Prediction, PredictionRequest
+```
+
 Methods:
 
-- <code title="post /predictions">client.predictions.<a href="./src/replicate/resources/predictions.py">create</a>(\*\*<a href="src/replicate/types/prediction_create_params.py">params</a>) -> <a href="./src/replicate/types/prediction_response.py">PredictionResponse</a></code>
-- <code title="get /predictions">client.predictions.<a href="./src/replicate/resources/predictions.py">list</a>(\*\*<a href="src/replicate/types/prediction_list_params.py">params</a>) -> <a href="./src/replicate/types/prediction_response.py">SyncCursorURLPage[PredictionResponse]</a></code>
+- <code title="post /predictions">client.predictions.<a href="./src/replicate/resources/predictions.py">create</a>(\*\*<a href="src/replicate/types/prediction_create_params.py">params</a>) -> <a href="./src/replicate/types/prediction.py">Prediction</a></code>
+- <code title="get /predictions">client.predictions.<a href="./src/replicate/resources/predictions.py">list</a>(\*\*<a href="src/replicate/types/prediction_list_params.py">params</a>) -> <a href="./src/replicate/types/prediction.py">SyncCursorURLPage[Prediction]</a></code>
 - <code title="post /predictions/{prediction_id}/cancel">client.predictions.<a href="./src/replicate/resources/predictions.py">cancel</a>(prediction_id) -> None</code>
-- <code title="get /predictions/{prediction_id}">client.predictions.<a href="./src/replicate/resources/predictions.py">list_by_id</a>(prediction_id) -> <a href="./src/replicate/types/prediction_response.py">PredictionResponse</a></code>
+- <code title="get /predictions/{prediction_id}">client.predictions.<a href="./src/replicate/resources/predictions.py">list_by_id</a>(prediction_id) -> <a href="./src/replicate/types/prediction.py">Prediction</a></code>
 
 # Trainings
 

src/replicate/resources/deployments/predictions.py

@@ -22,8 +22,8 @@
     async_to_streamed_response_wrapper,
 )
 from ..._base_client import make_request_options
+from ...types.prediction import Prediction
 from ...types.deployments import prediction_create_params
-from ...types.prediction_response import PredictionResponse
 
 __all__ = ["PredictionsResource", "AsyncPredictionsResource"]
 
@@ -64,7 +64,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction for the deployment and inputs you provide.
 
@@ -181,7 +181,7 @@ def create(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 
@@ -221,7 +221,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction for the deployment and inputs you provide.
 
@@ -338,7 +338,7 @@ async def create(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 

src/replicate/resources/models/models.py

@@ -31,7 +31,7 @@
     async_to_streamed_response_wrapper,
 )
 from ..._base_client import make_request_options
-from ...types.prediction_response import PredictionResponse
+from ...types.prediction import Prediction
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 
@@ -388,7 +388,7 @@ def create_prediction(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction using an
         [official model](https://replicate.com/changelog/2025-01-29-official-models).
@@ -509,7 +509,7 @@ def create_prediction(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 
@@ -865,7 +865,7 @@ async def create_prediction(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction using an
         [official model](https://replicate.com/changelog/2025-01-29-official-models).
@@ -986,7 +986,7 @@ async def create_prediction(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 

src/replicate/resources/predictions.py

@@ -25,7 +25,7 @@
 )
 from ..pagination import SyncCursorURLPage, AsyncCursorURLPage
 from .._base_client import AsyncPaginator, make_request_options
-from ..types.prediction_response import PredictionResponse
+from ..types.prediction import Prediction
 
 __all__ = ["PredictionsResource", "AsyncPredictionsResource"]
 
@@ -65,7 +65,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction for the model version and inputs you provide.
 
@@ -186,7 +186,7 @@ def create(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
     def list(
@@ -200,7 +200,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncCursorURLPage[PredictionResponse]:
+    ) -> SyncCursorURLPage[Prediction]:
         """
         Get a paginated list of all predictions created by the user or organization
         associated with the provided API token.
@@ -287,7 +287,7 @@ def list(
         """
         return self._get_api_list(
             "/predictions",
-            page=SyncCursorURLPage[PredictionResponse],
+            page=SyncCursorURLPage[Prediction],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -301,7 +301,7 @@ def list(
                     prediction_list_params.PredictionListParams,
                 ),
             ),
-            model=PredictionResponse,
+            model=Prediction,
         )
 
     def cancel(
@@ -348,7 +348,7 @@ def list_by_id(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Get the current state of a prediction.
 
@@ -437,7 +437,7 @@ def list_by_id(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 
@@ -476,7 +476,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Create a prediction for the model version and inputs you provide.
 
@@ -597,7 +597,7 @@ async def create(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
     def list(
@@ -611,7 +611,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[PredictionResponse, AsyncCursorURLPage[PredictionResponse]]:
+    ) -> AsyncPaginator[Prediction, AsyncCursorURLPage[Prediction]]:
         """
         Get a paginated list of all predictions created by the user or organization
         associated with the provided API token.
@@ -698,7 +698,7 @@ def list(
         """
         return self._get_api_list(
             "/predictions",
-            page=AsyncCursorURLPage[PredictionResponse],
+            page=AsyncCursorURLPage[Prediction],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -712,7 +712,7 @@ def list(
                     prediction_list_params.PredictionListParams,
                 ),
             ),
-            model=PredictionResponse,
+            model=Prediction,
         )
 
     async def cancel(
@@ -759,7 +759,7 @@ async def list_by_id(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionResponse:
+    ) -> Prediction:
         """
         Get the current state of a prediction.
 
@@ -848,7 +848,7 @@ async def list_by_id(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=PredictionResponse,
+            cast_to=Prediction,
         )
 
 

src/replicate/types/__init__.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+from .prediction import Prediction as Prediction
 from .model_create_params import ModelCreateParams as ModelCreateParams
-from .prediction_response import PredictionResponse as PredictionResponse
 from .account_list_response import AccountListResponse as AccountListResponse
 from .hardware_list_response import HardwareListResponse as HardwareListResponse
 from .prediction_list_params import PredictionListParams as PredictionListParams

src/replicate/types/prediction_response.py renamed to src/replicate/types/prediction.py

@@ -6,7 +6,7 @@
 
 from .._models import BaseModel
 
-__all__ = ["PredictionResponse", "URLs"]
+__all__ = ["Prediction", "URLs"]
 
 
 class URLs(BaseModel):
@@ -20,7 +20,7 @@ class URLs(BaseModel):
     """An event source to stream the output of the prediction"""
 
 
-class PredictionResponse(BaseModel):
+class Prediction(BaseModel):
     id: str
 
     created_at: datetime

tests/api_resources/deployments/test_predictions.py

@@ -9,7 +9,7 @@
 
 from replicate import ReplicateClient, AsyncReplicateClient
 from tests.utils import assert_matches_type
-from replicate.types import PredictionResponse
+from replicate.types import Prediction
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
@@ -25,7 +25,7 @@ def test_method_create(self, client: ReplicateClient) -> None:
             deployment_owner="deployment_owner",
             input={},
         )
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -39,7 +39,7 @@ def test_method_create_with_all_params(self, client: ReplicateClient) -> None:
             webhook_events_filter=["start"],
             prefer="wait=5",
         )
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -53,7 +53,7 @@ def test_raw_response_create(self, client: ReplicateClient) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         prediction = response.parse()
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -67,7 +67,7 @@ def test_streaming_response_create(self, client: ReplicateClient) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = response.parse()
-            assert_matches_type(PredictionResponse, prediction, path=["response"])
+            assert_matches_type(Prediction, prediction, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -100,7 +100,7 @@ async def test_method_create(self, async_client: AsyncReplicateClient) -> None:
             deployment_owner="deployment_owner",
             input={},
         )
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -114,7 +114,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicateC
             webhook_events_filter=["start"],
             prefer="wait=5",
         )
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -128,7 +128,7 @@ async def test_raw_response_create(self, async_client: AsyncReplicateClient) ->
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         prediction = await response.parse()
-        assert_matches_type(PredictionResponse, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -142,7 +142,7 @@ async def test_streaming_response_create(self, async_client: AsyncReplicateClien
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = await response.parse()
-            assert_matches_type(PredictionResponse, prediction, path=["response"])
+            assert_matches_type(Prediction, prediction, path=["response"])
 
         assert cast(Any, response.is_closed) is True