| | | 1 | | using Microsoft.Agents.AI; |
| | | 2 | | using Microsoft.Extensions.DependencyInjection; |
| | | 3 | | |
| | | 4 | | using NexusLabs.Needlr.AgentFramework; |
| | | 5 | | using NexusLabs.Needlr.AgentFramework.Diagnostics; |
| | | 6 | | using NexusLabs.Needlr.AgentFramework.Progress; |
| | | 7 | | |
| | | 8 | | namespace NexusLabs.Needlr.AgentFramework.Workflows.Diagnostics; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// <see cref="IAIAgentBuilderPlugin"/> that wires the diagnostics middleware layers |
| | | 12 | | /// into every agent created by the factory. Emits <see cref="IAgentMetrics"/> |
| | | 13 | | /// counters/histograms and <see cref="IProgressEvent"/> events in real-time. |
| | | 14 | | /// </summary> |
| | | 15 | | internal sealed class AgentDiagnosticsPlugin : IAIAgentBuilderPlugin |
| | | 16 | | { |
| | | 17 | | private readonly IServiceProvider _serviceProvider; |
| | | 18 | | |
| | 36 | 19 | | internal AgentDiagnosticsPlugin(IServiceProvider serviceProvider) |
| | | 20 | | { |
| | 36 | 21 | | _serviceProvider = serviceProvider; |
| | 36 | 22 | | } |
| | | 23 | | |
| | | 24 | | /// <inheritdoc /> |
| | | 25 | | public void Configure(AIAgentBuilderPluginOptions options) |
| | | 26 | | { |
| | 39 | 27 | | ArgumentNullException.ThrowIfNull(options); |
| | | 28 | | |
| | 39 | 29 | | var builder = options.AgentBuilder; |
| | | 30 | | |
| | 39 | 31 | | var writer = _serviceProvider.GetRequiredService<IAgentDiagnosticsWriter>(); |
| | 39 | 32 | | var metrics = _serviceProvider.GetRequiredService<IAgentMetrics>(); |
| | | 33 | | |
| | 39 | 34 | | var runMiddleware = new DiagnosticsAgentRunMiddleware("Agent", writer, metrics); |
| | 39 | 35 | | builder.Use( |
| | 39 | 36 | | runFunc: runMiddleware.HandleAsync, |
| | 39 | 37 | | runStreamingFunc: runMiddleware.HandleStreamingAsync); |
| | | 38 | | |
| | 39 | 39 | | var progressAccessor = _serviceProvider.GetRequiredService<IProgressReporterAccessor>(); |
| | 39 | 40 | | var toolCallCollector = _serviceProvider.GetService<IToolCallCollector>(); |
| | 39 | 41 | | DiagnosticsFunctionCallingMiddleware.Wire(builder, metrics, progressAccessor, toolCallCollector); |
| | 39 | 42 | | } |
| | | 43 | | } |