Function Example

pkg/telemetry/integration_example.go:11–94

Example 演示如何在 Agent 中集成 Telemetry

()

// Example 演示如何在 Agent 中集成 Telemetry
func Example() {
	// 1. 初始化 Tracer 和 Metrics
	tracer := NewSimpleTracer()
	metrics := NewSimpleMetrics()
	agentMetrics := NewAgentMetrics(metrics)

	// 设置为全局实例
	SetGlobalTracer(tracer)
	SetGlobalMetrics(metrics)

	// 2. 模拟 Agent 执行
	ctx := context.Background()
	agentID := "agent-001"

	// 开始追踪
	ctx, span := tracer.StartSpan(ctx, "agent.chat",
		WithSpanKind(SpanKindServer),
		WithAttributes(
			String("agent.id", agentID),
			String("user.id", "user-123"),
		),
	)
	defer span.End()

	startTime := time.Now()

	// 3. 执行业务逻辑
	span.AddEvent("processing_message", String("message", "Hello"))

	// 模拟模型调用
	_, modelSpan := tracer.StartSpan(ctx, "model.generate")
	modelSpan.SetAttributes(
		String("model", "claude-3-5-sonnet"),
		Int("input_tokens", 100),
	)
	time.Sleep(100 * time.Millisecond) // 模拟延迟
	modelSpan.End()

	// 记录 token 使用
	agentMetrics.RecordTokens(agentID, 100, 200)

	// 模拟工具调用
	_, toolSpan := tracer.StartSpan(ctx, "tool.execute")
	toolSpan.SetAttributes(String("tool.name", "search"))
	time.Sleep(50 * time.Millisecond)
	toolSpan.End()

	// 记录工具调用
	agentMetrics.RecordToolCall(agentID, "search", 50*time.Millisecond, true)

	// 4. 记录请求完成
	duration := time.Since(startTime)
	agentMetrics.RecordRequest(agentID, duration)
	span.SetStatus(StatusCodeOK, "completed")

	// 5. 输出指标快照
	snapshot := metrics.Snapshot()
	fmt.Printf("\n📊 Metrics Snapshot:\n")