Add budgeting endpoints (#2532)

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

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

.apigentools-info

@@ -4,13 +4,13 @@
     "spec_versions": {
         "v1": {
             "apigentools_version": "1.6.6",
-            "regenerated": "2025-04-29 18:57:43.853175",
-            "spec_repo_commit": "d1252b21"
+            "regenerated": "2025-04-30 13:01:34.246044",
+            "spec_repo_commit": "ca363d75"
         },
         "v2": {
             "apigentools_version": "1.6.6",
-            "regenerated": "2025-04-29 18:57:43.868755",
-            "spec_repo_commit": "d1252b21"
+            "regenerated": "2025-04-30 13:01:34.261200",
+            "spec_repo_commit": "ca363d75"
         }
     }
 }

.generator/schemas/v2/openapi.yaml

@@ -186,6 +186,13 @@ components:
       required: true
       schema:
         type: string
+    BudgetID:
+      description: Budget id.
+      in: path
+      name: budget_id
+      required: true
+      schema:
+        type: string
     CaseIDPathParameter:
       description: Case's UUID or key
       example: f98a5a5b-e0ff-45d4-b2f5-afe6e74de504
@@ -4720,6 +4727,135 @@ components:
         data:
           $ref: '#/components/schemas/BillingDimensionsMappingBody'
       type: object
+    Budget:
+      description: A budget.
+      properties:
+        attributes:
+          $ref: '#/components/schemas/BudgetAttributes'
+        id:
+          description: The id of the budget.
+          type: string
+        type:
+          description: The type of the object, must be `budget`.
+          type: string
+      type: object
+    BudgetArray:
+      description: An array of budgets.
+      example:
+        data:
+        - attributes:
+            created_at: 1741011342772
+            created_by: user1
+            end_month: 202502
+            metrics_query: aws.cost.amortized{service:ec2} by {service}
+            name: my budget
+            org_id: 123
+            start_month: 202501
+            total_amount: 1000
+            updated_at: 1741011342772
+            updated_by: user2
+          id: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: budget
+      properties:
+        data:
+          description: The `BudgetArray` `data`.
+          items:
+            $ref: '#/components/schemas/Budget'
+          type: array
+      type: object
+    BudgetAttributes:
+      description: The attributes of a budget.
+      properties:
+        created_at:
+          description: The timestamp when the budget was created.
+          example: 1738258683590
+          format: int64
+          type: integer
+        created_by:
+          description: The id of the user that created the budget.
+          example: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: string
+        end_month:
+          description: The month when the budget ends.
+          example: 202502
+          format: int64
+          type: integer
+        entries:
+          description: The entries of the budget.
+          items:
+            $ref: '#/components/schemas/BudgetEntry'
+          type: array
+        metrics_query:
+          description: The cost query used to track against the budget.
+          example: aws.cost.amortized{service:ec2} by {service}
+          type: string
+        name:
+          description: The name of the budget.
+          example: my budget
+          type: string
+        org_id:
+          description: The id of the org the budget belongs to.
+          example: 123
+          format: int64
+          type: integer
+        start_month:
+          description: The month when the budget starts.
+          example: 202501
+          format: int64
+          type: integer
+        total_amount:
+          description: The sum of all budget entries' amounts.
+          example: 1000
+          format: double
+          type: number
+        updated_at:
+          description: The timestamp when the budget was last updated.
+          example: 1738258683590
+          format: int64
+          type: integer
+        updated_by:
+          description: The id of the user that created the budget.
+          example: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: string
+      type: object
+    BudgetEntry:
+      description: The entry of a budget.
+      properties:
+        amount:
+          description: The `amount` of the budget entry.
+          example: 500
+          format: double
+          type: number
+        month:
+          description: The `month` of the budget entry.
+          example: 202501
+          format: int64
+          type: integer
+        tag_filters:
+          description: The `tag_filters` of the budget entry.
+          items:
+            $ref: '#/components/schemas/TagFilter'
+          type: array
+      type: object
+    BudgetWithEntries:
+      description: The definition of the `BudgetWithEntries` object.
+      properties:
+        data:
+          $ref: '#/components/schemas/BudgetWithEntriesData'
+      type: object
+    BudgetWithEntriesData:
+      description: A budget and all its entries.
+      properties:
+        attributes:
+          $ref: '#/components/schemas/BudgetAttributes'
+        id:
+          description: The `BudgetWithEntriesData` `id`.
+          example: 00000000-0a0a-0a0a-aaa0-00000000000a
+          type: string
+        type:
+          description: The type of the object, must be `budget`.
+          type: string
+      type: object
     BulkMuteFindingsRequest:
       description: The new bulk mute finding request.
       properties:
@@ -36898,6 +37034,18 @@ components:
           format: double
           type: number
       type: object
+    TagFilter:
+      description: Tag filter for the budget's entries.
+      properties:
+        tag_key:
+          description: The key of the tag.
+          example: service
+          type: string
+        tag_value:
+          description: The value of the tag.
+          example: ec2
+          type: string
+      type: object
     TagsEventAttribute:
       description: Array of tags associated with your event.
       example:
@@ -43463,6 +43611,105 @@ paths:
         operator: OR
         permissions:
         - cloud_cost_management_write
+  /api/v2/cost/budget:
+    put:
+      description: Create a new budget or update an existing one.
+      operationId: UpsertBudget
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/BudgetWithEntries'
+        required: true
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/BudgetWithEntries'
+          description: OK
+        '400':
+          $ref: '#/components/responses/BadRequestResponse'
+        '404':
+          $ref: '#/components/responses/NotFoundResponse'
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ:
+        - cloud_cost_management_write
+      summary: Create or update a budget
+      tags:
+      - Cloud Cost Management
+  /api/v2/cost/budget/{budget_id}:
+    delete:
+      description: Delete a budget.
+      operationId: DeleteBudget
+      parameters:
+      - $ref: '#/components/parameters/BudgetID'
+      responses:
+        '204':
+          description: No Content
+        '400':
+          $ref: '#/components/responses/BadRequestResponse'
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ:
+        - cloud_cost_management_write
+      summary: Delete a budget
+      tags:
+      - Cloud Cost Management
+    get:
+      description: Get a budget.
+      operationId: GetBudget
+      parameters:
+      - $ref: '#/components/parameters/BudgetID'
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/BudgetWithEntries'
+          description: OK
+        '400':
+          $ref: '#/components/responses/BadRequestResponse'
+        '404':
+          $ref: '#/components/responses/NotFoundResponse'
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ:
+        - cloud_cost_management_read
+      summary: Get a budget
+      tags:
+      - Cloud Cost Management
+  /api/v2/cost/budgets:
+    get:
+      description: List budgets.
+      operationId: ListBudgets
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/BudgetArray'
+          description: OK
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ:
+        - cloud_cost_management_read
+      summary: List budgets
+      tags:
+      - Cloud Cost Management
   /api/v2/cost/custom_costs:
     get:
       description: List the Custom Costs files.

docs/datadog_api_client.v2.model.rst

@@ -1796,6 +1796,48 @@ datadog\_api\_client.v2.model.billing\_dimensions\_mapping\_response module
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.model.budget module
+-------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.budget\_array module
+--------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget_array
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.budget\_attributes module
+-------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.budget\_entry module
+--------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget_entry
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.budget\_with\_entries module
+----------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget_with_entries
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.budget\_with\_entries\_data module
+----------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.budget_with_entries_data
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.model.bulk\_mute\_findings\_request module
 ------------------------------------------------------------------
 
@@ -16020,6 +16062,13 @@ datadog\_api\_client.v2.model.step\_display\_bounds module
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.model.tag\_filter module
+------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.tag_filter
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.model.tags\_event\_attribute module
 -----------------------------------------------------------
 

examples/v2/cloud-cost-management/DeleteBudget.py

@@ -0,0 +1,13 @@
+"""
+Delete a budget returns "No Content" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    api_instance.delete_budget(
+        budget_id="budget_id",
+    )

examples/v2/cloud-cost-management/GetBudget.py

@@ -0,0 +1,15 @@
+"""
+Get a budget returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    response = api_instance.get_budget(
+        budget_id="budget_id",
+    )
+
+    print(response)