Skip to content

Commit ec45807

Browse files
author
Liudmila Molkova
authored
add az.namespace to Azure AI Inference spans (Azure#48022)
1 parent fe16c3a commit ec45807

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

sdk/ai/Azure.AI.Inference/src/Telemetry/OpenTelemetryConstants.cs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ internal class OpenTelemetryConstants
1111
// https://github.com/open-telemetry/semantic-conventions/tree/v1.27.0/docs/gen-ai
1212

1313
public const string ErrorTypeKey = "error.type";
14+
15+
public const string AzNamespaceKey = "az.namespace";
1416
public const string ServerAddressKey = "server.address";
1517
public const string ServerPortKey = "server.port";
1618

@@ -39,6 +41,8 @@ internal class OpenTelemetryConstants
3941
public const string GenAiEventContent = "gen_ai.event.content";
4042
public const string GenAiChoice = "gen_ai.choice";
4143

44+
public const string AzureRpNamespaceValue = "Microsoft.CognitiveServices";
45+
4246
public const string ClientName = "Azure.AI.Inference.ChatCompletionsClient";
4347
public const string EnableOpenTelemetrySwitch = "Azure.Experimental.EnableActivitySource";
4448
public const string TraceContentsSwitch = "Azure.Experimental.TraceGenAIMessageContent";

sdk/ai/Azure.AI.Inference/src/Telemetry/OpenTelemetryScope.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ private OpenTelemetryScope(ChatCompletionsOptions requestOptions, Uri endpoint)
8888
{
8989
_activity?.SetTag(kv.Key, kv.Value);
9090
}
91+
_activity?.SetTag(AzNamespaceKey, AzureRpNamespaceValue);
9192
SetTagMaybe(GenAiRequestMaxTokensKey, requestOptions.MaxTokens);
9293
SetTagMaybe(GenAiRequestTemperatureKey, requestOptions.Temperature);
9394
SetTagMaybe(GenAiRequestTopPKey, requestOptions.NucleusSamplingFactor);

sdk/ai/Azure.AI.Inference/tests/Utilities/ValidatingActivityListener.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ public void ValidateResponse(RecordedResponse response, string errorType, string
9393
ValidateTag(activity, GenAiResponseFinishReasonsKey, response.FinishReasons);
9494
ValidateIntTag(activity, GenAiUsageOutputTokensKey, response.CompletionTokens);
9595
ValidateIntTag(activity, GenAiUsageInputTokensKey, response.PromptTokens);
96+
ValidateTag(activity, AzNamespaceKey, AzureRpNamespaceValue);
9697

9798
HashSet<string> expectedChoices = new HashSet<string>(response.Choices.Select(c => JsonSerializer.Serialize(c, options: s_jsonOptions)));
9899
for (int i = 0; i < actualChoices.Count; i++)

0 commit comments

Comments
 (0)