add prop

Author: nina-kollman

sample-app/generate_joke_workflow_example.go

@@ -149,6 +149,8 @@ func historyJokesTool(ctx context.Context, agent *sdk.Agent, client *openai.Clie
 		Name:        "history_jokes",
 		Description: "Get some history jokes",
 		Parameters:  map[string]interface{}{},
+	}, map[string]string{
+		"user_id": "user_12345",
 	})
 	defer tool.End()
 

sample-app/recipe_agent_example.go

@@ -9,6 +9,11 @@ import (
 	sdk "github.com/traceloop/go-openllmetry/traceloop-sdk"
 )
 
+var associationProperties = map[string]string{
+	"user_id": "user_67890",
+	"ab_testing_variant": "variant_a",
+}
+
 // ingredientValidatorTool validates that requested ingredients are available/safe
 func ingredientValidatorTool(ctx context.Context, agent *sdk.Agent, client *openai.Client, ingredients string) (string, error) {
 	tool := agent.NewTool("ingredient_validator", "function", sdk.ToolFunction{
@@ -23,7 +28,7 @@ func ingredientValidatorTool(ctx context.Context, agent *sdk.Agent, client *open
 				},
 			},
 		},
-	})
+	}, associationProperties)
 	defer tool.End()
 
 	prompt := sdk.Prompt{
@@ -91,7 +96,7 @@ func nutritionCalculatorTool(ctx context.Context, agent *sdk.Agent, client *open
 				},
 			},
 		},
-	})
+	}, associationProperties)
 	defer tool.End()
 
 	prompt := sdk.Prompt{
@@ -159,6 +164,8 @@ func cookingTimeEstimatorTool(ctx context.Context, agent *sdk.Agent, client *ope
 				},
 			},
 		},
+	}, map[string]string{
+		"user_id": "user_67890",
 	})
 	defer tool.End()
 
@@ -229,10 +236,7 @@ func runRecipeAgent() {
 	client := openai.NewClient(os.Getenv("OPENAI_API_KEY"))
 
 	// Create standalone agent with association properties
-	agent := traceloop.NewAgent(ctx, "recipe_generator", map[string]string{
-		"ab_testing_variant": "variant_a",
-		"user_id":            "user_67890",
-	})
+	agent := traceloop.NewAgent(ctx, "recipe_generator", associationProperties)
 	defer agent.End()
 
 	// User request

traceloop-sdk/agent.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	semconvai "github.com/traceloop/go-openllmetry/semconv-ai"
+	"go.opentelemetry.io/otel/attribute"
 	"go.opentelemetry.io/otel/trace"
 	"github.com/traceloop/go-openllmetry/traceloop-sdk/model"
 )
@@ -42,13 +43,24 @@ func (agent *Agent) LogPrompt(prompt Prompt) LLMSpan {
 	return agent.sdk.LogPrompt(agent.ctx, prompt, contextAttrs)
 }
 
-func (agent *Agent) NewTool(name string, toolType string, toolFunction ToolFunction) *Tool {
+func (agent *Agent) NewTool(name string, toolType string, toolFunction ToolFunction, associationProperties map[string]string) *Tool {
 	toolCtx, span := agent.sdk.getTracer().Start(agent.ctx, fmt.Sprintf("%s.tool", name))
-	span.SetAttributes(
+	attrs := []attribute.KeyValue{
 		semconvai.LLMAgentName.String(agent.Attributes.Name),
 		semconvai.TraceloopSpanKind.String(string(model.SpanKindTool)),
 		semconvai.TraceloopEntityName.String(name),
-	)
+	}
+
+	for key, value := range agent.Attributes.AssociationProperties {
+		attrs = append(attrs, attribute.String("traceloop.association.properties."+key, value))
+	}
+
+	// Add tool-specific association properties
+	for key, value := range associationProperties {
+		attrs = append(attrs, attribute.String("traceloop.association.properties."+key, value))
+	}
+
+	span.SetAttributes(attrs...)
 
 	return &Tool{
 		agent:    *agent,