feat(api): update via SDK Studio

stainless-app[bot] · stainless-app[bot] · commit c04249614917 · 2025-06-17T13:26:37.000Z
diff --git a/.stats.yml b/.stats.yml
@@ -1,4 +1,4 @@
-configured_endpoints: 59
+configured_endpoints: 58
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fdigitalocean-genai-sdk-6e449984986e066baea73af5c2726811e74a284f0d68d49926ec5c7821c7ed31.yml
 openapi_spec_hash: 78f43f68f46df0d81891ae2ff66bf3a0
-config_hash: 3e04a2c7a4b0b9b16bd2956a3208b942
+config_hash: 84ba29fbded3618d3cc3994639c82547
diff --git a/api.md b/api.md
@@ -332,4 +332,4 @@ from gradientai.types import Model, ModelListResponse
 Methods:
 
 - <code title="get /models/{model}">client.models.<a href="./src/gradientai/resources/models.py">retrieve</a>(model) -> <a href="./src/gradientai/types/model.py">Model</a></code>
-- <code title="get /models">client.models.<a href="./src/gradientai/resources/models.py">list</a>() -> <a href="./src/gradientai/types/model_list_response.py">ModelListResponse</a></code>
+- <code title="get /v2/gen-ai/models">client.models.<a href="./src/gradientai/resources/models.py">list</a>(\*\*<a href="src/gradientai/types/model_list_params.py">params</a>) -> <a href="./src/gradientai/types/model_list_response.py">ModelListResponse</a></code>
diff --git a/src/gradientai/resources/models.py b/src/gradientai/resources/models.py
@@ -2,9 +2,14 @@
 
 from __future__ import annotations
 
+from typing import List
+from typing_extensions import Literal
+
 import httpx
 
+from ..types import model_list_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -77,6 +82,21 @@ def retrieve(
     def list(
         self,
         *,
+        page: int | NotGiven = NOT_GIVEN,
+        per_page: int | NotGiven = NOT_GIVEN,
+        public_only: bool | NotGiven = NOT_GIVEN,
+        usecases: List[
+            Literal[
+                "MODEL_USECASE_UNKNOWN",
+                "MODEL_USECASE_AGENT",
+                "MODEL_USECASE_FINETUNED",
+                "MODEL_USECASE_KNOWLEDGEBASE",
+                "MODEL_USECASE_GUARDRAIL",
+                "MODEL_USECASE_REASONING",
+                "MODEL_USECASE_SERVERLESS",
+            ]
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -85,13 +105,50 @@ def list(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ModelListResponse:
         """
-        Lists the currently available models, and provides basic information about each
-        one such as the owner and availability.
+        To list all models, send a GET request to `/v2/gen-ai/models`.
+
+        Args:
+          page: page number.
+
+          per_page: items per page.
+
+          public_only: only include models that are publicly available.
+
+          usecases: include only models defined for the listed usecases.
+
+              - MODEL_USECASE_UNKNOWN: The use case of the model is unknown
+              - MODEL_USECASE_AGENT: The model maybe used in an agent
+              - MODEL_USECASE_FINETUNED: The model maybe used for fine tuning
+              - MODEL_USECASE_KNOWLEDGEBASE: The model maybe used for knowledge bases
+                (embedding models)
+              - MODEL_USECASE_GUARDRAIL: The model maybe used for guardrails
+              - MODEL_USECASE_REASONING: The model usecase for reasoning
+              - MODEL_USECASE_SERVERLESS: The model usecase for serverless inference
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._get(
-            "/models",
+            "/v2/gen-ai/models",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                        "public_only": public_only,
+                        "usecases": usecases,
+                    },
+                    model_list_params.ModelListParams,
+                ),
             ),
             cast_to=ModelListResponse,
         )
@@ -154,6 +211,21 @@ async def retrieve(
     async def list(
         self,
         *,
+        page: int | NotGiven = NOT_GIVEN,
+        per_page: int | NotGiven = NOT_GIVEN,
+        public_only: bool | NotGiven = NOT_GIVEN,
+        usecases: List[
+            Literal[
+                "MODEL_USECASE_UNKNOWN",
+                "MODEL_USECASE_AGENT",
+                "MODEL_USECASE_FINETUNED",
+                "MODEL_USECASE_KNOWLEDGEBASE",
+                "MODEL_USECASE_GUARDRAIL",
+                "MODEL_USECASE_REASONING",
+                "MODEL_USECASE_SERVERLESS",
+            ]
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -162,13 +234,50 @@ async def list(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ModelListResponse:
         """
-        Lists the currently available models, and provides basic information about each
-        one such as the owner and availability.
+        To list all models, send a GET request to `/v2/gen-ai/models`.
+
+        Args:
+          page: page number.
+
+          per_page: items per page.
+
+          public_only: only include models that are publicly available.
+
+          usecases: include only models defined for the listed usecases.
+
+              - MODEL_USECASE_UNKNOWN: The use case of the model is unknown
+              - MODEL_USECASE_AGENT: The model maybe used in an agent
+              - MODEL_USECASE_FINETUNED: The model maybe used for fine tuning
+              - MODEL_USECASE_KNOWLEDGEBASE: The model maybe used for knowledge bases
+                (embedding models)
+              - MODEL_USECASE_GUARDRAIL: The model maybe used for guardrails
+              - MODEL_USECASE_REASONING: The model usecase for reasoning
+              - MODEL_USECASE_SERVERLESS: The model usecase for serverless inference
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._get(
-            "/models",
+            "/v2/gen-ai/models",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                        "public_only": public_only,
+                        "usecases": usecases,
+                    },
+                    model_list_params.ModelListParams,
+                ),
             ),
             cast_to=ModelListResponse,
         )
diff --git a/src/gradientai/types/__init__.py b/src/gradientai/types/__init__.py
@@ -9,6 +9,7 @@
 from .api_indexing_job import APIIndexingJob as APIIndexingJob
 from .agent_list_params import AgentListParams as AgentListParams
 from .api_model_version import APIModelVersion as APIModelVersion
+from .model_list_params import ModelListParams as ModelListParams
 from .api_knowledge_base import APIKnowledgeBase as APIKnowledgeBase
 from .region_list_params import RegionListParams as RegionListParams
 from .agent_create_params import AgentCreateParams as AgentCreateParams
diff --git a/src/gradientai/types/model_list_params.py b/src/gradientai/types/model_list_params.py
@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["ModelListParams"]
+
+
+class ModelListParams(TypedDict, total=False):
+    page: int
+    """page number."""
+
+    per_page: int
+    """items per page."""
+
+    public_only: bool
+    """only include models that are publicly available."""
+
+    usecases: List[
+        Literal[
+            "MODEL_USECASE_UNKNOWN",
+            "MODEL_USECASE_AGENT",
+            "MODEL_USECASE_FINETUNED",
+            "MODEL_USECASE_KNOWLEDGEBASE",
+            "MODEL_USECASE_GUARDRAIL",
+            "MODEL_USECASE_REASONING",
+            "MODEL_USECASE_SERVERLESS",
+        ]
+    ]
+    """include only models defined for the listed usecases.
+
+    - MODEL_USECASE_UNKNOWN: The use case of the model is unknown
+    - MODEL_USECASE_AGENT: The model maybe used in an agent
+    - MODEL_USECASE_FINETUNED: The model maybe used for fine tuning
+    - MODEL_USECASE_KNOWLEDGEBASE: The model maybe used for knowledge bases
+      (embedding models)
+    - MODEL_USECASE_GUARDRAIL: The model maybe used for guardrails
+    - MODEL_USECASE_REASONING: The model usecase for reasoning
+    - MODEL_USECASE_SERVERLESS: The model usecase for serverless inference
+    """
diff --git a/src/gradientai/types/model_list_response.py b/src/gradientai/types/model_list_response.py
@@ -1,15 +1,42 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List
-from typing_extensions import Literal
+from typing import List, Optional
+from datetime import datetime
 
-from .model import Model
 from .._models import BaseModel
+from .api_agreement import APIAgreement
+from .agents.api_meta import APIMeta
+from .agents.api_links import APILinks
+from .api_model_version import APIModelVersion
 
-__all__ = ["ModelListResponse"]
+__all__ = ["ModelListResponse", "Model"]
+
+
+class Model(BaseModel):
+    agreement: Optional[APIAgreement] = None
+
+    created_at: Optional[datetime] = None
+
+    is_foundational: Optional[bool] = None
+
+    name: Optional[str] = None
+
+    parent_uuid: Optional[str] = None
+
+    updated_at: Optional[datetime] = None
+
+    upload_complete: Optional[bool] = None
+
+    url: Optional[str] = None
+
+    uuid: Optional[str] = None
+
+    version: Optional[APIModelVersion] = None
 
 
 class ModelListResponse(BaseModel):
-    data: List[Model]
+    links: Optional[APILinks] = None
+
+    meta: Optional[APIMeta] = None
 
-    object: Literal["list"]
+    models: Optional[List[Model]] = None
diff --git a/tests/api_resources/test_models.py b/tests/api_resources/test_models.py
@@ -65,6 +65,17 @@ def test_method_list(self, client: GradientAI) -> None:
         model = client.models.list()
         assert_matches_type(ModelListResponse, model, path=["response"])
 
+    @pytest.mark.skip()
+    @parametrize
+    def test_method_list_with_all_params(self, client: GradientAI) -> None:
+        model = client.models.list(
+            page=0,
+            per_page=0,
+            public_only=True,
+            usecases=["MODEL_USECASE_UNKNOWN"],
+        )
+        assert_matches_type(ModelListResponse, model, path=["response"])
+
     @pytest.mark.skip()
     @parametrize
     def test_raw_response_list(self, client: GradientAI) -> None:
@@ -139,6 +150,17 @@ async def test_method_list(self, async_client: AsyncGradientAI) -> None:
         model = await async_client.models.list()
         assert_matches_type(ModelListResponse, model, path=["response"])
 
+    @pytest.mark.skip()
+    @parametrize
+    async def test_method_list_with_all_params(self, async_client: AsyncGradientAI) -> None:
+        model = await async_client.models.list(
+            page=0,
+            per_page=0,
+            public_only=True,
+            usecases=["MODEL_USECASE_UNKNOWN"],
+        )
+        assert_matches_type(ModelListResponse, model, path=["response"])
+
     @pytest.mark.skip()
     @parametrize
     async def test_raw_response_list(self, async_client: AsyncGradientAI) -> None:
