loop_span

type AgentStep struct {
	// 基础属性
	ID *string `json:"id,omitempty"`
	// 父ID， trace导入时取parent_span_id
	ParentID *string `json:"parent_id,omitempty"`
	// name，trace导入时取span_name
	Name *string `json:"name,omitempty"`
	// 输入
	Input *string `json:"input,omitempty"`
	// 输出
	Output *string `json:"output,omitempty"`
	// 子节点，agent执行内部经历了哪些步骤
	Steps []*Step `json:"steps"`
	// 系统属性
	Metadata    map[string]string `json:"metadata,omitempty"`
	BasicInfo   *BasicInfo        `json:"basic_info,omitempty"`
	MetricsInfo *MetricsInfo      `json:"metrics_info,omitempty"`
}

type Annotation struct {
	ID              string
	SpanID          string
	TraceID         string
	StartTime       time.Time // start time of span
	WorkspaceID     string
	AnnotationType  AnnotationType
	AnnotationIndex []string
	Key             string
	Value           AnnotationValue
	Reasoning       string
	Corrections     []AnnotationCorrection
	Metadata        any
	Status          AnnotationStatus
	CreatedAt       time.Time
	CreatedBy       string
	UpdatedAt       time.Time
	UpdatedBy       string
	IsDeleted       bool
}

type AnnotationCorrection struct {
	Reasoning string                   `json:"reasoning,omitempty"`
	Value     AnnotationValue          `json:"value"`
	Type      AnnotationCorrectionType `json:"type"`
	UpdateAt  time.Time                `json:"update_at"`
	UpdatedBy string                   `json:"updated_by"`
}

type AnnotationCorrectionType string

type AnnotationList []*Annotation

type AnnotationManualFeedback struct {
	TagKeyId   int64  // 标签Key的ID
	TagKeyName string // 标签Key的名称
	TagValueId *int64 // 标签值的名称，自由文本/数值没有ID
	TagValue   string // 显示的标签值
}

type AnnotationStatus string

type AnnotationType string

type AnnotationValue struct {
	ValueType   AnnotationValueType `json:"value_type,omitempty"` // 类型
	LongValue   int64               `json:"long_value,omitempty"`
	StringValue string              `json:"string_value,omitempty"`
	FloatValue  float64             `json:"float_value,omitempty"`
	BoolValue   bool                `json:"bool_value,omitempty"`
}

type AnnotationValueType string

type Attachment struct {
	Field  string `json:"field"`
	Name   string `json:"name"`
	Type   string `json:"type"`
	TosKey string `json:"tos_key"`
}

type AttrTos struct {
	InputDataURL   string
	OutputDataURL  string
	MultimodalData map[string]string
}

type AutoEvaluateMetadata struct {
	TaskID             int64 `json:"task_id"`
	EvaluatorRecordID  int64 `json:"evaluator_record_id"`
	EvaluatorVersionID int64 `json:"evaluator_version_id"`
}

type BasicInfo struct {
	// 单位毫秒
	StartedAt string `json:"started_at"`
	// 单位毫秒
	Duration string `json:"duration"`
	Error    *Error `json:"error,omitempty"`
}

type Error struct {
	Code int32  `json:"code"`
	Msg  string `json:"msg"`
}

type Field struct {
	// contains filtered or unexported fields
}

type FieldOptions struct {
	I64List    []int64   `mapstructure:"i64_list" json:"i64_list"`
	F64List    []float64 `mapstructure:"f64_list" json:"f64_list"`
	StringList []string  `mapstructure:"string_list" json:"string_list"`
}

type FieldType string

type FilterField struct {
	FieldName  string            `mapstructure:"field_name" json:"field_name"`
	FieldType  FieldType         `mapstructure:"field_type" json:"field_type"`
	Values     []string          `mapstructure:"values" json:"values"`
	QueryType  *QueryTypeEnum    `mapstructure:"query_type" json:"query_type"`
	QueryAndOr *QueryAndOrEnum   `mapstructure:"query_and_or" json:"query_and_or"`
	SubFilter  *FilterFields     `mapstructure:"sub_filter" json:"sub_filter"`
	IsSystem   bool              `mapstructure:"is_system" json:"is_system"`
	IsCustom   bool              `mapstructure:"is_custom" json:"is_custom"`
	Hidden     bool              `mapstructure:"hidden" json:"hidden"`
	ExtraInfo  map[string]string `mapstructure:"extra_info" json:"extra_info"`
}

type FilterFields struct {
	QueryAndOr   *QueryAndOrEnum `mapstructure:"query_and_or" json:"query_and_or"`
	FilterFields []*FilterField  `mapstructure:"filter_fields" json:"filter_fields"`
}

type FilterObject interface {
	GetFieldValue(fieldName string, isSystem, isCustom bool) any
}

type InputFilter struct {
	KeyWhiteList []string `mapstructure:"key_white_list" json:"key_white_list"`
	// contains filtered or unexported fields
}

type ManualDatasetMetadata struct{}

type MetricsInfo struct {
	// 单位毫秒
	LlmDuration *string `json:"llm_duration,omitempty"`
	// 单位毫秒
	ToolDuration *string `json:"tool_duration,omitempty"`
	// Tool错误分布，格式为：错误码-->list<ToolStepID>
	ToolErrors map[int32][]string `json:"tool_errors,omitempty"`
	// Tool错误率
	ToolErrorRate *float64 `json:"tool_error_rate,omitempty"`
	// Model错误分布，格式为：错误码-->list<ModelStepID>
	ModelErrors map[int32][]string `json:"model_errors,omitempty"`
	// Model错误率
	ModelErrorRate *float64 `json:"model_error_rate,omitempty"`
	// Tool Step占比(分母是总子Step)
	ToolStepProportion *float64 `json:"tool_step_proportion,omitempty"`
	// 输入token数
	InputTokens *int32 `json:"input_tokens,omitempty"`
	// 输出token数
	OutputTokens *int32 `json:"output_tokens,omitempty"`
}

type ModelInfo struct {
	InputTokens               int64  `json:"input_tokens"`
	OutputTokens              int64  `json:"output_tokens"`
	LatencyFirstResp          string `json:"latency_first_resp"` // 单位毫秒
	ReasoningTokens           int64  `json:"reasoning_tokens"`
	InputReadCachedTokens     int64  `json:"input_read_cached_tokens"`
	InputCreationCachedTokens int64  `json:"input_creation_cached_tokens"`
}

type ObjectStorage struct {
	InputTosKey  string        `json:"input_tos_key"`
	OutputTosKey string        `json:"output_tos_key"`
	Attachments  []*Attachment `json:"Attachments"`
}

type OutputFilter struct {
	KeyWhiteList []string `mapstructure:"key_white_list" json:"key_white_list"`
	// contains filtered or unexported fields
}

type PlatformType string

type QueryAndOrEnum string

type QueryTypeEnum string

type RootStep struct {
	// 唯一ID，trace导入时取span_id
	ID *string `json:"id,omitempty"`
	// name，trace导入时取span_name
	Name *string `json:"name,omitempty"`
	// 输入
	Input *string `json:"input,omitempty"`
	// 输出
	Output *string `json:"output,omitempty"`
	// 系统属性
	Metadata    map[string]string `json:"metadata,omitempty"`
	BasicInfo   *BasicInfo        `json:"basic_info,omitempty"`
	MetricsInfo *MetricsInfo      `json:"metrics_info,omitempty"`
}

type Span struct {
	StartTime      int64  `json:"start_time"` // us
	SpanID         string `json:"span_id"`
	ParentID       string `json:"parent_id"`
	TraceID        string `json:"trace_id"`
	DurationMicros int64  `json:"duration_micros"` // us
	CallType       string `json:"call_type"`
	PSM            string `json:"psm"`
	LogID          string `json:"log_id"`
	WorkspaceID    string `json:"space_id"`
	SpanName       string `json:"span_name"`
	SpanType       string `json:"span_type"`
	Method         string `json:"method"`
	StatusCode     int32  `json:"status_code"`
	Input          string `json:"input"`
	Output         string `json:"output"`
	ObjectStorage  string `json:"object_storage"`

	SystemTagsString map[string]string  `json:"system_tags_string"`
	SystemTagsLong   map[string]int64   `json:"system_tags_long"`
	SystemTagsDouble map[string]float64 `json:"system_tags_double"`

	TagsString map[string]string  `json:"tags_string"`
	TagsLong   map[string]int64   `json:"tags_long"`
	TagsDouble map[string]float64 `json:"tags_double"`

	TagsBool map[string]bool   `json:"tags_bool"`
	TagsByte map[string]string `json:"tags_byte"`

	AttrTos         *AttrTos       `json:"-"`
	LogicDeleteTime int64          `json:"-"` // us
	Annotations     AnnotationList `json:"-"`
}

type SpanList []*Span

type SpanListType string

type SpanTransCfgList []*SpanTransConfig

type SpanTransConfig struct {
	SpanFilter   *FilterFields `mapstructure:"span_filter" json:"span_filter"`
	TagFilter    *TagFilter    `mapstructure:"tag_filter" json:"tag_filter"`
	InputFilter  *InputFilter  `mapstructure:"input_filter" json:"input_filter"`
	OutputFilter *OutputFilter `mapstructure:"output_filter" json:"output_filter"`
	// contains filtered or unexported fields
}

type Step struct {
	// 基础属性
	ID *string `json:"id,omitempty"`
	// 父ID， trace导入时取parent_span_id
	ParentID *string `json:"parent_id,omitempty"`
	// 类型
	Type *StepType `json:"type,omitempty"`
	// name，trace导入时取span_name
	Name *string `json:"name,omitempty"`
	// 输入
	Input *string `json:"input,omitempty"`
	// 输出
	Output *string `json:"output,omitempty"`
	// 各种类型补充信息
	ModelInfo *ModelInfo `json:"model_info,omitempty"`
	// 系统属性
	Metadata  map[string]string `json:"metadata,omitempty"`
	BasicInfo *BasicInfo        `json:"basic_info,omitempty"`
}

type StepType = string

type StringWrapper struct {
	Role    string `json:"role"`
	Content string `json:"content"`
	Type    string `json:"type"`
}

type Struct struct {
	// contains filtered or unexported fields
}

type TTL string

type Tag struct {
	ValueType TagValueType
	Key       string
	Value     *TagValue
}

type TagFilter struct {
	KeyBlackList []string `mapstructure:"key_black_list" json:"key_black_list"`
	// contains filtered or unexported fields
}

type TagValue struct {
	VBool   *bool
	VLong   *int64
	VDouble *float64
	VStr    *string
}

type TagValueType int

type TraceAdvanceInfo struct {
	TraceId    string
	InputCost  int64
	OutputCost int64
}

type Trajectory struct {
	// trace_id
	ID *string `json:"id,omitempty"`
	// 根节点，记录整个轨迹的信息
	RootStep *RootStep `json:"root_step,omitempty"`
	// agent step列表，记录轨迹中agent执行信息
	AgentSteps []*AgentStep `json:"agent_steps"`
}

type TrajectoryList []*Trajectory