Uncomment edit dataset block, add dataset limitations into endpoint descriptions (#2563)

Co-authored-by: ci.datadog-api-spec <[email protected]>

Author: api-clients-generation-pipeline[bot]

.generated-info

@@ -1,4 +1,4 @@
 {
-  "spec_repo_commit": "d51e9a8",
-  "generated": "2025-07-25 13:23:06.075"
+  "spec_repo_commit": "e4f653f",
+  "generated": "2025-07-25 14:06:59.037"
 }

.generator/schemas/v2/openapi.yaml

@@ -12478,7 +12478,14 @@ components:
       - type
       type: object
     Dataset:
-      description: Dataset object.
+      description: "Dataset object.\n\n### Datasets Constraints\n- **Tag Limit per
+        Dataset**:\n  - Each restricted dataset supports a maximum of 10 key:value
+        pairs per product.\n\n- **Tag Key Rules per Telemetry Type**:\n  - Only one
+        tag key or attribute may be used to define access within a single telemetry
+        type.\n  - The same or different tag key may be used across different telemetry
+        types.\n\n- **Tag Value Uniqueness**:\n  - Tag values must be unique within
+        a single dataset.\n  - A tag value used in one dataset cannot be reused in
+        another dataset of the same telemetry type."
       properties:
         attributes:
           $ref: '#/components/schemas/DatasetAttributes'
@@ -12556,6 +12563,14 @@ components:
       required:
       - data
       type: object
+    DatasetUpdateRequest:
+      description: Edit request for a dataset.
+      properties:
+        data:
+          $ref: '#/components/schemas/Dataset'
+      required:
+      - data
+      type: object
     Date:
       description: Date as Unix timestamp in milliseconds.
       example: 1722439510282
@@ -16083,10 +16098,9 @@ components:
           type: array
         product:
           description: 'Name of the product the dataset is for. Possible values are
-            ''apm'', ''rum'', ''synthetics'',
+            ''apm'', ''rum'',
 
-            ''metrics'', ''logs'', ''sd_repoinfo'', ''error_tracking'', ''cloud_cost'',
-            and ''ml_obs''.'
+            ''metrics'', ''logs'', ''error_tracking'', and ''cloud_cost''.'
           example: logs
           type: string
       required:
@@ -48282,6 +48296,44 @@ paths:
       x-permission:
         operator: OPEN
         permissions: []
+    put:
+      description: Edits the dataset associated with the ID.
+      operationId: UpdateDataset
+      parameters:
+      - $ref: '#/components/parameters/DatasetID'
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/DatasetUpdateRequest'
+        description: Dataset payload
+        required: true
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/DatasetResponseSingle'
+          description: OK
+        '400':
+          $ref: '#/components/responses/BadRequestResponse'
+        '403':
+          $ref: '#/components/responses/NotAuthorizedResponse'
+        '404':
+          $ref: '#/components/responses/NotFoundResponse'
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ: []
+      summary: Edit a dataset
+      tags:
+      - Datasets
+      x-codegen-request-body-name: body
+      x-permission:
+        operator: OPEN
+        permissions: []
   /api/v2/deletion/data/{product}:
     post:
       description: Creates a data deletion request by providing a query and a timeframe

features/v2/datasets.feature

@@ -53,6 +53,30 @@ Feature: Datasets
     When the request is sent
     Then the response status is 404 Not Found
 
+  @generated @skip @team:DataDog/aaa-granular-access
+  Scenario: Edit a dataset returns "Bad Request" response
+    Given new "UpdateDataset" request
+    And request contains "dataset_id" parameter from "REPLACE.ME"
+    And body with value {"data": {"attributes": {"created_at": null, "name": "Security Audit Dataset", "principals": ["role:86245fce-0a4e-11f0-92bd-da7ad0900002"], "product_filters": [{"filters": ["@application.id:ABCD"], "product": "logs"}]}, "id": "123e4567-e89b-12d3-a456-426614174000", "type": "dataset"}}
+    When the request is sent
+    Then the response status is 400 Bad Request
+
+  @generated @skip @team:DataDog/aaa-granular-access
+  Scenario: Edit a dataset returns "Not Found" response
+    Given new "UpdateDataset" request
+    And request contains "dataset_id" parameter from "REPLACE.ME"
+    And body with value {"data": {"attributes": {"created_at": null, "name": "Security Audit Dataset", "principals": ["role:86245fce-0a4e-11f0-92bd-da7ad0900002"], "product_filters": [{"filters": ["@application.id:ABCD"], "product": "logs"}]}, "id": "123e4567-e89b-12d3-a456-426614174000", "type": "dataset"}}
+    When the request is sent
+    Then the response status is 404 Not Found
+
+  @generated @skip @team:DataDog/aaa-granular-access
+  Scenario: Edit a dataset returns "OK" response
+    Given new "UpdateDataset" request
+    And request contains "dataset_id" parameter from "REPLACE.ME"
+    And body with value {"data": {"attributes": {"created_at": null, "name": "Security Audit Dataset", "principals": ["role:86245fce-0a4e-11f0-92bd-da7ad0900002"], "product_filters": [{"filters": ["@application.id:ABCD"], "product": "logs"}]}, "id": "123e4567-e89b-12d3-a456-426614174000", "type": "dataset"}}
+    When the request is sent
+    Then the response status is 200 OK
+
   @generated @skip @team:DataDog/aaa-granular-access
   Scenario: Get a single dataset by ID returns "Bad Request" response
     Given new "GetDataset" request

features/v2/undo.json

@@ -888,6 +888,12 @@
       "type": "safe"
     }
   },
+  "UpdateDataset": {
+    "tag": "Datasets",
+    "undo": {
+      "type": "idempotent"
+    }
+  },
   "CreateDataDeletionRequest": {
     "tag": "Data Deletion",
     "undo": {

private/bdd_runner/src/support/scenarios_model_mapping.ts

@@ -4590,6 +4590,17 @@ export const ScenariosModelMappings: { [key: string]: OperationMapping } = {
     },
     operationResponseType: "DatasetResponseSingle",
   },
+  "DatasetsApi.V2.UpdateDataset": {
+    datasetId: {
+      type: "string",
+      format: "",
+    },
+    body: {
+      type: "DatasetUpdateRequest",
+      format: "",
+    },
+    operationResponseType: "DatasetResponseSingle",
+  },
   "DatasetsApi.V2.DeleteDataset": {
     datasetId: {
       type: "string",

services/datasets/src/v2/DatasetsApi.ts

@@ -26,6 +26,7 @@ import { APIErrorResponse } from "./models/APIErrorResponse";
 import { DatasetCreateRequest } from "./models/DatasetCreateRequest";
 import { DatasetResponseMulti } from "./models/DatasetResponseMulti";
 import { DatasetResponseSingle } from "./models/DatasetResponseSingle";
+import { DatasetUpdateRequest } from "./models/DatasetUpdateRequest";
 import { version } from "../version";
 
 export class DatasetsApiRequestFactory extends BaseAPIRequestFactory {
@@ -213,6 +214,66 @@ export class DatasetsApiRequestFactory extends BaseAPIRequestFactory {
 
     return requestContext;
   }
+
+  public async updateDataset(
+    datasetId: string,
+    body: DatasetUpdateRequest,
+    _options?: Configuration,
+  ): Promise<RequestContext> {
+    const _config = _options || this.configuration;
+
+    // verify required parameter 'datasetId' is not null or undefined
+    if (datasetId === null || datasetId === undefined) {
+      throw new RequiredError("datasetId", "updateDataset");
+    }
+
+    // verify required parameter 'body' is not null or undefined
+    if (body === null || body === undefined) {
+      throw new RequiredError("body", "updateDataset");
+    }
+
+    // Path Params
+    const localVarPath = "/api/v2/datasets/{dataset_id}".replace(
+      "{dataset_id}",
+      encodeURIComponent(String(datasetId)),
+    );
+
+    // Make Request Context
+    const { server, overrides } = _config.getServerAndOverrides(
+      "DatasetsApi.v2.updateDataset",
+      DatasetsApi.operationServers,
+    );
+    const requestContext = server.makeRequestContext(
+      localVarPath,
+      HttpMethod.PUT,
+      overrides,
+    );
+    requestContext.setHeaderParam("Accept", "application/json");
+    requestContext.setHttpConfig(_config.httpConfig);
+
+    // Set User-Agent
+    if (this.userAgent) {
+      requestContext.setHeaderParam("User-Agent", this.userAgent);
+    }
+
+    // Body Params
+    const contentType = getPreferredMediaType(["application/json"]);
+    requestContext.setHeaderParam("Content-Type", contentType);
+    const serializedBody = stringify(
+      serialize(body, TypingInfo, "DatasetUpdateRequest", ""),
+      contentType,
+    );
+    requestContext.setBody(serializedBody);
+
+    // Apply auth methods
+    applySecurityAuthentication(_config, requestContext, [
+      "apiKeyAuth",
+      "appKeyAuth",
+      "AuthZ",
+    ]);
+
+    return requestContext;
+  }
 }
 
 export class DatasetsApiResponseProcessor {
@@ -441,6 +502,67 @@ export class DatasetsApiResponseProcessor {
       'Unknown API Status Code!\nBody: "' + body + '"',
     );
   }
+
+  /**
+   * Unwraps the actual response sent by the server from the response context and deserializes the response content
+   * to the expected objects
+   *
+   * @params response Response returned by the server for a request to updateDataset
+   * @throws ApiException if the response code was not in [200, 299]
+   */
+  public async updateDataset(
+    response: ResponseContext,
+  ): Promise<DatasetResponseSingle> {
+    const contentType = normalizeMediaType(response.headers["content-type"]);
+    if (response.httpStatusCode === 200) {
+      const body: DatasetResponseSingle = deserialize(
+        parse(await response.body.text(), contentType),
+        TypingInfo,
+        "DatasetResponseSingle",
+      ) as DatasetResponseSingle;
+      return body;
+    }
+    if (
+      response.httpStatusCode === 400 ||
+      response.httpStatusCode === 403 ||
+      response.httpStatusCode === 404 ||
+      response.httpStatusCode === 429
+    ) {
+      const bodyText = parse(await response.body.text(), contentType);
+      let body: APIErrorResponse;
+      try {
+        body = deserialize(
+          bodyText,
+          TypingInfo,
+          "APIErrorResponse",
+        ) as APIErrorResponse;
+      } catch (error) {
+        logger.debug(`Got error deserializing error: ${error}`);
+        throw new ApiException<APIErrorResponse>(
+          response.httpStatusCode,
+          bodyText,
+        );
+      }
+      throw new ApiException<APIErrorResponse>(response.httpStatusCode, body);
+    }
+
+    // Work around for missing responses in specification, e.g. for petstore.yaml
+    if (response.httpStatusCode >= 200 && response.httpStatusCode <= 299) {
+      const body: DatasetResponseSingle = deserialize(
+        parse(await response.body.text(), contentType),
+        TypingInfo,
+        "DatasetResponseSingle",
+        "",
+      ) as DatasetResponseSingle;
+      return body;
+    }
+
+    const body = (await response.body.text()) || "";
+    throw new ApiException<string>(
+      response.httpStatusCode,
+      'Unknown API Status Code!\nBody: "' + body + '"',
+    );
+  }
 }
 
 export interface DatasetsApiCreateDatasetRequest {
@@ -467,6 +589,19 @@ export interface DatasetsApiGetDatasetRequest {
   datasetId: string;
 }
 
+export interface DatasetsApiUpdateDatasetRequest {
+  /**
+   * The ID of a defined dataset.
+   * @type string
+   */
+  datasetId: string;
+  /**
+   * Dataset payload
+   * @type DatasetUpdateRequest
+   */
+  body: DatasetUpdateRequest;
+}
+
 export class DatasetsApi {
   private requestFactory: DatasetsApiRequestFactory;
   private responseProcessor: DatasetsApiResponseProcessor;
@@ -565,4 +700,26 @@ export class DatasetsApi {
         });
     });
   }
+
+  /**
+   * Edits the dataset associated with the ID.
+   * @param param The request object
+   */
+  public updateDataset(
+    param: DatasetsApiUpdateDatasetRequest,
+    options?: Configuration,
+  ): Promise<DatasetResponseSingle> {
+    const requestContextPromise = this.requestFactory.updateDataset(
+      param.datasetId,
+      param.body,
+      options,
+    );
+    return requestContextPromise.then((requestContext) => {
+      return this.configuration.httpApi
+        .send(requestContext)
+        .then((responseContext) => {
+          return this.responseProcessor.updateDataset(responseContext);
+        });
+    });
+  }
 }

services/datasets/src/v2/index.ts

@@ -2,6 +2,7 @@ export {
   DatasetsApiCreateDatasetRequest,
   DatasetsApiDeleteDatasetRequest,
   DatasetsApiGetDatasetRequest,
+  DatasetsApiUpdateDatasetRequest,
   DatasetsApi,
 } from "./DatasetsApi";
 
@@ -11,4 +12,5 @@ export { DatasetAttributes } from "./models/DatasetAttributes";
 export { DatasetCreateRequest } from "./models/DatasetCreateRequest";
 export { DatasetResponseMulti } from "./models/DatasetResponseMulti";
 export { DatasetResponseSingle } from "./models/DatasetResponseSingle";
+export { DatasetUpdateRequest } from "./models/DatasetUpdateRequest";
 export { FiltersPerProduct } from "./models/FiltersPerProduct";

services/datasets/src/v2/models/Dataset.ts

@@ -4,6 +4,18 @@ import { DatasetAttributes } from "./DatasetAttributes";
 
 /**
  * Dataset object.
+ *
+ * ### Datasets Constraints
+ * - **Tag Limit per Dataset**:
+ *   - Each restricted dataset supports a maximum of 10 key:value pairs per product.
+ *
+ * - **Tag Key Rules per Telemetry Type**:
+ *   - Only one tag key or attribute may be used to define access within a single telemetry type.
+ *   - The same or different tag key may be used across different telemetry types.
+ *
+ * - **Tag Value Uniqueness**:
+ *   - Tag values must be unique within a single dataset.
+ *   - A tag value used in one dataset cannot be reused in another dataset of the same telemetry type.
  */
 export class Dataset {
   /**

services/datasets/src/v2/models/DatasetCreateRequest.ts

@@ -8,6 +8,18 @@ import { Dataset } from "./Dataset";
 export class DatasetCreateRequest {
   /**
    * Dataset object.
+   *
+   * ### Datasets Constraints
+   * - **Tag Limit per Dataset**:
+   *   - Each restricted dataset supports a maximum of 10 key:value pairs per product.
+   *
+   * - **Tag Key Rules per Telemetry Type**:
+   *   - Only one tag key or attribute may be used to define access within a single telemetry type.
+   *   - The same or different tag key may be used across different telemetry types.
+   *
+   * - **Tag Value Uniqueness**:
+   *   - Tag values must be unique within a single dataset.
+   *   - A tag value used in one dataset cannot be reused in another dataset of the same telemetry type.
    */
   "data": Dataset;
   /**