feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 35
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-49078e632f99a7f21fdcd7763517de7118c680b2ff2b90bee601eb3d49ace321.yml
-openapi_spec_hash: baf550db5880018d99637a11dee48380
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-f5aff38eef8d7b245a7af062bf02920ae75e5b9b3dc822416aeb48966c2c6874.yml
+openapi_spec_hash: c0a966beaf5ae95c6bdddd4a933bd4aa
 config_hash: 12536d2bf978a995771d076a4647c17d

api.md

@@ -61,7 +61,7 @@ Methods:
 Types:
 
 ```python
-from replicate.types import ModelListResponse, ModelGetResponse
+from replicate.types import ModelListResponse, ModelGetResponse, ModelSearchResponse
 ```
 
 Methods:
@@ -70,7 +70,7 @@ Methods:
 - <code title="get /models">replicate.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}">replicate.models.<a href="./src/replicate/resources/models/models.py">delete</a>(\*, model_owner, model_name) -> None</code>
 - <code title="get /models/{model_owner}/{model_name}">replicate.models.<a href="./src/replicate/resources/models/models.py">get</a>(\*, model_owner, model_name) -> <a href="./src/replicate/types/model_get_response.py">ModelGetResponse</a></code>
-- <code title="query /models">replicate.models.<a href="./src/replicate/resources/models/models.py">search</a>(\*\*<a href="src/replicate/types/model_search_params.py">params</a>) -> SyncCursorURLPage[object]</code>
+- <code title="query /models">replicate.models.<a href="./src/replicate/resources/models/models.py">search</a>(\*\*<a href="src/replicate/types/model_search_params.py">params</a>) -> <a href="./src/replicate/types/model_search_response.py">SyncCursorURLPage[ModelSearchResponse]</a></code>
 
 ## Examples
 

src/replicate/resources/models/models.py

@@ -53,6 +53,7 @@
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.model_get_response import ModelGetResponse
 from ...types.model_list_response import ModelListResponse
+from ...types.model_search_response import ModelSearchResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 
@@ -414,7 +415,7 @@ def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncCursorURLPage[object]:
+    ) -> SyncCursorURLPage[ModelSearchResponse]:
         """
         Get a list of public models matching a search query.
 
@@ -447,12 +448,12 @@ def search(
         """
         return self._get_api_list(
             "/models",
-            page=SyncCursorURLPage[object],
+            page=SyncCursorURLPage[ModelSearchResponse],
             body=maybe_transform(body, model_search_params.ModelSearchParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            model=object,
+            model=ModelSearchResponse,
             method="query",
         )
 
@@ -814,7 +815,7 @@ def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[object, AsyncCursorURLPage[object]]:
+    ) -> AsyncPaginator[ModelSearchResponse, AsyncCursorURLPage[ModelSearchResponse]]:
         """
         Get a list of public models matching a search query.
 
@@ -847,12 +848,12 @@ def search(
         """
         return self._get_api_list(
             "/models",
-            page=AsyncCursorURLPage[object],
+            page=AsyncCursorURLPage[ModelSearchResponse],
             body=maybe_transform(body, model_search_params.ModelSearchParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            model=object,
+            model=ModelSearchResponse,
             method="query",
         )
 

src/replicate/types/__init__.py

@@ -13,6 +13,7 @@
 from .account_get_response import AccountGetResponse as AccountGetResponse
 from .file_create_response import FileCreateResponse as FileCreateResponse
 from .file_download_params import FileDownloadParams as FileDownloadParams
+from .model_search_response import ModelSearchResponse as ModelSearchResponse
 from .training_get_response import TrainingGetResponse as TrainingGetResponse
 from .hardware_list_response import HardwareListResponse as HardwareListResponse
 from .prediction_list_params import PredictionListParams as PredictionListParams

src/replicate/types/model_search_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__ = ["ModelSearchResponse"]
+
+
+class ModelSearchResponse(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,11 @@
 
 from replicate import Replicate, AsyncReplicate
 from tests.utils import assert_matches_type
-from replicate.types import ModelGetResponse, ModelListResponse
+from replicate.types import (
+    ModelGetResponse,
+    ModelListResponse,
+    ModelSearchResponse,
+)
 from replicate.pagination import SyncCursorURLPage, AsyncCursorURLPage
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
@@ -215,7 +219,7 @@ def test_method_search(self, client: Replicate) -> None:
         model = client.models.search(
             body="body",
         )
-        assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
+        assert_matches_type(SyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -227,7 +231,7 @@ def test_raw_response_search(self, client: Replicate) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         model = response.parse()
-        assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
+        assert_matches_type(SyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -239,7 +243,7 @@ def test_streaming_response_search(self, client: Replicate) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = response.parse()
-            assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
+            assert_matches_type(SyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -446,7 +450,7 @@ async def test_method_search(self, async_client: AsyncReplicate) -> None:
         model = await async_client.models.search(
             body="body",
         )
-        assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -458,7 +462,7 @@ async def test_raw_response_search(self, async_client: AsyncReplicate) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         model = await response.parse()
-        assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -470,6 +474,6 @@ async def test_streaming_response_search(self, async_client: AsyncReplicate) ->
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = await response.parse()
-            assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
+            assert_matches_type(AsyncCursorURLPage[ModelSearchResponse], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True