Skip to content

Commit 0733396

Browse files
author
ci.datadog-api-spec
committed
Regenerate client from commit df2bc66 of spec repo
1 parent 3714b60 commit 0733396

76 files changed

Lines changed: 7434 additions & 49 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.generator/schemas/v2/openapi.yaml‎

Lines changed: 1073 additions & 7 deletions
Large diffs are not rendered by default.

‎examples/v2/agent-observability/CreateLLMObsPrompt.rb‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
body = DatadogAPIClient::V2::LLMObsCreatePromptRequest.new({
1010
data: DatadogAPIClient::V2::LLMObsCreatePromptData.new({
1111
attributes: DatadogAPIClient::V2::LLMObsCreatePromptDataAttributes.new({
12+
config: DatadogAPIClient::V2::LLMObsPromptConfig.new({}),
1213
env_ids: [],
1314
labels: [
1415
DatadogAPIClient::V2::LLMObsPromptVersionLabel::PRODUCTION,

‎examples/v2/agent-observability/CreateLLMObsPromptVersion.rb‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
body = DatadogAPIClient::V2::LLMObsCreatePromptVersionRequest.new({
1010
data: DatadogAPIClient::V2::LLMObsCreatePromptVersionData.new({
1111
attributes: DatadogAPIClient::V2::LLMObsCreatePromptVersionDataAttributes.new({
12+
config: DatadogAPIClient::V2::LLMObsPromptConfig.new({}),
1213
env_ids: [],
1314
labels: [
1415
DatadogAPIClient::V2::LLMObsPromptVersionLabel::PRODUCTION,

‎examples/v2/databricks-integration/CreateDatabricksIntegrationAccount.rb‎

Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
# Create a Databricks integration account returns "Created" response
2+
3+
require "datadog_api_client"
4+
DatadogAPIClient.configure do |config|
5+
config.unstable_operations["v2.create_databricks_integration_account".to_sym] = true
6+
end
7+
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new
8+
9+
body = DatadogAPIClient::V2::DatabricksIntegrationAccountCreateRequest.new({
10+
data: DatadogAPIClient::V2::DatabricksIntegrationAccountCreateData.new({
11+
attributes: DatadogAPIClient::V2::DatabricksIntegrationAccountCreateAttributes.new({
12+
authentication: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthRequest.new({
13+
auth_type: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthType::DATABRICKS_OAUTH,
14+
azure_tenant_id: "4d3bac44-0230-4732-9e70-cc00736f0a97",
15+
client_id: "5c10654a-b3a3-4840-b37f-f477590c70a0",
16+
client_secret: "your-client-secret",
17+
}),
18+
dataflows: DatadogAPIClient::V2::DatabricksIntegrationDataflowsRequest.new({
19+
databricks_cloud_cost_metrics: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowRequest.new({
20+
enabled: true,
21+
settings: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest.new({
22+
ccm_collect_all_workspaces: true,
23+
}),
24+
}),
25+
databricks_data_observability_jobs_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest.new({
26+
enabled: true,
27+
settings: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest.new({
28+
dd_api_key_id: "fe383f4e-09fc-46bf-8e10-4efdd453a646",
29+
dd_api_key_secret: "your-datadog-api-key",
30+
djm_global_init_script_enabled: true,
31+
script_gpum_enabled: true,
32+
script_logs_enabled: true,
33+
serverless_jobs_enabled: true,
34+
}),
35+
}),
36+
databricks_data_observability_quality_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest.new({
37+
enabled: true,
38+
settings: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest.new({
39+
do_crawlers_cron: "0 * * * *",
40+
sync_system_catalog: true,
41+
}),
42+
}),
43+
databricks_model_serving_metrics: DatadogAPIClient::V2::DatabricksModelServingMetricsIntegrationDataflowRequest.new({
44+
enabled: true,
45+
}),
46+
}),
47+
name: "My Databricks Workspace",
48+
settings: DatadogAPIClient::V2::DatabricksIntegrationAccountSettingsRequest.new({
49+
system_tables_sql_warehouse_id: "aba7c023d4172910",
50+
workspace_url: "https://dbc-1234abcd.cloud.databricks.com",
51+
}),
52+
}),
53+
type: DatadogAPIClient::V2::IntegrationAccountType::INTEGRATION_ACCOUNT,
54+
}),
55+
})
56+
p api_instance.create_databricks_integration_account(body)

‎examples/v2/databricks-integration/DeleteDatabricksIntegrationAccount.rb‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
# Delete a Databricks integration account returns "OK" response
2+
3+
require "datadog_api_client"
4+
DatadogAPIClient.configure do |config|
5+
config.unstable_operations["v2.delete_databricks_integration_account".to_sym] = true
6+
end
7+
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new
8+
p api_instance.delete_databricks_integration_account("account_id")

‎examples/v2/databricks-integration/GetDatabricksIntegrationAccount.rb‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
# Get a Databricks integration account returns "OK" response
2+
3+
require "datadog_api_client"
4+
DatadogAPIClient.configure do |config|
5+
config.unstable_operations["v2.get_databricks_integration_account".to_sym] = true
6+
end
7+
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new
8+
p api_instance.get_databricks_integration_account("account_id")

‎examples/v2/databricks-integration/ListDatabricksIntegrationAccounts.rb‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
# List Databricks integration accounts returns "OK" response
2+
3+
require "datadog_api_client"
4+
DatadogAPIClient.configure do |config|
5+
config.unstable_operations["v2.list_databricks_integration_accounts".to_sym] = true
6+
end
7+
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new
8+
p api_instance.list_databricks_integration_accounts()

‎examples/v2/databricks-integration/UpdateDatabricksIntegrationAccount.rb‎

Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,57 @@
1+
# Update a Databricks integration account returns "OK" response
2+
3+
require "datadog_api_client"
4+
DatadogAPIClient.configure do |config|
5+
config.unstable_operations["v2.update_databricks_integration_account".to_sym] = true
6+
end
7+
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new
8+
9+
body = DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateRequest.new({
10+
data: DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateData.new({
11+
attributes: DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateAttributes.new({
12+
authentication: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthUpdate.new({
13+
auth_type: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthType::DATABRICKS_OAUTH,
14+
azure_tenant_id: "4d3bac44-0230-4732-9e70-cc00736f0a97",
15+
client_id: "5c10654a-b3a3-4840-b37f-f477590c70a0",
16+
client_secret: "your-client-secret",
17+
}),
18+
dataflows: DatadogAPIClient::V2::DatabricksIntegrationDataflowsRequest.new({
19+
databricks_cloud_cost_metrics: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowRequest.new({
20+
enabled: true,
21+
settings: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest.new({
22+
ccm_collect_all_workspaces: true,
23+
}),
24+
}),
25+
databricks_data_observability_jobs_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest.new({
26+
enabled: true,
27+
settings: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest.new({
28+
dd_api_key_id: "fe383f4e-09fc-46bf-8e10-4efdd453a646",
29+
dd_api_key_secret: "your-datadog-api-key",
30+
djm_global_init_script_enabled: true,
31+
script_gpum_enabled: true,
32+
script_logs_enabled: true,
33+
serverless_jobs_enabled: true,
34+
}),
35+
}),
36+
databricks_data_observability_quality_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest.new({
37+
enabled: true,
38+
settings: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest.new({
39+
do_crawlers_cron: "0 * * * *",
40+
sync_system_catalog: true,
41+
}),
42+
}),
43+
databricks_model_serving_metrics: DatadogAPIClient::V2::DatabricksModelServingMetricsIntegrationDataflowRequest.new({
44+
enabled: true,
45+
}),
46+
}),
47+
name: "My Databricks Workspace",
48+
settings: DatadogAPIClient::V2::DatabricksIntegrationAccountSettingsUpdate.new({
49+
system_tables_sql_warehouse_id: "aba7c023d4172910",
50+
workspace_url: "https://dbc-1234abcd.cloud.databricks.com",
51+
}),
52+
}),
53+
id: "a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
54+
type: DatadogAPIClient::V2::IntegrationAccountType::INTEGRATION_ACCOUNT,
55+
}),
56+
})
57+
p api_instance.update_databricks_integration_account("account_id", body)

‎features/scenarios_model_mapping.rb‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4080,6 +4080,19 @@
40804080
"id" => "UUID",
40814081
"body" => "IncidentTimestampOverridePatchRequest",
40824082
},
4083+
"v2.CreateDatabricksIntegrationAccount" => {
4084+
"body" => "DatabricksIntegrationAccountCreateRequest",
4085+
},
4086+
"v2.DeleteDatabricksIntegrationAccount" => {
4087+
"account_id" => "String",
4088+
},
4089+
"v2.GetDatabricksIntegrationAccount" => {
4090+
"account_id" => "String",
4091+
},
4092+
"v2.UpdateDatabricksIntegrationAccount" => {
4093+
"account_id" => "String",
4094+
"body" => "DatabricksIntegrationAccountUpdateRequest",
4095+
},
40834096
"v2.CreateElasticCloudIntegrationAccount" => {
40844097
"body" => "ElasticCloudIntegrationAccountCreateRequest",
40854098
},

‎features/v2/agent_observability.feature‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ Feature: Agent Observability
184184
Given operation "CreateLLMObsPromptVersion" enabled
185185
And new "CreateLLMObsPromptVersion" request
186186
And request contains "prompt_id" parameter from "REPLACE.ME"
187-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
187+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
188188
When the request is sent
189189
Then the response status is 400 Bad Request
190190

@@ -193,7 +193,7 @@ Feature: Agent Observability
193193
Given operation "CreateLLMObsPromptVersion" enabled
194194
And new "CreateLLMObsPromptVersion" request
195195
And request contains "prompt_id" parameter from "REPLACE.ME"
196-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
196+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
197197
When the request is sent
198198
Then the response status is 404 Not Found
199199

@@ -202,7 +202,7 @@ Feature: Agent Observability
202202
Given operation "CreateLLMObsPromptVersion" enabled
203203
And new "CreateLLMObsPromptVersion" request
204204
And request contains "prompt_id" parameter from "REPLACE.ME"
205-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
205+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-template-versions"}}
206206
When the request is sent
207207
Then the response status is 200 OK
208208

@@ -310,23 +310,23 @@ Feature: Agent Observability
310310
Scenario: Create an Agent Observability prompt returns "Bad Request" response
311311
Given operation "CreateLLMObsPrompt" enabled
312312
And new "CreateLLMObsPrompt" request
313-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
313+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
314314
When the request is sent
315315
Then the response status is 400 Bad Request
316316

317317
@generated @skip @team:DataDog/ml-observability
318318
Scenario: Create an Agent Observability prompt returns "Conflict" response
319319
Given operation "CreateLLMObsPrompt" enabled
320320
And new "CreateLLMObsPrompt" request
321-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
321+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
322322
When the request is sent
323323
Then the response status is 409 Conflict
324324

325325
@generated @skip @team:DataDog/ml-observability
326326
Scenario: Create an Agent Observability prompt returns "OK" response
327327
Given operation "CreateLLMObsPrompt" enabled
328328
And new "CreateLLMObsPrompt" request
329-
And body with value {"data": {"attributes": {"env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
329+
And body with value {"data": {"attributes": {"config": {"model": "provider-model", "response_format": {"type": "json_object"}, "temperature": 0.2}, "env_ids": [], "labels": ["production"], "prompt_id": "customer-support-assistant", "template": "You are a helpful assistant for {{audience}}."}, "type": "prompt-templates"}}
330330
When the request is sent
331331
Then the response status is 200 OK
332332

0 commit comments

Comments
 (0)