Skip to content

Add versioned prompt configuration #4004

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
api-clients-generation-pipeline merged 1 commit into from
Sep 24, 2026
Merged

Add versioned prompt configuration #4004

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
View file
Open in desktop

Large diffs are not rendered by default.

4 changes: 4 additions & 0 deletions examples/v2/agent-observability/CreateLLMObsPrompt.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,16 @@
from datadog_api_client.v2.model.llm_obs_create_prompt_data import LLMObsCreatePromptData
from datadog_api_client.v2.model.llm_obs_create_prompt_data_attributes import LLMObsCreatePromptDataAttributes
from datadog_api_client.v2.model.llm_obs_create_prompt_request import LLMObsCreatePromptRequest
from datadog_api_client.v2.model.llm_obs_prompt_config import LLMObsPromptConfig
from datadog_api_client.v2.model.llm_obs_prompt_type import LLMObsPromptType
from datadog_api_client.v2.model.llm_obs_prompt_version_label import LLMObsPromptVersionLabel

body = LLMObsCreatePromptRequest(
data=LLMObsCreatePromptData(
attributes=LLMObsCreatePromptDataAttributes(
config=LLMObsPromptConfig(
[("model", "provider-model"), ("response_format", "{'type': 'json_object'}"), ("temperature", "0.2")]
),
env_ids=[],
labels=[
LLMObsPromptVersionLabel.PRODUCTION,
Expand Down
4 changes: 4 additions & 0 deletions examples/v2/agent-observability/CreateLLMObsPromptVersion.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,16 @@
LLMObsCreatePromptVersionDataAttributes,
)
from datadog_api_client.v2.model.llm_obs_create_prompt_version_request import LLMObsCreatePromptVersionRequest
from datadog_api_client.v2.model.llm_obs_prompt_config import LLMObsPromptConfig
from datadog_api_client.v2.model.llm_obs_prompt_version_label import LLMObsPromptVersionLabel
from datadog_api_client.v2.model.llm_obs_prompt_version_type import LLMObsPromptVersionType

body = LLMObsCreatePromptVersionRequest(
data=LLMObsCreatePromptVersionData(
attributes=LLMObsCreatePromptVersionDataAttributes(
config=LLMObsPromptConfig(
[("model", "provider-model"), ("response_format", "{'type': 'json_object'}"), ("temperature", "0.2")]
),
env_ids=[],
labels=[
LLMObsPromptVersionLabel.PRODUCTION,
Expand Down
103 changes: 103 additions & 0 deletions examples/v2/databricks-integration/CreateDatabricksIntegrationAccount.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,103 @@
"""
Create a Databricks integration account returns "Created" response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
DatabricksCloudCostMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_integration_account_create_attributes import (
DatabricksIntegrationAccountCreateAttributes,
)
from datadog_api_client.v2.model.databricks_integration_account_create_data import (
DatabricksIntegrationAccountCreateData,
)
from datadog_api_client.v2.model.databricks_integration_account_create_request import (
DatabricksIntegrationAccountCreateRequest,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_request import (
DatabricksIntegrationAccountOAuthAuthRequest,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
DatabricksIntegrationAccountOAuthAuthType,
)
from datadog_api_client.v2.model.databricks_integration_account_settings_request import (
DatabricksIntegrationAccountSettingsRequest,
)
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
DatabricksModelServingMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType

body = DatabricksIntegrationAccountCreateRequest(
data=DatabricksIntegrationAccountCreateData(
attributes=DatabricksIntegrationAccountCreateAttributes(
authentication=DatabricksIntegrationAccountOAuthAuthRequest(
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
client_secret="your-client-secret",
),
dataflows=DatabricksIntegrationDataflowsRequest(
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
enabled=True,
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
ccm_collect_all_workspaces=True,
),
),
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
enabled=True,
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
dd_api_key_secret="your-datadog-api-key",
djm_global_init_script_enabled=True,
script_gpum_enabled=True,
script_logs_enabled=True,
serverless_jobs_enabled=True,
),
),
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
enabled=True,
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
do_crawlers_cron="0 * * * *",
sync_system_catalog=True,
),
),
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
enabled=True,
),
),
name="My Databricks Workspace",
settings=DatabricksIntegrationAccountSettingsRequest(
system_tables_sql_warehouse_id="aba7c023d4172910",
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
),
),
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
),
)

configuration = Configuration()
configuration.unstable_operations["create_databricks_integration_account"] = True
with ApiClient(configuration) as api_client:
api_instance = DatabricksIntegrationApi(api_client)
response = api_instance.create_databricks_integration_account(body=body)

print(response)
14 changes: 14 additions & 0 deletions examples/v2/databricks-integration/DeleteDatabricksIntegrationAccount.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
"""
Delete a Databricks integration account returns "OK" response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi

configuration = Configuration()
configuration.unstable_operations["delete_databricks_integration_account"] = True
with ApiClient(configuration) as api_client:
api_instance = DatabricksIntegrationApi(api_client)
api_instance.delete_databricks_integration_account(
account_id="account_id",
)
16 changes: 16 additions & 0 deletions examples/v2/databricks-integration/GetDatabricksIntegrationAccount.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
"""
Get a Databricks integration account returns "OK" response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi

configuration = Configuration()
configuration.unstable_operations["get_databricks_integration_account"] = True
with ApiClient(configuration) as api_client:
api_instance = DatabricksIntegrationApi(api_client)
response = api_instance.get_databricks_integration_account(
account_id="account_id",
)

print(response)
14 changes: 14 additions & 0 deletions examples/v2/databricks-integration/ListDatabricksIntegrationAccounts.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
"""
List Databricks integration accounts returns "OK" response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi

configuration = Configuration()
configuration.unstable_operations["list_databricks_integration_accounts"] = True
with ApiClient(configuration) as api_client:
api_instance = DatabricksIntegrationApi(api_client)
response = api_instance.list_databricks_integration_accounts()

print(response)
104 changes: 104 additions & 0 deletions examples/v2/databricks-integration/UpdateDatabricksIntegrationAccount.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
"""
Update a Databricks integration account returns "OK" response
"""

from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
DatabricksCloudCostMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
DatabricksIntegrationAccountOAuthAuthType,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_update import (
DatabricksIntegrationAccountOAuthAuthUpdate,
)
from datadog_api_client.v2.model.databricks_integration_account_settings_update import (
DatabricksIntegrationAccountSettingsUpdate,
)
from datadog_api_client.v2.model.databricks_integration_account_update_attributes import (
DatabricksIntegrationAccountUpdateAttributes,
)
from datadog_api_client.v2.model.databricks_integration_account_update_data import (
DatabricksIntegrationAccountUpdateData,
)
from datadog_api_client.v2.model.databricks_integration_account_update_request import (
DatabricksIntegrationAccountUpdateRequest,
)
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
DatabricksModelServingMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType

body = DatabricksIntegrationAccountUpdateRequest(
data=DatabricksIntegrationAccountUpdateData(
attributes=DatabricksIntegrationAccountUpdateAttributes(
authentication=DatabricksIntegrationAccountOAuthAuthUpdate(
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
client_secret="your-client-secret",
),
dataflows=DatabricksIntegrationDataflowsRequest(
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
enabled=True,
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
ccm_collect_all_workspaces=True,
),
),
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
enabled=True,
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
dd_api_key_secret="your-datadog-api-key",
djm_global_init_script_enabled=True,
script_gpum_enabled=True,
script_logs_enabled=True,
serverless_jobs_enabled=True,
),
),
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
enabled=True,
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
do_crawlers_cron="0 * * * *",
sync_system_catalog=True,
),
),
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
enabled=True,
),
),
name="My Databricks Workspace",
settings=DatabricksIntegrationAccountSettingsUpdate(
system_tables_sql_warehouse_id="aba7c023d4172910",
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
),
),
id="a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
),
)

configuration = Configuration()
configuration.unstable_operations["update_databricks_integration_account"] = True
with ApiClient(configuration) as api_client:
api_instance = DatabricksIntegrationApi(api_client)
response = api_instance.update_databricks_integration_account(account_id="account_id", body=body)

print(response)
5 changes: 5 additions & 0 deletions src/datadog_api_client/configuration.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -655,6 +655,11 @@ def __init__(
"v2.update_incident_user_defined_field": False,
"v2.update_incident_user_defined_role": False,
"v2.update_timestamp_override": False,
"v2.create_databricks_integration_account": False,
"v2.delete_databricks_integration_account": False,
"v2.get_databricks_integration_account": False,
"v2.list_databricks_integration_accounts": False,
"v2.update_databricks_integration_account": False,
"v2.create_elastic_cloud_integration_account": False,
"v2.delete_elastic_cloud_integration_account": False,
"v2.get_elastic_cloud_integration_account": False,
Expand Down
11 changes: 6 additions & 5 deletions src/datadog_api_client/v2/api/agent_observability_api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -78,10 +78,11 @@
from datadog_api_client.v2.model.llm_obs_delete_projects_request import LLMObsDeleteProjectsRequest
from datadog_api_client.v2.model.llm_obs_project_update_request import LLMObsProjectUpdateRequest
from datadog_api_client.v2.model.llm_obs_prompts_response import LLMObsPromptsResponse
from datadog_api_client.v2.model.llm_obs_prompt_response import LLMObsPromptResponse
from datadog_api_client.v2.model.llm_obs_create_prompt_response import LLMObsCreatePromptResponse
from datadog_api_client.v2.model.llm_obs_create_prompt_request import LLMObsCreatePromptRequest
from datadog_api_client.v2.model.llm_obs_deleted_prompt_response import LLMObsDeletedPromptResponse
from datadog_api_client.v2.model.llm_obs_prompt_sdk_response import LLMObsPromptSDKResponse
from datadog_api_client.v2.model.llm_obs_prompt_response import LLMObsPromptResponse
from datadog_api_client.v2.model.llm_obs_update_prompt_request import LLMObsUpdatePromptRequest
from datadog_api_client.v2.model.llm_obs_prompt_versions_response import LLMObsPromptVersionsResponse
from datadog_api_client.v2.model.llm_obs_prompt_version_response import LLMObsPromptVersionResponse
Expand Down Expand Up @@ -418,7 +419,7 @@ def __init__(self, api_client=None):

self._create_llm_obs_prompt_endpoint = _Endpoint(
settings={
"response_type": (LLMObsPromptResponse,),
"response_type": (LLMObsCreatePromptResponse,),
"auth": ["apiKeyAuth", "appKeyAuth"],
"endpoint_path": "/api/v2/llm-obs/v1/prompts",
"operation_id": "create_llm_obs_prompt",
Expand Down Expand Up @@ -2476,14 +2477,14 @@ def create_llm_obs_project(
def create_llm_obs_prompt(
self,
body: LLMObsCreatePromptRequest,
) -> LLMObsPromptResponse:
) -> LLMObsCreatePromptResponse:
"""Create an Agent Observability prompt.

Create a new prompt (and its first version) in the Agent Observability prompt registry.

:param body: Create prompt payload.
:type body: LLMObsCreatePromptRequest
:rtype: LLMObsPromptResponse
:rtype: LLMObsCreatePromptResponse
"""
kwargs: Dict[str, Any] = {}
kwargs["body"] = body
Expand Down Expand Up @@ -2990,7 +2991,7 @@ def get_llm_obs_prompt_version(
) -> LLMObsPromptVersionResponse:
"""Get a specific Agent Observability prompt version.

Get the full template of a single, specific version of an Agent Observability prompt.
Get the full template and configuration of a single, specific version of an Agent Observability prompt.

:param prompt_id: The customer-provided identifier of the Agent Observability prompt.
:type prompt_id: str
Expand Down
Loading
Loading