Create or update usage quotas

"""
Create or update usage quotas returns "OK. The response includes each item's result; see each item's `error` attribute
for any that failed to write." response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
from datadog_api_client.v2.model.usage_quota_create_attributes import UsageQuotaCreateAttributes
from datadog_api_client.v2.model.usage_quota_create_data import UsageQuotaCreateData
from datadog_api_client.v2.model.usage_quota_request_scope import UsageQuotaRequestScope
from datadog_api_client.v2.model.usage_quota_type import UsageQuotaType
from datadog_api_client.v2.model.usage_quotas_create_request import UsageQuotasCreateRequest

body = UsageQuotasCreateRequest(
    data=[
        UsageQuotaCreateData(
            attributes=UsageQuotaCreateAttributes(
                enforced=True,
                scope=UsageQuotaRequestScope(
                    user_handle="jane@example.com",
                ),
                usage_limit=100000,
            ),
            type=UsageQuotaType.QUOTAS,
        ),
    ],
)

configuration = Configuration()
configuration.unstable_operations["create_quotas"] = True
with ApiClient(configuration) as api_client:
    api_instance = UsageMeteringApi(api_client)
    response = api_instance.create_quotas(quota_namespace="ai_credits", body=body)

    print(response)

# Create or update usage quotas returns "OK. The response includes each item's result; see each item's `error` attribute
# for any that failed to write." response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.unstable_operations["v2.create_quotas".to_sym] = true
end
api_instance = DatadogAPIClient::V2::UsageMeteringAPI.new

body = DatadogAPIClient::V2::UsageQuotasCreateRequest.new({
  data: [
    DatadogAPIClient::V2::UsageQuotaCreateData.new({
      attributes: DatadogAPIClient::V2::UsageQuotaCreateAttributes.new({
        enforced: true,
        scope: {
          user_handle: "jane@example.com",
        },
        usage_limit: 100000,
      }),
      type: DatadogAPIClient::V2::UsageQuotaType::QUOTAS,
    }),
  ],
})
p api_instance.create_quotas("ai_credits", body)

// Create or update usage quotas returns "OK. The response includes each item's result; see each item's `error` attribute
// for any that failed to write." 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.UsageQuotasCreateRequest{
		Data: []datadogV2.UsageQuotaCreateData{
			{
				Attributes: datadogV2.UsageQuotaCreateAttributes{
					Enforced: true,
					Scope: map[string]string{
						"user_handle": "jane@example.com",
					},
					UsageLimit: 100000,
				},
				Type: datadogV2.USAGEQUOTATYPE_QUOTAS,
			},
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.CreateQuotas", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewUsageMeteringApi(apiClient)
	resp, r, err := api.CreateQuotas(ctx, "ai_credits", body, *datadogV2.NewCreateQuotasOptionalParameters())

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

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

// Create or update usage quotas returns "OK. The response includes each item's result; see each
// item's `error` attribute
// for any that failed to write." response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.UsageMeteringApi;
import com.datadog.api.client.v2.model.UsageQuotaCreateAttributes;
import com.datadog.api.client.v2.model.UsageQuotaCreateData;
import com.datadog.api.client.v2.model.UsageQuotaType;
import com.datadog.api.client.v2.model.UsageQuotasBulkResponse;
import com.datadog.api.client.v2.model.UsageQuotasCreateRequest;
import java.util.Collections;
import java.util.Map;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setUnstableOperationEnabled("v2.createQuotas", true);
    UsageMeteringApi apiInstance = new UsageMeteringApi(defaultClient);

    UsageQuotasCreateRequest body =
        new UsageQuotasCreateRequest()
            .data(
                Collections.singletonList(
                    new UsageQuotaCreateData()
                        .attributes(
                            new UsageQuotaCreateAttributes()
                                .enforced(true)
                                .scope(Map.ofEntries(Map.entry("user_handle", "jane@example.com")))
                                .usageLimit(100000L))
                        .type(UsageQuotaType.QUOTAS)));

    try {
      UsageQuotasBulkResponse result = apiInstance.createQuotas("ai_credits", body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling UsageMeteringApi#createQuotas");
      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 usage quotas returns "OK. The response includes each item's
// result; see each item's `error` attribute for any that failed to write."
// response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_usage_metering::CreateQuotasOptionalParams;
use datadog_api_client::datadogV2::api_usage_metering::UsageMeteringAPI;
use datadog_api_client::datadogV2::model::UsageQuotaCreateAttributes;
use datadog_api_client::datadogV2::model::UsageQuotaCreateData;
use datadog_api_client::datadogV2::model::UsageQuotaType;
use datadog_api_client::datadogV2::model::UsageQuotasCreateRequest;
use std::collections::BTreeMap;

#[tokio::main]
async fn main() {
    let body = UsageQuotasCreateRequest::new(vec![UsageQuotaCreateData::new(
        UsageQuotaCreateAttributes::new(true, 100000).scope(BTreeMap::from([(
            "user_handle".to_string(),
            "jane@example.com".to_string(),
        )])),
        UsageQuotaType::QUOTAS,
    )]);
    let mut configuration = datadog::Configuration::new();
    configuration.set_unstable_operation_enabled("v2.CreateQuotas", true);
    let api = UsageMeteringAPI::with_config(configuration);
    let resp = api
        .create_quotas(
            "ai_credits".to_string(),
            body,
            CreateQuotasOptionalParams::default(),
        )
        .await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Create or update usage quotas returns "OK. The response includes each item's result; see each item's `error` attribute
 * for any that failed to write." response
 */

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

const configuration = client.createConfiguration();
configuration.unstableOperations["v2.createQuotas"] = true;
const apiInstance = new v2.UsageMeteringApi(configuration);

const params: v2.UsageMeteringApiCreateQuotasRequest = {
  body: {
    data: [
      {
        attributes: {
          enforced: true,
          scope: {
            user_handle: "jane@example.com",
          },
          usageLimit: 100000,
        },
        type: "quotas",
      },
    ],
  },
  quotaNamespace: "ai_credits",
};

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