Create or update a budget

// Create or update a budget returns "OK" response

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)

func main() {
	body := datadogV2.BudgetWithEntries{
		Data: &datadogV2.BudgetWithEntriesData{
			Attributes: &datadogV2.BudgetAttributes{
				Costs: &datadogV2.BudgetAttributesCosts{
					Actual:       *datadog.NewNullableFloat64(nil),
					Amount:       *datadog.NewNullableFloat64(nil),
					Forecast:     *datadog.NewNullableFloat64(nil),
					OotbForecast: *datadog.NewNullableFloat64(nil),
				},
				CostsUnit: &datadogV2.BudgetAttributesCostsUnit{
					Id: datadog.PtrInt64(42),
				},
				CreatedAt: datadog.PtrInt64(1738258683590),
				CreatedBy: datadog.PtrString("00000000-0a0a-0a0a-aaa0-00000000000a"),
				EndMonth:  datadog.PtrInt64(202502),
				Entries: []datadogV2.BudgetWithEntriesDataAttributesEntriesItems{
					{
						Costs: &datadogV2.BudgetWithEntriesDataAttributesEntriesItemsCosts{
							Actual:         *datadog.NewNullableFloat64(nil),
							Amount:         *datadog.NewNullableFloat64(nil),
							CustomForecast: *datadog.NewNullableFloat64(nil),
							Forecast:       *datadog.NewNullableFloat64(nil),
							OotbForecast:   *datadog.NewNullableFloat64(nil),
						},
						TagFilters: []datadogV2.BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems{
							{},
						},
					},
				},
				MetricsQuery: datadog.PtrString("aws.cost.amortized{service:ec2} by {service}"),
				Name:         datadog.PtrString("my budget"),
				OrgId:        datadog.PtrInt64(123),
				StartMonth:   datadog.PtrInt64(202501),
				Tags: []string{
					"service",
				},
				TotalAmount: datadog.PtrFloat64(1000),
				UpdatedAt:   datadog.PtrInt64(1738258683590),
				UpdatedBy:   datadog.PtrString("00000000-0a0a-0a0a-aaa0-00000000000a"),
			},
			Id:   datadog.PtrString("00000000-0a0a-0a0a-aaa0-00000000000a"),
			Type: datadog.PtrString(""),
		},
		Meta: &datadogV2.BudgetWithEntriesMeta{
			Error: "",
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewCloudCostManagementApi(apiClient)
	resp, r, err := api.UpsertBudget(ctx, body)

	if err != nil {
		fmt.Fprintf(os.Stderr, "Error when calling `CloudCostManagementApi.UpsertBudget`: %v\n", err)
		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
	}

	responseContent, _ := json.MarshalIndent(resp, "", "  ")
	fmt.Fprintf(os.Stdout, "Response from `CloudCostManagementApi.UpsertBudget`:\n%s\n", responseContent)
}

// Create or update a budget returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.CloudCostManagementApi;
import com.datadog.api.client.v2.model.BudgetAttributes;
import com.datadog.api.client.v2.model.BudgetAttributesCosts;
import com.datadog.api.client.v2.model.BudgetAttributesCostsUnit;
import com.datadog.api.client.v2.model.BudgetWithEntries;
import com.datadog.api.client.v2.model.BudgetWithEntriesData;
import com.datadog.api.client.v2.model.BudgetWithEntriesDataAttributesEntriesItems;
import com.datadog.api.client.v2.model.BudgetWithEntriesDataAttributesEntriesItemsCosts;
import com.datadog.api.client.v2.model.BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems;
import com.datadog.api.client.v2.model.BudgetWithEntriesMeta;
import java.util.Collections;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setAccessToken(System.getenv("DD_BEARER_TOKEN"));
    CloudCostManagementApi apiInstance = new CloudCostManagementApi(defaultClient);

    BudgetWithEntries body =
        new BudgetWithEntries()
            .data(
                new BudgetWithEntriesData()
                    .attributes(
                        new BudgetAttributes()
                            .costs(
                                new BudgetAttributesCosts()
                                    .actual(null)
                                    .amount(null)
                                    .forecast(null)
                                    .ootbForecast(null))
                            .costsUnit(new BudgetAttributesCostsUnit().id(42L))
                            .createdAt(1738258683590L)
                            .createdBy("00000000-0a0a-0a0a-aaa0-00000000000a")
                            .endMonth(202502L)
                            .entries(
                                Collections.singletonList(
                                    new BudgetWithEntriesDataAttributesEntriesItems()
                                        .costs(
                                            new BudgetWithEntriesDataAttributesEntriesItemsCosts()
                                                .actual(null)
                                                .amount(null)
                                                .customForecast(null)
                                                .forecast(null)
                                                .ootbForecast(null))
                                        .tagFilters(
                                            Collections.singletonList(
                                                new BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems()))))
                            .metricsQuery("aws.cost.amortized{service:ec2} by {service}")
                            .name("my budget")
                            .orgId(123L)
                            .startMonth(202501L)
                            .tags(Collections.singletonList("service"))
                            .totalAmount(1000.0)
                            .updatedAt(1738258683590L)
                            .updatedBy("00000000-0a0a-0a0a-aaa0-00000000000a"))
                    .id("00000000-0a0a-0a0a-aaa0-00000000000a")
                    .type(""))
            .meta(new BudgetWithEntriesMeta().error(""));

    try {
      BudgetWithEntries result = apiInstance.upsertBudget(body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling CloudCostManagementApi#upsertBudget");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

"""
Create or update a budget returns "OK" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
from datadog_api_client.v2.model.budget_attributes import BudgetAttributes
from datadog_api_client.v2.model.budget_attributes_costs import BudgetAttributesCosts
from datadog_api_client.v2.model.budget_attributes_costs_unit import BudgetAttributesCostsUnit
from datadog_api_client.v2.model.budget_with_entries import BudgetWithEntries
from datadog_api_client.v2.model.budget_with_entries_data import BudgetWithEntriesData
from datadog_api_client.v2.model.budget_with_entries_data_attributes_entries_items import (
    BudgetWithEntriesDataAttributesEntriesItems,
)
from datadog_api_client.v2.model.budget_with_entries_data_attributes_entries_items_costs import (
    BudgetWithEntriesDataAttributesEntriesItemsCosts,
)
from datadog_api_client.v2.model.budget_with_entries_data_attributes_entries_items_tag_filters_items import (
    BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems,
)
from datadog_api_client.v2.model.budget_with_entries_meta import BudgetWithEntriesMeta

body = BudgetWithEntries(
    data=BudgetWithEntriesData(
        attributes=BudgetAttributes(
            costs=BudgetAttributesCosts(
                actual=None,
                amount=None,
                forecast=None,
                ootb_forecast=None,
            ),
            costs_unit=BudgetAttributesCostsUnit(
                id=42,
            ),
            created_at=1738258683590,
            created_by="00000000-0a0a-0a0a-aaa0-00000000000a",
            end_month=202502,
            entries=[
                BudgetWithEntriesDataAttributesEntriesItems(
                    costs=BudgetWithEntriesDataAttributesEntriesItemsCosts(
                        actual=None,
                        amount=None,
                        custom_forecast=None,
                        forecast=None,
                        ootb_forecast=None,
                    ),
                    tag_filters=[
                        BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems(),
                    ],
                ),
            ],
            metrics_query="aws.cost.amortized{service:ec2} by {service}",
            name="my budget",
            org_id=123,
            start_month=202501,
            tags=[
                "service",
            ],
            total_amount=1000.0,
            updated_at=1738258683590,
            updated_by="00000000-0a0a-0a0a-aaa0-00000000000a",
        ),
        id="00000000-0a0a-0a0a-aaa0-00000000000a",
        type="",
    ),
    meta=BudgetWithEntriesMeta(
        error="",
    ),
)

configuration = Configuration()
configuration.access_token = environ["DD_BEARER_TOKEN"]
with ApiClient(configuration) as api_client:
    api_instance = CloudCostManagementApi(api_client)
    response = api_instance.upsert_budget(body=body)

    print(response)

# Create or update a budget returns "OK" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.access_token = ENV["DD_BEARER_TOKEN"]
end
api_instance = DatadogAPIClient::V2::CloudCostManagementAPI.new

body = DatadogAPIClient::V2::BudgetWithEntries.new({
  data: DatadogAPIClient::V2::BudgetWithEntriesData.new({
    attributes: DatadogAPIClient::V2::BudgetAttributes.new({
      costs: DatadogAPIClient::V2::BudgetAttributesCosts.new({
        actual: nil,
        amount: nil,
        forecast: nil,
        ootb_forecast: nil,
      }),
      costs_unit: DatadogAPIClient::V2::BudgetAttributesCostsUnit.new({
        id: 42,
      }),
      created_at: 1738258683590,
      created_by: "00000000-0a0a-0a0a-aaa0-00000000000a",
      end_month: 202502,
      entries: [
        DatadogAPIClient::V2::BudgetWithEntriesDataAttributesEntriesItems.new({
          costs: DatadogAPIClient::V2::BudgetWithEntriesDataAttributesEntriesItemsCosts.new({
            actual: nil,
            amount: nil,
            custom_forecast: nil,
            forecast: nil,
            ootb_forecast: nil,
          }),
          tag_filters: [
            DatadogAPIClient::V2::BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems.new({}),
          ],
        }),
      ],
      metrics_query: "aws.cost.amortized{service:ec2} by {service}",
      name: "my budget",
      org_id: 123,
      start_month: 202501,
      tags: [
        "service",
      ],
      total_amount: 1000,
      updated_at: 1738258683590,
      updated_by: "00000000-0a0a-0a0a-aaa0-00000000000a",
    }),
    id: "00000000-0a0a-0a0a-aaa0-00000000000a",
    type: "",
  }),
  meta: DatadogAPIClient::V2::BudgetWithEntriesMeta.new({
    error: "",
  }),
})
p api_instance.upsert_budget(body)

// Create or update a budget returns "OK" response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_cloud_cost_management::CloudCostManagementAPI;
use datadog_api_client::datadogV2::model::BudgetAttributes;
use datadog_api_client::datadogV2::model::BudgetAttributesCosts;
use datadog_api_client::datadogV2::model::BudgetAttributesCostsUnit;
use datadog_api_client::datadogV2::model::BudgetWithEntries;
use datadog_api_client::datadogV2::model::BudgetWithEntriesData;
use datadog_api_client::datadogV2::model::BudgetWithEntriesDataAttributesEntriesItems;
use datadog_api_client::datadogV2::model::BudgetWithEntriesDataAttributesEntriesItemsCosts;
use datadog_api_client::datadogV2::model::BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems;
use datadog_api_client::datadogV2::model::BudgetWithEntriesMeta;

#[tokio::main]
async fn main() {
    let body = BudgetWithEntries::new()
        .data(
            BudgetWithEntriesData::new()
                .attributes(
                    BudgetAttributes::new()
                        .costs(
                            BudgetAttributesCosts::new()
                                .actual(None)
                                .amount(None)
                                .forecast(None)
                                .ootb_forecast(None),
                        )
                        .costs_unit(BudgetAttributesCostsUnit::new().id(42))
                        .created_at(1738258683590)
                        .created_by("00000000-0a0a-0a0a-aaa0-00000000000a".to_string())
                        .end_month(202502)
                        .entries(vec![BudgetWithEntriesDataAttributesEntriesItems::new()
                            .costs(
                                BudgetWithEntriesDataAttributesEntriesItemsCosts::new()
                                    .actual(None)
                                    .amount(None)
                                    .custom_forecast(None)
                                    .forecast(None)
                                    .ootb_forecast(None),
                            )
                            .tag_filters(vec![
                                BudgetWithEntriesDataAttributesEntriesItemsTagFiltersItems::new(),
                            ])])
                        .metrics_query("aws.cost.amortized{service:ec2} by {service}".to_string())
                        .name("my budget".to_string())
                        .org_id(123)
                        .start_month(202501)
                        .tags(vec!["service".to_string()])
                        .total_amount(1000.0 as f64)
                        .updated_at(1738258683590)
                        .updated_by("00000000-0a0a-0a0a-aaa0-00000000000a".to_string()),
                )
                .id("00000000-0a0a-0a0a-aaa0-00000000000a".to_string())
                .type_("".to_string()),
        )
        .meta(BudgetWithEntriesMeta::new("".to_string()));
    let configuration = datadog::Configuration::new();
    let api = CloudCostManagementAPI::with_config(configuration);
    let resp = api.upsert_budget(body).await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Create or update a budget returns "OK" response
 */

import { client, v2 } from "@datadog/datadog-api-client";

const configuration = client.createConfiguration({
  authMethods: {
    AuthZ: {
      accessToken: process.env.DD_BEARER_TOKEN as string,
    },
  },
});
const apiInstance = new v2.CloudCostManagementApi(configuration);

const params: v2.CloudCostManagementApiUpsertBudgetRequest = {
  body: {
    data: {
      attributes: {
        costs: {
          actual: undefined,
          amount: undefined,
          forecast: undefined,
          ootbForecast: undefined,
        },
        costsUnit: {
          id: 42,
        },
        createdAt: 1738258683590,
        createdBy: "00000000-0a0a-0a0a-aaa0-00000000000a",
        endMonth: 202502,
        entries: [
          {
            costs: {
              actual: undefined,
              amount: undefined,
              customForecast: undefined,
              forecast: undefined,
              ootbForecast: undefined,
            },
            tagFilters: [{}],
          },
        ],
        metricsQuery: "aws.cost.amortized{service:ec2} by {service}",
        name: "my budget",
        orgId: 123,
        startMonth: 202501,
        tags: ["service"],
        totalAmount: 1000,
        updatedAt: 1738258683590,
        updatedBy: "00000000-0a0a-0a0a-aaa0-00000000000a",
      },
      id: "00000000-0a0a-0a0a-aaa0-00000000000a",
      type: "",
    },
    meta: {
      error: "",
    },
  },
};

apiInstance
  .upsertBudget(params)
  .then((data: v2.BudgetWithEntries) => {
    console.log(
      "API called successfully. Returned data: " + JSON.stringify(data)
    );
  })
  .catch((error: any) => console.error(error));