feat(api): update via SDK Studio (#16)

Author: stainless-app[bot]

api.md

@@ -80,16 +80,10 @@ Methods:
 
 # Predictions
 
-Types:
-
-```python
-from replicate.types import PredictionListResponse
-```
-
 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_list_response.py">PredictionListResponse</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/{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>
 

src/replicate/pagination.py

@@ -0,0 +1,52 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Generic, TypeVar, Optional
+from typing_extensions import override
+
+import httpx
+
+from ._base_client import BasePage, PageInfo, BaseSyncPage, BaseAsyncPage
+
+__all__ = ["SyncCursorURLPage", "AsyncCursorURLPage"]
+
+_T = TypeVar("_T")
+
+
+class SyncCursorURLPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    results: List[_T]
+    next: Optional[str] = None
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        results = self.results
+        if not results:
+            return []
+        return results
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        url = self.next
+        if url is None:
+            return None
+
+        return PageInfo(url=httpx.URL(url))
+
+
+class AsyncCursorURLPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    results: List[_T]
+    next: Optional[str] = None
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        results = self.results
+        if not results:
+            return []
+        return results
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        url = self.next
+        if url is None:
+            return None
+
+        return PageInfo(url=httpx.URL(url))

src/replicate/resources/predictions.py

@@ -23,9 +23,9 @@
     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.prediction_response import PredictionResponse
-from ..types.prediction_list_response import PredictionListResponse
 
 __all__ = ["PredictionsResource", "AsyncPredictionsResource"]
 
@@ -200,7 +200,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionListResponse:
+    ) -> SyncCursorURLPage[PredictionResponse]:
         """
         Get a paginated list of all predictions created by the user or organization
         associated with the provided API token.
@@ -285,8 +285,9 @@ def list(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/predictions",
+            page=SyncCursorURLPage[PredictionResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -300,7 +301,7 @@ def list(
                     prediction_list_params.PredictionListParams,
                 ),
             ),
-            cast_to=PredictionListResponse,
+            model=PredictionResponse,
         )
 
     def cancel(
@@ -599,7 +600,7 @@ async def create(
             cast_to=PredictionResponse,
         )
 
-    async def list(
+    def list(
         self,
         *,
         created_after: Union[str, datetime] | NotGiven = NOT_GIVEN,
@@ -610,7 +611,7 @@ async def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> PredictionListResponse:
+    ) -> AsyncPaginator[PredictionResponse, AsyncCursorURLPage[PredictionResponse]]:
         """
         Get a paginated list of all predictions created by the user or organization
         associated with the provided API token.
@@ -695,22 +696,23 @@ async def list(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/predictions",
+            page=AsyncCursorURLPage[PredictionResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform(
+                query=maybe_transform(
                     {
                         "created_after": created_after,
                         "created_before": created_before,
                     },
                     prediction_list_params.PredictionListParams,
                 ),
             ),
-            cast_to=PredictionListResponse,
+            model=PredictionResponse,
         )
 
     async def cancel(

src/replicate/types/__init__.py

@@ -11,7 +11,6 @@
 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 .prediction_list_response import PredictionListResponse as PredictionListResponse
 from .deployment_create_response import DeploymentCreateResponse as DeploymentCreateResponse
 from .deployment_update_response import DeploymentUpdateResponse as DeploymentUpdateResponse
 from .deployment_retrieve_response import DeploymentRetrieveResponse as DeploymentRetrieveResponse

src/replicate/types/prediction_list_response.py

@@ -9,8 +9,9 @@
 
 from replicate import ReplicateClient, AsyncReplicateClient
 from tests.utils import assert_matches_type
-from replicate.types import PredictionResponse, PredictionListResponse
+from replicate.types import PredictionResponse
 from replicate._utils import parse_datetime
+from replicate.pagination import SyncCursorURLPage, AsyncCursorURLPage
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
@@ -72,7 +73,7 @@ def test_streaming_response_create(self, client: ReplicateClient) -> None:
     @parametrize
     def test_method_list(self, client: ReplicateClient) -> None:
         prediction = client.predictions.list()
-        assert_matches_type(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(SyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -81,7 +82,7 @@ def test_method_list_with_all_params(self, client: ReplicateClient) -> None:
             created_after=parse_datetime("2025-01-01T00:00:00Z"),
             created_before=parse_datetime("2025-02-01T00:00:00Z"),
         )
-        assert_matches_type(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(SyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -91,7 +92,7 @@ def test_raw_response_list(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(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(SyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -101,7 +102,7 @@ def test_streaming_response_list(self, client: ReplicateClient) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = response.parse()
-            assert_matches_type(PredictionListResponse, prediction, path=["response"])
+            assert_matches_type(SyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -247,7 +248,7 @@ async def test_streaming_response_create(self, async_client: AsyncReplicateClien
     @parametrize
     async def test_method_list(self, async_client: AsyncReplicateClient) -> None:
         prediction = await async_client.predictions.list()
-        assert_matches_type(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -256,7 +257,7 @@ async def test_method_list_with_all_params(self, async_client: AsyncReplicateCli
             created_after=parse_datetime("2025-01-01T00:00:00Z"),
             created_before=parse_datetime("2025-02-01T00:00:00Z"),
         )
-        assert_matches_type(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -266,7 +267,7 @@ async def test_raw_response_list(self, async_client: AsyncReplicateClient) -> No
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         prediction = await response.parse()
-        assert_matches_type(PredictionListResponse, prediction, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -276,7 +277,7 @@ async def test_streaming_response_list(self, async_client: AsyncReplicateClient)
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = await response.parse()
-            assert_matches_type(PredictionListResponse, prediction, path=["response"])
+            assert_matches_type(AsyncCursorURLPage[PredictionResponse], prediction, path=["response"])
 
         assert cast(Any, response.is_closed) is True