Agent Observability 데이터세트 레코드 일괄 업데이트

"""
Batch update Agent Observability dataset records returns "OK" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.agent_observability_api import AgentObservabilityApi
from datadog_api_client.v2.model.llm_obs_dataset_batch_update_data_attributes_request import (
    LLMObsDatasetBatchUpdateDataAttributesRequest,
)
from datadog_api_client.v2.model.llm_obs_dataset_batch_update_data_request import LLMObsDatasetBatchUpdateDataRequest
from datadog_api_client.v2.model.llm_obs_dataset_batch_update_insert_record import LLMObsDatasetBatchUpdateInsertRecord
from datadog_api_client.v2.model.llm_obs_dataset_batch_update_request import LLMObsDatasetBatchUpdateRequest
from datadog_api_client.v2.model.llm_obs_dataset_batch_update_update_record import LLMObsDatasetBatchUpdateUpdateRecord
from datadog_api_client.v2.model.llm_obs_dataset_record_tag_operations import LLMObsDatasetRecordTagOperations
from datadog_api_client.v2.model.llm_obs_dataset_type import LLMObsDatasetType

body = LLMObsDatasetBatchUpdateRequest(
    data=LLMObsDatasetBatchUpdateDataRequest(
        attributes=LLMObsDatasetBatchUpdateDataAttributesRequest(
            create_new_version=True,
            delete_records=[],
            insert_records=[
                LLMObsDatasetBatchUpdateInsertRecord(
                    expected_output=None,
                    id="rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
                    input=None,
                    tag_operations=LLMObsDatasetRecordTagOperations(
                        add=[],
                        remove=[],
                        set=[],
                    ),
                    tags=[],
                ),
            ],
            tags=[],
            update_records=[
                LLMObsDatasetBatchUpdateUpdateRecord(
                    expected_output=None,
                    id="rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
                    input=None,
                    tag_operations=LLMObsDatasetRecordTagOperations(
                        add=[],
                        remove=[],
                        set=[],
                    ),
                ),
            ],
        ),
        id="9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d",
        type=LLMObsDatasetType.DATASETS,
    ),
)

configuration = Configuration()
configuration.access_token = environ["DD_BEARER_TOKEN"]
configuration.unstable_operations["batch_update_llm_obs_dataset"] = True
with ApiClient(configuration) as api_client:
    api_instance = AgentObservabilityApi(api_client)
    response = api_instance.batch_update_llm_obs_dataset(project_id="project_id", dataset_id="dataset_id", body=body)

    print(response)

# Batch update Agent Observability dataset records returns "OK" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.access_token = ENV["DD_BEARER_TOKEN"]
  config.unstable_operations["v2.batch_update_llm_obs_dataset".to_sym] = true
end
api_instance = DatadogAPIClient::V2::AgentObservabilityAPI.new

body = DatadogAPIClient::V2::LLMObsDatasetBatchUpdateRequest.new({
  data: DatadogAPIClient::V2::LLMObsDatasetBatchUpdateDataRequest.new({
    attributes: DatadogAPIClient::V2::LLMObsDatasetBatchUpdateDataAttributesRequest.new({
      create_new_version: true,
      delete_records: [],
      insert_records: [
        DatadogAPIClient::V2::LLMObsDatasetBatchUpdateInsertRecord.new({
          expected_output: nil,
          id: "rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
          input: nil,
          tag_operations: DatadogAPIClient::V2::LLMObsDatasetRecordTagOperations.new({
            add: [],
            remove: [],
            set: [],
          }),
          tags: [],
        }),
      ],
      tags: [],
      update_records: [
        DatadogAPIClient::V2::LLMObsDatasetBatchUpdateUpdateRecord.new({
          expected_output: nil,
          id: "rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
          input: nil,
          tag_operations: DatadogAPIClient::V2::LLMObsDatasetRecordTagOperations.new({
            add: [],
            remove: [],
            set: [],
          }),
        }),
      ],
    }),
    id: "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d",
    type: DatadogAPIClient::V2::LLMObsDatasetType::DATASETS,
  }),
})
p api_instance.batch_update_llm_obs_dataset("project_id", "dataset_id", body)

// Batch update Agent Observability dataset records returns "OK" response

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)

func main() {
	body := datadogV2.LLMObsDatasetBatchUpdateRequest{
		Data: datadogV2.LLMObsDatasetBatchUpdateDataRequest{
			Attributes: datadogV2.LLMObsDatasetBatchUpdateDataAttributesRequest{
				CreateNewVersion: datadog.PtrBool(true),
				DeleteRecords:    []string{},
				InsertRecords: []datadogV2.LLMObsDatasetBatchUpdateInsertRecord{
					{
						ExpectedOutput: *datadogV2.NewNullableAnyValue(nil),
						Id:             datadog.PtrString("rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c"),
						Input:          *datadogV2.NewNullableAnyValue(nil),
						TagOperations: &datadogV2.LLMObsDatasetRecordTagOperations{
							Add:    []string{},
							Remove: []string{},
							Set:    []string{},
						},
						Tags: []string{},
					},
				},
				Tags: []string{},
				UpdateRecords: []datadogV2.LLMObsDatasetBatchUpdateUpdateRecord{
					{
						ExpectedOutput: *datadogV2.NewNullableAnyValue(nil),
						Id:             "rec-7c3f5a1b-9e2d-4f8a-b1c6-3d7e9f0a2b4c",
						Input:          *datadogV2.NewNullableAnyValue(nil),
						TagOperations: &datadogV2.LLMObsDatasetRecordTagOperations{
							Add:    []string{},
							Remove: []string{},
							Set:    []string{},
						},
					},
				},
			},
			Id:   "9f64e5c7-dc5a-45c8-a17c-1b85f0bec97d",
			Type: datadogV2.LLMOBSDATASETTYPE_DATASETS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.BatchUpdateLLMObsDataset", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewAgentObservabilityApi(apiClient)
	resp, r, err := api.BatchUpdateLLMObsDataset(ctx, "project_id", "dataset_id", body)

	if err != nil {
		fmt.Fprintf(os.Stderr, "Error when calling `AgentObservabilityApi.BatchUpdateLLMObsDataset`: %v\n", err)
		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
	}

	responseContent, _ := json.MarshalIndent(resp, "", "  ")
	fmt.Fprintf(os.Stdout, "Response from `AgentObservabilityApi.BatchUpdateLLMObsDataset`:\n%s\n", responseContent)
}