feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 9
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-17ec32fdd85f93d924b2bab3f9af3688f619999da15a92fbcc7bf6db9f479b37.yml
-openapi_spec_hash: 72fcd3031f757ec9289b9138fb34081f
+configured_endpoints: 13
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-490b6c951dbe764981862a42c1067201cca1473d6ea2b5950d6b52246db9c455.yml
+openapi_spec_hash: 25f3fb0ed209b3ddd390b3b1df365f5f
 config_hash: 017a31b2ac0856f4d4152a5173e6fa18

README.md

@@ -29,11 +29,9 @@ const client = new Lemma({
   apiKey: process.env['LEMMA_API_KEY'], // This is the default and can be omitted
 });
 
-const response = await client.datasets.generateDataset({
-  dataset_id: 'dataset_id',
-  entries_required: 0,
-  schema: { foo: 'bar' },
-});
+const dataset = await client.datasets.retrieve('dataset_id');
+
+console.log(dataset.id);
 ```
 
 ### Request & Response types
@@ -48,12 +46,7 @@ const client = new Lemma({
   apiKey: process.env['LEMMA_API_KEY'], // This is the default and can be omitted
 });
 
-const params: Lemma.DatasetGenerateDatasetParams = {
-  dataset_id: 'dataset_id',
-  entries_required: 0,
-  schema: { foo: 'bar' },
-};
-const response: Lemma.DatasetGenerateDatasetResponse = await client.datasets.generateDataset(params);
+const dataset: Lemma.Dataset = await client.datasets.retrieve('dataset_id');
 ```
 
 Documentation for each method, request param, and response field are available in docstrings and will appear on hover in most modern editors.
@@ -95,17 +88,15 @@ a subclass of `APIError` will be thrown:
 
 <!-- prettier-ignore -->
 ```ts
-const response = await client.datasets
-  .generateDataset({ dataset_id: 'dataset_id', entries_required: 0, schema: { foo: 'bar' } })
-  .catch(async (err) => {
-    if (err instanceof Lemma.APIError) {
-      console.log(err.status); // 400
-      console.log(err.name); // BadRequestError
-      console.log(err.headers); // {server: 'nginx', ...}
-    } else {
-      throw err;
-    }
-  });
+const dataset = await client.datasets.retrieve('dataset_id').catch(async (err) => {
+  if (err instanceof Lemma.APIError) {
+    console.log(err.status); // 400
+    console.log(err.name); // BadRequestError
+    console.log(err.headers); // {server: 'nginx', ...}
+  } else {
+    throw err;
+  }
+});
 ```
 
 Error codes are as follows:
@@ -137,7 +128,7 @@ const client = new Lemma({
 });
 
 // Or, configure per-request:
-await client.datasets.generateDataset({ dataset_id: 'dataset_id', entries_required: 0, schema: { foo: 'bar' } }, {
+await client.datasets.retrieve('dataset_id', {
   maxRetries: 5,
 });
 ```
@@ -154,7 +145,7 @@ const client = new Lemma({
 });
 
 // Override per-request:
-await client.datasets.generateDataset({ dataset_id: 'dataset_id', entries_required: 0, schema: { foo: 'bar' } }, {
+await client.datasets.retrieve('dataset_id', {
   timeout: 5 * 1000,
 });
 ```
@@ -177,17 +168,13 @@ Unlike `.asResponse()` this method consumes the body, returning once it is parse
 ```ts
 const client = new Lemma();
 
-const response = await client.datasets
-  .generateDataset({ dataset_id: 'dataset_id', entries_required: 0, schema: { foo: 'bar' } })
-  .asResponse();
+const response = await client.datasets.retrieve('dataset_id').asResponse();
 console.log(response.headers.get('X-My-Header'));
 console.log(response.statusText); // access the underlying Response object
 
-const { data: response, response: raw } = await client.datasets
-  .generateDataset({ dataset_id: 'dataset_id', entries_required: 0, schema: { foo: 'bar' } })
-  .withResponse();
+const { data: dataset, response: raw } = await client.datasets.retrieve('dataset_id').withResponse();
 console.log(raw.headers.get('X-My-Header'));
-console.log(response);
+console.log(dataset.id);
 ```
 
 ### Logging
@@ -267,7 +254,7 @@ parameter. This library doesn't validate at runtime that the request matches the
 send will be sent as-is.
 
 ```ts
-client.datasets.generateDataset({
+client.datasets.retrieve({
   // ...
   // @ts-expect-error baz is not yet public
   baz: 'undocumented option',

api.md

@@ -3,12 +3,19 @@
 Types:
 
 - <code><a href="./src/resources/datasets/datasets.ts">Dataset</a></code>
+- <code><a href="./src/resources/datasets/datasets.ts">DatasetListResponse</a></code>
+- <code><a href="./src/resources/datasets/datasets.ts">DatasetDeleteResponse</a></code>
+- <code><a href="./src/resources/datasets/datasets.ts">DatasetCreateDownloadURLResponse</a></code>
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetGenerateDatasetResponse</a></code>
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetGenerateSchemaResponse</a></code>
 - <code><a href="./src/resources/datasets/datasets.ts">DatasetGenerateValidatorsResponse</a></code>
 
 Methods:
 
+- <code title="get /datasets/{dataset_id}">client.datasets.<a href="./src/resources/datasets/datasets.ts">retrieve</a>(datasetID) -> Dataset</code>
+- <code title="get /datasets/">client.datasets.<a href="./src/resources/datasets/datasets.ts">list</a>({ ...params }) -> DatasetListResponse</code>
+- <code title="delete /datasets/{dataset_id}">client.datasets.<a href="./src/resources/datasets/datasets.ts">delete</a>(datasetID) -> DatasetDeleteResponse</code>
+- <code title="get /datasets/{dataset_id}/create-download-url">client.datasets.<a href="./src/resources/datasets/datasets.ts">createDownloadURL</a>(datasetID) -> string</code>
 - <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>

packages/mcp-server/README.md

@@ -187,7 +187,7 @@ http://localhost:3000?client=cursor&capability=tool-name-length%3D40
 import { server, endpoints, init } from "lemma-mcp/server";
 
 // import a specific tool
-import generateDatasetDatasets from "lemma-mcp/tools/datasets/generate-dataset-datasets";
+import retrieveDatasets from "lemma-mcp/tools/datasets/retrieve-datasets";
 
 // initialize the server and all endpoints
 init({ server, endpoints });
@@ -212,7 +212,7 @@ const myCustomEndpoint = {
 };
 
 // initialize the server with your custom endpoints
-init({ server: myServer, endpoints: [generateDatasetDatasets, myCustomEndpoint] });
+init({ server: myServer, endpoints: [retrieveDatasets, myCustomEndpoint] });
 ```
 
 ## Available Tools
@@ -221,6 +221,23 @@ The following tools are available in this MCP server.
 
 ### Resource `datasets`:
 
+- `retrieve_datasets` (`read`): Get a dataset by ID.
+
+  Retrieves a specific dataset by its ID. The dataset must belong to the
+  specified project, which must belong to the current user's tenant.
+
+- `list_datasets` (`read`): List all datasets for the specified project.
+
+  Retrieves all datasets that belong to the specified project.
+  The project must belong to the current user's tenant.
+
+- `delete_datasets` (`write`): Delete a dataset by ID.
+- `create_download_url_datasets` (`read`): Create a download URL for a dataset.
+
+  Creates a secure, time-limited download URL that allows downloading
+  the dataset data. The URL can be used to access the dataset file
+  without requiring authentication.
+
 - `generate_dataset_datasets` (`write`): Generate a dataset using AI.
 
   Initiates the generation of a dataset based on the provided schema and

packages/mcp-server/src/tools/datasets/create-download-url-datasets.ts

@@ -0,0 +1,50 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { maybeFilter } from 'lemma-mcp/filtering';
+import { Metadata, asTextContentResult } from 'lemma-mcp/tools/types';
+
+import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import Lemma from 'lemma';
+
+export const metadata: Metadata = {
+  resource: 'datasets',
+  operation: 'read',
+  tags: [],
+  httpMethod: 'get',
+  httpPath: '/datasets/{dataset_id}/create-download-url',
+  operationId: 'create_download_url_datasets__dataset_id__create_download_url_get',
+};
+
+export const tool: Tool = {
+  name: 'create_download_url_datasets',
+  description:
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nCreate a download URL for a dataset.\n\nCreates a secure, time-limited download URL that allows downloading\nthe dataset data. The URL can be used to access the dataset file\nwithout requiring authentication.\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/dataset_create_download_url_response',\n  $defs: {\n    dataset_create_download_url_response: {\n      type: 'string',\n      title: 'Response Create Download Url Datasets  Dataset Id  Create Download Url Get'\n    }\n  }\n}\n```",
+  inputSchema: {
+    type: 'object',
+    properties: {
+      dataset_id: {
+        type: 'string',
+        title: 'Dataset Id',
+      },
+      jq_filter: {
+        type: 'string',
+        title: 'jq Filter',
+        description:
+          'A jq filter to apply to the response to include certain fields. Consult the output schema in the tool description to see the fields that are available.\n\nFor example: to include only the `name` field in every object of a results array, you can provide ".results[].name".\n\nFor more information, see the [jq documentation](https://jqlang.org/manual/).',
+      },
+    },
+    required: ['dataset_id'],
+  },
+  annotations: {
+    readOnlyHint: true,
+  },
+};
+
+export const handler = async (client: Lemma, args: Record<string, unknown> | undefined) => {
+  const { dataset_id, jq_filter, ...body } = args as any;
+  return asTextContentResult(
+    await maybeFilter(jq_filter, await client.datasets.createDownloadURL(dataset_id)),
+  );
+};
+
+export default { metadata, tool, handler };

packages/mcp-server/src/tools/datasets/delete-datasets.ts

@@ -0,0 +1,48 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { maybeFilter } from 'lemma-mcp/filtering';
+import { Metadata, asTextContentResult } from 'lemma-mcp/tools/types';
+
+import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import Lemma from 'lemma';
+
+export const metadata: Metadata = {
+  resource: 'datasets',
+  operation: 'write',
+  tags: [],
+  httpMethod: 'delete',
+  httpPath: '/datasets/{dataset_id}',
+  operationId: 'delete_dataset_datasets__dataset_id__delete',
+};
+
+export const tool: Tool = {
+  name: 'delete_datasets',
+  description:
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nDelete a dataset by ID.\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/dataset_delete_response',\n  $defs: {\n    dataset_delete_response: {\n      type: 'boolean',\n      title: 'Response Delete Dataset Datasets  Dataset Id  Delete'\n    }\n  }\n}\n```",
+  inputSchema: {
+    type: 'object',
+    properties: {
+      dataset_id: {
+        type: 'string',
+        title: 'Dataset Id',
+      },
+      jq_filter: {
+        type: 'string',
+        title: 'jq Filter',
+        description:
+          'A jq filter to apply to the response to include certain fields. Consult the output schema in the tool description to see the fields that are available.\n\nFor example: to include only the `name` field in every object of a results array, you can provide ".results[].name".\n\nFor more information, see the [jq documentation](https://jqlang.org/manual/).',
+      },
+    },
+    required: ['dataset_id'],
+  },
+  annotations: {
+    idempotentHint: true,
+  },
+};
+
+export const handler = async (client: Lemma, args: Record<string, unknown> | undefined) => {
+  const { dataset_id, jq_filter, ...body } = args as any;
+  return asTextContentResult(await maybeFilter(jq_filter, await client.datasets.delete(dataset_id)));
+};
+
+export default { metadata, tool, handler };

packages/mcp-server/src/tools/datasets/list-datasets.ts

@@ -0,0 +1,49 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { maybeFilter } from 'lemma-mcp/filtering';
+import { Metadata, asTextContentResult } from 'lemma-mcp/tools/types';
+
+import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import Lemma from 'lemma';
+
+export const metadata: Metadata = {
+  resource: 'datasets',
+  operation: 'read',
+  tags: [],
+  httpMethod: 'get',
+  httpPath: '/datasets/',
+  operationId: 'list_datasets_datasets__get',
+};
+
+export const tool: Tool = {
+  name: 'list_datasets',
+  description:
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all datasets for the specified project.\n\nRetrieves all datasets that belong to the specified project.\nThe project must belong to the current user's tenant.\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/dataset_list_response',\n  $defs: {\n    dataset_list_response: {\n      type: 'array',\n      title: 'Response List Datasets Datasets  Get',\n      items: {\n        $ref: '#/$defs/dataset'\n      }\n    },\n    dataset: {\n      type: 'object',\n      title: 'Dataset',\n      description: 'Dataset database model.',\n      properties: {\n        name: {\n          type: 'string',\n          title: 'Name'\n        },\n        projectId: {\n          type: 'string',\n          title: 'Projectid'\n        },\n        id: {\n          type: 'string',\n          title: 'Id'\n        },\n        createdAt: {\n          type: 'string',\n          title: 'Createdat',\n          description: 'Timestamp when the record was created',\n          format: 'date-time'\n        },\n        description: {\n          type: 'string',\n          title: 'Description'\n        },\n        entries: {\n          type: 'integer',\n          title: 'Entries'\n        },\n        schema: {\n          type: 'string',\n          title: 'Schema'\n        },\n        source: {\n          type: 'string',\n          title: 'DatasetSourceEnum',\n          enum: [            'UPLOADED',\n            'GENERATED'\n          ]\n        },\n        status: {\n          type: 'string',\n          title: 'DatasetStatus',\n          enum: [            'READY',\n            'EMPTY',\n            'GENERATING'\n          ]\n        },\n        storageBytes: {\n          type: 'integer',\n          title: 'Storagebytes'\n        },\n        updatedAt: {\n          type: 'string',\n          title: 'Updatedat',\n          description: 'Timestamp when the record was last updated',\n          format: 'date-time'\n        }\n      },\n      required: [        'name',\n        'projectId'\n      ]\n    }\n  }\n}\n```",
+  inputSchema: {
+    type: 'object',
+    properties: {
+      projectId: {
+        type: 'string',
+        title: 'Projectid',
+        description: 'The ID of the project',
+      },
+      jq_filter: {
+        type: 'string',
+        title: 'jq Filter',
+        description:
+          'A jq filter to apply to the response to include certain fields. Consult the output schema in the tool description to see the fields that are available.\n\nFor example: to include only the `name` field in every object of a results array, you can provide ".results[].name".\n\nFor more information, see the [jq documentation](https://jqlang.org/manual/).',
+      },
+    },
+    required: ['projectId'],
+  },
+  annotations: {
+    readOnlyHint: true,
+  },
+};
+
+export const handler = async (client: Lemma, args: Record<string, unknown> | undefined) => {
+  const { jq_filter, ...body } = args as any;
+  return asTextContentResult(await maybeFilter(jq_filter, await client.datasets.list(body)));
+};
+
+export default { metadata, tool, handler };

packages/mcp-server/src/tools/datasets/retrieve-datasets.ts

@@ -0,0 +1,48 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { maybeFilter } from 'lemma-mcp/filtering';
+import { Metadata, asTextContentResult } from 'lemma-mcp/tools/types';
+
+import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import Lemma from 'lemma';
+
+export const metadata: Metadata = {
+  resource: 'datasets',
+  operation: 'read',
+  tags: [],
+  httpMethod: 'get',
+  httpPath: '/datasets/{dataset_id}',
+  operationId: 'get_dataset_datasets__dataset_id__get',
+};
+
+export const tool: Tool = {
+  name: 'retrieve_datasets',
+  description:
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nGet a dataset by ID.\n\nRetrieves a specific dataset by its ID. The dataset must belong to the\nspecified project, which must belong to the current user's tenant.\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/dataset',\n  $defs: {\n    dataset: {\n      type: 'object',\n      title: 'Dataset',\n      description: 'Dataset database model.',\n      properties: {\n        name: {\n          type: 'string',\n          title: 'Name'\n        },\n        projectId: {\n          type: 'string',\n          title: 'Projectid'\n        },\n        id: {\n          type: 'string',\n          title: 'Id'\n        },\n        createdAt: {\n          type: 'string',\n          title: 'Createdat',\n          description: 'Timestamp when the record was created',\n          format: 'date-time'\n        },\n        description: {\n          type: 'string',\n          title: 'Description'\n        },\n        entries: {\n          type: 'integer',\n          title: 'Entries'\n        },\n        schema: {\n          type: 'string',\n          title: 'Schema'\n        },\n        source: {\n          type: 'string',\n          title: 'DatasetSourceEnum',\n          enum: [            'UPLOADED',\n            'GENERATED'\n          ]\n        },\n        status: {\n          type: 'string',\n          title: 'DatasetStatus',\n          enum: [            'READY',\n            'EMPTY',\n            'GENERATING'\n          ]\n        },\n        storageBytes: {\n          type: 'integer',\n          title: 'Storagebytes'\n        },\n        updatedAt: {\n          type: 'string',\n          title: 'Updatedat',\n          description: 'Timestamp when the record was last updated',\n          format: 'date-time'\n        }\n      },\n      required: [        'name',\n        'projectId'\n      ]\n    }\n  }\n}\n```",
+  inputSchema: {
+    type: 'object',
+    properties: {
+      dataset_id: {
+        type: 'string',
+        title: 'Dataset Id',
+      },
+      jq_filter: {
+        type: 'string',
+        title: 'jq Filter',
+        description:
+          'A jq filter to apply to the response to include certain fields. Consult the output schema in the tool description to see the fields that are available.\n\nFor example: to include only the `name` field in every object of a results array, you can provide ".results[].name".\n\nFor more information, see the [jq documentation](https://jqlang.org/manual/).',
+      },
+    },
+    required: ['dataset_id'],
+  },
+  annotations: {
+    readOnlyHint: true,
+  },
+};
+
+export const handler = async (client: Lemma, args: Record<string, unknown> | undefined) => {
+  const { dataset_id, jq_filter, ...body } = args as any;
+  return asTextContentResult(await maybeFilter(jq_filter, await client.datasets.retrieve(dataset_id)));
+};
+
+export default { metadata, tool, handler };