comments

Author: nina-kollman

sample-app/recipe_agent_example.go

@@ -15,7 +15,7 @@ var associationProperties = map[string]string{
 }
 
 var abTest = &model.ABTest{
-	VarientsKeys: map[string]bool{
+	VarientKeys: map[string]bool{
 		"variant_a": false,
 		"variant_b": true,
 	},
@@ -167,9 +167,7 @@ func cookingTimeEstimatorTool(ctx context.Context, agent *sdk.Agent, client *ope
 				},
 			},
 		},
-	}, map[string]string{
-		"user_id": "user_67890",
-	})
+	}, associationProperties)
 	defer tool.End()
 
 	prompt := sdk.Prompt{
@@ -238,9 +236,9 @@ func runRecipeAgent() {
 
 	// Create standalone agent with association properties
 	agent := traceloop.NewAgent(ctx, "recipe_generator", sdk.AgentAttributes{
-		Name: "recipe_generator",
+		Name:                  "recipe_generator",
 		AssociationProperties: associationProperties,
-		ABTest: abTest,
+		ABTest:                abTest,
 	})
 	defer agent.End()
 

traceloop-sdk/agent.go

@@ -24,6 +24,7 @@ func (agent *Agent) End() {
 func (agent *Agent) LogPrompt(prompt Prompt) LLMSpan {
 	// Merge workflow and agent association properties
 	contextAttrs := ContextAttributes{
+		AgentName:             &agent.Attributes.Name,
 		AssociationProperties: make(map[string]string),
 	}
 

traceloop-sdk/model/types.go

@@ -3,13 +3,13 @@ package model
 type SpanKind string
 
 const (
-	SpanKindTool SpanKind = "tool"
-	SpanKindAgent SpanKind = "agent"
-	SpanKindTask SpanKind = "task"
+	SpanKindTool     SpanKind = "tool"
+	SpanKindAgent    SpanKind = "agent"
+	SpanKindTask     SpanKind = "task"
 	SpanKindWorkflow SpanKind = "workflow"
 )
 
 // The varient that is active will be added to the trace.
 type ABTest struct {
-	VarientsKeys map[string]bool `json:"varients_keys"`
+	VarientKeys map[string]bool `json:"varient_keys"`
 }

traceloop-sdk/sdk.go

@@ -21,7 +21,6 @@ import (
 
 const PromptsPath = "/v1/traceloop/prompts"
 
-
 type Traceloop struct {
 	config         Config
 	promptRegistry model.PromptRegistry
@@ -30,7 +29,6 @@ type Traceloop struct {
 	http.Client
 }
 
-
 func NewClient(ctx context.Context, config Config) (*Traceloop, error) {
 	instance := Traceloop{
 		config:         config,
@@ -153,7 +151,7 @@ func (instance *Traceloop) NewAgent(ctx context.Context, name string, agentAttrs
 	}
 
 	if agentAttrs.ABTest != nil {
-		for key, activeVarient := range agentAttrs.ABTest.VarientsKeys {
+		for key, activeVarient := range agentAttrs.ABTest.VarientKeys {
 			if activeVarient {
 				agentAttrs.AssociationProperties["ab_testing_variant"] = key
 				break
@@ -169,9 +167,9 @@ func (instance *Traceloop) NewAgent(ctx context.Context, name string, agentAttrs
 	span.SetAttributes(attrs...)
 
 	return &Agent{
-		sdk:      instance,
-		workflow: nil,
-		ctx:      aCtx,
+		sdk:        instance,
+		workflow:   nil,
+		ctx:        aCtx,
 		Attributes: agentAttrs,
 	}
 }
@@ -199,7 +197,6 @@ func (instance *Traceloop) LogPrompt(ctx context.Context, prompt Prompt, context
 	for key, value := range contextAttrs.AssociationProperties {
 		attrs = append(attrs, attribute.String("traceloop.association.properties."+key, value))
 	}
-	
 
 	span.SetAttributes(attrs...)
 	setMessagesAttribute(span, "llm.prompts", prompt.Messages)

traceloop-sdk/workflow.go

@@ -4,8 +4,8 @@ import (
 	"context"
 	"fmt"
 
-	"github.com/traceloop/go-openllmetry/traceloop-sdk/model"
 	semconvai "github.com/traceloop/go-openllmetry/semconv-ai"
+	"github.com/traceloop/go-openllmetry/traceloop-sdk/model"
 	"go.opentelemetry.io/otel/attribute"
 	"go.opentelemetry.io/otel/trace"
 )
@@ -26,12 +26,12 @@ func (instance *Traceloop) NewWorkflow(ctx context.Context, attrs WorkflowAttrib
 	)
 
 	if attrs.ABTest != nil {
-		for key, activeVarient := range attrs.ABTest.VarientsKeys {
+		for key, activeVarient := range attrs.ABTest.VarientKeys {
 			if activeVarient {
 				span.SetAttributes(attribute.String("traceloop.association.properties.ab_testing_variant", key))
 				break
 			}
-		}	
+		}
 	}
 
 	return &Workflow{
@@ -79,7 +79,7 @@ func (workflow *Workflow) NewAgent(name string, associationProperties map[string
 	}
 
 	if workflow.Attributes.ABTest != nil {
-		for key, activeVarient := range workflow.Attributes.ABTest.VarientsKeys {
+		for key, activeVarient := range workflow.Attributes.ABTest.VarientKeys {
 			if activeVarient {
 				associationProperties["ab_testing_variant"] = key
 			}
@@ -102,4 +102,3 @@ func (workflow *Workflow) NewAgent(name string, associationProperties map[string
 		},
 	}
 }
-