observability_trace_service.go

// Copyright (c) 2025 coze-dev Authors
// SPDX-License-Identifier: Apache-2.0

// Code generated by hertz generator.

package apis

import (
	"context"

	"github.com/cloudwego/hertz/pkg/app"
	"github.com/cloudwego/hertz/pkg/protocol/consts"

	"github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/apis/observabilitytraceservice"
	trace "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/trace"
)

var observabilityClient observabilitytraceservice.Client

// ListSpans .
// @router /api/observability/v1/traces/spans [POST]
func ListSpans(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListSpans)
}

// GetTrace .
// @router /api/observability/v1/trace/:trace_id [GET]
func GetTrace(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.GetTrace)
}

// BatchGetTracesAdvanceInfo .
// @router /api/observability/v1/traces/batch_get_advance_info [POST]
func BatchGetTracesAdvanceInfo(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.BatchGetTracesAdvanceInfo)
}

// GetTracesMetaInfo .
// @router /api/observability/v1/traces/meta_info [GET]
func GetTracesMetaInfo(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.GetTracesMetaInfo)
}

// CreateView .
// @router /api/observability/v1/views/create_view [POST]
func CreateView(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.CreateView)
}

// UpdateView .
// @router /api/observability/v1/views/update_view [POST]
func UpdateView(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.UpdateView)
}

// DeleteView .
// @router /api/observability/v1/views/delete_view [POST]
func DeleteView(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.DeleteView)
}

// ListViews .
// @router /api/observability/v1/views/list_view [POST]
func ListViews(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListViews)
}

// CreateManualAnnotation .
// @router /api/observability/v1/annotations [POST]
func CreateManualAnnotation(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.CreateManualAnnotation)
}

// UpdateManualAnnotation .
// @router /api/observability/v1/annotations/:annotation_id [PUT]
func UpdateManualAnnotation(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.UpdateManualAnnotation)
}

// DeleteManualAnnotation .
// @router /api/observability/v1/annotations/:annotation_id [DELETE]
func DeleteManualAnnotation(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.DeleteManualAnnotation)
}

// ListAnnotations .
// @router /api/observability/v1/annotations/list [POST]
func ListAnnotations(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListAnnotations)
}

// ExportTracesToDataset .
// @router /api/observation/v1/traces/export_to_dataset [POST]
func ExportTracesToDataset(ctx context.Context, c *app.RequestContext) {
	var err error
	var req trace.ExportTracesToDatasetRequest
	err = c.BindAndValidate(&req)
	if err != nil {
		c.String(consts.StatusBadRequest, err.Error())
		return
	}

	resp := new(trace.ExportTracesToDatasetResponse)

	c.JSON(consts.StatusOK, resp)
}

// PreviewExportTracesToDataset .
// @router /api/observation/v1/traces/preview_export_to_dataset [POST]
func PreviewExportTracesToDataset(ctx context.Context, c *app.RequestContext) {
	var err error
	var req trace.PreviewExportTracesToDatasetRequest
	err = c.BindAndValidate(&req)
	if err != nil {
		c.String(consts.StatusBadRequest, err.Error())
		return
	}

	resp := new(trace.PreviewExportTracesToDatasetResponse)

	c.JSON(consts.StatusOK, resp)
}

// SearchTraceTree .
// @router /api/observability/v1/traces/search_tree [POST]
func SearchTraceTree(ctx context.Context, c *app.RequestContext) {
	var err error
	var req trace.SearchTraceTreeRequest
	err = c.BindAndValidate(&req)
	if err != nil {
		c.String(consts.StatusBadRequest, err.Error())
		return
	}

	resp := new(trace.SearchTraceTreeResponse)

	c.JSON(consts.StatusOK, resp)
}

// ChangeEvaluatorScore .
// @router /api/observability/v1/annotations/change_eEvaluator_sScore [POST]
func ChangeEvaluatorScore(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ChangeEvaluatorScore)
}

// ListAnnotationEvaluators .
// @router /api/observability/v1/annotations/lis_annotation_evaluators [POST]
func ListAnnotationEvaluators(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListAnnotationEvaluators)
}

// ExtractSpanInfo .
// @router /api/observability/v1/traces/extract_span_info [POST]
func ExtractSpanInfo(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ExtractSpanInfo)
}

// ListPreSpan .
// @router /api/observability/v1/spans/pre_list [POST]
func ListPreSpan(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListPreSpan)
}

// UpsertTrajectoryConfig .
// @router /api/observability/v1/trajectory_config [POST]
func UpsertTrajectoryConfig(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.UpsertTrajectoryConfig)
}

// GetTrajectoryConfig .
// @router /api/observability/v1/trajectory_config [GET]
func GetTrajectoryConfig(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.GetTrajectoryConfig)
}

// ListTrajectory .
// @router /api/observability/v1/traces/trajectory [POST]
func ListTrajectory(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListTrajectory)
}

// ListWorkspaceAnnotations .
// @router /api/observability/v1/annotations/list_by_workspace [POST]
func ListWorkspaceAnnotations(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListWorkspaceAnnotations)
}

// ListMetadata .
// @router /api/observability/v1/traces/metadata/list [POST]
func ListMetadata(ctx context.Context, c *app.RequestContext) {
	invokeAndRender(ctx, c, observabilityClient.ListMetadata)
}