|
16 | 16 | using OpenTelemetry.Metrics; |
17 | 17 | using OpenTelemetry.Resources; |
18 | 18 | using OpenTelemetry.Trace; |
19 | | -using System.Collections.Concurrent; |
20 | 19 | using System.Diagnostics; |
21 | 20 | using System.Diagnostics.Metrics; |
22 | 21 | using System.Text.Json; |
@@ -67,7 +66,6 @@ public sealed class OpenAITelemetryPlugin( |
67 | 66 | private LanguageModelPricesLoader? _loader; |
68 | 67 | private MeterProvider? _meterProvider; |
69 | 68 | private TracerProvider? _tracerProvider; |
70 | | - private readonly ConcurrentDictionary<string, List<OpenAITelemetryPluginReportModelUsageInformation>> _modelUsage = []; |
71 | 69 |
|
72 | 70 | public override string Name => nameof(OpenAITelemetryPlugin); |
73 | 71 |
|
@@ -204,11 +202,9 @@ public override Task AfterRecordingStopAsync(RecordingArgs e, CancellationToken |
204 | 202 | Environment = Configuration.Environment, |
205 | 203 | Currency = Configuration.Currency, |
206 | 204 | IncludeCosts = Configuration.IncludeCosts, |
207 | | - ModelUsage = _modelUsage.ToDictionary() |
208 | 205 | }; |
209 | 206 |
|
210 | 207 | StoreReport(report, e); |
211 | | - _modelUsage.Clear(); |
212 | 208 |
|
213 | 209 | Logger.LogTrace("Left {Name}", nameof(AfterRecordingStopAsync)); |
214 | 210 | return Task.CompletedTask; |
@@ -858,8 +854,6 @@ private void RecordUsageMetrics(Activity activity, OpenAIRequest request, OpenAI |
858 | 854 | CompletionTokens = usage.CompletionTokens, |
859 | 855 | CachedTokens = usage.PromptTokensDetails?.CachedTokens ?? 0L |
860 | 856 | }; |
861 | | - var usagePerModel = _modelUsage.GetOrAdd(response.Model, model => []); |
862 | | - usagePerModel.Add(reportModelUsageInformation); |
863 | 857 |
|
864 | 858 | if (!Configuration.IncludeCosts || Configuration.Prices is null) |
865 | 859 | { |
|
0 commit comments