feat: enhance image generation metrics

Signed-off-by: Hrushikesh Patil <[email protected]>

Author: nutanix-Hrushikesh

internal/metrics/genai.go

@@ -24,6 +24,11 @@ const (
 	genaiAttributeTokenType     = "gen_ai.token.type" //nolint:gosec // metric name, not credential
 	genaiAttributeErrorType     = "error.type"
 
+	// Image generation specific attributes (currently used in main codebase)
+	genaiAttributeImageCount = "gen_ai.image.count"
+	genaiAttributeImageModel = "gen_ai.image.model"
+	genaiAttributeImageSize  = "gen_ai.image.size"
+
 	genaiOperationChat            = "chat"
 	genaiOperationEmbedding       = "embeddings"
 	genaiOperationImageGeneration = "image_generation"

internal/metrics/image_generation_metrics.go

@@ -87,9 +87,9 @@ func (i *imageGeneration) RecordImageGeneration(ctx context.Context, imageCount
 	// Add image-specific attributes
 	extendedAttrs := attribute.NewSet(
 		append(attrs.ToSlice(),
-			attribute.Key("gen_ai.image.count").Int(imageCount),
-			attribute.Key("gen_ai.image.model").String(model),
-			attribute.Key("gen_ai.image.size").String(size),
+			attribute.Key(genaiAttributeImageCount).Int(imageCount),
+			attribute.Key(genaiAttributeImageModel).String(model),
+			attribute.Key(genaiAttributeImageSize).String(size),
 		)...,
 	)