Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1,080 changes: 1,073 additions & 7 deletions .generator/schemas/v2/openapi.yaml

Large diffs are not rendered by default.

10 changes: 8 additions & 2 deletions examples/v2/agent-observability/CreateLLMObsPrompt.java
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,12 @@
import com.datadog.api.client.v2.model.LLMObsCreatePromptData;
import com.datadog.api.client.v2.model.LLMObsCreatePromptDataAttributes;
import com.datadog.api.client.v2.model.LLMObsCreatePromptRequest;
import com.datadog.api.client.v2.model.LLMObsPromptResponse;
import com.datadog.api.client.v2.model.LLMObsCreatePromptResponse;
import com.datadog.api.client.v2.model.LLMObsPromptTemplate;
import com.datadog.api.client.v2.model.LLMObsPromptType;
import com.datadog.api.client.v2.model.LLMObsPromptVersionLabel;
import java.util.Collections;
import java.util.Map;

public class Example {
public static void main(String[] args) {
Expand All @@ -24,14 +25,19 @@ public static void main(String[] args) {
new LLMObsCreatePromptData()
.attributes(
new LLMObsCreatePromptDataAttributes()
.config(
Map.ofEntries(
Map.entry("model", "provider-model"),
Map.entry("response_format", "{'type': 'json_object'}"),
Map.entry("temperature", "0.2")))
.labels(Collections.singletonList(LLMObsPromptVersionLabel.PRODUCTION))
.promptId("customer-support-assistant")
.template(
new LLMObsPromptTemplate("You are a helpful assistant for .")))
.type(LLMObsPromptType.PROMPT_TEMPLATES));

try {
LLMObsPromptResponse result = apiInstance.createLLMObsPrompt(body);
LLMObsCreatePromptResponse result = apiInstance.createLLMObsPrompt(body);
System.out.println(result);
} catch (ApiException e) {
System.err.println("Exception when calling AgentObservabilityApi#createLLMObsPrompt");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
import com.datadog.api.client.v2.model.LLMObsPromptVersionResponse;
import com.datadog.api.client.v2.model.LLMObsPromptVersionType;
import java.util.Collections;
import java.util.Map;

public class Example {
public static void main(String[] args) {
Expand All @@ -24,6 +25,11 @@ public static void main(String[] args) {
new LLMObsCreatePromptVersionData()
.attributes(
new LLMObsCreatePromptVersionDataAttributes()
.config(
Map.ofEntries(
Map.entry("model", "provider-model"),
Map.entry("response_format", "{'type': 'json_object'}"),
Map.entry("temperature", "0.2")))
.labels(Collections.singletonList(LLMObsPromptVersionLabel.PRODUCTION))
.template(
new LLMObsPromptTemplate("You are a helpful assistant for .")))
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,95 @@
// Create a Databricks integration account returns "Created" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.DatabricksIntegrationApi;
import com.datadog.api.client.v2.model.DatabricksCloudCostMetricsIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountAuthenticationRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountCreateAttributes;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountCreateData;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountCreateRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountOAuthAuthRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountOAuthAuthType;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountResponse;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationDataflowsRequest;
import com.datadog.api.client.v2.model.DatabricksModelServingMetricsIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.IntegrationAccountType;

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

DatabricksIntegrationAccountCreateRequest body =
new DatabricksIntegrationAccountCreateRequest()
.data(
new DatabricksIntegrationAccountCreateData()
.attributes(
new DatabricksIntegrationAccountCreateAttributes()
.authentication(
new DatabricksIntegrationAccountAuthenticationRequest(
new DatabricksIntegrationAccountOAuthAuthRequest()
.authType(
DatabricksIntegrationAccountOAuthAuthType
.DATABRICKS_OAUTH)
.azureTenantId("4d3bac44-0230-4732-9e70-cc00736f0a97")
.clientId("5c10654a-b3a3-4840-b37f-f477590c70a0")
.clientSecret("your-client-secret")))
.dataflows(
new DatabricksIntegrationDataflowsRequest()
.databricksCloudCostMetrics(
new DatabricksCloudCostMetricsIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest()
.ccmCollectAllWorkspaces(true)))
.databricksDataObservabilityJobsMonitoring(
new DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest()
.ddApiKeyId(
"fe383f4e-09fc-46bf-8e10-4efdd453a646")
.ddApiKeySecret("your-datadog-api-key")
.djmGlobalInitScriptEnabled(true)
.scriptGpumEnabled(true)
.scriptLogsEnabled(true)
.serverlessJobsEnabled(true)))
.databricksDataObservabilityQualityMonitoring(
new DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest()
.doCrawlersCron("0 * * * *")
.syncSystemCatalog(true)))
.databricksModelServingMetrics(
new DatabricksModelServingMetricsIntegrationDataflowRequest()
.enabled(true)))
.name("My Databricks Workspace")
.settings(
new DatabricksIntegrationAccountSettingsRequest()
.systemTablesSqlWarehouseId("aba7c023d4172910")
.workspaceUrl("https://dbc-1234abcd.cloud.databricks.com")))
.type(IntegrationAccountType.INTEGRATION_ACCOUNT));

try {
DatabricksIntegrationAccountResponse result =
apiInstance.createDatabricksIntegrationAccount(body);
System.out.println(result);
} catch (ApiException e) {
System.err.println(
"Exception when calling DatabricksIntegrationApi#createDatabricksIntegrationAccount");
System.err.println("Status code: " + e.getCode());
System.err.println("Reason: " + e.getResponseBody());
System.err.println("Response headers: " + e.getResponseHeaders());
e.printStackTrace();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
// Delete a Databricks integration account returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.DatabricksIntegrationApi;

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

try {
apiInstance.deleteDatabricksIntegrationAccount("account_id");
} catch (ApiException e) {
System.err.println(
"Exception when calling DatabricksIntegrationApi#deleteDatabricksIntegrationAccount");
System.err.println("Status code: " + e.getCode());
System.err.println("Reason: " + e.getResponseBody());
System.err.println("Response headers: " + e.getResponseHeaders());
e.printStackTrace();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
// Get a Databricks integration account returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.DatabricksIntegrationApi;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountResponse;

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

try {
DatabricksIntegrationAccountResponse result =
apiInstance.getDatabricksIntegrationAccount("account_id");
System.out.println(result);
} catch (ApiException e) {
System.err.println(
"Exception when calling DatabricksIntegrationApi#getDatabricksIntegrationAccount");
System.err.println("Status code: " + e.getCode());
System.err.println("Reason: " + e.getResponseBody());
System.err.println("Response headers: " + e.getResponseHeaders());
e.printStackTrace();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
// List Databricks integration accounts returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.DatabricksIntegrationApi;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountsResponse;

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

try {
DatabricksIntegrationAccountsResponse result =
apiInstance.listDatabricksIntegrationAccounts();
System.out.println(result);
} catch (ApiException e) {
System.err.println(
"Exception when calling DatabricksIntegrationApi#listDatabricksIntegrationAccounts");
System.err.println("Status code: " + e.getCode());
System.err.println("Reason: " + e.getResponseBody());
System.err.println("Response headers: " + e.getResponseHeaders());
e.printStackTrace();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
// Update a Databricks integration account returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.DatabricksIntegrationApi;
import com.datadog.api.client.v2.model.DatabricksCloudCostMetricsIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountAuthenticationUpdate;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountOAuthAuthType;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountOAuthAuthUpdate;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountResponse;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountSettingsUpdate;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountUpdateAttributes;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountUpdateData;
import com.datadog.api.client.v2.model.DatabricksIntegrationAccountUpdateRequest;
import com.datadog.api.client.v2.model.DatabricksIntegrationDataflowsRequest;
import com.datadog.api.client.v2.model.DatabricksModelServingMetricsIntegrationDataflowRequest;
import com.datadog.api.client.v2.model.IntegrationAccountType;

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

DatabricksIntegrationAccountUpdateRequest body =
new DatabricksIntegrationAccountUpdateRequest()
.data(
new DatabricksIntegrationAccountUpdateData()
.attributes(
new DatabricksIntegrationAccountUpdateAttributes()
.authentication(
new DatabricksIntegrationAccountAuthenticationUpdate(
new DatabricksIntegrationAccountOAuthAuthUpdate()
.authType(
DatabricksIntegrationAccountOAuthAuthType
.DATABRICKS_OAUTH)
.azureTenantId("4d3bac44-0230-4732-9e70-cc00736f0a97")
.clientId("5c10654a-b3a3-4840-b37f-f477590c70a0")
.clientSecret("your-client-secret")))
.dataflows(
new DatabricksIntegrationDataflowsRequest()
.databricksCloudCostMetrics(
new DatabricksCloudCostMetricsIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest()
.ccmCollectAllWorkspaces(true)))
.databricksDataObservabilityJobsMonitoring(
new DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest()
.ddApiKeyId(
"fe383f4e-09fc-46bf-8e10-4efdd453a646")
.ddApiKeySecret("your-datadog-api-key")
.djmGlobalInitScriptEnabled(true)
.scriptGpumEnabled(true)
.scriptLogsEnabled(true)
.serverlessJobsEnabled(true)))
.databricksDataObservabilityQualityMonitoring(
new DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest()
.enabled(true)
.settings(
new DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest()
.doCrawlersCron("0 * * * *")
.syncSystemCatalog(true)))
.databricksModelServingMetrics(
new DatabricksModelServingMetricsIntegrationDataflowRequest()
.enabled(true)))
.name("My Databricks Workspace")
.settings(
new DatabricksIntegrationAccountSettingsUpdate()
.systemTablesSqlWarehouseId("aba7c023d4172910")
.workspaceUrl("https://dbc-1234abcd.cloud.databricks.com")))
.id("a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f")
.type(IntegrationAccountType.INTEGRATION_ACCOUNT));

try {
DatabricksIntegrationAccountResponse result =
apiInstance.updateDatabricksIntegrationAccount("account_id", body);
System.out.println(result);
} catch (ApiException e) {
System.err.println(
"Exception when calling DatabricksIntegrationApi#updateDatabricksIntegrationAccount");
System.err.println("Status code: " + e.getCode());
System.err.println("Reason: " + e.getResponseBody());
System.err.println("Response headers: " + e.getResponseHeaders());
e.printStackTrace();
}
}
}
5 changes: 5 additions & 0 deletions src/main/java/com/datadog/api/client/ApiClient.java
Original file line number Diff line number Diff line change
Expand Up @@ -1308,6 +1308,11 @@ public class ApiClient {
put("v2.updateIncidentUserDefinedField", false);
put("v2.updateIncidentUserDefinedRole", false);
put("v2.updateTimestampOverride", false);
put("v2.createDatabricksIntegrationAccount", false);
put("v2.deleteDatabricksIntegrationAccount", false);
put("v2.getDatabricksIntegrationAccount", false);
put("v2.listDatabricksIntegrationAccounts", false);
put("v2.updateDatabricksIntegrationAccount", false);
put("v2.createElasticCloudIntegrationAccount", false);
put("v2.deleteElasticCloudIntegrationAccount", false);
put("v2.getElasticCloudIntegrationAccount", false);
Expand Down
Loading
Loading