otel add psm

Author: JqRrt

backend/modules/observability/application/convertor/trace/trajectory.go

@@ -135,6 +135,8 @@ func MetricsInfoDO2DTO(info *loop_span.MetricsInfo) *dtotrajectory.MetricsInfo {
 		ModelErrors:        info.ModelErrors,
 		ModelErrorRate:     info.ModelErrorRate,
 		ToolStepProportion: info.ToolStepProportion,
+		InputTokens:        info.InputTokens,
+		OutputTokens:       info.OutputTokens,
 	}
 }
 

backend/modules/observability/domain/trace/entity/loop_span/trajectory.go

@@ -138,14 +138,13 @@ func BuildTrajectoryFromSpans(spanList SpanList) *Trajectory {
 		spanMap[span.SpanID] = span
 	}
 
-	var trajectoryID *string
+	trajectoryID := ptr.Of(spanList[0].TraceID)
 
 	// 找到root节点
 	var rootSpan *Span
 	for _, span := range spanList {
 		if span.ParentID == "" || span.ParentID == "0" {
 			rootSpan = span
-			trajectoryID = &span.SpanID
 			break
 		}
 	}
@@ -181,9 +180,6 @@ func BuildTrajectoryFromSpans(spanList SpanList) *Trajectory {
 		if agentSpan == nil {
 			continue
 		}
-		if trajectoryID == nil {
-			trajectoryID = &agentSpan.SpanID
-		}
 		agentStep := &AgentStep{
 			ID:        &agentSpan.SpanID,
 			ParentID:  &agentSpan.ParentID,

backend/modules/observability/lib/otel/otel_convert.go

@@ -66,9 +66,9 @@ var (
 			DataType:  dataTypeString,
 		},
 		"psm": {
-			attributeKey: []string{"service.name"},
-			isTag:        false,
-			dataType:     dataTypeString,
+			AttributeKey: []string{"service.name"},
+			IsTag:        false,
+			DataType:     dataTypeString,
 		},
 
 		// model