Agent Observability spans を検索します

"""
Search Agent Observability spans returns "OK" response
"""

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_search_spans_request import LLMObsSearchSpansRequest
from datadog_api_client.v2.model.llm_obs_search_spans_request_attributes import LLMObsSearchSpansRequestAttributes
from datadog_api_client.v2.model.llm_obs_search_spans_request_data import LLMObsSearchSpansRequestData
from datadog_api_client.v2.model.llm_obs_search_spans_request_type import LLMObsSearchSpansRequestType
from datadog_api_client.v2.model.llm_obs_span_filter import LLMObsSpanFilter
from datadog_api_client.v2.model.llm_obs_span_page_query import LLMObsSpanPageQuery
from datadog_api_client.v2.model.llm_obs_span_search_options import LLMObsSpanSearchOptions

body = LLMObsSearchSpansRequest(
    data=LLMObsSearchSpansRequestData(
        attributes=LLMObsSearchSpansRequestAttributes(
            filter=LLMObsSpanFilter(
                _from="now-900s",
                ml_app="my-llm-app",
                query="@session_id:abc123def456",
                span_id="abc123def456",
                span_kind="llm",
                span_name="llm_call",
                to="now",
                trace_id="trace-9a8b7c6d5e4f",
            ),
            options=LLMObsSpanSearchOptions(
                include_attachments=True,
                time_offset=0,
            ),
            page=LLMObsSpanPageQuery(
                cursor="eyJzdGFydCI6MTAwfQ==",
                limit=10,
            ),
            sort="-start_ns",
        ),
        type=LLMObsSearchSpansRequestType.SPANS,
    ),
)

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

    print(response)

# Search Agent Observability spans returns "OK" response

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

body = DatadogAPIClient::V2::LLMObsSearchSpansRequest.new({
  data: DatadogAPIClient::V2::LLMObsSearchSpansRequestData.new({
    attributes: DatadogAPIClient::V2::LLMObsSearchSpansRequestAttributes.new({
      filter: DatadogAPIClient::V2::LLMObsSpanFilter.new({
        from: "now-900s",
        ml_app: "my-llm-app",
        query: "@session_id:abc123def456",
        span_id: "abc123def456",
        span_kind: "llm",
        span_name: "llm_call",
        to: "now",
        trace_id: "trace-9a8b7c6d5e4f",
      }),
      options: DatadogAPIClient::V2::LLMObsSpanSearchOptions.new({
        include_attachments: true,
        time_offset: 0,
      }),
      page: DatadogAPIClient::V2::LLMObsSpanPageQuery.new({
        cursor: "eyJzdGFydCI6MTAwfQ==",
        limit: 10,
      }),
      sort: "-start_ns",
    }),
    type: DatadogAPIClient::V2::LLMObsSearchSpansRequestType::SPANS,
  }),
})
p api_instance.search_llm_obs_spans(body)

// Search Agent Observability spans 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.LLMObsSearchSpansRequest{
		Data: datadogV2.LLMObsSearchSpansRequestData{
			Attributes: datadogV2.LLMObsSearchSpansRequestAttributes{
				Filter: &datadogV2.LLMObsSpanFilter{
					From:     datadog.PtrString("now-900s"),
					MlApp:    datadog.PtrString("my-llm-app"),
					Query:    datadog.PtrString("@session_id:abc123def456"),
					SpanId:   datadog.PtrString("abc123def456"),
					SpanKind: datadog.PtrString("llm"),
					SpanName: datadog.PtrString("llm_call"),
					To:       datadog.PtrString("now"),
					TraceId:  datadog.PtrString("trace-9a8b7c6d5e4f"),
				},
				Options: &datadogV2.LLMObsSpanSearchOptions{
					IncludeAttachments: datadog.PtrBool(true),
					TimeOffset:         datadog.PtrInt64(0),
				},
				Page: &datadogV2.LLMObsSpanPageQuery{
					Cursor: datadog.PtrString("eyJzdGFydCI6MTAwfQ=="),
					Limit:  datadog.PtrInt64(10),
				},
				Sort: datadog.PtrString("-start_ns"),
			},
			Type: datadogV2.LLMOBSSEARCHSPANSREQUESTTYPE_SPANS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.SearchLLMObsSpans", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewAgentObservabilityApi(apiClient)
	resp, r, err := api.SearchLLMObsSpans(ctx, body)

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

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

// Search Agent Observability spans returns "OK" 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.LLMObsSearchSpansRequest;
import com.datadog.api.client.v2.model.LLMObsSearchSpansRequestAttributes;
import com.datadog.api.client.v2.model.LLMObsSearchSpansRequestData;
import com.datadog.api.client.v2.model.LLMObsSearchSpansRequestType;
import com.datadog.api.client.v2.model.LLMObsSpanFilter;
import com.datadog.api.client.v2.model.LLMObsSpanPageQuery;
import com.datadog.api.client.v2.model.LLMObsSpanSearchOptions;
import com.datadog.api.client.v2.model.LLMObsSpansResponse;

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

    LLMObsSearchSpansRequest body =
        new LLMObsSearchSpansRequest()
            .data(
                new LLMObsSearchSpansRequestData()
                    .attributes(
                        new LLMObsSearchSpansRequestAttributes()
                            .filter(
                                new LLMObsSpanFilter()
                                    .from("now-900s")
                                    .mlApp("my-llm-app")
                                    .query("@session_id:abc123def456")
                                    .spanId("abc123def456")
                                    .spanKind("llm")
                                    .spanName("llm_call")
                                    .to("now")
                                    .traceId("trace-9a8b7c6d5e4f"))
                            .options(
                                new LLMObsSpanSearchOptions()
                                    .includeAttachments(true)
                                    .timeOffset(0L))
                            .page(
                                new LLMObsSpanPageQuery().cursor("eyJzdGFydCI6MTAwfQ==").limit(10L))
                            .sort("-start_ns"))
                    .type(LLMObsSearchSpansRequestType.SPANS));

    try {
      LLMObsSpansResponse result = apiInstance.searchLLMObsSpans(body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling AgentObservabilityApi#searchLLMObsSpans");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}