Add OpenAPI spec for Bits AI SRE investigations

docs/datadog_api_client.v2.api.rst

@@ -88,6 +88,13 @@ datadog\_api\_client.v2.api.aws\_logs\_integration\_api module
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.api.bits\_ai\_api module
+------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.api.bits_ai_api
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.api.case\_management\_api module
 --------------------------------------------------------
 

examples/v2/bits-ai/GetInvestigation.py

@@ -0,0 +1,16 @@
+"""
+Get a Bits AI investigation returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.bits_ai_api import BitsAIApi
+
+configuration = Configuration()
+configuration.unstable_operations["get_investigation"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = BitsAIApi(api_client)
+    response = api_instance.get_investigation(
+        id="id",
+    )
+
+    print(response)

examples/v2/bits-ai/ListInvestigations.py

@@ -0,0 +1,14 @@
+"""
+List Bits AI investigations returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.bits_ai_api import BitsAIApi
+
+configuration = Configuration()
+configuration.unstable_operations["list_investigations"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = BitsAIApi(api_client)
+    response = api_instance.list_investigations()
+
+    print(response)

examples/v2/bits-ai/ListInvestigations_130750454.py

@@ -0,0 +1,14 @@
+"""
+List Bits AI investigations returns "OK" response with pagination
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.bits_ai_api import BitsAIApi
+
+configuration = Configuration()
+configuration.unstable_operations["list_investigations"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = BitsAIApi(api_client)
+    items = api_instance.list_investigations_with_pagination()
+    for item in items:
+        print(item)

examples/v2/bits-ai/TriggerInvestigation.py

@@ -0,0 +1,39 @@
+"""
+Trigger a Bits AI investigation returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.bits_ai_api import BitsAIApi
+from datadog_api_client.v2.model.monitor_alert_trigger_attributes import MonitorAlertTriggerAttributes
+from datadog_api_client.v2.model.trigger_attributes import TriggerAttributes
+from datadog_api_client.v2.model.trigger_investigation_request import TriggerInvestigationRequest
+from datadog_api_client.v2.model.trigger_investigation_request_data import TriggerInvestigationRequestData
+from datadog_api_client.v2.model.trigger_investigation_request_data_attributes import (
+    TriggerInvestigationRequestDataAttributes,
+)
+from datadog_api_client.v2.model.trigger_investigation_request_type import TriggerInvestigationRequestType
+from datadog_api_client.v2.model.trigger_type import TriggerType
+
+body = TriggerInvestigationRequest(
+    data=TriggerInvestigationRequestData(
+        attributes=TriggerInvestigationRequestDataAttributes(
+            trigger=TriggerAttributes(
+                monitor_alert_trigger=MonitorAlertTriggerAttributes(
+                    event_id="1234567890123456789",
+                    event_ts=1700000000000,
+                    monitor_id=12345678,
+                ),
+                type=TriggerType.MONITOR_ALERT_TRIGGER,
+            ),
+        ),
+        type=TriggerInvestigationRequestType.TRIGGER_INVESTIGATION_REQUEST,
+    ),
+)
+
+configuration = Configuration()
+configuration.unstable_operations["trigger_investigation"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = BitsAIApi(api_client)
+    response = api_instance.trigger_investigation(body=body)
+
+    print(response)

src/datadog_api_client/configuration.py

@@ -266,6 +266,9 @@ def __init__(
                 "v2.get_open_api": False,
                 "v2.list_apis": False,
                 "v2.update_open_api": False,
+                "v2.get_investigation": False,
+                "v2.list_investigations": False,
+                "v2.trigger_investigation": False,
                 "v2.create_case_jira_issue": False,
                 "v2.create_case_notebook": False,
                 "v2.create_case_service_now_ticket": False,
@@ -280,11 +283,9 @@ def __init__(
                 "v2.update_change_request": False,
                 "v2.update_change_request_decision": False,
                 "v2.delete_test_optimization_service_settings": False,
-                "v2.get_flaky_tests_management_policies": False,
                 "v2.get_test_optimization_service_settings": False,
                 "v2.search_flaky_tests": False,
                 "v2.update_flaky_tests": False,
-                "v2.update_flaky_tests_management_policies": False,
                 "v2.update_test_optimization_service_settings": False,
                 "v2.create_aws_cloud_auth_persona_mapping": False,
                 "v2.delete_aws_cloud_auth_persona_mapping": False,