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-661f1eceefb9f933f9834ee8fc503ed35473904eceb2467852157fae379b3641.yml
-openapi_spec_hash: 73a5fc50881bce84daef22ba4620c499
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate%2Freplicate-client-49078e632f99a7f21fdcd7763517de7118c680b2ff2b90bee601eb3d49ace321.yml
+openapi_spec_hash: baf550db5880018d99637a11dee48380
 config_hash: 12536d2bf978a995771d076a4647c17d

api.md

@@ -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>) -> 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>) -> SyncCursorURLPage[object]</code>
 
 ## Examples
 

src/replicate/resources/models/models.py

@@ -414,7 +414,7 @@ def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> SyncCursorURLPage[object]:
         """
         Get a list of public models matching a search query.
 
@@ -445,13 +445,15 @@ def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._query(
+        return self._get_api_list(
             "/models",
+            page=SyncCursorURLPage[object],
             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
             ),
-            cast_to=object,
+            model=object,
+            method="query",
         )
 
 
@@ -802,7 +804,7 @@ async def get(
             cast_to=ModelGetResponse,
         )
 
-    async def search(
+    def search(
         self,
         *,
         body: str,
@@ -812,7 +814,7 @@ async def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> AsyncPaginator[object, AsyncCursorURLPage[object]]:
         """
         Get a list of public models matching a search query.
 
@@ -843,13 +845,15 @@ async def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._query(
+        return self._get_api_list(
             "/models",
-            body=await async_maybe_transform(body, model_search_params.ModelSearchParams),
+            page=AsyncCursorURLPage[object],
+            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
             ),
-            cast_to=object,
+            model=object,
+            method="query",
         )
 
 

tests/api_resources/test_models.py

@@ -215,7 +215,7 @@ def test_method_search(self, client: Replicate) -> None:
         model = client.models.search(
             body="body",
         )
-        assert_matches_type(object, model, path=["response"])
+        assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -227,7 +227,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(object, model, path=["response"])
+        assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -239,7 +239,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(object, model, path=["response"])
+            assert_matches_type(SyncCursorURLPage[object], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -446,7 +446,7 @@ async def test_method_search(self, async_client: AsyncReplicate) -> None:
         model = await async_client.models.search(
             body="body",
         )
-        assert_matches_type(object, model, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -458,7 +458,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(object, model, path=["response"])
+        assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
 
     @pytest.mark.skip(reason="Prism doesn't support query methods yet")
     @parametrize
@@ -470,6 +470,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(object, model, path=["response"])
+            assert_matches_type(AsyncCursorURLPage[object], model, path=["response"])
 
         assert cast(Any, response.is_closed) is True