Update a specific LLM Observability prompt version

// Update a specific LLM Observability prompt version returns "OK" response

package main

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

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

func main() {
	// there is a valid "prompt" in the system
	PromptDataAttributesPromptID := os.Getenv("PROMPT_DATA_ATTRIBUTES_PROMPT_ID")

	// there is a valid "prompt_version" in the system
	PromptVersionDataAttributesVersion, _ := strconv.ParseInt(os.Getenv("PROMPT_VERSION_DATA_ATTRIBUTES_VERSION"), 10, 64)

	body := datadogV2.LLMObsUpdatePromptVersionRequest{
		Data: datadogV2.LLMObsUpdatePromptVersionData{
			Attributes: datadogV2.LLMObsUpdatePromptVersionDataAttributes{
				Description: datadog.PtrString("Give concise answers and cite relevant help-center articles."),
			},
			Type: datadogV2.LLMOBSPROMPTVERSIONTYPE_PROMPT_TEMPLATE_VERSIONS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.UpdateLLMObsPromptVersion", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewLLMObservabilityApi(apiClient)
	resp, r, err := api.UpdateLLMObsPromptVersion(ctx, PromptDataAttributesPromptID, PromptVersionDataAttributesVersion, body)

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

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

// Update a specific LLM Observability prompt version returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.LlmObservabilityApi;
import com.datadog.api.client.v2.model.LLMObsPromptVersionResponse;
import com.datadog.api.client.v2.model.LLMObsPromptVersionType;
import com.datadog.api.client.v2.model.LLMObsUpdatePromptVersionData;
import com.datadog.api.client.v2.model.LLMObsUpdatePromptVersionDataAttributes;
import com.datadog.api.client.v2.model.LLMObsUpdatePromptVersionRequest;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setUnstableOperationEnabled("v2.updateLLMObsPromptVersion", true);
    LlmObservabilityApi apiInstance = new LlmObservabilityApi(defaultClient);

    // there is a valid "prompt" in the system
    String PROMPT_DATA_ATTRIBUTES_PROMPT_ID = System.getenv("PROMPT_DATA_ATTRIBUTES_PROMPT_ID");

    // there is a valid "prompt_version" in the system
    Long PROMPT_VERSION_DATA_ATTRIBUTES_VERSION =
        Long.parseLong(System.getenv("PROMPT_VERSION_DATA_ATTRIBUTES_VERSION"));

    LLMObsUpdatePromptVersionRequest body =
        new LLMObsUpdatePromptVersionRequest()
            .data(
                new LLMObsUpdatePromptVersionData()
                    .attributes(
                        new LLMObsUpdatePromptVersionDataAttributes()
                            .description(
                                "Give concise answers and cite relevant help-center articles."))
                    .type(LLMObsPromptVersionType.PROMPT_TEMPLATE_VERSIONS));

    try {
      LLMObsPromptVersionResponse result =
          apiInstance.updateLLMObsPromptVersion(
              PROMPT_DATA_ATTRIBUTES_PROMPT_ID, PROMPT_VERSION_DATA_ATTRIBUTES_VERSION, body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling LlmObservabilityApi#updateLLMObsPromptVersion");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

"""
Update a specific LLM Observability prompt version returns "OK" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.llm_observability_api import LLMObservabilityApi
from datadog_api_client.v2.model.llm_obs_prompt_version_type import LLMObsPromptVersionType
from datadog_api_client.v2.model.llm_obs_update_prompt_version_data import LLMObsUpdatePromptVersionData
from datadog_api_client.v2.model.llm_obs_update_prompt_version_data_attributes import (
    LLMObsUpdatePromptVersionDataAttributes,
)
from datadog_api_client.v2.model.llm_obs_update_prompt_version_request import LLMObsUpdatePromptVersionRequest

# there is a valid "prompt" in the system
PROMPT_DATA_ATTRIBUTES_PROMPT_ID = environ["PROMPT_DATA_ATTRIBUTES_PROMPT_ID"]

# there is a valid "prompt_version" in the system
PROMPT_VERSION_DATA_ATTRIBUTES_VERSION = environ["PROMPT_VERSION_DATA_ATTRIBUTES_VERSION"]

body = LLMObsUpdatePromptVersionRequest(
    data=LLMObsUpdatePromptVersionData(
        attributes=LLMObsUpdatePromptVersionDataAttributes(
            description="Give concise answers and cite relevant help-center articles.",
        ),
        type=LLMObsPromptVersionType.PROMPT_TEMPLATE_VERSIONS,
    ),
)

configuration = Configuration()
configuration.unstable_operations["update_llm_obs_prompt_version"] = True
with ApiClient(configuration) as api_client:
    api_instance = LLMObservabilityApi(api_client)
    response = api_instance.update_llm_obs_prompt_version(
        prompt_id=PROMPT_DATA_ATTRIBUTES_PROMPT_ID, version=int(PROMPT_VERSION_DATA_ATTRIBUTES_VERSION), body=body
    )

    print(response)

# Update a specific LLM Observability prompt version returns "OK" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.unstable_operations["v2.update_llm_obs_prompt_version".to_sym] = true
end
api_instance = DatadogAPIClient::V2::LLMObservabilityAPI.new

# there is a valid "prompt" in the system
PROMPT_DATA_ATTRIBUTES_PROMPT_ID = ENV["PROMPT_DATA_ATTRIBUTES_PROMPT_ID"]

# there is a valid "prompt_version" in the system
PROMPT_VERSION_DATA_ATTRIBUTES_VERSION = ENV["PROMPT_VERSION_DATA_ATTRIBUTES_VERSION"]

body = DatadogAPIClient::V2::LLMObsUpdatePromptVersionRequest.new({
  data: DatadogAPIClient::V2::LLMObsUpdatePromptVersionData.new({
    attributes: DatadogAPIClient::V2::LLMObsUpdatePromptVersionDataAttributes.new({
      description: "Give concise answers and cite relevant help-center articles.",
    }),
    type: DatadogAPIClient::V2::LLMObsPromptVersionType::PROMPT_TEMPLATE_VERSIONS,
  }),
})
p api_instance.update_llm_obs_prompt_version(PROMPT_DATA_ATTRIBUTES_PROMPT_ID, PROMPT_VERSION_DATA_ATTRIBUTES_VERSION.to_i, body)

// Update a specific LLM Observability prompt version returns "OK" response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_llm_observability::LLMObservabilityAPI;
use datadog_api_client::datadogV2::model::LLMObsPromptVersionType;
use datadog_api_client::datadogV2::model::LLMObsUpdatePromptVersionData;
use datadog_api_client::datadogV2::model::LLMObsUpdatePromptVersionDataAttributes;
use datadog_api_client::datadogV2::model::LLMObsUpdatePromptVersionRequest;

#[tokio::main]
async fn main() {
    // there is a valid "prompt" in the system
    let prompt_data_attributes_prompt_id =
        std::env::var("PROMPT_DATA_ATTRIBUTES_PROMPT_ID").unwrap();

    // there is a valid "prompt_version" in the system
    let prompt_version_data_attributes_version: i64 =
        std::env::var("PROMPT_VERSION_DATA_ATTRIBUTES_VERSION")
            .unwrap()
            .parse()
            .unwrap();
    let body = LLMObsUpdatePromptVersionRequest::new(LLMObsUpdatePromptVersionData::new(
        LLMObsUpdatePromptVersionDataAttributes::new().description(
            "Give concise answers and cite relevant help-center articles.".to_string(),
        ),
        LLMObsPromptVersionType::PROMPT_TEMPLATE_VERSIONS,
    ));
    let mut configuration = datadog::Configuration::new();
    configuration.set_unstable_operation_enabled("v2.UpdateLLMObsPromptVersion", true);
    let api = LLMObservabilityAPI::with_config(configuration);
    let resp = api
        .update_llm_obs_prompt_version(
            prompt_data_attributes_prompt_id.clone(),
            prompt_version_data_attributes_version.clone(),
            body,
        )
        .await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Update a specific LLM Observability prompt version returns "OK" response
 */

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

const configuration = client.createConfiguration();
configuration.unstableOperations["v2.updateLLMObsPromptVersion"] = true;
const apiInstance = new v2.LLMObservabilityApi(configuration);

// there is a valid "prompt" in the system
const PROMPT_DATA_ATTRIBUTES_PROMPT_ID = process.env
  .PROMPT_DATA_ATTRIBUTES_PROMPT_ID as string;

// there is a valid "prompt_version" in the system
const PROMPT_VERSION_DATA_ATTRIBUTES_VERSION = parseInt(
  process.env.PROMPT_VERSION_DATA_ATTRIBUTES_VERSION as string
);

const params: v2.LLMObservabilityApiUpdateLLMObsPromptVersionRequest = {
  body: {
    data: {
      attributes: {
        description:
          "Give concise answers and cite relevant help-center articles.",
      },
      type: "prompt-template-versions",
    },
  },
  promptId: PROMPT_DATA_ATTRIBUTES_PROMPT_ID,
  version: PROMPT_VERSION_DATA_ATTRIBUTES_VERSION,
};

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