Add SmartAgent sample and tracing improvements

Add a new Samples/SmartAgent.cs LLM-powered agent with per-ID system prompts and Claude45Haiku injection. Introduce IMonitorSourceProvider and related MonitorPoll request/result/feed DTOs for polling monitor sources. Enhance telemetry and tracing: expose CorrelationId/TraceId/ParentSpanId/TraceSampled on TelegramBotUpdate, capture Activity in Telegram Program handler, and use a short-timeout HttpClient in WebhookSetupService. Configure OpenTelemetry sampling (configurable default 0.2), set a ParentBased TraceIdRatio sampler, and filter HTTP client instrumentation to only emit client spans when the parent is sampled. Add token usage fields to AgentReply and populate them from LLM responses; surface modelId and usage in OrleansAgentChatClient. Change AgentV2 memory write to TryAdd and return updated state. Update IAW.AppHost and IAWExtensions to optionally wait for LLM resources, add an mcp direct HTTP endpoint, tweak startup waits and logging defaults. Update various tools (IAW.MCP) to target Samples.SmartAgent grain class. Misc: appsettings and logging adjustments, add iaw entry to mcp.json, and small .claude command additions.

Co-Authored-By: Eugene <59283295+ScientistFromMars@users.noreply.github.com>

Author: LeftTwixWand

.claude/settings.local.json

@@ -56,7 +56,9 @@
       "Bash(wmic process:*)",
       "Bash(aspire mcp:*)",
       "Bash(docker ps:*)",
-      "Bash(ping:*)"
+      "Bash(ping:*)",
+      "Bash(wc:*)",
+      "mcp__aspire__list_trace_structured_logs"
     ]
   },
   "enabledMcpjsonServers": [

.mcp.json

@@ -1,5 +1,8 @@
 {
   "mcpServers": {
+    "iaw": {
+      "url": "http://localhost:5300"
+    },
     "aspire": {
       "command": "aspire",
       "args": ["mcp", "start", "--non-interactive"]

samples/Samples/SmartAgent.cs

@@ -0,0 +1,96 @@
+using Core;
+using Core.AI;
+using Core.AI.Models;
+using Core.V2;
+using Microsoft.Extensions.AI;
+using Orleans.Journaling;
+
+namespace Samples;
+
+public class SmartAgent(
+    [Memory("v2-messages")] IDurableList<AgentMessage> messages,
+    [Memory("v2-memory")] IDurableDictionary<string, string> memory,
+    [Memory("v2-events")] IDurableList<AgentEvent> events,
+    [Memory("v2-subscriptions")] IDurableDictionary<string, List<string>> subscriptions,
+    [Memory("v2-notifications")] IDurableList<NotificationRecord> notifications,
+    [Memory("v2-tracking")] IDurableDictionary<string, string> tracking,
+    [Llm<Claude45Haiku>] IChatClient chatClient)
+    : Agent(messages, memory, events, subscriptions, notifications, tracking)
+{
+    static readonly Dictionary<string, (string DisplayName, string Prompt)> Profiles = new(StringComparer.OrdinalIgnoreCase)
+    {
+        ["personal-assistant"] = ("Personal Assistant", """
+            You are a personal assistant and team lead for a software development team.
+            You help users with tasks by understanding their needs, breaking down complex requests,
+            and coordinating work. You can answer questions, provide guidance, and help plan work.
+            Be concise, helpful, and proactive.
+            """),
+        ["roslyn"] = ("Roslyn", """
+            You are a C# code intelligence agent powered by Roslyn.
+            You help with code analysis, understanding syntax trees, finding type information,
+            detecting patterns, and analyzing architecture. You provide detailed technical answers
+            about C# code structure and semantics.
+            """),
+        ["dotnet"] = ("DotNet", """
+            You are a .NET toolchain agent. You help with building, testing, and formatting
+            .NET projects. You understand MSBuild, dotnet CLI, project files, and the .NET ecosystem.
+            Provide practical commands and solutions for build issues.
+            """),
+        ["nuget"] = ("NuGet", """
+            You are a NuGet package management agent. You help find packages, check versions,
+            resolve dependency conflicts, and manage package references. You know the NuGet ecosystem well.
+            """),
+        ["github"] = ("GitHub", """
+            You are a GitHub agent. You help with pull requests, issues, releases, and repository management.
+            You understand GitHub workflows, Actions, and collaboration patterns.
+            """),
+        ["reviewer"] = ("Reviewer", """
+            You are a code review agent. You analyze code for quality, correctness, security,
+            performance, and maintainability. You provide actionable feedback with specific suggestions.
+            Focus on important issues, not style nitpicks.
+            """),
+        ["fs"] = ("FileSystem", """
+            You are a file system agent. You help with reading, writing, and searching files.
+            You understand project structures, file formats, and can help navigate codebases.
+            """),
+        ["shell"] = ("Shell", """
+            You are a shell command agent. You help execute and explain shell commands.
+            You understand bash, PowerShell, and common CLI tools. You prioritize safe, correct commands.
+            """),
+        ["git"] = ("Git", """
+            You are a Git version control agent. You help with branches, commits, merges, rebases,
+            and repository management. You understand Git workflows and best practices.
+            """),
+        ["build"] = ("Build", """
+            You are a build runner agent. You help with build systems, CI/CD pipelines,
+            and automated builds. You understand MSBuild, Make, and other build tools.
+            """),
+        ["knowledge"] = ("Knowledge", """
+            You are a project knowledge agent. You store and retrieve project information including
+            architecture decisions, tech stack details, patterns, and conventions.
+            You help maintain institutional knowledge.
+            """),
+        ["user"] = ("User", """
+            You are a user preferences agent. You help manage user settings, preferences,
+            and memories. You remember what users tell you and recall it when relevant.
+            """),
+        ["planning"] = ("Planning", """
+            You are a planning agent. You help create execution plans for software development tasks.
+            You break down complex tasks into steps, identify dependencies, and estimate effort.
+            You produce clear, actionable plans.
+            """),
+        ["notification"] = ("Notification", """
+            You are a notification agent. You help manage alerts and notifications for the user.
+            You can summarize important updates and help configure notification preferences.
+            """),
+    };
+
+    public override string DisplayName => Profiles.TryGetValue(AgentId, out var p) ? p.DisplayName : AgentId;
+
+    public override string SystemPrompt => Profiles.TryGetValue(AgentId, out var p)
+        ? p.Prompt
+        : "You are a helpful AI assistant. Answer questions clearly and concisely.";
+
+    protected override Task<AgentReply> OnRespondAsync(AgentRequest request, CancellationToken ct)
+        => RespondWithLlmAsync(chatClient, request, ct);
+}

src/Clients.Telegram.Bot/ITelegramConversation.cs

@@ -34,6 +34,10 @@ public sealed class TelegramBotUpdate
     [Id(8)] public long? FromUserId { get; set; }
     [Id(9)] public string? VoiceFileId { get; set; }
     [Id(10)] public int VoiceDuration { get; set; }
+    [Id(11)] public string? CorrelationId { get; set; }
+    [Id(12)] public string? TraceId { get; set; }
+    [Id(13)] public string? ParentSpanId { get; set; }
+    [Id(14)] public bool TraceSampled { get; set; }
 }
 
 [GenerateSerializer]

src/Clients.Telegram.Bot/Program.cs

@@ -3,6 +3,7 @@
 using Microsoft.Extensions.Options;
 using Orleans.Journaling;
 using ServiceDefaults;
+using System.Diagnostics;
 using Telegram.BotAPI;
 using TelegramBot;
 using TelegramBot.Services;
@@ -75,6 +76,7 @@
     if (chatId == 0)
         return Results.Ok();
 
+    var currentActivity = Activity.Current;
     var botUpdate = new TelegramBotUpdate
     {
         ChatId = chatId,
@@ -87,7 +89,11 @@
         FirstName = update.Message?.From?.FirstName ?? update.CallbackQuery?.From?.FirstName,
         FromUserId = update.Message?.From?.Id ?? update.CallbackQuery?.From?.Id,
         VoiceFileId = update.Message?.Voice?.FileId,
-        VoiceDuration = update.Message?.Voice?.Duration ?? 0
+        VoiceDuration = update.Message?.Voice?.Duration ?? 0,
+        CorrelationId = currentActivity?.TraceId.ToHexString() ?? context.TraceIdentifier,
+        TraceId = currentActivity?.TraceId.ToHexString(),
+        ParentSpanId = currentActivity?.SpanId.ToHexString(),
+        TraceSampled = currentActivity?.Recorded ?? false
     };
 
     var conversation = grains.GetGrain<Core.ITelegramConversation>($"conversation-{chatId}");

src/Clients.Telegram.Bot/WebhookSetupService.cs

@@ -16,7 +16,6 @@ public sealed class TelegramBotOptions
 public sealed class WebhookSetupService(
     IGrainFactory grains,
     IOptions<TelegramBotOptions> options,
-    IHttpClientFactory httpClientFactory,
     ILogger<WebhookSetupService> logger) : BackgroundService
 {
     protected override async Task ExecuteAsync(CancellationToken stoppingToken)
@@ -48,7 +47,12 @@ protected override async Task ExecuteAsync(CancellationToken stoppingToken)
 
         try
         {
-            using var http = httpClientFactory.CreateClient();
+            // Use a direct short-timeout client here to avoid noisy resilience retries
+            // when ngrok is intentionally unavailable in local/dev runs.
+            using var http = new HttpClient
+            {
+                Timeout = TimeSpan.FromSeconds(3)
+            };
             var json = await http.GetFromJsonAsync<JsonElement>(
                 $"{ngrokApiUrl.TrimEnd('/')}/api/tunnels", ct);
 

src/Clients.Telegram.Bot/appsettings.json

@@ -1,8 +1,18 @@
 {
   "Logging": {
     "LogLevel": {
-      "Default": "Information",
-      "Microsoft.AspNetCore": "Warning"
+      "Default": "Warning",
+      "TelegramBot": "Information",
+      "Microsoft.AspNetCore": "Warning",
+      "Microsoft.Hosting.Lifetime": "Information",
+      "Orleans": "Warning",
+      "System.Net.Http.HttpClient": "Warning",
+      "Polly": "Error"
+    }
+  },
+  "Telemetry": {
+    "Tracing": {
+      "SampleRatio": 1.0
     }
   },
   "Telegram": {

src/Core/IMonitorSourceProvider.cs

@@ -0,0 +1,37 @@
+namespace Core;
+
+public interface IMonitorSourceProvider : IGrainWithStringKey
+{
+    Task<MonitorPollResult> PollAsync(MonitorPollRequest request, CancellationToken ct = default);
+}
+
+[GenerateSerializer]
+public sealed class MonitorPollRequest
+{
+    [Id(0)] public string Source { get; set; } = string.Empty;
+    [Id(1)] public string RawQuery { get; set; } = string.Empty;
+    [Id(2)] public string? Cursor { get; set; }
+    [Id(3)] public int MaxItems { get; set; } = 5;
+    [Id(4)] public bool EmitInitialItems { get; set; }
+}
+
+[GenerateSerializer]
+public sealed class MonitorPollResult
+{
+    [Id(0)] public bool Success { get; set; }
+    [Id(1)] public string ProviderId { get; set; } = string.Empty;
+    [Id(2)] public string Status { get; set; } = string.Empty;
+    [Id(3)] public string? NextCursor { get; set; }
+    [Id(4)] public List<MonitorFeedItem> NewItems { get; set; } = [];
+    [Id(5)] public DateTimeOffset CheckedAtUtc { get; set; } = DateTimeOffset.UtcNow;
+}
+
+[GenerateSerializer]
+public sealed class MonitorFeedItem
+{
+    [Id(0)] public string Id { get; set; } = string.Empty;
+    [Id(1)] public string Title { get; set; } = string.Empty;
+    [Id(2)] public string Url { get; set; } = string.Empty;
+    [Id(3)] public DateTimeOffset? PublishedAtUtc { get; set; }
+    [Id(4)] public string Summary { get; set; } = string.Empty;
+}

src/Core/V2/AgentReply.cs

@@ -14,4 +14,13 @@ public sealed class AgentReply
 
     [Id(3)]
     public DateTimeOffset TimestampUtc { get; set; } = DateTimeOffset.UtcNow;
+
+    [Id(4)]
+    public long? InputTokens { get; set; }
+
+    [Id(5)]
+    public long? OutputTokens { get; set; }
+
+    [Id(6)]
+    public long? TotalTokens { get; set; }
 }

src/Core/V2/AgentV2.cs

@@ -173,8 +173,7 @@ public async Task SetMemoryAsync(string key, string value, CancellationToken ct
         if (string.IsNullOrWhiteSpace(key))
             throw new ArgumentException("Memory key cannot be empty.", nameof(key));
 
-        ct.ThrowIfCancellationRequested();
-        memory[key] = value;
+        memory.TryAdd(key, value);       
         await WriteStateAsync(ct);
     }
 
@@ -435,7 +434,14 @@ protected async Task<AgentReply> RespondWithLlmAsync(IChatClient chatClient, Age
         var options = new ChatOptions { Tools = [.. DefineTools()] };
         var response = await chatClient.GetResponseAsync(chatMessages, options, ct);
         var output = string.Join("", response.Messages.Where(m => m.Role == ChatRole.Assistant).Select(m => m.Text));
-        return new AgentReply { Output = output, ModelId = response.ModelId };
+        return new AgentReply
+        {
+            Output = output,
+            ModelId = response.ModelId,
+            InputTokens = response.Usage?.InputTokenCount,
+            OutputTokens = response.Usage?.OutputTokenCount,
+            TotalTokens = response.Usage?.TotalTokenCount
+        };
     }
 
     // -- IRemindable --