feat(api): manual updates

Move models to platform api

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 77
+configured_endpoints: 76
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-e8b3cbc80e18e4f7f277010349f25e1319156704f359911dc464cc21a0d077a6.yml
 openapi_spec_hash: c773d792724f5647ae25a5ae4ccec208
-config_hash: 94c621a4008714c6ab5f773670265bcc
+config_hash: f0976fbc552ea878bb527447b5e663c9

api.md

@@ -367,10 +367,8 @@ Types:
 - <code><a href="./src/resources/models.ts">APIAgreement</a></code>
 - <code><a href="./src/resources/models.ts">APIModel</a></code>
 - <code><a href="./src/resources/models.ts">APIModelVersion</a></code>
-- <code><a href="./src/resources/models.ts">Model</a></code>
 - <code><a href="./src/resources/models.ts">ModelListResponse</a></code>
 
 Methods:
 
-- <code title="get /models/{model}">client.models.<a href="./src/resources/models.ts">retrieve</a>(model) -> Model</code>
-- <code title="get /models">client.models.<a href="./src/resources/models.ts">list</a>() -> ModelListResponse</code>
+- <code title="get /v2/gen-ai/models">client.models.<a href="./src/resources/models.ts">list</a>({ ...params }) -> ModelListResponse</code>

src/client.ts

@@ -21,7 +21,7 @@ import {
   APIAgreement,
   APIModel,
   APIModelVersion,
-  Model,
+  ModelListParams,
   ModelListResponse,
   Models,
 } from './resources/models';
@@ -833,8 +833,8 @@ export declare namespace GradientAI {
     type APIAgreement as APIAgreement,
     type APIModel as APIModel,
     type APIModelVersion as APIModelVersion,
-    type Model as Model,
     type ModelListResponse as ModelListResponse,
+    type ModelListParams as ModelListParams,
   };
 
   export type APILinks = API.APILinks;

src/resources/index.ts

@@ -42,7 +42,7 @@ export {
   type APIAgreement,
   type APIModel,
   type APIModelVersion,
-  type Model,
   type ModelListResponse,
+  type ModelListParams,
 } from './models';
 export { Regions, type RegionListResponse, type RegionListParams } from './regions';

src/resources/models.ts

@@ -1,29 +1,24 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import { APIResource } from '../core/resource';
+import * as Shared from './shared';
 import { APIPromise } from '../core/api-promise';
 import { RequestOptions } from '../internal/request-options';
-import { path } from '../internal/utils/path';
 
 export class Models extends APIResource {
   /**
-   * Retrieves a model instance, providing basic information about the model such as
-   * the owner and permissioning.
+   * To list all models, send a GET request to `/v2/gen-ai/models`.
    */
-  retrieve(model: string, options?: RequestOptions): APIPromise<Model> {
-    return this._client.get(path`/models/${model}`, {
-      defaultBaseURL: 'https://inference.do-ai.run/v1',
+  list(
+    query: ModelListParams | null | undefined = {},
+    options?: RequestOptions,
+  ): APIPromise<ModelListResponse> {
+    return this._client.get('/v2/gen-ai/models', {
+      query,
+      defaultBaseURL: 'https://api.digitalocean.com',
       ...options,
     });
   }
-
-  /**
-   * Lists the currently available models, and provides basic information about each
-   * one such as the owner and availability.
-   */
-  list(options?: RequestOptions): APIPromise<ModelListResponse> {
-    return this._client.get('/models', { defaultBaseURL: 'https://inference.do-ai.run/v1', ...options });
-  }
 }
 
 export interface APIAgreement {
@@ -66,43 +61,59 @@ export interface APIModelVersion {
   patch?: number;
 }
 
-/**
- * Describes a model offering that can be used with the API.
- */
-export interface Model {
+export interface ModelListResponse {
+  links?: Shared.APILinks;
+
+  meta?: Shared.APIMeta;
+
+  models?: Array<APIModel>;
+}
+
+export interface ModelListParams {
   /**
-   * The model identifier, which can be referenced in the API endpoints.
+   * page number.
    */
-  id: string;
+  page?: number;
 
   /**
-   * The Unix timestamp (in seconds) when the model was created.
+   * items per page.
    */
-  created: number;
+  per_page?: number;
 
   /**
-   * The object type, which is always "model".
+   * only include models that are publicly available.
    */
-  object: 'model';
+  public_only?: boolean;
 
   /**
-   * The organization that owns the model.
+   * 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
    */
-  owned_by: string;
-}
-
-export interface ModelListResponse {
-  data: Array<Model>;
-
-  object: 'list';
+  usecases?: Array<
+    | 'MODEL_USECASE_UNKNOWN'
+    | 'MODEL_USECASE_AGENT'
+    | 'MODEL_USECASE_FINETUNED'
+    | 'MODEL_USECASE_KNOWLEDGEBASE'
+    | 'MODEL_USECASE_GUARDRAIL'
+    | 'MODEL_USECASE_REASONING'
+    | 'MODEL_USECASE_SERVERLESS'
+  >;
 }
 
 export declare namespace Models {
   export {
     type APIAgreement as APIAgreement,
     type APIModel as APIModel,
     type APIModelVersion as APIModelVersion,
-    type Model as Model,
     type ModelListResponse as ModelListResponse,
+    type ModelListParams as ModelListParams,
   };
 }

tests/api-resources/models.test.ts

@@ -10,8 +10,8 @@ const client = new GradientAI({
 
 describe('resource models', () => {
   // skipped: tests are disabled for the time being
-  test.skip('retrieve', async () => {
-    const responsePromise = client.models.retrieve('llama3-8b-instruct');
+  test.skip('list', async () => {
+    const responsePromise = client.models.list();
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
     const response = await responsePromise;
@@ -22,14 +22,13 @@ describe('resource models', () => {
   });
 
   // skipped: tests are disabled for the time being
-  test.skip('list', async () => {
-    const responsePromise = client.models.list();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
+  test.skip('list: request options and params are passed correctly', async () => {
+    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
+    await expect(
+      client.models.list(
+        { page: 0, per_page: 0, public_only: true, usecases: ['MODEL_USECASE_UNKNOWN'] },
+        { path: '/_stainless_unknown_path' },
+      ),
+    ).rejects.toThrow(GradientAI.NotFoundError);
   });
 });