Enhance LitAgentRunner and related classes for improved execution and parallelism support

Author: KSemenenko

MIGRATION_PLAN.md

@@ -37,7 +37,7 @@ This plan tracks parity work between `external/microsoft-agent-lightning` (Pytho
 | LightningStore (async) | `store/base.py` | ✅ | `ILightningStore` exposes start/enqueue/start-attempt, span sequencing, and wait semantics |
 | In-memory store | `store/memory.py` | ✅ | Expanded store handles attempts, spans, resources, and polling waits with thread-safe state |
 | Client/server bridge | `store/client_server.py` | ❓ | Decide ASP.NET hosting approach |
-| Runner execution strategies | `execution/*` | 🚧 | C# runner tracks worker ids, retries via store requeue; parallel orchestration still pending |
+| Runner execution strategies | `execution/*` | 🚧 | C# runner supports single-step execution, retry-aware polling, and resource hydration; parallel orchestration still pending |
 
 ## Algorithms & Training Pipelines
 

src/ManagedCode.AgentLightning.AgentRuntime/LightningAgent.cs

@@ -1,5 +1,8 @@
+using System.Collections.Generic;
 using System.Collections.ObjectModel;
+using System.Linq;
 using ManagedCode.AgentLightning.Core.Models;
+using ManagedCode.AgentLightning.Core.Resources;
 using Microsoft.Extensions.AI;
 using Microsoft.Extensions.Logging;
 
@@ -41,7 +44,20 @@ public LightningAgent(
         _timeProvider = timeProvider ?? TimeProvider.System;
     }
 
-    protected override async Task<LightningExecutionResult> RolloutAsync(object taskInput, CancellationToken cancellationToken)
+    public Task<LightningExecutionResult> ExecuteAsync(
+        object taskInput,
+        NamedResources? resources,
+        string? resourcesId,
+        RolloutMode? mode = null,
+        CancellationToken cancellationToken = default) =>
+        base.ExecuteAsync(taskInput, resources, mode, resourcesId, cancellationToken);
+
+    protected override async Task<LightningExecutionResult> RolloutAsync(
+        object taskInput,
+        NamedResources? resources,
+        RolloutMode? mode,
+        string? resourcesId,
+        CancellationToken cancellationToken)
     {
         ThrowIfDisposed();
 
@@ -52,15 +68,34 @@ protected override async Task<LightningExecutionResult> RolloutAsync(object task
             taskInput,
             startTimestamp,
             config: _options.RolloutConfig,
-            mode: null,
-            resourcesId: null);
+            mode: mode,
+            resourcesId: resourcesId);
 
         var attempt = new Attempt(
             rolloutId,
             $"{rolloutId}:attempt:1",
             sequenceId: 1,
             startTimestamp);
 
+        if (!string.IsNullOrEmpty(resourcesId))
+        {
+            var id = resourcesId!;
+            rollout.AddMetadata("resources.id", id);
+            attempt.AddMetadata("resources.id", id);
+        }
+
+        if (resources is { Count: > 0 })
+        {
+            var resourceNames = resources.Keys.ToArray();
+            rollout.AddMetadata("resources.names", resourceNames);
+            attempt.AddMetadata("resources.names", resourceNames);
+        }
+
+        if (mode is { } rolloutMode)
+        {
+            rollout.AddMetadata("mode", rolloutMode.ToString());
+        }
+
         var context = new LightningContext(this, this, rollout);
         await InvokeHooksAsync(h => h.OnTraceStartAsync(context, cancellationToken), cancellationToken).ConfigureAwait(false);
 

src/ManagedCode.AgentLightning.AgentRuntime/LitAgentBase.cs

@@ -1,4 +1,6 @@
 using ManagedCode.AgentLightning.AgentRuntime.Runner;
+using ManagedCode.AgentLightning.Core.Models;
+using ManagedCode.AgentLightning.Core.Resources;
 
 namespace ManagedCode.AgentLightning.AgentRuntime;
 
@@ -26,10 +28,25 @@ internal void AttachRunner(LitAgentRunner runner)
 
     public virtual bool IsAsync => true;
 
-    public async Task<LightningExecutionResult> ExecuteAsync(TTask task, CancellationToken cancellationToken = default)
+    public Task<LightningExecutionResult> ExecuteAsync(TTask task, CancellationToken cancellationToken = default) =>
+        ExecuteAsync(task, resources: null, mode: null, resourcesId: null, cancellationToken);
+
+    public Task<LightningExecutionResult> ExecuteAsync(
+        TTask task,
+        NamedResources? resources,
+        RolloutMode? mode,
+        CancellationToken cancellationToken = default) =>
+        ExecuteAsync(task, resources, mode, resourcesId: null, cancellationToken);
+
+    public async Task<LightningExecutionResult> ExecuteAsync(
+        TTask task,
+        NamedResources? resources,
+        RolloutMode? mode,
+        string? resourcesId,
+        CancellationToken cancellationToken = default)
     {
         await OnRolloutStartAsync(task, cancellationToken).ConfigureAwait(false);
-        var result = await RolloutAsync(task, cancellationToken).ConfigureAwait(false);
+        var result = await RolloutAsync(task, resources, mode, resourcesId, cancellationToken).ConfigureAwait(false);
         await OnRolloutEndAsync(task, result, cancellationToken).ConfigureAwait(false);
         return result;
     }
@@ -39,5 +56,10 @@ public async Task<LightningExecutionResult> ExecuteAsync(TTask task, Cancellatio
     protected virtual Task OnRolloutEndAsync(TTask task, LightningExecutionResult result, CancellationToken cancellationToken) =>
         Task.CompletedTask;
 
-    protected abstract Task<LightningExecutionResult> RolloutAsync(TTask task, CancellationToken cancellationToken);
+    protected abstract Task<LightningExecutionResult> RolloutAsync(
+        TTask task,
+        NamedResources? resources,
+        RolloutMode? mode,
+        string? resourcesId,
+        CancellationToken cancellationToken);
 }