Specify accepted types for UpsertRows values (#3218)

Co-authored-by: ci.datadog-api-spec <packages@datadoghq.com>

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

.generator/schemas/v2/openapi.yaml

@@ -7075,19 +7075,27 @@ components:
     BatchUpsertRowsRequestDataAttributes:
       description: Attributes containing row data values for row creation or update
         operations.
+      example:
+        values: {}
       properties:
         values:
           additionalProperties:
-            x-required-field: true
-          description: Key-value pairs representing row data, where keys are field
-            names from the schema.
-          example:
-            example_key_value: primary_key_value
-            name: row_name
+            $ref: '#/components/schemas/BatchUpsertRowsRequestDataAttributesValue'
+          description: Key-value pairs representing row data, where keys are schema
+            field names and values match the corresponding column types.
           type: object
       required:
       - values
       type: object
+    BatchUpsertRowsRequestDataAttributesValue:
+      description: Types allowed for Reference Table row values.
+      oneOf:
+      - example: row_name
+        type: string
+      - example: 25
+        format: int32
+        maximum: 2147483647
+        type: integer
     BillConfig:
       description: Bill config.
       properties:
@@ -79483,6 +79491,18 @@ paths:
       requestBody:
         content:
           application/json:
+            examples:
+              happy_path:
+                summary: Upsert a row with mixed string and int values
+                value:
+                  data:
+                  - attributes:
+                      values:
+                        age: 25
+                        example_key_value: primary_key_value
+                        name: row_name
+                    id: primary_key_value
+                    type: row
             schema:
               $ref: '#/components/schemas/BatchUpsertRowsRequestArray'
         required: true

features/v2/reference_tables.feature

@@ -148,22 +148,22 @@ Feature: Reference Tables
   Scenario: Upsert rows returns "Bad Request" response
     Given new "UpsertRows" request
     And request contains "id" parameter from "REPLACE.ME"
-    And body with value {"data": [{"attributes": {"values": {"example_key_value": "primary_key_value", "name": "row_name"}}, "id": "primary_key_value", "type": "row"}]}
+    And body with value {"data": [{"attributes": {"values": {}}, "id": "primary_key_value", "type": "row"}]}
     When the request is sent
     Then the response status is 400 Bad Request
 
   @generated @skip @team:DataDog/redapl-experiences
   Scenario: Upsert rows returns "Not Found" response
     Given new "UpsertRows" request
     And request contains "id" parameter from "REPLACE.ME"
-    And body with value {"data": [{"attributes": {"values": {"example_key_value": "primary_key_value", "name": "row_name"}}, "id": "primary_key_value", "type": "row"}]}
+    And body with value {"data": [{"attributes": {"values": {}}, "id": "primary_key_value", "type": "row"}]}
     When the request is sent
     Then the response status is 404 Not Found
 
   @generated @skip @team:DataDog/redapl-experiences
   Scenario: Upsert rows returns "Rows created or updated successfully" response
     Given new "UpsertRows" request
     And request contains "id" parameter from "REPLACE.ME"
-    And body with value {"data": [{"attributes": {"values": {"example_key_value": "primary_key_value", "name": "row_name"}}, "id": "primary_key_value", "type": "row"}]}
+    And body with value {"data": [{"attributes": {"values": {}}, "id": "primary_key_value", "type": "row"}]}
     When the request is sent
     Then the response status is 200 Rows created or updated successfully

services/reference_tables/src/v2/index.ts

@@ -17,6 +17,7 @@ export { BatchDeleteRowsRequestData } from "./models/BatchDeleteRowsRequestData"
 export { BatchUpsertRowsRequestArray } from "./models/BatchUpsertRowsRequestArray";
 export { BatchUpsertRowsRequestData } from "./models/BatchUpsertRowsRequestData";
 export { BatchUpsertRowsRequestDataAttributes } from "./models/BatchUpsertRowsRequestDataAttributes";
+export { BatchUpsertRowsRequestDataAttributesValue } from "./models/BatchUpsertRowsRequestDataAttributesValue";
 export { CreateTableRequest } from "./models/CreateTableRequest";
 export { CreateTableRequestData } from "./models/CreateTableRequestData";
 export { CreateTableRequestDataAttributes } from "./models/CreateTableRequestDataAttributes";

services/reference_tables/src/v2/models/BatchUpsertRowsRequestDataAttributes.ts

@@ -1,13 +1,15 @@
 import { AttributeTypeMap } from "@datadog/datadog-api-client";
 
+import { BatchUpsertRowsRequestDataAttributesValue } from "./BatchUpsertRowsRequestDataAttributesValue";
+
 /**
  * Attributes containing row data values for row creation or update operations.
  */
 export class BatchUpsertRowsRequestDataAttributes {
   /**
-   * Key-value pairs representing row data, where keys are field names from the schema.
+   * Key-value pairs representing row data, where keys are schema field names and values match the corresponding column types.
    */
-  "values": { [key: string]: any };
+  "values": { [key: string]: BatchUpsertRowsRequestDataAttributesValue };
   /**
    * A container for additional, undeclared properties.
    * This is a holder for any undeclared properties as specified with
@@ -25,7 +27,7 @@ export class BatchUpsertRowsRequestDataAttributes {
   static readonly attributeTypeMap: AttributeTypeMap = {
     values: {
       baseName: "values",
-      type: "{ [key: string]: any; }",
+      type: "{ [key: string]: BatchUpsertRowsRequestDataAttributesValue; }",
       required: true,
     },
     additionalProperties: {

services/reference_tables/src/v2/models/BatchUpsertRowsRequestDataAttributesValue.ts

@@ -0,0 +1,9 @@
+import { UnparsedObject } from "@datadog/datadog-api-client";
+
+/**
+ * Types allowed for Reference Table row values.
+ */
+export type BatchUpsertRowsRequestDataAttributesValue =
+  | string
+  | number
+  | UnparsedObject;

services/reference_tables/src/v2/models/TypingInfo.ts

@@ -90,6 +90,7 @@ export const TypingInfo: ModelTypingInfo = {
     TableRowResourceDataType: ["row"],
   },
   oneOfMap: {
+    BatchUpsertRowsRequestDataAttributesValue: ["string", "number"],
     CreateTableRequestDataAttributesFileMetadata: [
       "CreateTableRequestDataAttributesFileMetadataCloudStorage",
       "CreateTableRequestDataAttributesFileMetadataLocalFile",