Skip to content

Commit ffe128f

Browse files
Add model tagging to OpenAITelemetryPlugin for request and response (#1642)
1 parent 190765a commit ffe128f

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

DevProxy.Plugins/Inspection/OpenAITelemetryPlugin.cs

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -848,6 +848,11 @@ private void AddCommonRequestTags(Activity activity, OpenAIRequest openAiRequest
848848
{
849849
Logger.LogTrace("AddCommonRequestTags() called");
850850

851+
if (!string.IsNullOrEmpty(openAiRequest.Model))
852+
{
853+
_ = activity.SetTag(SemanticConvention.GEN_AI_REQUEST_MODEL, openAiRequest.Model);
854+
}
855+
851856
if (openAiRequest.Temperature.HasValue)
852857
{
853858
_ = activity.SetTag(SemanticConvention.GEN_AI_REQUEST_TEMPERATURE,
@@ -915,6 +920,11 @@ private void RecordUsageMetrics(Activity activity, OpenAIRequest request, OpenAI
915920
new(SemanticConvention.GEN_AI_RESPONSE_MODEL, response.Model)
916921
]);
917922

923+
if (!string.IsNullOrEmpty(response.Model))
924+
{
925+
_ = activity.SetTag(SemanticConvention.GEN_AI_RESPONSE_MODEL, response.Model);
926+
}
927+
918928
_ = activity.SetTag(SemanticConvention.GEN_AI_USAGE_INPUT_TOKENS, usage.PromptTokens)
919929
.SetTag(SemanticConvention.GEN_AI_USAGE_OUTPUT_TOKENS, usage.CompletionTokens)
920930
.SetTag(SemanticConvention.GEN_AI_USAGE_TOTAL_TOKENS, usage.TotalTokens);

0 commit comments

Comments
 (0)