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

api.md

@@ -61,16 +61,16 @@ Methods:
 Types:
 
 ```python
-from replicate.types import ModelListResponse
+from replicate.types import ModelListResponse, ModelGetResponse
 ```
 
 Methods:
 
 - <code title="post /models">replicate.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">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) -> None</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>) -> 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>
 
 ## Examples
 

src/replicate/resources/models/models.py

@@ -51,6 +51,7 @@
 )
 from ...pagination import SyncCursorURLPage, AsyncCursorURLPage
 from ..._base_client import AsyncPaginator, make_request_options
+from ...types.model_get_response import ModelGetResponse
 from ...types.model_list_response import ModelListResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
@@ -306,7 +307,7 @@ def get(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> ModelGetResponse:
         """
         Example cURL request:
 
@@ -395,13 +396,12 @@ def get(
             raise ValueError(f"Expected a non-empty value for `model_owner` but received {model_owner!r}")
         if not model_name:
             raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._get(
             f"/models/{model_owner}/{model_name}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=ModelGetResponse,
         )
 
     def search(
@@ -414,7 +414,7 @@ def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> object:
         """
         Get a list of public models matching a search query.
 
@@ -445,14 +445,13 @@ def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._query(
             "/models",
             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=NoneType,
+            cast_to=object,
         )
 
 
@@ -706,7 +705,7 @@ async def get(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> ModelGetResponse:
         """
         Example cURL request:
 
@@ -795,13 +794,12 @@ async def get(
             raise ValueError(f"Expected a non-empty value for `model_owner` but received {model_owner!r}")
         if not model_name:
             raise ValueError(f"Expected a non-empty value for `model_name` but received {model_name!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._get(
             f"/models/{model_owner}/{model_name}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=ModelGetResponse,
         )
 
     async def search(
@@ -814,7 +812,7 @@ async def search(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> object:
         """
         Get a list of public models matching a search query.
 
@@ -845,14 +843,13 @@ async def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._query(
             "/models",
             body=await async_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=NoneType,
+            cast_to=object,
         )
 
 

src/replicate/types/__init__.py

@@ -6,6 +6,7 @@
 from .file_get_response import FileGetResponse as FileGetResponse
 from .file_create_params import FileCreateParams as FileCreateParams
 from .file_list_response import FileListResponse as FileListResponse
+from .model_get_response import ModelGetResponse as ModelGetResponse
 from .model_create_params import ModelCreateParams as ModelCreateParams
 from .model_list_response import ModelListResponse as ModelListResponse
 from .model_search_params import ModelSearchParams as ModelSearchParams

src/replicate/types/model_get_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__ = ["ModelGetResponse"]
+
+
+class ModelGetResponse(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,7 @@
 
 from replicate import Replicate, AsyncReplicate
 from tests.utils import assert_matches_type
-from replicate.types import ModelListResponse
+from replicate.types import ModelGetResponse, ModelListResponse
 from replicate.pagination import SyncCursorURLPage, AsyncCursorURLPage
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
@@ -164,7 +164,7 @@ def test_method_get(self, client: Replicate) -> None:
             model_owner="model_owner",
             model_name="model_name",
         )
-        assert model is None
+        assert_matches_type(ModelGetResponse, model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -177,7 +177,7 @@ def test_raw_response_get(self, client: Replicate) -> 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(ModelGetResponse, model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -190,7 +190,7 @@ def test_streaming_response_get(self, client: Replicate) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = response.parse()
-            assert model is None
+            assert_matches_type(ModelGetResponse, model, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -215,7 +215,7 @@ def test_method_search(self, client: Replicate) -> None:
         model = client.models.search(
             body="body",
         )
-        assert model is None
+        assert_matches_type(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 model is None
+        assert_matches_type(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 model is None
+            assert_matches_type(object, model, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -395,7 +395,7 @@ async def test_method_get(self, async_client: AsyncReplicate) -> None:
             model_owner="model_owner",
             model_name="model_name",
         )
-        assert model is None
+        assert_matches_type(ModelGetResponse, model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -408,7 +408,7 @@ async def test_raw_response_get(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 model is None
+        assert_matches_type(ModelGetResponse, model, path=["response"])
 
     @pytest.mark.skip()
     @parametrize
@@ -421,7 +421,7 @@ async def test_streaming_response_get(self, async_client: AsyncReplicate) -> Non
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             model = await response.parse()
-            assert model is None
+            assert_matches_type(ModelGetResponse, 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 model is None
+        assert_matches_type(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 model is None
+        assert_matches_type(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 model is None
+            assert_matches_type(object, model, path=["response"])
 
         assert cast(Any, response.is_closed) is True