diff options
Diffstat (limited to 'src/Agent.Mcp/QueuedTaskWorker.cs')
| -rw-r--r-- | src/Agent.Mcp/QueuedTaskWorker.cs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/Agent.Mcp/QueuedTaskWorker.cs b/src/Agent.Mcp/QueuedTaskWorker.cs index edea1ad..799d5d8 100644 --- a/src/Agent.Mcp/QueuedTaskWorker.cs +++ b/src/Agent.Mcp/QueuedTaskWorker.cs @@ -11,10 +11,10 @@ public sealed class QueuedTaskWorker(TaskRegistry registry, AgentRuntime runtime await foreach (Guid taskId in registry.ReadQueuedAsync(stoppingToken).ConfigureAwait(false)) { if (!registry.Transition(taskId, AgentTaskState.Planning, "planning")) continue; - registry.Transition(taskId, AgentTaskState.Running, "ollama chat"); using var linked = CancellationTokenSource.CreateLinkedTokenSource(stoppingToken, registry.GetCancellationToken(taskId)); try { + // ExecuteAsync handles Planning → Running transition internally. await runtime.ExecuteAsync(taskId, linked.Token).ConfigureAwait(false); } catch (OperationCanceledException) when (linked.IsCancellationRequested) |