Publish temporary AI-credit grant request APIs in preview

examples/v2/usage-metering/CreateTemporaryGrantRequest.py

@@ -0,0 +1,33 @@
+"""
+Create a temporary grant request returns "Created" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+from datadog_api_client.v2.model.temporary_grant_request_create_attributes import TemporaryGrantRequestCreateAttributes
+from datadog_api_client.v2.model.temporary_grant_request_create_data import TemporaryGrantRequestCreateData
+from datadog_api_client.v2.model.temporary_grant_request_create_request import TemporaryGrantRequestCreateRequest
+from datadog_api_client.v2.model.temporary_grant_request_type import TemporaryGrantRequestType
+from datetime import datetime
+from dateutil.tz import tzutc
+
+body = TemporaryGrantRequestCreateRequest(
+    data=TemporaryGrantRequestCreateData(
+        attributes=TemporaryGrantRequestCreateAttributes(
+            expires_at=datetime(2028, 2, 16, 0, 0, tzinfo=tzutc()),
+            justification="Credits needed to finish the incident investigation.",
+            total=200,
+        ),
+        type=TemporaryGrantRequestType.TEMPORARY_GRANT_REQUESTS,
+    ),
+)
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["create_temporary_grant_request"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.create_temporary_grant_request(body=body)
+
+    print(response)

examples/v2/usage-metering/DeleteRequestedTemporaryGrant.py

@@ -0,0 +1,17 @@
+"""
+Revoke a requested temporary grant returns "No Content" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+from uuid import UUID
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["delete_requested_temporary_grant"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    api_instance.delete_requested_temporary_grant(
+        request_id=UUID("12345678-1234-1234-1234-123456789012"),
+    )

examples/v2/usage-metering/DeleteTemporaryGrantRequest.py

@@ -0,0 +1,17 @@
+"""
+Cancel your temporary grant request returns "No Content" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+from uuid import UUID
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["delete_temporary_grant_request"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    api_instance.delete_temporary_grant_request(
+        request_id=UUID("12345678-1234-1234-1234-123456789012"),
+    )

examples/v2/usage-metering/GetMyTemporaryGrantRequest.py

@@ -0,0 +1,21 @@
+"""
+Get your temporary grant request returns "OK" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+# there is a valid "temporary_grant_request" in the system
+TEMPORARY_GRANT_REQUEST_DATA_ID = environ["TEMPORARY_GRANT_REQUEST_DATA_ID"]
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["get_my_temporary_grant_request"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.get_my_temporary_grant_request(
+        request_id=TEMPORARY_GRANT_REQUEST_DATA_ID,
+    )
+
+    print(response)

examples/v2/usage-metering/GetTemporaryGrantRequest.py

@@ -0,0 +1,21 @@
+"""
+Get an organization temporary grant request returns "OK" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+# there is a valid "temporary_grant_request" in the system
+TEMPORARY_GRANT_REQUEST_DATA_ID = environ["TEMPORARY_GRANT_REQUEST_DATA_ID"]
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["get_temporary_grant_request"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.get_temporary_grant_request(
+        request_id=TEMPORARY_GRANT_REQUEST_DATA_ID,
+    )
+
+    print(response)

examples/v2/usage-metering/ListMyTemporaryGrantRequests.py

@@ -0,0 +1,16 @@
+"""
+List your temporary grant requests returns "OK" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["list_my_temporary_grant_requests"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.list_my_temporary_grant_requests()
+
+    print(response)

examples/v2/usage-metering/ListMyTemporaryGrantRequests_3885243213.py

@@ -0,0 +1,16 @@
+"""
+List your temporary grant requests returns "OK" response with pagination
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["list_my_temporary_grant_requests"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    items = api_instance.list_my_temporary_grant_requests_with_pagination()
+    for item in items:
+        print(item)

examples/v2/usage-metering/ListTemporaryGrantRequests.py

@@ -0,0 +1,16 @@
+"""
+List organization temporary grant requests returns "OK" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["list_temporary_grant_requests"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.list_temporary_grant_requests()
+
+    print(response)

examples/v2/usage-metering/ListTemporaryGrantRequests_3760725651.py

@@ -0,0 +1,16 @@
+"""
+List organization temporary grant requests returns "OK" response with pagination
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["list_temporary_grant_requests"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    items = api_instance.list_temporary_grant_requests_with_pagination()
+    for item in items:
+        print(item)

examples/v2/usage-metering/UpdateTemporaryGrantRequest.py

@@ -0,0 +1,36 @@
+"""
+Decide a temporary grant request returns "OK" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.usage_metering_api import UsageMeteringApi
+from datadog_api_client.v2.model.temporary_grant_request_decision import TemporaryGrantRequestDecision
+from datadog_api_client.v2.model.temporary_grant_request_decision_attributes import (
+    TemporaryGrantRequestDecisionAttributes,
+)
+from datadog_api_client.v2.model.temporary_grant_request_decision_data import TemporaryGrantRequestDecisionData
+from datadog_api_client.v2.model.temporary_grant_request_decision_request import TemporaryGrantRequestDecisionRequest
+from datadog_api_client.v2.model.temporary_grant_request_type import TemporaryGrantRequestType
+
+# there is a valid "temporary_grant_request" in the system
+TEMPORARY_GRANT_REQUEST_DATA_ID = environ["TEMPORARY_GRANT_REQUEST_DATA_ID"]
+
+body = TemporaryGrantRequestDecisionRequest(
+    data=TemporaryGrantRequestDecisionData(
+        attributes=TemporaryGrantRequestDecisionAttributes(
+            decision=TemporaryGrantRequestDecision.DENIED,
+        ),
+        id=TEMPORARY_GRANT_REQUEST_DATA_ID,
+        type=TemporaryGrantRequestType.TEMPORARY_GRANT_REQUESTS,
+    ),
+)
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+configuration.unstable_operations["update_temporary_grant_request"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = UsageMeteringApi(api_client)
+    response = api_instance.update_temporary_grant_request(request_id=TEMPORARY_GRANT_REQUEST_DATA_ID, body=body)
+
+    print(response)

src/datadog_api_client/configuration.py

@@ -493,10 +493,18 @@ def __init__(
                 "v2.search_cost_recommendations": False,
                 "v2.update_unit_cost": False,
                 "v2.create_quotas": False,
+                "v2.create_temporary_grant_request": False,
                 "v2.delete_pending_quota": False,
                 "v2.delete_quota": False,
+                "v2.delete_requested_temporary_grant": False,
+                "v2.delete_temporary_grant_request": False,
+                "v2.get_my_temporary_grant_request": False,
+                "v2.get_temporary_grant_request": False,
+                "v2.list_my_temporary_grant_requests": False,
                 "v2.list_quotas": False,
+                "v2.list_temporary_grant_requests": False,
                 "v2.update_quota": False,
+                "v2.update_temporary_grant_request": False,
                 "v2.create_ownership_feedback": False,
                 "v2.get_ownership_evidence": False,
                 "v2.get_ownership_inference": False,