Add integration field to StartSpanConfig and related functions

Author: jboolean

internal/llmobs/llmobs.go

@@ -688,6 +688,7 @@ func (l *LLMObs) StartSpan(ctx context.Context, kind SpanKind, name string, cfg
 	span.mlApp = cfg.MLApp
 	span.spanKind = kind
 	span.sessionID = cfg.SessionID
+	span.integration = cfg.Integration
 
 	span.llmCtx = llmobsContext{
 		modelName:     cfg.ModelName,

internal/llmobs/span.go

@@ -30,6 +30,8 @@ type StartSpanConfig struct {
 	MLApp string
 	// StartTime sets a custom start time for the span. If zero, uses current time.
 	StartTime time.Time
+	// Name of the tracing integration.
+	Integration string
 }
 
 // FinishSpanConfig contains configuration options for finishing an LLMObs span.

llmobs/option.go

@@ -68,6 +68,12 @@ func WithModelName(modelName string) StartSpanOption {
 	}
 }
 
+func WithIntegration(integration string) StartSpanOption {
+	return func(c *illmobs.StartSpanConfig) {
+		c.Integration = integration
+	}
+}
+
 // ------------- Finish options -------------
 
 // FinishSpanOption configures span finishing. Use with span.Finish().