Fix Agent Observability request types

.generator/schemas/v2/openapi.yaml

@@ -69285,7 +69285,7 @@ components:
         attributes:
           $ref: "#/components/schemas/LLMObsDatasetRecordsDataAttributesRequest"
         type:
-          $ref: "#/components/schemas/LLMObsRecordType"
+          $ref: "#/components/schemas/LLMObsDatasetType"
       required:
         - type
         - attributes
@@ -69350,7 +69350,7 @@ components:
         attributes:
           $ref: "#/components/schemas/LLMObsDatasetRecordsUpdateDataAttributesRequest"
         type:
-          $ref: "#/components/schemas/LLMObsRecordType"
+          $ref: "#/components/schemas/LLMObsDatasetType"
       required:
         - type
         - attributes
@@ -69687,7 +69687,7 @@ components:
         attributes:
           $ref: "#/components/schemas/LLMObsDeleteDatasetRecordsDataAttributesRequest"
         type:
-          $ref: "#/components/schemas/LLMObsRecordType"
+          $ref: "#/components/schemas/LLMObsDatasetType"
       required:
         - type
         - attributes
@@ -69926,14 +69926,6 @@ components:
       type: string
       x-enum-varnames:
         - DISPLAY_BLOCK
-    LLMObsEventType:
-      description: Resource type for Agent Observability experiment events.
-      enum:
-        - events
-      example: events
-      type: string
-      x-enum-varnames:
-        - EVENTS
     LLMObsExperimentDataAttributesRequest:
       description: Attributes for creating an Agent Observability experiment.
       properties:
@@ -70189,7 +70181,7 @@ components:
         attributes:
           $ref: "#/components/schemas/LLMObsExperimentEventsDataAttributesRequest"
         type:
-          $ref: "#/components/schemas/LLMObsEventType"
+          $ref: "#/components/schemas/LLMObsExperimentType"
       required:
         - type
         - attributes
@@ -73735,14 +73727,6 @@ components:
       required:
         - data
       type: object
-    LLMObsRecordType:
-      description: Resource type of Agent Observability dataset records.
-      enum:
-        - records
-      example: records
-      type: string
-      x-enum-varnames:
-        - RECORDS
     LLMObsSearchSpansRequest:
       description: Request body for searching Agent Observability spans.
       properties:
@@ -192310,7 +192294,7 @@ paths:
                           start_ns: 1705314600000000000
                           status: ok
                           trace_id: abc123def456
-                    type: events
+                    type: experiments
             schema:
               $ref: "#/components/schemas/LLMObsExperimentEventsRequest"
         description: Experiment events payload.
@@ -195577,7 +195561,7 @@ paths:
                     attributes:
                       records:
                         - id: rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c
-                    type: records
+                    type: datasets
             schema:
               $ref: "#/components/schemas/LLMObsDatasetRecordsUpdateRequest"
         description: Update records payload.
@@ -195652,7 +195636,7 @@ paths:
               default:
                 value:
                   data:
-                    type: records
+                    type: datasets
             schema:
               $ref: "#/components/schemas/LLMObsDatasetRecordsRequest"
         description: Append records payload.
@@ -195751,7 +195735,7 @@ paths:
                     attributes:
                       record_ids:
                         - rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c
-                    type: records
+                    type: datasets
             schema:
               $ref: "#/components/schemas/LLMObsDeleteDatasetRecordsRequest"
         description: Delete records payload.

examples/v2/agent-observability/CreateLLMObsDatasetRecords.py

@@ -11,7 +11,7 @@
 )
 from datadog_api_client.v2.model.llm_obs_dataset_records_data_request import LLMObsDatasetRecordsDataRequest
 from datadog_api_client.v2.model.llm_obs_dataset_records_request import LLMObsDatasetRecordsRequest
-from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
 body = LLMObsDatasetRecordsRequest(
     data=LLMObsDatasetRecordsDataRequest(
@@ -23,7 +23,7 @@
                 ),
             ],
         ),
-        type=LLMObsRecordType.RECORDS,
+        type=LLMObsDatasetType.DATASETS,
     ),
 )
 

examples/v2/agent-observability/CreateLLMObsExperimentEvents.py

@@ -5,7 +5,6 @@
 from os import environ
 from datadog_api_client import ApiClient, Configuration
 from datadog_api_client.v2.api.agent_observability_api import AgentObservabilityApi
-from datadog_api_client.v2.model.llm_obs_event_type import LLMObsEventType
 from datadog_api_client.v2.model.llm_obs_experiment_events_data_attributes_request import (
     LLMObsExperimentEventsDataAttributesRequest,
 )
@@ -17,6 +16,7 @@
 from datadog_api_client.v2.model.llm_obs_experiment_span_error import LLMObsExperimentSpanError
 from datadog_api_client.v2.model.llm_obs_experiment_span_meta import LLMObsExperimentSpanMeta
 from datadog_api_client.v2.model.llm_obs_experiment_span_status import LLMObsExperimentSpanStatus
+from datadog_api_client.v2.model.llm_obs_experiment_type import LLMObsExperimentType
 from datadog_api_client.v2.model.llm_obs_metric_assessment import LLMObsMetricAssessment
 from datadog_api_client.v2.model.llm_obs_metric_score_type import LLMObsMetricScoreType
 
@@ -57,7 +57,7 @@
                 ),
             ],
         ),
-        type=LLMObsEventType.EVENTS,
+        type=LLMObsExperimentType.EXPERIMENTS,
     ),
 )
 

examples/v2/agent-observability/DeleteLLMObsDatasetRecords.py

@@ -5,14 +5,14 @@
 from os import environ
 from datadog_api_client import ApiClient, Configuration
 from datadog_api_client.v2.api.agent_observability_api import AgentObservabilityApi
+from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 from datadog_api_client.v2.model.llm_obs_delete_dataset_records_data_attributes_request import (
     LLMObsDeleteDatasetRecordsDataAttributesRequest,
 )
 from datadog_api_client.v2.model.llm_obs_delete_dataset_records_data_request import (
     LLMObsDeleteDatasetRecordsDataRequest,
 )
 from datadog_api_client.v2.model.llm_obs_delete_dataset_records_request import LLMObsDeleteDatasetRecordsRequest
-from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
 
 body = LLMObsDeleteDatasetRecordsRequest(
     data=LLMObsDeleteDatasetRecordsDataRequest(
@@ -21,7 +21,7 @@
                 "rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
             ],
         ),
-        type=LLMObsRecordType.RECORDS,
+        type=LLMObsDatasetType.DATASETS,
     ),
 )
 

examples/v2/agent-observability/UpdateLLMObsDatasetRecords.py

@@ -13,7 +13,7 @@
     LLMObsDatasetRecordsUpdateDataRequest,
 )
 from datadog_api_client.v2.model.llm_obs_dataset_records_update_request import LLMObsDatasetRecordsUpdateRequest
-from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
 body = LLMObsDatasetRecordsUpdateRequest(
     data=LLMObsDatasetRecordsUpdateDataRequest(
@@ -26,7 +26,7 @@
                 ),
             ],
         ),
-        type=LLMObsRecordType.RECORDS,
+        type=LLMObsDatasetType.DATASETS,
     ),
 )
 

src/datadog_api_client/v2/model/llm_obs_dataset_records_data_request.py

@@ -15,7 +15,7 @@
     from datadog_api_client.v2.model.llm_obs_dataset_records_data_attributes_request import (
         LLMObsDatasetRecordsDataAttributesRequest,
     )
-    from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+    from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
 
 class LLMObsDatasetRecordsDataRequest(ModelNormal):
@@ -24,27 +24,27 @@ def openapi_types(_):
         from datadog_api_client.v2.model.llm_obs_dataset_records_data_attributes_request import (
             LLMObsDatasetRecordsDataAttributesRequest,
         )
-        from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+        from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
         return {
             "attributes": (LLMObsDatasetRecordsDataAttributesRequest,),
-            "type": (LLMObsRecordType,),
+            "type": (LLMObsDatasetType,),
         }
 
     attribute_map = {
         "attributes": "attributes",
         "type": "type",
     }
 
-    def __init__(self_, attributes: LLMObsDatasetRecordsDataAttributesRequest, type: LLMObsRecordType, **kwargs):
+    def __init__(self_, attributes: LLMObsDatasetRecordsDataAttributesRequest, type: LLMObsDatasetType, **kwargs):
         """
         Data object for appending records to an Agent Observability dataset.
 
         :param attributes: Attributes for appending records to an Agent Observability dataset.
         :type attributes: LLMObsDatasetRecordsDataAttributesRequest
 
-        :param type: Resource type of Agent Observability dataset records.
-        :type type: LLMObsRecordType
+        :param type: Resource type of an Agent Observability dataset.
+        :type type: LLMObsDatasetType
         """
         super().__init__(kwargs)
 

src/datadog_api_client/v2/model/llm_obs_dataset_records_update_data_request.py

@@ -15,7 +15,7 @@
     from datadog_api_client.v2.model.llm_obs_dataset_records_update_data_attributes_request import (
         LLMObsDatasetRecordsUpdateDataAttributesRequest,
     )
-    from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+    from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
 
 class LLMObsDatasetRecordsUpdateDataRequest(ModelNormal):
@@ -24,27 +24,27 @@ def openapi_types(_):
         from datadog_api_client.v2.model.llm_obs_dataset_records_update_data_attributes_request import (
             LLMObsDatasetRecordsUpdateDataAttributesRequest,
         )
-        from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+        from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
         return {
             "attributes": (LLMObsDatasetRecordsUpdateDataAttributesRequest,),
-            "type": (LLMObsRecordType,),
+            "type": (LLMObsDatasetType,),
         }
 
     attribute_map = {
         "attributes": "attributes",
         "type": "type",
     }
 
-    def __init__(self_, attributes: LLMObsDatasetRecordsUpdateDataAttributesRequest, type: LLMObsRecordType, **kwargs):
+    def __init__(self_, attributes: LLMObsDatasetRecordsUpdateDataAttributesRequest, type: LLMObsDatasetType, **kwargs):
         """
         Data object for updating records in an Agent Observability dataset.
 
         :param attributes: Attributes for updating records in an Agent Observability dataset.
         :type attributes: LLMObsDatasetRecordsUpdateDataAttributesRequest
 
-        :param type: Resource type of Agent Observability dataset records.
-        :type type: LLMObsRecordType
+        :param type: Resource type of an Agent Observability dataset.
+        :type type: LLMObsDatasetType
         """
         super().__init__(kwargs)
 

src/datadog_api_client/v2/model/llm_obs_delete_dataset_records_data_request.py

@@ -15,7 +15,7 @@
     from datadog_api_client.v2.model.llm_obs_delete_dataset_records_data_attributes_request import (
         LLMObsDeleteDatasetRecordsDataAttributesRequest,
     )
-    from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+    from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
 
 class LLMObsDeleteDatasetRecordsDataRequest(ModelNormal):
@@ -24,27 +24,27 @@ def openapi_types(_):
         from datadog_api_client.v2.model.llm_obs_delete_dataset_records_data_attributes_request import (
             LLMObsDeleteDatasetRecordsDataAttributesRequest,
         )
-        from datadog_api_client.v2.model.llm_obs_record_type import LLMObsRecordType
+        from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType
 
         return {
             "attributes": (LLMObsDeleteDatasetRecordsDataAttributesRequest,),
-            "type": (LLMObsRecordType,),
+            "type": (LLMObsDatasetType,),
         }
 
     attribute_map = {
         "attributes": "attributes",
         "type": "type",
     }
 
-    def __init__(self_, attributes: LLMObsDeleteDatasetRecordsDataAttributesRequest, type: LLMObsRecordType, **kwargs):
+    def __init__(self_, attributes: LLMObsDeleteDatasetRecordsDataAttributesRequest, type: LLMObsDatasetType, **kwargs):
         """
         Data object for deleting records from an Agent Observability dataset.
 
         :param attributes: Attributes for deleting records from an Agent Observability dataset.
         :type attributes: LLMObsDeleteDatasetRecordsDataAttributesRequest
 
-        :param type: Resource type of Agent Observability dataset records.
-        :type type: LLMObsRecordType
+        :param type: Resource type of an Agent Observability dataset.
+        :type type: LLMObsDatasetType
         """
         super().__init__(kwargs)
 

src/datadog_api_client/v2/model/llm_obs_experiment_events_data_request.py

@@ -15,7 +15,7 @@
     from datadog_api_client.v2.model.llm_obs_experiment_events_data_attributes_request import (
         LLMObsExperimentEventsDataAttributesRequest,
     )
-    from datadog_api_client.v2.model.llm_obs_event_type import LLMObsEventType
+    from datadog_api_client.v2.model.llm_obs_experiment_type import LLMObsExperimentType
 
 
 class LLMObsExperimentEventsDataRequest(ModelNormal):
@@ -24,27 +24,27 @@ def openapi_types(_):
         from datadog_api_client.v2.model.llm_obs_experiment_events_data_attributes_request import (
             LLMObsExperimentEventsDataAttributesRequest,
         )
-        from datadog_api_client.v2.model.llm_obs_event_type import LLMObsEventType
+        from datadog_api_client.v2.model.llm_obs_experiment_type import LLMObsExperimentType
 
         return {
             "attributes": (LLMObsExperimentEventsDataAttributesRequest,),
-            "type": (LLMObsEventType,),
+            "type": (LLMObsExperimentType,),
         }
 
     attribute_map = {
         "attributes": "attributes",
         "type": "type",
     }
 
-    def __init__(self_, attributes: LLMObsExperimentEventsDataAttributesRequest, type: LLMObsEventType, **kwargs):
+    def __init__(self_, attributes: LLMObsExperimentEventsDataAttributesRequest, type: LLMObsExperimentType, **kwargs):
         """
         Data object for pushing experiment events.
 
         :param attributes: Attributes for pushing experiment events including spans and metrics.
         :type attributes: LLMObsExperimentEventsDataAttributesRequest
 
-        :param type: Resource type for Agent Observability experiment events.
-        :type type: LLMObsEventType
+        :param type: Resource type of an Agent Observability experiment.
+        :type type: LLMObsExperimentType
         """
         super().__init__(kwargs)