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-b45f922f6a041550870a96f5acec02aa6d8830046fc98b95a275c6486f7586fc.yml
-openapi_spec_hash: ef7fddfb49b4d9c440b0635d2c86f341
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-37bb31ed76da599d3bded543a3765f745c8575d105c13554df7f8361c3641482.yml
+openapi_spec_hash: 15bdec12ca84042768bfb28cc48dfce3
 config_hash: 810de4c2eee1a7649263cff01f00da7c

api.md

@@ -59,11 +59,17 @@ Methods:
 
 # Models
 
+Types:
+
+```python
+from replicate.types import ModelListResponse
+```
+
 Methods:
 
 - <code title="post /models">client.models.<a href="./src/replicate/resources/models/models.py">create</a>(\*\*<a href="src/replicate/types/model_create_params.py">params</a>) -> None</code>
 - <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="get /models">client.models.<a href="./src/replicate/resources/models/models.py">list</a>() -> <a href="./src/replicate/types/model_list_response.py">SyncCursorURLPage[ModelListResponse]</a></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.py">Prediction</a></code>
 

src/replicate/resources/models/models.py

@@ -30,8 +30,10 @@
     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 import Prediction
+from ...types.model_list_response import ModelListResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 
@@ -290,7 +292,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> SyncCursorURLPage[ModelListResponse]:
         """
         Get a paginated list of public models.
 
@@ -307,13 +309,13 @@ def list(
         See the [`models.get`](#models.get) docs for more details about the model
         object.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return self._get(
+        return self._get_api_list(
             "/models",
+            page=SyncCursorURLPage[ModelListResponse],
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            model=ModelListResponse,
         )
 
     def delete(
@@ -758,7 +760,7 @@ async def retrieve(
             cast_to=NoneType,
         )
 
-    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.
@@ -767,7 +769,7 @@ async def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AsyncPaginator[ModelListResponse, AsyncCursorURLPage[ModelListResponse]]:
         """
         Get a paginated list of public models.
 
@@ -784,13 +786,13 @@ async def list(
         See the [`models.get`](#models.get) docs for more details about the model
         object.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return await self._get(
+        return self._get_api_list(
             "/models",
+            page=AsyncCursorURLPage[ModelListResponse],
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            model=ModelListResponse,
         )
 
     async def delete(

src/replicate/types/__init__.py

@@ -5,6 +5,7 @@
 from .prediction import Prediction as Prediction
 from .prediction_output import PredictionOutput as PredictionOutput
 from .model_create_params import ModelCreateParams as ModelCreateParams
+from .model_list_response import ModelListResponse as ModelListResponse
 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/model_list_response.py

@@ -0,0 +1,46 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["ModelListResponse"]
+
+
+class ModelListResponse(BaseModel):
+    cover_image_url: Optional[str] = None
+    """A URL for the model's cover image"""
+
+    default_example: Optional[object] = None
+    """The model's default example prediction"""
+
+    description: Optional[str] = None
+    """A description of the model"""
+
+    github_url: Optional[str] = None
+    """A URL for the model's source code on GitHub"""
+
+    latest_version: Optional[object] = None
+    """The model's latest version"""
+
+    license_url: Optional[str] = None
+    """A URL for the model's license"""
+
+    name: Optional[str] = None
+    """The name of the model"""
+
+    owner: Optional[str] = None
+    """The name of the user or organization that owns the model"""
+
+    paper_url: Optional[str] = None
+    """A URL for the model's paper"""
+
+    run_count: Optional[int] = None
+    """The number of times the model has been run"""
+
+    url: Optional[str] = None
+    """The URL of the model on Replicate"""
+
+    visibility: Optional[Literal["public", "private"]] = None
+    """Whether the model is public or private"""

tests/api_resources/test_models.py

@@ -9,7 +9,8 @@
 
 from replicate import ReplicateClient, AsyncReplicateClient
 from tests.utils import assert_matches_type
-from replicate.types import Prediction
+from replicate.types import Prediction, ModelListResponse
+from replicate.pagination import SyncCursorURLPage, AsyncCursorURLPage
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
@@ -132,7 +133,7 @@ def test_path_params_retrieve(self, client: ReplicateClient) -> None:
     @parametrize
     def test_method_list(self, client: ReplicateClient) -> None:
         model = client.models.list()
-        assert model is None
+        assert_matches_type(SyncCursorURLPage[ModelListResponse], model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -142,7 +143,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"
         model = response.parse()
-        assert model is None
+        assert_matches_type(SyncCursorURLPage[ModelListResponse], model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -152,7 +153,7 @@ def test_streaming_response_list(self, client: ReplicateClient) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = response.parse()
-            assert model is None
+            assert_matches_type(SyncCursorURLPage[ModelListResponse], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -398,7 +399,7 @@ async def test_path_params_retrieve(self, async_client: AsyncReplicateClient) ->
     @parametrize
     async def test_method_list(self, async_client: AsyncReplicateClient) -> None:
         model = await async_client.models.list()
-        assert model is None
+        assert_matches_type(AsyncCursorURLPage[ModelListResponse], model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -408,7 +409,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"
         model = await response.parse()
-        assert model is None
+        assert_matches_type(AsyncCursorURLPage[ModelListResponse], model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -418,7 +419,7 @@ async def test_streaming_response_list(self, async_client: AsyncReplicateClient)
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = await response.parse()
-            assert model is None
+            assert_matches_type(AsyncCursorURLPage[ModelListResponse], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True