Add OpenAPI for LLM Observability dataset export batch_update clone restore and records upload SSE

examples/v2/llm-observability/BatchUpdateLLMObsDataset.java

@@ -0,0 +1,59 @@
+// Batch update LLM Observability dataset records returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LlmObservabilityApi;
+import com.datadog.api.client.v2.model.LLMObsDatasetBatchUpdateDataAttributesRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetBatchUpdateDataRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetBatchUpdateInsertRecord;
+import com.datadog.api.client.v2.model.LLMObsDatasetBatchUpdateRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetBatchUpdateUpdateRecord;
+import com.datadog.api.client.v2.model.LLMObsDatasetRecordTagOperations;
+import com.datadog.api.client.v2.model.LLMObsDatasetRecordsMutationResponse;
+import com.datadog.api.client.v2.model.LLMObsDatasetType;
+import java.util.Collections;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.batchUpdateLLMObsDataset", true);
+    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);
+
+    LLMObsDatasetBatchUpdateRequest body =
+        new LLMObsDatasetBatchUpdateRequest()
+            .data(
+                new LLMObsDatasetBatchUpdateDataRequest()
+                    .attributes(
+                        new LLMObsDatasetBatchUpdateDataAttributesRequest()
+                            .createNewVersion(true)
+                            .insertRecords(
+                                Collections.singletonList(
+                                    new LLMObsDatasetBatchUpdateInsertRecord()
+                                        .expectedOutput(null)
+                                        .id("rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c")
+                                        .input(null)
+                                        .tagOperations(new LLMObsDatasetRecordTagOperations())))
+                            .updateRecords(
+                                Collections.singletonList(
+                                    new LLMObsDatasetBatchUpdateUpdateRecord()
+                                        .expectedOutput(null)
+                                        .id("rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c")
+                                        .input(null)
+                                        .tagOperations(new LLMObsDatasetRecordTagOperations()))))
+                    .id("9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d")
+                    .type(LLMObsDatasetType.DATASETS));
+
+    try {
+      LLMObsDatasetRecordsMutationResponse result =
+          apiInstance.batchUpdateLLMObsDataset(
+              "a33671aa-24fd-4dcd-9b33-a8ec7dde7751", "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d", body);
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LlmObservabilityApi#batchUpdateLLMObsDataset");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/llm-observability/CloneLLMObsDataset.java

@@ -0,0 +1,42 @@
+// Clone an LLM Observability dataset returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LlmObservabilityApi;
+import com.datadog.api.client.v2.model.LLMObsDatasetCloneDataAttributesRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetCloneDataRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetCloneRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetResponse;
+import com.datadog.api.client.v2.model.LLMObsDatasetType;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.cloneLLMObsDataset", true);
+    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);
+
+    LLMObsDatasetCloneRequest body =
+        new LLMObsDatasetCloneRequest()
+            .data(
+                new LLMObsDatasetCloneDataRequest()
+                    .attributes(
+                        new LLMObsDatasetCloneDataAttributesRequest()
+                            .description("Clone of the original dataset for experimentation.")
+                            .name("My cloned dataset"))
+                    .id("9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d")
+                    .type(LLMObsDatasetType.DATASETS));
+
+    try {
+      LLMObsDatasetResponse result =
+          apiInstance.cloneLLMObsDataset(
+              "a33671aa-24fd-4dcd-9b33-a8ec7dde7751", "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d", body);
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LlmObservabilityApi#cloneLLMObsDataset");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/llm-observability/ExportLLMObsDataset.java

@@ -0,0 +1,26 @@
+// Export an LLM Observability dataset returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LlmObservabilityApi;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.exportLLMObsDataset", true);
+    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);
+
+    try {
+      String result =
+          apiInstance.exportLLMObsDataset(
+              "a33671aa-24fd-4dcd-9b33-a8ec7dde7751", "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LlmObservabilityApi#exportLLMObsDataset");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/llm-observability/RestoreLLMObsDatasetVersion.java

@@ -0,0 +1,37 @@
+// Restore an LLM Observability dataset version returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LlmObservabilityApi;
+import com.datadog.api.client.v2.model.LLMObsDatasetRestoreVersionDataAttributesRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetRestoreVersionDataRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetRestoreVersionRequest;
+import com.datadog.api.client.v2.model.LLMObsDatasetType;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.restoreLLMObsDatasetVersion", true);
+    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);
+
+    LLMObsDatasetRestoreVersionRequest body =
+        new LLMObsDatasetRestoreVersionRequest()
+            .data(
+                new LLMObsDatasetRestoreVersionDataRequest()
+                    .attributes(
+                        new LLMObsDatasetRestoreVersionDataAttributesRequest().datasetVersion(1))
+                    .id("9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d")
+                    .type(LLMObsDatasetType.DATASETS));
+
+    try {
+      apiInstance.restoreLLMObsDatasetVersion(
+          "a33671aa-24fd-4dcd-9b33-a8ec7dde7751", "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d", body);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LlmObservabilityApi#restoreLLMObsDatasetVersion");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/llm-observability/UploadLLMObsDatasetRecordsFile.java

@@ -0,0 +1,27 @@
+// Upload records to an LLM Observability dataset returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LlmObservabilityApi;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.uploadLLMObsDatasetRecordsFile", true);
+    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);
+
+    try {
+      String result =
+          apiInstance.uploadLLMObsDatasetRecordsFile(
+              "a33671aa-24fd-4dcd-9b33-a8ec7dde7751", "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LlmObservabilityApi#uploadLLMObsDatasetRecordsFile");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

src/main/java/com/datadog/api/client/ApiClient.java

@@ -774,6 +774,8 @@ public class ApiClient {
           put("v2.triggerFleetSchedule", false);
           put("v2.updateFleetSchedule", false);
           put("v2.aggregateLLMObsExperimentation", false);
+          put("v2.batchUpdateLLMObsDataset", false);
+          put("v2.cloneLLMObsDataset", false);
           put("v2.createLLMObsAnnotationQueue", false);
           put("v2.createLLMObsAnnotationQueueInteractions", false);
           put("v2.createLLMObsDataset", false);
@@ -790,6 +792,7 @@ public class ApiClient {
           put("v2.deleteLLMObsDatasets", false);
           put("v2.deleteLLMObsExperiments", false);
           put("v2.deleteLLMObsProjects", false);
+          put("v2.exportLLMObsDataset", false);
           put("v2.getLLMObsAnnotatedInteractions", false);
           put("v2.getLLMObsAnnotatedInteractionsByTraceIDs", false);
           put("v2.getLLMObsAnnotationQueueLabelSchema", false);
@@ -806,6 +809,7 @@ public class ApiClient {
           put("v2.listLLMObsProjects", false);
           put("v2.listLLMObsSpans", false);
           put("v2.lockLLMObsDatasetDraftState", false);
+          put("v2.restoreLLMObsDatasetVersion", false);
           put("v2.searchLLMObsExperimentation", false);
           put("v2.searchLLMObsSpans", false);
           put("v2.simpleSearchLLMObsExperimentation", false);
@@ -817,6 +821,7 @@ public class ApiClient {
           put("v2.updateLLMObsDatasetRecords", false);
           put("v2.updateLLMObsExperiment", false);
           put("v2.updateLLMObsProject", false);
+          put("v2.uploadLLMObsDatasetRecordsFile", false);
           put("v2.anonymizeUsers", false);
           put("v2.validate", false);
           put("v2.createOpenAPI", false);