feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 49
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-490b6c951dbe764981862a42c1067201cca1473d6ea2b5950d6b52246db9c455.yml
-openapi_spec_hash: 25f3fb0ed209b3ddd390b3b1df365f5f
+configured_endpoints: 48
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-dc1b1744bf9a13daa82065ecf5dc75e3870d611ab3fabf2b0f91a7b36815b2a4.yml
+openapi_spec_hash: ae1eb47bf32f7b2c225338a245298e02
 config_hash: 929863a9ffccfd7c8065b0deffba83c3

api.md

@@ -10,7 +10,6 @@ Types:
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetGenerateSchemaResponse</a></code>
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetGenerateValidatorsResponse</a></code>
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetInsertEntryResponse</a></code>
-- <code><a href="./src/resources/datasets/datasets.ts">DatasetUploadJSONLResponse</a></code>
 
 Methods:
 
@@ -23,8 +22,7 @@ Methods:
 - <code title="post /datasets/generate-dataset">client.datasets.<a href="./src/resources/datasets/datasets.ts">generateDataset</a>({ ...params }) -> DatasetGenerateDatasetResponse</code>
 - <code title="post /datasets/generate-schema">client.datasets.<a href="./src/resources/datasets/datasets.ts">generateSchema</a>({ ...params }) -> DatasetGenerateSchemaResponse</code>
 - <code title="post /datasets/generate-validators">client.datasets.<a href="./src/resources/datasets/datasets.ts">generateValidators</a>({ ...params }) -> DatasetGenerateValidatorsResponse</code>
-- <code title="post /datasets/{dataset_id}/insert-entry">client.datasets.<a href="./src/resources/datasets/datasets.ts">insertEntry</a>(datasetID, { ...params }) -> string</code>
-- <code title="post /datasets/{dataset_id}/upload-jsonl">client.datasets.<a href="./src/resources/datasets/datasets.ts">uploadJSONL</a>(datasetID, { ...params }) -> DatasetUploadJSONLResponse</code>
+- <code title="post /datasets/{dataset_id}/insert-entry">client.datasets.<a href="./src/resources/datasets/datasets.ts">insertEntry</a>(datasetID, { ...params }) -> DatasetInsertEntryResponse</code>
 
 ## Generations
 

src/client.ts

@@ -41,8 +41,6 @@ import {
   DatasetListParams,
   DatasetListResponse,
   DatasetUpdateParams,
-  DatasetUploadJSONLParams,
-  DatasetUploadJSONLResponse,
   Datasets,
 } from './resources/datasets/datasets';
 import {
@@ -774,15 +772,13 @@ export declare namespace Lemma {
     type DatasetGenerateSchemaResponse as DatasetGenerateSchemaResponse,
     type DatasetGenerateValidatorsResponse as DatasetGenerateValidatorsResponse,
     type DatasetInsertEntryResponse as DatasetInsertEntryResponse,
-    type DatasetUploadJSONLResponse as DatasetUploadJSONLResponse,
     type DatasetCreateParams as DatasetCreateParams,
     type DatasetUpdateParams as DatasetUpdateParams,
     type DatasetListParams as DatasetListParams,
     type DatasetGenerateDatasetParams as DatasetGenerateDatasetParams,
     type DatasetGenerateSchemaParams as DatasetGenerateSchemaParams,
     type DatasetGenerateValidatorsParams as DatasetGenerateValidatorsParams,
     type DatasetInsertEntryParams as DatasetInsertEntryParams,
-    type DatasetUploadJSONLParams as DatasetUploadJSONLParams,
   };
 
   export {

src/resources/datasets/datasets.ts

@@ -128,28 +128,10 @@ export class Datasets extends APIResource {
     datasetID: string,
     params: DatasetInsertEntryParams,
     options?: RequestOptions,
-  ): APIPromise<string> {
+  ): APIPromise<DatasetInsertEntryResponse> {
     const { body } = params;
     return this._client.post(path`/datasets/${datasetID}/insert-entry`, { body: body, ...options });
   }
-
-  /**
-   * Upload a JSONL file to a dataset.
-   *
-   * Accepts a JSONL (JSON Lines) file where each line is a valid JSON object. Each
-   * entry is stored in S3 and a JSON schema is automatically generated from all
-   * entries using GenSON. The dataset status will be updated to READY.
-   */
-  uploadJSONL(
-    datasetID: string,
-    body: DatasetUploadJSONLParams,
-    options?: RequestOptions,
-  ): APIPromise<DatasetUploadJSONLResponse> {
-    return this._client.post(
-      path`/datasets/${datasetID}/upload-jsonl`,
-      multipartFormRequestOptions({ body, ...options }, this._client),
-    );
-  }
 }
 
 /**
@@ -207,9 +189,20 @@ export interface DatasetGenerateValidatorsResponse {
   validators: Array<string>;
 }
 
-export type DatasetInsertEntryResponse = string;
+/**
+ * Response model for inserting an entry into a dataset.
+ */
+export interface DatasetInsertEntryResponse {
+  /**
+   * The ID of the inserted entry
+   */
+  entry_id: string;
 
-export type DatasetUploadJSONLResponse = { [key: string]: unknown };
+  /**
+   * The storage bytes of the inserted entry
+   */
+  storage_bytes: number;
+}
 
 export interface DatasetCreateParams {
   /**
@@ -288,10 +281,6 @@ export interface DatasetInsertEntryParams {
   body: { [key: string]: unknown };
 }
 
-export interface DatasetUploadJSONLParams {
-  file: Uploadable;
-}
-
 Datasets.Generations = Generations;
 
 export declare namespace Datasets {
@@ -304,15 +293,13 @@ export declare namespace Datasets {
     type DatasetGenerateSchemaResponse as DatasetGenerateSchemaResponse,
     type DatasetGenerateValidatorsResponse as DatasetGenerateValidatorsResponse,
     type DatasetInsertEntryResponse as DatasetInsertEntryResponse,
-    type DatasetUploadJSONLResponse as DatasetUploadJSONLResponse,
     type DatasetCreateParams as DatasetCreateParams,
     type DatasetUpdateParams as DatasetUpdateParams,
     type DatasetListParams as DatasetListParams,
     type DatasetGenerateDatasetParams as DatasetGenerateDatasetParams,
     type DatasetGenerateSchemaParams as DatasetGenerateSchemaParams,
     type DatasetGenerateValidatorsParams as DatasetGenerateValidatorsParams,
     type DatasetInsertEntryParams as DatasetInsertEntryParams,
-    type DatasetUploadJSONLParams as DatasetUploadJSONLParams,
   };
 
   export {

src/resources/datasets/index.ts

@@ -10,15 +10,13 @@ export {
   type DatasetGenerateSchemaResponse,
   type DatasetGenerateValidatorsResponse,
   type DatasetInsertEntryResponse,
-  type DatasetUploadJSONLResponse,
   type DatasetCreateParams,
   type DatasetUpdateParams,
   type DatasetListParams,
   type DatasetGenerateDatasetParams,
   type DatasetGenerateSchemaParams,
   type DatasetGenerateValidatorsParams,
   type DatasetInsertEntryParams,
-  type DatasetUploadJSONLParams,
 } from './datasets';
 export {
   Generations,

src/resources/index.ts

@@ -10,15 +10,13 @@ export {
   type DatasetGenerateSchemaResponse,
   type DatasetGenerateValidatorsResponse,
   type DatasetInsertEntryResponse,
-  type DatasetUploadJSONLResponse,
   type DatasetCreateParams,
   type DatasetUpdateParams,
   type DatasetListParams,
   type DatasetGenerateDatasetParams,
   type DatasetGenerateSchemaParams,
   type DatasetGenerateValidatorsParams,
   type DatasetInsertEntryParams,
-  type DatasetUploadJSONLParams,
 } from './datasets/datasets';
 export {
   Logs,

tests/api-resources/datasets/datasets.test.ts

@@ -188,25 +188,4 @@ describe('resource datasets', () => {
   test.skip('insertEntry: required and optional params', async () => {
     const response = await client.datasets.insertEntry('dataset_id', { body: { foo: 'bar' } });
   });
-
-  // Prism tests are disabled
-  test.skip('uploadJSONL: only required params', async () => {
-    const responsePromise = client.datasets.uploadJSONL('dataset_id', {
-      file: await toFile(Buffer.from('# my file contents'), 'README.md'),
-    });
-    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);
-  });
-
-  // Prism tests are disabled
-  test.skip('uploadJSONL: required and optional params', async () => {
-    const response = await client.datasets.uploadJSONL('dataset_id', {
-      file: await toFile(Buffer.from('# my file contents'), 'README.md'),
-    });
-  });
 });