Create or update annotations

// Create or update annotations returns "OK — annotations created or updated. For mixed batches, denied items and other
// per-item errors are listed in `errors`." 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.LLMObsAnnotationsRequest{
		Data: datadogV2.LLMObsAnnotationsDataRequest{
			Attributes: datadogV2.LLMObsAnnotationsDataAttributesRequest{
				Annotations: []datadogV2.LLMObsUpsertAnnotationItem{
					{
						InteractionId: "00000000-0000-0000-0000-000000000001",
						LabelValues: []datadogV2.LLMObsAnnotationLabelValue{
							{
								LabelSchemaId: "abc-123",
								Value: datadogV2.LLMObsAnnotationLabelValueValue{
									AnyValueString: datadog.PtrString("good")},
							},
							{
								LabelSchemaId: "ef56gh78",
								Value: datadogV2.LLMObsAnnotationLabelValueValue{
									AnyValueString: datadog.PtrString("positive")},
							},
						},
					},
				},
			},
			Type: datadogV2.LLMOBSANNOTATIONSTYPE_ANNOTATIONS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.UpsertLLMObsAnnotations", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewAgentObservabilityApi(apiClient)
	resp, r, err := api.UpsertLLMObsAnnotations(ctx, "queue_id", body)

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

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

// Create or update annotations returns "OK — annotations created or updated. For mixed batches,
// denied items and other
// per-item errors are listed in `errors`." response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.AgentObservabilityApi;
import com.datadog.api.client.v2.model.LLMObsAnnotationLabelValue;
import com.datadog.api.client.v2.model.LLMObsAnnotationLabelValueValue;
import com.datadog.api.client.v2.model.LLMObsAnnotationsDataAttributesRequest;
import com.datadog.api.client.v2.model.LLMObsAnnotationsDataRequest;
import com.datadog.api.client.v2.model.LLMObsAnnotationsRequest;
import com.datadog.api.client.v2.model.LLMObsAnnotationsResponse;
import com.datadog.api.client.v2.model.LLMObsAnnotationsType;
import com.datadog.api.client.v2.model.LLMObsUpsertAnnotationItem;
import java.util.Arrays;
import java.util.Collections;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setAccessToken(System.getenv("DD_BEARER_TOKEN"));
    defaultClient.setUnstableOperationEnabled("v2.upsertLLMObsAnnotations", true);
    AgentObservabilityApi apiInstance = new AgentObservabilityApi(defaultClient);

    LLMObsAnnotationsRequest body =
        new LLMObsAnnotationsRequest()
            .data(
                new LLMObsAnnotationsDataRequest()
                    .attributes(
                        new LLMObsAnnotationsDataAttributesRequest()
                            .annotations(
                                Collections.singletonList(
                                    new LLMObsUpsertAnnotationItem()
                                        .interactionId("00000000-0000-0000-0000-000000000001")
                                        .labelValues(
                                            Arrays.asList(
                                                new LLMObsAnnotationLabelValue()
                                                    .labelSchemaId("abc-123")
                                                    .value(
                                                        new LLMObsAnnotationLabelValueValue(
                                                            "good")),
                                                new LLMObsAnnotationLabelValue()
                                                    .labelSchemaId("ef56gh78")
                                                    .value(
                                                        new LLMObsAnnotationLabelValueValue(
                                                            "positive")))))))
                    .type(LLMObsAnnotationsType.ANNOTATIONS));

    try {
      LLMObsAnnotationsResponse result =
          apiInstance.upsertLLMObsAnnotations("00000000-0000-0000-0000-000000000001", body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling AgentObservabilityApi#upsertLLMObsAnnotations");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

"""
Create or update annotations returns "OK — annotations created or updated. For mixed batches, denied items and other
per-item errors are listed in `errors`." 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_annotation_label_value import LLMObsAnnotationLabelValue
from datadog_api_client.v2.model.llm_obs_annotations_data_attributes_request import (
    LLMObsAnnotationsDataAttributesRequest,
)
from datadog_api_client.v2.model.llm_obs_annotations_data_request import LLMObsAnnotationsDataRequest
from datadog_api_client.v2.model.llm_obs_annotations_request import LLMObsAnnotationsRequest
from datadog_api_client.v2.model.llm_obs_annotations_type import LLMObsAnnotationsType
from datadog_api_client.v2.model.llm_obs_upsert_annotation_item import LLMObsUpsertAnnotationItem

body = LLMObsAnnotationsRequest(
    data=LLMObsAnnotationsDataRequest(
        attributes=LLMObsAnnotationsDataAttributesRequest(
            annotations=[
                LLMObsUpsertAnnotationItem(
                    interaction_id="00000000-0000-0000-0000-000000000001",
                    label_values=[
                        LLMObsAnnotationLabelValue(
                            label_schema_id="abc-123",
                            value="good",
                        ),
                        LLMObsAnnotationLabelValue(
                            label_schema_id="ef56gh78",
                            value="positive",
                        ),
                    ],
                ),
            ],
        ),
        type=LLMObsAnnotationsType.ANNOTATIONS,
    ),
)

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

    print(response)

# Create or update annotations returns "OK — annotations created or updated. For mixed batches, denied items and other
# per-item errors are listed in `errors`." response

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

body = DatadogAPIClient::V2::LLMObsAnnotationsRequest.new({
  data: DatadogAPIClient::V2::LLMObsAnnotationsDataRequest.new({
    attributes: DatadogAPIClient::V2::LLMObsAnnotationsDataAttributesRequest.new({
      annotations: [
        DatadogAPIClient::V2::LLMObsUpsertAnnotationItem.new({
          interaction_id: "00000000-0000-0000-0000-000000000001",
          label_values: [
            DatadogAPIClient::V2::LLMObsAnnotationLabelValue.new({
              label_schema_id: "abc-123",
              value: "good",
            }),
            DatadogAPIClient::V2::LLMObsAnnotationLabelValue.new({
              label_schema_id: "ef56gh78",
              value: "positive",
            }),
          ],
        }),
      ],
    }),
    type: DatadogAPIClient::V2::LLMObsAnnotationsType::ANNOTATIONS,
  }),
})
p api_instance.upsert_llm_obs_annotations("queue_id", body)

// Create or update annotations returns "OK — annotations created or updated. For
// mixed batches, denied items and other per-item errors are listed in `errors`."
// response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_agent_observability::AgentObservabilityAPI;
use datadog_api_client::datadogV2::model::LLMObsAnnotationLabelValue;
use datadog_api_client::datadogV2::model::LLMObsAnnotationLabelValueValue;
use datadog_api_client::datadogV2::model::LLMObsAnnotationsDataAttributesRequest;
use datadog_api_client::datadogV2::model::LLMObsAnnotationsDataRequest;
use datadog_api_client::datadogV2::model::LLMObsAnnotationsRequest;
use datadog_api_client::datadogV2::model::LLMObsAnnotationsType;
use datadog_api_client::datadogV2::model::LLMObsUpsertAnnotationItem;

#[tokio::main]
async fn main() {
    let body = LLMObsAnnotationsRequest::new(LLMObsAnnotationsDataRequest::new(
        LLMObsAnnotationsDataAttributesRequest::new(vec![LLMObsUpsertAnnotationItem::new(
            "00000000-0000-0000-0000-000000000001".to_string(),
            vec![
                LLMObsAnnotationLabelValue::new(
                    "abc-123".to_string(),
                    LLMObsAnnotationLabelValueValue::AnyValueString("good".to_string()),
                ),
                LLMObsAnnotationLabelValue::new(
                    "ef56gh78".to_string(),
                    LLMObsAnnotationLabelValueValue::AnyValueString("positive".to_string()),
                ),
            ],
        )]),
        LLMObsAnnotationsType::ANNOTATIONS,
    ));
    let mut configuration = datadog::Configuration::new();
    configuration.set_unstable_operation_enabled("v2.upsert_llm_obs_annotations", true);
    let api = AgentObservabilityAPI::with_config(configuration);
    let resp = api
        .upsert_llm_obs_annotations("queue_id".to_string(), body)
        .await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Create or update annotations returns "OK — annotations created or updated. For mixed batches, denied items and other
 * per-item errors are listed in `errors`." response
 */

import { client, v2 } from "@datadog/datadog-api-client";

const configuration = client.createConfiguration({
  authMethods: {
    AuthZ: {
      accessToken: process.env.DD_BEARER_TOKEN as string,
    },
  },
});
configuration.unstableOperations["v2.upsertLLMObsAnnotations"] = true;
const apiInstance = new v2.AgentObservabilityApi(configuration);

const params: v2.AgentObservabilityApiUpsertLLMObsAnnotationsRequest = {
  body: {
    data: {
      attributes: {
        annotations: [
          {
            interactionId: "00000000-0000-0000-0000-000000000001",
            labelValues: [
              {
                labelSchemaId: "abc-123",
                value: "good",
              },
              {
                labelSchemaId: "ef56gh78",
                value: "positive",
              },
            ],
          },
        ],
      },
      type: "annotations",
    },
  },
  queueId: "queue_id",
};

apiInstance
  .upsertLLMObsAnnotations(params)
  .then((data: v2.LLMObsAnnotationsResponse) => {
    console.log(
      "API called successfully. Returned data: " + JSON.stringify(data)
    );
  })
  .catch((error: any) => console.error(error));