go fmt

Author: shunjiazhu

examples/observability/agent.go

@@ -21,8 +21,7 @@ import (
 
 func main() {
 	ctx := context.Background()
-
-	// Important: Always call Shutdown to flush spans and metrics
+	// Shutdown to flush spans and metrics
 	defer observability.Shutdown(ctx)
 
 	cfg := &veagent.Config{

observability/constant.go

@@ -86,7 +86,7 @@ const (
 	AttrSessionId          = "session.id"    // Alias of gen_ai.session.id for CozeLoop/TLS platforms
 	AttrInvocationId       = "invocation.id" // Alias of gen_ai.invocation.id for CozeLoop platform
 
-	AttrErrorType          = "error.type"
+	AttrErrorType            = "error.type"
 	AttrCozeloopReportSource = "cozeloop.report.source" // Fixed value: veadk
 	AttrCozeloopCallType     = "cozeloop.call_type"     // CozeLoop call type
 
@@ -153,14 +153,14 @@ const (
 
 // Tool attributes
 const (
-	AttrGenAIOperationName = "gen_ai.operation.name"
-	AttrGenAIOperationType = "gen_ai.operation.type"
-	AttrGenAIToolName      = "gen_ai.tool.name"
+	AttrGenAIOperationName   = "gen_ai.operation.name"
+	AttrGenAIOperationType   = "gen_ai.operation.type"
+	AttrGenAIToolName        = "gen_ai.tool.name"
 	AttrGenAIToolDescription = "gen_ai.tool.description"
-	AttrGenAIToolInput     = "gen_ai.tool.input"
-	AttrGenAIToolOutput    = "gen_ai.tool.output"
-	AttrGenAISpanKind      = "gen_ai.span.kind"
-	AttrGenAIToolCallID    = "gen_ai.tool.call.id"
+	AttrGenAIToolInput       = "gen_ai.tool.input"
+	AttrGenAIToolOutput      = "gen_ai.tool.output"
+	AttrGenAISpanKind        = "gen_ai.span.kind"
+	AttrGenAIToolCallID      = "gen_ai.tool.call.id"
 
 	// Platform specific
 	AttrCozeloopInput  = "cozeloop.input"

observability/registry.go

@@ -286,7 +286,7 @@ func (r *TraceRegistry) ScheduleCleanup(adkTraceID trace.TraceID, internalRunID
 		adkTraceID:    adkTraceID,
 		internalRunID: internalRunID,
 		veadkSpanID:   veadkSpanID,
-		deadline:      time.Now().Add(5 * time.Minute),
+		deadline:      time.Now().Add(2 * time.Minute),
 	}:
 	default:
 		log.Warn("trace cleanup queue is full")