Add OpenAPI for Cloud Cost Management custom forecast endpoints (#1773)

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

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

.generator/schemas/v2/openapi.yaml

@@ -27158,6 +27158,156 @@ components:
             $ref: "#/components/schemas/CustomDestinationResponseDefinition"
           type: array
       type: object
+    CustomForecastEntry:
+      description: A monthly entry of a custom budget forecast.
+      properties:
+        amount:
+          description: Forecast amount for the month.
+          example: 400
+          format: double
+          type: number
+        month:
+          description: Month the custom forecast entry applies to, in `YYYYMM` format.
+          example: 202501
+          format: int64
+          type: integer
+        tag_filters:
+          description: Tag filters that scope this custom forecast entry to specific resources.
+          items:
+            $ref: "#/components/schemas/CustomForecastEntryTagFilter"
+          type: array
+      required:
+        - month
+        - amount
+        - tag_filters
+      type: object
+    CustomForecastEntryTagFilter:
+      description: A tag filter that scopes a custom forecast entry to specific resource tags.
+      properties:
+        tag_key:
+          description: The tag key to filter on.
+          example: service
+          type: string
+        tag_value:
+          description: The tag value to filter on.
+          example: ec2
+          type: string
+      required:
+        - tag_key
+        - tag_value
+      type: object
+    CustomForecastResponse:
+      description: Response object containing the custom forecast for a budget.
+      properties:
+        data:
+          $ref: "#/components/schemas/CustomForecastResponseData"
+      required:
+        - data
+      type: object
+    CustomForecastResponseData:
+      description: Custom forecast resource wrapper in a response.
+      properties:
+        attributes:
+          $ref: "#/components/schemas/CustomForecastResponseDataAttributes"
+        id:
+          description: The unique identifier of the custom forecast.
+          example: 11111111-1111-1111-1111-111111111111
+          type: string
+        type:
+          $ref: "#/components/schemas/CustomForecastType"
+      required:
+        - id
+        - type
+        - attributes
+      type: object
+    CustomForecastResponseDataAttributes:
+      description: Attributes of a custom forecast.
+      properties:
+        budget_uid:
+          description: The UUID of the budget that this custom forecast belongs to.
+          example: 00000000-0000-0000-0000-000000000001
+          type: string
+        created_at:
+          description: Timestamp the custom forecast was created, in Unix milliseconds.
+          example: 1738258683590
+          format: int64
+          type: integer
+        created_by:
+          description: The id of the user that created the custom forecast.
+          example: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: string
+        entries:
+          description: Monthly custom forecast entries.
+          items:
+            $ref: "#/components/schemas/CustomForecastEntry"
+          type: array
+        updated_at:
+          description: Timestamp the custom forecast was last updated, in Unix milliseconds.
+          example: 1738258683590
+          format: int64
+          type: integer
+        updated_by:
+          description: The id of the user that last updated the custom forecast.
+          example: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: string
+      required:
+        - budget_uid
+        - created_at
+        - updated_at
+        - created_by
+        - updated_by
+        - entries
+      type: object
+    CustomForecastType:
+      default: custom_forecast
+      description: The type of the custom forecast resource. Must be `custom_forecast`.
+      enum:
+        - custom_forecast
+      example: custom_forecast
+      type: string
+      x-enum-varnames:
+        - CUSTOM_FORECAST
+    CustomForecastUpsertRequest:
+      description: Request body to upsert (create or replace) the custom forecast for a budget.
+      properties:
+        data:
+          $ref: "#/components/schemas/CustomForecastUpsertRequestData"
+      required:
+        - data
+      type: object
+    CustomForecastUpsertRequestData:
+      description: Custom forecast resource wrapper in an upsert request.
+      properties:
+        attributes:
+          $ref: "#/components/schemas/CustomForecastUpsertRequestDataAttributes"
+        id:
+          description: Unused on upsert; the resource is keyed by `budget_uid`. Send an empty string.
+          example: ""
+          type: string
+        type:
+          $ref: "#/components/schemas/CustomForecastType"
+      required:
+        - type
+        - attributes
+      type: object
+    CustomForecastUpsertRequestDataAttributes:
+      description: Attributes of a custom forecast upsert request.
+      properties:
+        budget_uid:
+          description: The UUID of the budget that this custom forecast belongs to.
+          example: 00000000-0000-0000-0000-000000000001
+          type: string
+        entries:
+          description: |-
+            Monthly custom forecast entries. An empty list deletes any existing
+            custom forecast for the budget.
+          items:
+            $ref: "#/components/schemas/CustomForecastEntry"
+          type: array
+      required:
+        - budget_uid
+        - entries
+      type: object
     CustomFrameworkControl:
       description: Framework Control.
       properties:
@@ -124191,6 +124341,82 @@ paths:
       summary: Validate CSV budget
       tags:
         - Cloud Cost Management
+  /api/v2/cost/budget/custom-forecast:
+    put:
+      description: |-
+        Create or replace the custom forecast for an existing budget.
+        Pass an empty `entries` list to delete the custom forecast for the budget.
+      operationId: UpsertCustomForecast
+      requestBody:
+        content:
+          application/json:
+            examples:
+              default:
+                value:
+                  data:
+                    attributes:
+                      budget_uid: 00000000-0000-0000-0000-000000000001
+                      entries:
+                        - amount: 400
+                          month: 202501
+                          tag_filters:
+                            - tag_key: service
+                              tag_value: ec2
+                        - amount: 450
+                          month: 202502
+                          tag_filters:
+                            - tag_key: service
+                              tag_value: ec2
+                    id: ""
+                    type: custom_forecast
+            schema:
+              $ref: "#/components/schemas/CustomForecastUpsertRequest"
+        required: true
+      responses:
+        "200":
+          content:
+            application/json:
+              examples:
+                default:
+                  value:
+                    data:
+                      attributes:
+                        budget_uid: 00000000-0000-0000-0000-000000000001
+                        created_at: 1738258683590
+                        created_by: 00000000-0a0a-0a0a-aaa0-00000000000a
+                        entries:
+                          - amount: 400
+                            month: 202501
+                            tag_filters:
+                              - tag_key: service
+                                tag_value: ec2
+                          - amount: 450
+                            month: 202502
+                            tag_filters:
+                              - tag_key: service
+                                tag_value: ec2
+                        updated_at: 1738258683590
+                        updated_by: 00000000-0a0a-0a0a-aaa0-00000000000a
+                      id: 11111111-1111-1111-1111-111111111111
+                      type: custom_forecast
+              schema:
+                $ref: "#/components/schemas/CustomForecastResponse"
+          description: OK
+        "400":
+          $ref: "#/components/responses/BadRequestResponse"
+        "404":
+          $ref: "#/components/responses/NotFoundResponse"
+        "429":
+          $ref: "#/components/responses/TooManyRequestsResponse"
+      security:
+        - apiKeyAuth: []
+          appKeyAuth: []
+      summary: Create or replace a budget's custom forecast
+      tags:
+        - Cloud Cost Management
+      x-unstable: |-
+        **Note**: This endpoint is in preview and is subject to change.
+        If you have any feedback, contact [Datadog support](https://docs.datadoghq.com/help/).
   /api/v2/cost/budget/validate:
     post:
       description: Validate a budget configuration without creating or modifying it
@@ -124313,6 +124539,30 @@ paths:
       summary: Get budget
       tags:
         - Cloud Cost Management
+  /api/v2/cost/budget/{budget_id}/custom-forecast:
+    delete:
+      description: Delete the custom forecast for a budget.
+      operationId: DeleteCustomForecast
+      parameters:
+        - $ref: "#/components/parameters/BudgetID"
+      responses:
+        "204":
+          description: No Content
+        "400":
+          $ref: "#/components/responses/BadRequestResponse"
+        "404":
+          $ref: "#/components/responses/NotFoundResponse"
+        "429":
+          $ref: "#/components/responses/TooManyRequestsResponse"
+      security:
+        - apiKeyAuth: []
+          appKeyAuth: []
+      summary: Delete a budget's custom forecast
+      tags:
+        - Cloud Cost Management
+      x-unstable: |-
+        **Note**: This endpoint is in preview and is subject to change.
+        If you have any feedback, contact [Datadog support](https://docs.datadoghq.com/help/).
   /api/v2/cost/budgets:
     get:
       description: List budgets.

examples/v2_cloud-cost-management_DeleteCustomForecast.rs

@@ -0,0 +1,16 @@
+// Delete a budget's custom forecast returns "No Content" response
+use datadog_api_client::datadog;
+use datadog_api_client::datadogV2::api_cloud_cost_management::CloudCostManagementAPI;
+
+#[tokio::main]
+async fn main() {
+    let mut configuration = datadog::Configuration::new();
+    configuration.set_unstable_operation_enabled("v2.DeleteCustomForecast", true);
+    let api = CloudCostManagementAPI::with_config(configuration);
+    let resp = api.delete_custom_forecast("budget_id".to_string()).await;
+    if let Ok(value) = resp {
+        println!("{:#?}", value);
+    } else {
+        println!("{:#?}", resp.unwrap_err());
+    }
+}

examples/v2_cloud-cost-management_UpsertCustomForecast.rs

@@ -0,0 +1,39 @@
+// Create or replace a budget's custom forecast returns "OK" response
+use datadog_api_client::datadog;
+use datadog_api_client::datadogV2::api_cloud_cost_management::CloudCostManagementAPI;
+use datadog_api_client::datadogV2::model::CustomForecastEntry;
+use datadog_api_client::datadogV2::model::CustomForecastEntryTagFilter;
+use datadog_api_client::datadogV2::model::CustomForecastType;
+use datadog_api_client::datadogV2::model::CustomForecastUpsertRequest;
+use datadog_api_client::datadogV2::model::CustomForecastUpsertRequestData;
+use datadog_api_client::datadogV2::model::CustomForecastUpsertRequestDataAttributes;
+
+#[tokio::main]
+async fn main() {
+    let body = CustomForecastUpsertRequest::new(
+        CustomForecastUpsertRequestData::new(
+            CustomForecastUpsertRequestDataAttributes::new(
+                "00000000-0000-0000-0000-000000000001".to_string(),
+                vec![CustomForecastEntry::new(
+                    400.0,
+                    202501,
+                    vec![CustomForecastEntryTagFilter::new(
+                        "service".to_string(),
+                        "ec2".to_string(),
+                    )],
+                )],
+            ),
+            CustomForecastType::CUSTOM_FORECAST,
+        )
+        .id("".to_string()),
+    );
+    let mut configuration = datadog::Configuration::new();
+    configuration.set_unstable_operation_enabled("v2.UpsertCustomForecast", true);
+    let api = CloudCostManagementAPI::with_config(configuration);
+    let resp = api.upsert_custom_forecast(body).await;
+    if let Ok(value) = resp {
+        println!("{:#?}", value);
+    } else {
+        println!("{:#?}", resp.unwrap_err());
+    }
+}

src/datadog/configuration.rs

@@ -485,6 +485,7 @@ impl Default for Configuration {
             ("v2.get_code_coverage_branch_summary".to_owned(), false),
             ("v2.get_code_coverage_commit_summary".to_owned(), false),
             ("v2.get_rule_based_view".to_owned(), false),
+            ("v2.delete_custom_forecast".to_owned(), false),
             ("v2.get_commitments_commitment_list".to_owned(), false),
             ("v2.get_commitments_coverage_scalar".to_owned(), false),
             ("v2.get_commitments_coverage_timeseries".to_owned(), false),
@@ -508,6 +509,7 @@ impl Default for Configuration {
             ("v2.list_cost_tag_metadata_months".to_owned(), false),
             ("v2.list_cost_tag_metadata_orchestrators".to_owned(), false),
             ("v2.search_cost_recommendations".to_owned(), false),
+            ("v2.upsert_custom_forecast".to_owned(), false),
             ("v2.create_ownership_feedback".to_owned(), false),
             ("v2.get_ownership_evidence".to_owned(), false),
             ("v2.get_ownership_inference".to_owned(), false),