mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
7ebe00ec3d
* Updates to async run loop. * fix: Workflow Onwership can be release by nonowner * fix: Incorrect handling of blockOnPending in StreamingRun Depending on whether we are running in streaming on non-streaming mode, we may be using the StreamingRun in different ways. Unfortunately, the only place we can really know what is the actual state of execution is in the RunEventStream implementations. This resulted in blocking where blocking was unneeded and occasionally not-blocking when blocking was needed. The fix is to move the logic of handling this blocking into RunEventStream implementations. * fix: Fix cleanup on error and end run This ensures we clean up the background resources correctly. * fix: Ensure we let the run loop proceed when shutting down * fix: Add timeout for Input Waiting * fix: Make the samples properly clean up `Run`s and `StreamingRun`s * fix: Simplify Declarative Workflow Run disposal pattern * Also fixes missing .Disposal() in Integration tests --------- Co-authored-by: Ben Thomas <ben.thomas@microsoft.com>
67 lines
2.0 KiB
C#
67 lines
2.0 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Agents.AI.Workflows.Checkpointing;
|
|
using Microsoft.Shared.Diagnostics;
|
|
|
|
namespace Microsoft.Agents.AI.Workflows;
|
|
|
|
/// <summary>
|
|
/// Represents a workflow run that supports checkpointing.
|
|
/// </summary>
|
|
/// <typeparam name="TRun">The type of the underlying workflow run handle.</typeparam>
|
|
/// <seealso cref="Run"/>
|
|
/// <seealso cref="StreamingRun"/>
|
|
public sealed class Checkpointed<TRun> : IAsyncDisposable
|
|
{
|
|
private readonly ICheckpointingHandle _runner;
|
|
|
|
internal Checkpointed(TRun run, ICheckpointingHandle runner)
|
|
{
|
|
this.Run = Throw.IfNull(run);
|
|
this._runner = Throw.IfNull(runner);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the workflow run associated with this <see cref="Checkpointed{TRun}"/> instance.
|
|
/// </summary>
|
|
/// <seealso cref="Run"/>
|
|
/// <seealso cref="StreamingRun"/>
|
|
public TRun Run { get; }
|
|
|
|
/// <inheritdoc cref="ICheckpointingHandle.Checkpoints"/>
|
|
public IReadOnlyList<CheckpointInfo> Checkpoints => this._runner.Checkpoints;
|
|
|
|
/// <summary>
|
|
/// Gets the most recent checkpoint information.
|
|
/// </summary>
|
|
public CheckpointInfo? LastCheckpoint
|
|
{
|
|
get
|
|
{
|
|
var checkpoints = this.Checkpoints;
|
|
return checkpoints.Count > 0 ? checkpoints[checkpoints.Count - 1] : null;
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
public async ValueTask DisposeAsync()
|
|
{
|
|
if (this.Run is IAsyncDisposable asyncDisposable)
|
|
{
|
|
await asyncDisposable.DisposeAsync().ConfigureAwait(false);
|
|
}
|
|
else if (this.Run is IDisposable disposable)
|
|
{
|
|
disposable.Dispose();
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc cref="ICheckpointingHandle.RestoreCheckpointAsync"/>
|
|
public ValueTask RestoreCheckpointAsync(CheckpointInfo checkpointInfo, CancellationToken cancellationToken = default)
|
|
=> this._runner.RestoreCheckpointAsync(checkpointInfo, cancellationToken);
|
|
}
|