Skip to content

Commit 096bda7

Browse files
author
ci.datadog-api-spec
committed
Regenerate client from commit 51461e5 of spec repo
1 parent 8f89a6b commit 096bda7

81 files changed

Lines changed: 5168 additions & 69 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: 1120 additions & 12 deletions
Large diffs are not rendered by default.

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,16 @@
77
from datadog_api_client.v2.model.llm_obs_create_prompt_data import LLMObsCreatePromptData
88
from datadog_api_client.v2.model.llm_obs_create_prompt_data_attributes import LLMObsCreatePromptDataAttributes
99
from datadog_api_client.v2.model.llm_obs_create_prompt_request import LLMObsCreatePromptRequest
10+
from datadog_api_client.v2.model.llm_obs_prompt_config import LLMObsPromptConfig
1011
from datadog_api_client.v2.model.llm_obs_prompt_type import LLMObsPromptType
1112
from datadog_api_client.v2.model.llm_obs_prompt_version_label import LLMObsPromptVersionLabel
1213

1314
body = LLMObsCreatePromptRequest(
1415
data=LLMObsCreatePromptData(
1516
attributes=LLMObsCreatePromptDataAttributes(
17+
config=LLMObsPromptConfig(
18+
[("model", "provider-model"), ("response_format", "{'type': 'json_object'}"), ("temperature", "0.2")]
19+
),
1620
env_ids=[],
1721
labels=[
1822
LLMObsPromptVersionLabel.PRODUCTION,

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,16 @@
99
LLMObsCreatePromptVersionDataAttributes,
1010
)
1111
from datadog_api_client.v2.model.llm_obs_create_prompt_version_request import LLMObsCreatePromptVersionRequest
12+
from datadog_api_client.v2.model.llm_obs_prompt_config import LLMObsPromptConfig
1213
from datadog_api_client.v2.model.llm_obs_prompt_version_label import LLMObsPromptVersionLabel
1314
from datadog_api_client.v2.model.llm_obs_prompt_version_type import LLMObsPromptVersionType
1415

1516
body = LLMObsCreatePromptVersionRequest(
1617
data=LLMObsCreatePromptVersionData(
1718
attributes=LLMObsCreatePromptVersionDataAttributes(
19+
config=LLMObsPromptConfig(
20+
[("model", "provider-model"), ("response_format", "{'type': 'json_object'}"), ("temperature", "0.2")]
21+
),
1822
env_ids=[],
1923
labels=[
2024
LLMObsPromptVersionLabel.PRODUCTION,

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

Lines changed: 103 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,103 @@
1+
"""
2+
Create a Databricks integration account returns "Created" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
7+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
8+
DatabricksCloudCostMetricsIntegrationDataflowRequest,
9+
)
10+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
12+
)
13+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
14+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
17+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
20+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
21+
)
22+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
23+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_integration_account_create_attributes import (
26+
DatabricksIntegrationAccountCreateAttributes,
27+
)
28+
from datadog_api_client.v2.model.databricks_integration_account_create_data import (
29+
DatabricksIntegrationAccountCreateData,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_create_request import (
32+
DatabricksIntegrationAccountCreateRequest,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_request import (
35+
DatabricksIntegrationAccountOAuthAuthRequest,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
38+
DatabricksIntegrationAccountOAuthAuthType,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_settings_request import (
41+
DatabricksIntegrationAccountSettingsRequest,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
44+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
45+
DatabricksModelServingMetricsIntegrationDataflowRequest,
46+
)
47+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
48+
49+
body = DatabricksIntegrationAccountCreateRequest(
50+
data=DatabricksIntegrationAccountCreateData(
51+
attributes=DatabricksIntegrationAccountCreateAttributes(
52+
authentication=DatabricksIntegrationAccountOAuthAuthRequest(
53+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
54+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
55+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
56+
client_secret="your-client-secret",
57+
),
58+
dataflows=DatabricksIntegrationDataflowsRequest(
59+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
60+
enabled=True,
61+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
62+
ccm_collect_all_workspaces=True,
63+
),
64+
),
65+
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
66+
enabled=True,
67+
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
68+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
69+
dd_api_key_secret="your-datadog-api-key",
70+
djm_global_init_script_enabled=True,
71+
script_gpum_enabled=True,
72+
script_logs_enabled=True,
73+
serverless_jobs_enabled=True,
74+
),
75+
),
76+
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
77+
enabled=True,
78+
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
79+
do_crawlers_cron="0 * * * *",
80+
sync_system_catalog=True,
81+
),
82+
),
83+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
84+
enabled=True,
85+
),
86+
),
87+
name="My Databricks Workspace",
88+
settings=DatabricksIntegrationAccountSettingsRequest(
89+
system_tables_sql_warehouse_id="aba7c023d4172910",
90+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
91+
),
92+
),
93+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
94+
),
95+
)
96+
97+
configuration = Configuration()
98+
configuration.unstable_operations["create_databricks_integration_account"] = True
99+
with ApiClient(configuration) as api_client:
100+
api_instance = DatabricksIntegrationApi(api_client)
101+
response = api_instance.create_databricks_integration_account(body=body)
102+
103+
print(response)

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

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""
2+
Delete a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["delete_databricks_integration_account"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationApi(api_client)
12+
api_instance.delete_databricks_integration_account(
13+
account_id="account_id",
14+
)

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

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
"""
2+
Get a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["get_databricks_integration_account"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationApi(api_client)
12+
response = api_instance.get_databricks_integration_account(
13+
account_id="account_id",
14+
)
15+
16+
print(response)

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

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""
2+
List Databricks integration accounts returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["list_databricks_integration_accounts"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationApi(api_client)
12+
response = api_instance.list_databricks_integration_accounts()
13+
14+
print(response)

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

Lines changed: 104 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,104 @@
1+
"""
2+
Update a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
7+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
8+
DatabricksCloudCostMetricsIntegrationDataflowRequest,
9+
)
10+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
12+
)
13+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
14+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
17+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
20+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
21+
)
22+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
23+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
26+
DatabricksIntegrationAccountOAuthAuthType,
27+
)
28+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_update import (
29+
DatabricksIntegrationAccountOAuthAuthUpdate,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_settings_update import (
32+
DatabricksIntegrationAccountSettingsUpdate,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_update_attributes import (
35+
DatabricksIntegrationAccountUpdateAttributes,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_update_data import (
38+
DatabricksIntegrationAccountUpdateData,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_update_request import (
41+
DatabricksIntegrationAccountUpdateRequest,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
44+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
45+
DatabricksModelServingMetricsIntegrationDataflowRequest,
46+
)
47+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
48+
49+
body = DatabricksIntegrationAccountUpdateRequest(
50+
data=DatabricksIntegrationAccountUpdateData(
51+
attributes=DatabricksIntegrationAccountUpdateAttributes(
52+
authentication=DatabricksIntegrationAccountOAuthAuthUpdate(
53+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
54+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
55+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
56+
client_secret="your-client-secret",
57+
),
58+
dataflows=DatabricksIntegrationDataflowsRequest(
59+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
60+
enabled=True,
61+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
62+
ccm_collect_all_workspaces=True,
63+
),
64+
),
65+
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
66+
enabled=True,
67+
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
68+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
69+
dd_api_key_secret="your-datadog-api-key",
70+
djm_global_init_script_enabled=True,
71+
script_gpum_enabled=True,
72+
script_logs_enabled=True,
73+
serverless_jobs_enabled=True,
74+
),
75+
),
76+
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
77+
enabled=True,
78+
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
79+
do_crawlers_cron="0 * * * *",
80+
sync_system_catalog=True,
81+
),
82+
),
83+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
84+
enabled=True,
85+
),
86+
),
87+
name="My Databricks Workspace",
88+
settings=DatabricksIntegrationAccountSettingsUpdate(
89+
system_tables_sql_warehouse_id="aba7c023d4172910",
90+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
91+
),
92+
),
93+
id="a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
94+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
95+
),
96+
)
97+
98+
configuration = Configuration()
99+
configuration.unstable_operations["update_databricks_integration_account"] = True
100+
with ApiClient(configuration) as api_client:
101+
api_instance = DatabricksIntegrationApi(api_client)
102+
response = api_instance.update_databricks_integration_account(account_id="account_id", body=body)
103+
104+
print(response)

‎src/datadog_api_client/configuration.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -655,6 +655,11 @@ def __init__(
655655
"v2.update_incident_user_defined_field": False,
656656
"v2.update_incident_user_defined_role": False,
657657
"v2.update_timestamp_override": False,
658+
"v2.create_databricks_integration_account": False,
659+
"v2.delete_databricks_integration_account": False,
660+
"v2.get_databricks_integration_account": False,
661+
"v2.list_databricks_integration_accounts": False,
662+
"v2.update_databricks_integration_account": False,
658663
"v2.create_elastic_cloud_integration_account": False,
659664
"v2.delete_elastic_cloud_integration_account": False,
660665
"v2.get_elastic_cloud_integration_account": False,

‎src/datadog_api_client/v2/api/agent_observability_api.py‎

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -78,10 +78,11 @@
7878
from datadog_api_client.v2.model.llm_obs_delete_projects_request import LLMObsDeleteProjectsRequest
7979
from datadog_api_client.v2.model.llm_obs_project_update_request import LLMObsProjectUpdateRequest
8080
from datadog_api_client.v2.model.llm_obs_prompts_response import LLMObsPromptsResponse
81-
from datadog_api_client.v2.model.llm_obs_prompt_response import LLMObsPromptResponse
81+
from datadog_api_client.v2.model.llm_obs_create_prompt_response import LLMObsCreatePromptResponse
8282
from datadog_api_client.v2.model.llm_obs_create_prompt_request import LLMObsCreatePromptRequest
8383
from datadog_api_client.v2.model.llm_obs_deleted_prompt_response import LLMObsDeletedPromptResponse
8484
from datadog_api_client.v2.model.llm_obs_prompt_sdk_response import LLMObsPromptSDKResponse
85+
from datadog_api_client.v2.model.llm_obs_prompt_response import LLMObsPromptResponse
8586
from datadog_api_client.v2.model.llm_obs_update_prompt_request import LLMObsUpdatePromptRequest
8687
from datadog_api_client.v2.model.llm_obs_prompt_versions_response import LLMObsPromptVersionsResponse
8788
from datadog_api_client.v2.model.llm_obs_prompt_version_response import LLMObsPromptVersionResponse
@@ -418,7 +419,7 @@ def __init__(self, api_client=None):
418419

419420
self._create_llm_obs_prompt_endpoint = _Endpoint(
420421
settings={
421-
"response_type": (LLMObsPromptResponse,),
422+
"response_type": (LLMObsCreatePromptResponse,),
422423
"auth": ["apiKeyAuth", "appKeyAuth"],
423424
"endpoint_path": "/api/v2/llm-obs/v1/prompts",
424425
"operation_id": "create_llm_obs_prompt",
@@ -2476,14 +2477,14 @@ def create_llm_obs_project(
24762477
def create_llm_obs_prompt(
24772478
self,
24782479
body: LLMObsCreatePromptRequest,
2479-
) -> LLMObsPromptResponse:
2480+
) -> LLMObsCreatePromptResponse:
24802481
"""Create an Agent Observability prompt.
24812482
24822483
Create a new prompt (and its first version) in the Agent Observability prompt registry.
24832484
24842485
:param body: Create prompt payload.
24852486
:type body: LLMObsCreatePromptRequest
2486-
:rtype: LLMObsPromptResponse
2487+
:rtype: LLMObsCreatePromptResponse
24872488
"""
24882489
kwargs: Dict[str, Any] = {}
24892490
kwargs["body"] = body
@@ -2990,7 +2991,7 @@ def get_llm_obs_prompt_version(
29902991
) -> LLMObsPromptVersionResponse:
29912992
"""Get a specific Agent Observability prompt version.
29922993
2993-
Get the full template of a single, specific version of an Agent Observability prompt.
2994+
Get the full template and configuration of a single, specific version of an Agent Observability prompt.
29942995
29952996
:param prompt_id: The customer-provided identifier of the Agent Observability prompt.
29962997
:type prompt_id: str

0 commit comments

Comments
 (0)